|
|
@@ -14,8 +14,8 @@ export interface Prompt {
|
|
|
|
|
|
export const SearchService = {
|
|
|
ready: false,
|
|
|
- builtinEngine: new Fuse<Prompt>([], { keys: ["title"] }),
|
|
|
- userEngine: new Fuse<Prompt>([], { keys: ["title"] }),
|
|
|
+ builtinEngine: new Fuse<Prompt>([], {keys: [ "title" ]}),
|
|
|
+ userEngine: new Fuse<Prompt>([], {keys: [ "title" ]}),
|
|
|
count: {
|
|
|
builtin: 0,
|
|
|
},
|
|
|
@@ -23,7 +23,7 @@ export const SearchService = {
|
|
|
builtinPrompts: [] as Prompt[],
|
|
|
|
|
|
init(builtinPrompts: Prompt[], userPrompts: Prompt[]) {
|
|
|
- if (this.ready) {
|
|
|
+ if ( this.ready ) {
|
|
|
return;
|
|
|
}
|
|
|
this.allPrompts = userPrompts.concat(builtinPrompts);
|
|
|
@@ -49,136 +49,140 @@ export const SearchService = {
|
|
|
};
|
|
|
|
|
|
export const usePromptStore = createPersistStore(
|
|
|
- {
|
|
|
- counter: 0,
|
|
|
- prompts: {} as Record<string, Prompt>,
|
|
|
- },
|
|
|
-
|
|
|
- (set, get) => ({
|
|
|
- add(prompt: Prompt) {
|
|
|
- const prompts = get().prompts;
|
|
|
- prompt.id = nanoid();
|
|
|
- prompt.isUser = true;
|
|
|
- prompt.createdAt = Date.now();
|
|
|
- prompts[prompt.id] = prompt;
|
|
|
-
|
|
|
- set(() => ({
|
|
|
- prompts: prompts,
|
|
|
- }));
|
|
|
-
|
|
|
- return prompt.id!;
|
|
|
+ {
|
|
|
+ counter: 0,
|
|
|
+ prompts: {} as Record<string, Prompt>,
|
|
|
},
|
|
|
|
|
|
- get(id: string) {
|
|
|
- const targetPrompt = get().prompts[id];
|
|
|
+ (set, get) => ({
|
|
|
+ add(prompt: Prompt) {
|
|
|
+ const prompts = get().prompts;
|
|
|
+ prompt.id = nanoid();
|
|
|
+ prompt.isUser = true;
|
|
|
+ prompt.createdAt = Date.now();
|
|
|
+ prompts[prompt.id] = prompt;
|
|
|
|
|
|
- if (!targetPrompt) {
|
|
|
- return SearchService.builtinPrompts.find((v) => v.id === id);
|
|
|
- }
|
|
|
+ set(() => ({
|
|
|
+ prompts: prompts,
|
|
|
+ }));
|
|
|
|
|
|
- return targetPrompt;
|
|
|
- },
|
|
|
+ return prompt.id!;
|
|
|
+ },
|
|
|
|
|
|
- remove(id: string) {
|
|
|
- const prompts = get().prompts;
|
|
|
- delete prompts[id];
|
|
|
+ get(id: string) {
|
|
|
+ const targetPrompt = get().prompts[id];
|
|
|
|
|
|
- Object.entries(prompts).some(([key, prompt]) => {
|
|
|
- if (prompt.id === id) {
|
|
|
- delete prompts[key];
|
|
|
- return true;
|
|
|
+ if ( !targetPrompt ) {
|
|
|
+ return SearchService.builtinPrompts.find((v) => v.id === id);
|
|
|
}
|
|
|
- return false;
|
|
|
- });
|
|
|
-
|
|
|
- SearchService.remove(id);
|
|
|
|
|
|
- set(() => ({
|
|
|
- prompts,
|
|
|
- counter: get().counter + 1,
|
|
|
- }));
|
|
|
- },
|
|
|
-
|
|
|
- getUserPrompts() {
|
|
|
- const userPrompts = Object.values(get().prompts ?? {});
|
|
|
- userPrompts.sort((a, b) =>
|
|
|
- b.id && a.id ? b.createdAt - a.createdAt : 0,
|
|
|
- );
|
|
|
- return userPrompts;
|
|
|
- },
|
|
|
+ return targetPrompt;
|
|
|
+ },
|
|
|
|
|
|
- updatePrompt(id: string, updater: (prompt: Prompt) => void) {
|
|
|
- const prompt = get().prompts[id] ?? {
|
|
|
- title: "",
|
|
|
- content: "",
|
|
|
- id,
|
|
|
- };
|
|
|
-
|
|
|
- SearchService.remove(id);
|
|
|
- updater(prompt);
|
|
|
- const prompts = get().prompts;
|
|
|
- prompts[id] = prompt;
|
|
|
- set(() => ({ prompts }));
|
|
|
- SearchService.add(prompt);
|
|
|
- },
|
|
|
-
|
|
|
- search(text: string) {
|
|
|
- if (text.length === 0) {
|
|
|
- // return all rompts
|
|
|
- return this.getUserPrompts().concat(SearchService.builtinPrompts);
|
|
|
- }
|
|
|
- return SearchService.search(text) as Prompt[];
|
|
|
- },
|
|
|
- }),
|
|
|
- {
|
|
|
- name: StoreKey.Prompt,
|
|
|
- version: 3,
|
|
|
+ remove(id: string) {
|
|
|
+ const prompts = get().prompts;
|
|
|
+ delete prompts[id];
|
|
|
|
|
|
- migrate(state, version) {
|
|
|
- const newState = JSON.parse(JSON.stringify(state)) as {
|
|
|
- prompts: Record<string, Prompt>;
|
|
|
- };
|
|
|
-
|
|
|
- if (version < 3) {
|
|
|
- Object.values(newState.prompts).forEach((p) => (p.id = nanoid()));
|
|
|
- }
|
|
|
-
|
|
|
- return newState as any;
|
|
|
- },
|
|
|
-
|
|
|
- onRehydrateStorage(state) {
|
|
|
- const PROMPT_URL = "./prompts.json";
|
|
|
-
|
|
|
- type PromptList = Array<[string, string]>;
|
|
|
-
|
|
|
- fetch(PROMPT_URL)
|
|
|
- .then((res) => res.json())
|
|
|
- .then((res) => {
|
|
|
- let fetchPrompts = [res.en, res.tw, res.cn];
|
|
|
- if (getLang() === "cn") {
|
|
|
- fetchPrompts = fetchPrompts.reverse();
|
|
|
+ Object.entries(prompts).some(([ key, prompt ]) => {
|
|
|
+ if ( prompt.id === id ) {
|
|
|
+ delete prompts[key];
|
|
|
+ return true;
|
|
|
}
|
|
|
- const builtinPrompts = fetchPrompts.filter(item => item).map((promptList: PromptList) => {
|
|
|
- return promptList.map(
|
|
|
- ([title, content]) =>
|
|
|
- ({
|
|
|
- id: nanoid(),
|
|
|
- title,
|
|
|
- content,
|
|
|
- createdAt: Date.now(),
|
|
|
- }) as Prompt,
|
|
|
- );
|
|
|
- });
|
|
|
-
|
|
|
- const userPrompts = usePromptStore.getState().getUserPrompts() ?? [];
|
|
|
-
|
|
|
- const allPromptsForSearch = builtinPrompts
|
|
|
- .reduce((pre, cur) => pre.concat(cur), [])
|
|
|
- .filter((v) => !!v.title && !!v.content);
|
|
|
- SearchService.count.builtin =
|
|
|
- res.en.length + res.cn.length + res.tw.length;
|
|
|
- SearchService.init(allPromptsForSearch, userPrompts);
|
|
|
+ return false;
|
|
|
});
|
|
|
+
|
|
|
+ SearchService.remove(id);
|
|
|
+
|
|
|
+ set(() => ({
|
|
|
+ prompts,
|
|
|
+ counter: get().counter + 1,
|
|
|
+ }));
|
|
|
+ },
|
|
|
+
|
|
|
+ getUserPrompts() {
|
|
|
+ const userPrompts = Object.values(get().prompts ?? {});
|
|
|
+ userPrompts.sort((a, b) =>
|
|
|
+ b.id && a.id ? b.createdAt - a.createdAt : 0,
|
|
|
+ );
|
|
|
+ return userPrompts;
|
|
|
+ },
|
|
|
+
|
|
|
+ updatePrompt(id: string, updater: (prompt: Prompt) => void) {
|
|
|
+ const prompt = get().prompts[id] ?? {
|
|
|
+ title: "",
|
|
|
+ content: "",
|
|
|
+ id,
|
|
|
+ };
|
|
|
+
|
|
|
+ SearchService.remove(id);
|
|
|
+ updater(prompt);
|
|
|
+ const prompts = get().prompts;
|
|
|
+ prompts[id] = prompt;
|
|
|
+ set(() => ({prompts}));
|
|
|
+ SearchService.add(prompt);
|
|
|
+ },
|
|
|
+
|
|
|
+ search(text: string) {
|
|
|
+ if ( text.length === 0 ) {
|
|
|
+ // return all rompts
|
|
|
+ return this.getUserPrompts().concat(SearchService.builtinPrompts);
|
|
|
+ }
|
|
|
+ return SearchService.search(text) as Prompt[];
|
|
|
+ },
|
|
|
+ }),
|
|
|
+ {
|
|
|
+ name: StoreKey.Prompt,
|
|
|
+ version: 3,
|
|
|
+
|
|
|
+ migrate(state, version) {
|
|
|
+ const newState = JSON.parse(JSON.stringify(state)) as {
|
|
|
+ prompts: Record<string, Prompt>;
|
|
|
+ };
|
|
|
+
|
|
|
+ if ( version < 3 ) {
|
|
|
+ Object.values(newState.prompts).forEach((p) => (p.id = nanoid()));
|
|
|
+ }
|
|
|
+
|
|
|
+ return newState as any;
|
|
|
+ },
|
|
|
+
|
|
|
+ onRehydrateStorage(state) {
|
|
|
+ const PROMPT_URL = "./prompts.json";
|
|
|
+
|
|
|
+ type PromptList = Array<[ string, string ]>; // [标题, 内容]的元组数组
|
|
|
+ // tw: Array<[string, string]>; // 已停用繁体中文提示词
|
|
|
+ // en: Array<[string, string]>; // 已停用英文提示词
|
|
|
+
|
|
|
+ fetch(PROMPT_URL)
|
|
|
+ .then((res) => res.json())
|
|
|
+ .then((res) => {
|
|
|
+ let fetchPrompts = [
|
|
|
+ // res.en, res.tw, // 注释掉英文和繁体中文提示词
|
|
|
+ res.cn ];
|
|
|
+ // if ( getLang() === "cn" ) {
|
|
|
+ // fetchPrompts = fetchPrompts.reverse();
|
|
|
+ // }
|
|
|
+ const builtinPrompts = fetchPrompts.filter(item => item).map((promptList: PromptList) => {
|
|
|
+ return promptList.map(
|
|
|
+ ([ title, content ]) =>
|
|
|
+ ({
|
|
|
+ id: nanoid(),
|
|
|
+ title,
|
|
|
+ content,
|
|
|
+ createdAt: Date.now(),
|
|
|
+ }) as Prompt,
|
|
|
+ );
|
|
|
+ });
|
|
|
+
|
|
|
+ const userPrompts = usePromptStore.getState().getUserPrompts() ?? [];
|
|
|
+
|
|
|
+ const allPromptsForSearch = builtinPrompts
|
|
|
+ .reduce((pre, cur) => pre.concat(cur), [])
|
|
|
+ .filter((v) => !!v.title && !!v.content);
|
|
|
+ SearchService.count.builtin =
|
|
|
+ res.cn.length; // 仅计算中文提示词数量
|
|
|
+ SearchService.init(allPromptsForSearch, userPrompts);
|
|
|
+ });
|
|
|
+ },
|
|
|
},
|
|
|
- },
|
|
|
);
|
Surface