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

fix(ui): fix missing 'none' on no-board cache updates

Browse Source
This commit is contained in:
psychedelicious 2023-07-21 15:53:41 +10:00
parent 5396e998b3
commit 8e98085530
1 changed files with 28 additions and 43 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
invokeai/frontend/web/src/services/api/endpoints/images.ts
View File

@ -225,7 +225,7 @@ export const imagesApi = api.injectEndpoints({
dispatch(
imagesApi.util.updateQueryData(
'listImages',
{ board_id, categories },
{ board_id: board_id ?? 'none', categories },
(draft) => {
const oldTotal = draft.total;
const newState = imagesAdapter.removeOne(draft, image_name);
@ -300,7 +300,7 @@ export const imagesApi = api.injectEndpoints({
dispatch(
imagesApi.util.updateQueryData(
'listImages',
{ board_id: imageDTO.board_id, categories },
{ board_id: imageDTO.board_id ?? 'none', categories },
(draft) => {
const oldTotal = draft.total;
const newState = imagesAdapter.removeOne(
@ -315,7 +315,10 @@ export const imagesApi = api.injectEndpoints({
);
} else {
// ELSE (it is being changed to a non-intermediate):
const queryArgs = { board_id: imageDTO.board_id, categories };
const queryArgs = {
board_id: imageDTO.board_id ?? 'none',
categories,
};
const currentCache = imagesApi.endpoints.listImages.select(queryArgs)(
getState()
@ -539,48 +542,30 @@ export const imagesApi = api.injectEndpoints({
)
);
if (imageDTO.board_id) {
// *remove* from old board_id/[images|assets]
patches.push(
dispatch(
imagesApi.util.updateQueryData(
'listImages',
{
board_id: imageDTO.board_id,
categories,
},
(draft) => {
const oldTotal = draft.total;
const newState = imagesAdapter.removeOne(
draft,
imageDTO.image_name
);
const delta = newState.total - oldTotal;
draft.total = draft.total + delta;
}
)
// *remove* from [no_board|board_id]/[images|assets]
patches.push(
dispatch(
imagesApi.util.updateQueryData(
'listImages',
{
board_id: imageDTO.board_id ?? 'none',
categories,
},
(draft) => {
const oldTotal = draft.total;
const newState = imagesAdapter.removeOne(
draft,
imageDTO.image_name
);
const delta = newState.total - oldTotal;
draft.total = draft.total + delta;
}
)
);
} else {
// *remove* from no_board/[images|assets]
patches.push(
dispatch(
imagesApi.util.updateQueryData(
'listImages',
{
board_id: 'none',
categories,
},
(draft) => {
imagesAdapter.removeOne(draft, imageDTO.image_name);
}
)
)
);
}
)
);
// $cache = board_id/[images|assets]
const queryArgs = { board_id, categories };
const queryArgs = { board_id: board_id ?? 'none', categories };
const currentCache = imagesApi.endpoints.listImages.select(queryArgs)(
getState()
);
@ -676,7 +661,7 @@ export const imagesApi = api.injectEndpoints({
imagesApi.util.updateQueryData(
'listImages',
{
board_id: imageDTO.board_id,
board_id: imageDTO.board_id ?? 'none',
categories,
},
(draft) => {