dsh-multimodal-node 0.0.0-stage → 0.2.2

package/lib/promptlibrary.js

@@ -0,0 +1,92 @@
+/**
+ * Saved prompts: a small per-account library of prompts a person wants to reuse.
+ *
+ * Why a separate file and not a workflow: a saved prompt is not a graph. It is the text a
+ * person typed (plus the negatives), reusable across workflows — the whole point of
+ * favouriting "白衬衫, 雨夜, 便利店" is using it on more than the graph that produced it.
+ *
+ * Storage shape is `{ byOwner: { <user>: [ {id, name, positive, negative, layers[], at} ] } }`,
+ * written atomically the same way the run index is, so a crash mid-save cannot truncate
+ * someone else's list.
+ */
+import { mkdir, readFile, rename, writeFile } from 'node:fs/promises';
+import { join } from 'node:path';
+import { randomBytes } from 'node:crypto';
+
+const MAX_PROMPTS = 200;
+
+export class PromptLibrary {
+    constructor(dataDir) {
+        this.file = join(dataDir, 'prompts.json');
+        this.byOwner = {};
+    }
+
+    async init() {
+        await mkdir(join(this.file, '..'), { recursive: true });
+        try {
+            const raw = await readFile(this.file, 'utf8');
+            const parsed = JSON.parse(raw);
+            this.byOwner = parsed !== null && typeof parsed.byOwner === 'object' && parsed.byOwner !== null ? parsed.byOwner : {};
+        } catch {
+            this.byOwner = {};
+            await this.save();
+        }
+    }
+
+    async save() {
+        const temp = `${this.file}.tmp`;
+        await writeFile(temp, `${JSON.stringify({ byOwner: this.byOwner }, null, 2)}\n`, 'utf8');
+        await rename(temp, this.file);
+    }
+
+    /** One account's prompts, newest first. */
+    list(owner) {
+        const rows = Array.isArray(this.byOwner[owner]) ? this.byOwner[owner] : [];
+        return rows.slice().sort((left, right) => (right.at ?? 0) - (left.at ?? 0));
+    }
+
+    /**
+     * Save one prompt. Saving the same name twice updates that entry instead of piling up
+     * duplicates — a person who fixes a typo in a favourite does not want two of them.
+     */
+    async add(owner, input) {
+        const name = String(input?.name ?? '').trim().slice(0, 60);
+        if (name === '') throw new Error('收藏需要一个名字');
+        const positive = String(input?.positive ?? '').trim();
+        const negative = String(input?.negative ?? '').trim();
+        const layers = Array.isArray(input?.layers) ? input.layers.map((v) => String(v ?? '')) : [];
+        if (positive === '' && negative === '' && layers.every((v) => v === '')) {
+            throw new Error('提示词是空的，没有可收藏的内容');
+        }
+        const rows = Array.isArray(this.byOwner[owner]) ? this.byOwner[owner] : [];
+        const existing = rows.find((row) => row.name === name);
+        const record = {
+            id: existing?.id ?? `p-${randomBytes(4).toString('hex')}`,
+            name,
+            positive,
+            negative,
+            layers,
+            at: Date.now(),
+        };
+        if (existing === undefined) {
+            rows.unshift(record);
+            // Newest-first order is maintained on read, but cap the stored list so a
+            // lifetime of favourites cannot grow the file without bound.
+            this.byOwner[owner] = rows.slice(0, MAX_PROMPTS);
+        } else {
+            rows[rows.indexOf(existing)] = record;
+            this.byOwner[owner] = rows;
+        }
+        await this.save();
+        return record;
+    }
+
+    async remove(owner, id) {
+        const rows = Array.isArray(this.byOwner[owner]) ? this.byOwner[owner] : [];
+        const next = rows.filter((row) => row.id !== id);
+        if (next.length === rows.length) throw new Error('没有这条收藏');
+        this.byOwner[owner] = next;
+        await this.save();
+        return { ok: true };
+    }
+}