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

implmenet custom sort to replace images adapter logic

Browse Source
This commit is contained in:
Mary Hipp 2024-06-23 19:26:04 -04:00 committed by psychedelicious
parent 719c066ac4
commit 7bbe236107
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/frontend/web/src/app/store/middleware/listenerMiddleware/listeners/socketio/socketInvocationComplete.ts
View File

@ -15,7 +15,7 @@ import { zNodeStatus } from 'features/nodes/types/invocation';
import { CANVAS_OUTPUT } from 'features/nodes/util/graph/constants'; import { CANVAS_OUTPUT } from 'features/nodes/util/graph/constants';
import { boardsApi } from 'services/api/endpoints/boards'; import { boardsApi } from 'services/api/endpoints/boards';
import { imagesApi } from 'services/api/endpoints/images'; import { imagesApi } from 'services/api/endpoints/images';
import { imagesAdapter } from 'services/api/util'; import { imageListDefaultSort, imagesAdapter } from 'services/api/util';
import { socketInvocationComplete } from 'services/events/actions'; import { socketInvocationComplete } from 'services/events/actions';
// These nodes output an image, but do not actually *save* an image, so we don't want to handle the gallery logic on them // These nodes output an image, but do not actually *save* an image, so we don't want to handle the gallery logic on them
@ -68,7 +68,8 @@ export const addInvocationCompleteEventListener = (startAppListening: AppStartLi
is_intermediate: false is_intermediate: false
}, },
(draft) => { (draft) => {
const updatedListLength = draft.items.unshift(imageDTO) const updatedListLength = draft.items.unshift(imageDTO);
draft.items.sort(imageListDefaultSort())
if (updatedListLength > IMAGE_LIMIT) { if (updatedListLength > IMAGE_LIMIT) {
draft.items.pop() draft.items.pop()
} }

13
invokeai/frontend/web/src/services/api/util.ts
View File

@ -75,6 +75,19 @@ export const imagesAdapter = createEntityAdapter<ImageDTO, string>({
}, },
}); });
export const imageListDefaultSort = () => {
return (a: ImageDTO, b: ImageDTO) => {
if (a.starred && !b.starred) {
return -1;
}
if (!a.starred && b.starred) {
return 1;
}
return dateComparator(b.created_at, a.created_at);
}
}
// Create selectors for the adapter. // Create selectors for the adapter.
export const imagesSelectors = imagesAdapter.getSelectors(undefined, getSelectorsOptions); export const imagesSelectors = imagesAdapter.getSelectors(undefined, getSelectorsOptions);