forked from akai/readest
Huang Xin
71 lines
2.5 KiB
TypeScript
71 lines
2.5 KiB
TypeScript
import { create } from 'zustand';
|
|
import { Book, BooksGroup } from '@/types/book';
|
|
import { EnvConfigType, isTauriAppPlatform } from '@/services/environment';
|
|
|
|
interface LibraryState {
|
|
library: Book[]; // might contain deleted books
|
|
isSyncing: boolean;
|
|
syncProgress: number;
|
|
checkOpenWithBooks: boolean;
|
|
checkLastOpenBooks: boolean;
|
|
currentBookshelf: (Book | BooksGroup)[];
|
|
selectedBooks: Set<string>; // hashes for books, ids for groups
|
|
setIsSyncing: (syncing: boolean) => void;
|
|
setSyncProgress: (progress: number) => void;
|
|
setSelectedBooks: (ids: string[]) => void;
|
|
getSelectedBooks: () => string[];
|
|
toggleSelectedBook: (id: string) => void;
|
|
getVisibleLibrary: () => Book[];
|
|
setCheckOpenWithBooks: (check: boolean) => void;
|
|
setCheckLastOpenBooks: (check: boolean) => void;
|
|
setLibrary: (books: Book[]) => void;
|
|
updateBook: (envConfig: EnvConfigType, book: Book) => void;
|
|
setCurrentBookshelf: (bookshelf: (Book | BooksGroup)[]) => void;
|
|
}
|
|
|
|
export const useLibraryStore = create<LibraryState>((set, get) => ({
|
|
library: [],
|
|
isSyncing: false,
|
|
syncProgress: 0,
|
|
currentBookshelf: [],
|
|
selectedBooks: new Set(),
|
|
checkOpenWithBooks: isTauriAppPlatform(),
|
|
checkLastOpenBooks: isTauriAppPlatform(),
|
|
setIsSyncing: (syncing: boolean) => set({ isSyncing: syncing }),
|
|
setSyncProgress: (progress: number) => set({ syncProgress: progress }),
|
|
getVisibleLibrary: () => get().library.filter((book) => !book.deletedAt),
|
|
setCurrentBookshelf: (bookshelf: (Book | BooksGroup)[]) => {
|
|
set({ currentBookshelf: bookshelf });
|
|
},
|
|
setCheckOpenWithBooks: (check) => set({ checkOpenWithBooks: check }),
|
|
setCheckLastOpenBooks: (check) => set({ checkLastOpenBooks: check }),
|
|
setLibrary: (books) => set({ library: books }),
|
|
updateBook: async (envConfig: EnvConfigType, book: Book) => {
|
|
const appService = await envConfig.getAppService();
|
|
const { library } = get();
|
|
const bookIndex = library.findIndex((b) => b.hash === book.hash);
|
|
if (bookIndex !== -1) {
|
|
library[bookIndex] = book;
|
|
}
|
|
set({ library: [...library] });
|
|
await appService.saveLibraryBooks(library);
|
|
},
|
|
setSelectedBooks: (ids: string[]) => {
|
|
set({ selectedBooks: new Set(ids) });
|
|
},
|
|
getSelectedBooks: () => {
|
|
return Array.from(get().selectedBooks);
|
|
},
|
|
toggleSelectedBook: (id: string) => {
|
|
set((state) => {
|
|
const newSelection = new Set(state.selectedBooks);
|
|
if (newSelection.has(id)) {
|
|
newSelection.delete(id);
|
|
} else {
|
|
newSelection.add(id);
|
|
}
|
|
return { selectedBooks: newSelection };
|
|
});
|
|
},
|
|
}));
|