Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ui): add rtk action type guard

Browse Source
This commit is contained in:
psychedelicious 2023-04-02 19:48:29 +10:00
parent 4ac447cf74
commit 662f1321f6
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
invokeai/frontend/web/src/services/invokeMiddleware.ts
View File

@ -1,14 +1,21 @@
import { Middleware, MiddlewareAPI } from '@reduxjs/toolkit'; import { isFulfilled, Middleware, MiddlewareAPI } from '@reduxjs/toolkit';
import { emitSubscribe } from 'app/nodesSocketio/actions'; import { emitSubscribe } from 'app/nodesSocketio/actions';
import { AppDispatch, RootState } from 'app/store'; import { AppDispatch, RootState } from 'app/store';
import { setSessionId } from './apiSlice'; import { setSessionId } from './apiSlice';
import { invokeSession } from './thunks/session'; import { createSession, invokeSession } from './thunks/session';
/**
* `redux-toolkit` provides nice matching utilities, which can be used as type guards
* See: https://redux-toolkit.js.org/api/matching-utilities
*/
const isFulfilledCreateSession = isFulfilled(createSession);
export const invokeMiddleware: Middleware = export const invokeMiddleware: Middleware =
(store: MiddlewareAPI<AppDispatch, RootState>) => (next) => (action) => { (store: MiddlewareAPI<AppDispatch, RootState>) => (next) => (action) => {
const { dispatch } = store; const { dispatch } = store;
if (action.type === 'api/createSession/fulfilled' && action?.payload?.id) { if (isFulfilledCreateSession(action)) {
const sessionId = action.payload.id; const sessionId = action.payload.id;
console.log('createSession.fulfilled'); console.log('createSession.fulfilled');