From 28916e69e03f88b3dfae9e5b8ba02c6b87991e7c Mon Sep 17 00:00:00 2001 From: kjqwer <2990346238@qq.com> Date: Sat, 15 Nov 2025 20:56:33 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=AF=8D=E5=BA=93=E4=B8=8D?= =?UTF-8?q?=E6=9B=B4=E6=96=B0=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/stores/promptStore.ts | 134 ++++++++++++++++++++++++++++++++++++-- 1 file changed, 130 insertions(+), 4 deletions(-) diff --git a/src/stores/promptStore.ts b/src/stores/promptStore.ts index d2b96cb..ce19c31 100644 --- a/src/stores/promptStore.ts +++ b/src/stores/promptStore.ts @@ -72,9 +72,25 @@ export const usePromptStore = defineStore('promptStore', { try { const bundle = JSON.parse(raw) as ExportBundle; if (bundle.dataset) { - this.dataset = bundle.dataset; + const diff = this.buildSnapshotDiff(deepClone(baseline!), bundle.dataset); + this.dataset = this.applyDiff(deepClone(baseline!), diff); + const newBundle: ExportBundle = { + version: 1, + savedAt: new Date().toISOString(), + customDiff: diff, + presets: bundle.presets || [], + extendedPresets: bundle.extendedPresets || [], + presetFolders: bundle.presetFolders || [], + presetManagement: bundle.presetManagement, + promptText: bundle.promptText, + selectedLang: bundle.selectedLang, + }; + localStorage.setItem(LS_KEY, JSON.stringify(newBundle)); } else if (bundle.customDiff) { - this.dataset = this.applyDiff(deepClone(baseline!), bundle.customDiff); + const purified = this.sanitizeDiff(bundle.customDiff); + this.dataset = this.applyDiff(deepClone(baseline!), purified); + const newBundle: ExportBundle = { ...bundle, customDiff: purified, savedAt: new Date().toISOString() }; + localStorage.setItem(LS_KEY, JSON.stringify(newBundle)); } else { this.dataset = deepClone(baseline!); } @@ -120,12 +136,12 @@ export const usePromptStore = defineStore('promptStore', { }, save() { if (!this.dataset) return; + const diff = baseline ? this.buildDiff(baseline, this.dataset) : { categories: [] }; const bundle: ExportBundle = { version: 1, savedAt: new Date().toISOString(), - dataset: deepClone(this.dataset), + customDiff: diff, presets: deepClone(this.presets), - // 扩展预设数据 extendedPresets: deepClone(this.extendedPresets), presetFolders: deepClone(this.presetFolders), presetManagement: deepClone(this.presetManagement), @@ -198,6 +214,34 @@ export const usePromptStore = defineStore('promptStore', { }, setSearch(q: string) { this.searchQuery = q; + const query = q.trim().toLowerCase(); + if (!query || !this.dataset) return; + const norm = (s: string) => s.toLowerCase().replace(/_/g, ' '); + for (let ci = 0; ci < this.dataset.categories.length; ci++) { + const cat = this.dataset.categories[ci]; + if (!cat) continue; + for (let gi = 0; gi < cat.groups.length; gi++) { + const grp = cat.groups[gi]; + if (!grp) continue; + for (const t of grp.tags) { + const keyLower = t.key.toLowerCase(); + const keyNorm = norm(t.key); + const trans = t.translation?.[this.selectedLang] ?? ''; + const transLower = trans.toLowerCase(); + const transNorm = norm(trans); + if ( + keyLower.includes(query) || + keyNorm.includes(query.replace(/_/g, ' ')) || + transLower.includes(query) || + transNorm.includes(query.replace(/_/g, ' ')) + ) { + this.selectedCategoryIndex = ci; + this.selectedGroupIndex = gi; + return; + } + } + } + } }, addTag(groupId: string, key = 'new_tag') { const grp = this.findGroupById(groupId); @@ -648,6 +692,88 @@ export const usePromptStore = defineStore('promptStore', { return target; }, + buildSnapshotDiff(base: PromptDataset, cur: PromptDataset): CustomDiff { + const categories: CustomDiff['categories'] = []; + const baseCatMap = new Map(base.categories.map((c) => [c.name, c])); + const curCatMap = new Map(cur.categories.map((c) => [c.name, c])); + for (const [name, curCat] of curCatMap.entries()) { + const baseCat = baseCatMap.get(name); + const catDiff: CustomDiff['categories'][number] = { name }; + if (!baseCat) { + catDiff.addedGroups = curCat.groups.map((g) => deepClone(g)); + } else { + const baseGrpMap = new Map(baseCat.groups.map((g) => [g.name, g])); + const curGrpMap = new Map(curCat.groups.map((g) => [g.name, g])); + const groupsDiff: NonNullable = []; + for (const [gname, curGrp] of curGrpMap.entries()) { + const baseGrp = baseGrpMap.get(gname); + if (!baseGrp) { + if (!catDiff.addedGroups) catDiff.addedGroups = []; + catDiff.addedGroups.push(deepClone(curGrp)); + continue; + } + const updated: { name: string; color?: string; added?: PromptTag[]; updated?: Array<{ key: string; translation?: Partial>; hidden?: boolean }>; order?: string[] } = { name: gname } as any; + if ((curGrp.color || '') !== (baseGrp.color || '')) updated.color = curGrp.color; + const baseTagMap = new Map(baseGrp.tags.map((t) => [t.key, t])); + const curTagMap = new Map(curGrp.tags.map((t) => [t.key, t])); + const addList: PromptTag[] = []; + const updList: Array<{ key: string; translation?: Partial>; hidden?: boolean }> = []; + for (const [key, curTag] of curTagMap.entries()) { + const baseTag = baseTagMap.get(key); + if (!baseTag) { + addList.push(deepClone(curTag)); + } else { + let changed = false; + const change: { key: string; translation?: Partial>; hidden?: boolean } = { key }; + const langs = Array.from(new Set([...(base.languages || []), ...(cur.languages || [])])); + for (const l of langs as LangCode[]) { + const a = baseTag.translation?.[l] ?? ''; + const b = curTag.translation?.[l] ?? ''; + if (a !== b) { + if (!change.translation) change.translation = {}; + change.translation[l] = b; + changed = true; + } + } + const aHidden = !!baseTag.hidden; + const bHidden = !!curTag.hidden; + if (aHidden !== bHidden) { change.hidden = bHidden; changed = true; } + if (changed) updList.push(change); + } + } + const baseOrder = baseGrp.tags.map((t) => t.key); + const curOrder = curGrp.tags.map((t) => t.key); + const orderChanged = baseOrder.length !== curOrder.length || baseOrder.some((k, i) => k !== curOrder[i]); + if (updated.color || addList.length || updList.length || orderChanged) { + if (updated.color) updated.color = updated.color; + if (addList.length) updated.added = addList; + if (updList.length) updated.updated = updList; + if (orderChanged) updated.order = curOrder; + groupsDiff.push(updated as any); + } + } + if (groupsDiff.length) catDiff.groups = groupsDiff; + } + if (catDiff.addedGroups?.length || catDiff.groups?.length) { + categories.push(catDiff); + } + } + return { categories }; + }, + + sanitizeDiff(diff: CustomDiff): CustomDiff { + const clean: CustomDiff = { categories: [] }; + for (const c of diff.categories || []) { + const entry: any = { name: c.name }; + if (c.addedGroups && c.addedGroups.length) entry.addedGroups = c.addedGroups; + if (c.groups && c.groups.length) { + entry.groups = c.groups.map((g) => ({ name: g.name, color: g.color, added: g.added, updated: g.updated, order: g.order })); + } + if (entry.addedGroups || entry.groups) clean.categories.push(entry); + } + return clean; + }, + // 扩展预设管理方法 initializeExtendedPresets() { // 如果没有扩展预设数据，初始化默认结构