mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
89 lines
2.8 KiB
TypeScript
89 lines
2.8 KiB
TypeScript
import {
|
|
ASSETS_CATEGORIES,
|
|
IMAGE_CATEGORIES,
|
|
} from 'features/gallery/store/types';
|
|
import { ImageCache, ImageDTO, ListImagesArgs } from './types';
|
|
import { createEntityAdapter } from '@reduxjs/toolkit';
|
|
import { dateComparator } from 'common/util/dateComparator';
|
|
import queryString from 'query-string';
|
|
|
|
export const getIsImageInDateRange = (
|
|
data: ImageCache | undefined,
|
|
imageDTO: ImageDTO
|
|
) => {
|
|
if (!data) {
|
|
return false;
|
|
}
|
|
|
|
const totalCachedImageDtos = imagesSelectors.selectAll(data);
|
|
|
|
if (totalCachedImageDtos.length <= 1) {
|
|
return true;
|
|
}
|
|
|
|
const cachedStarredImages = [];
|
|
const cachedUnstarredImages = [];
|
|
|
|
for (let index = 0; index < totalCachedImageDtos.length; index++) {
|
|
const image = totalCachedImageDtos[index];
|
|
if (image?.starred) {
|
|
cachedStarredImages.push(image);
|
|
}
|
|
if (!image?.starred) {
|
|
cachedUnstarredImages.push(image);
|
|
}
|
|
}
|
|
|
|
if (imageDTO.starred) {
|
|
const lastStarredImage =
|
|
cachedStarredImages[cachedStarredImages.length - 1];
|
|
// if starring or already starred, want to look in list of starred images
|
|
if (!lastStarredImage) {
|
|
return true;
|
|
} // no starred images showing, so always show this one
|
|
const createdDate = new Date(imageDTO.created_at);
|
|
const oldestDate = new Date(lastStarredImage.created_at);
|
|
return createdDate >= oldestDate;
|
|
} else {
|
|
const lastUnstarredImage =
|
|
cachedUnstarredImages[cachedUnstarredImages.length - 1];
|
|
// if unstarring or already unstarred, want to look in list of unstarred images
|
|
if (!lastUnstarredImage) {
|
|
return false;
|
|
} // no unstarred images showing, so don't show this one
|
|
const createdDate = new Date(imageDTO.created_at);
|
|
const oldestDate = new Date(lastUnstarredImage.created_at);
|
|
return createdDate >= oldestDate;
|
|
}
|
|
};
|
|
|
|
export const getCategories = (imageDTO: ImageDTO) => {
|
|
if (IMAGE_CATEGORIES.includes(imageDTO.image_category)) {
|
|
return IMAGE_CATEGORIES;
|
|
}
|
|
return ASSETS_CATEGORIES;
|
|
};
|
|
|
|
// The adapter is not actually the data store - it just provides helper functions to interact
|
|
// with some other store of data. We will use the RTK Query cache as that store.
|
|
export const imagesAdapter = createEntityAdapter<ImageDTO>({
|
|
selectId: (image) => image.image_name,
|
|
sortComparer: (a, b) => {
|
|
// Compare starred images first
|
|
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.
|
|
export const imagesSelectors = imagesAdapter.getSelectors();
|
|
|
|
// Helper to create the url for the listImages endpoint. Also we use it to create the cache key.
|
|
export const getListImagesUrl = (queryArgs: ListImagesArgs) =>
|
|
`images/?${queryString.stringify(queryArgs, { arrayFormat: 'none' })}`;
|