@musnows/scriverse 0.5.9 → 0.5.11

package/dist/store.js

@@ -7,6 +7,7 @@ import { currentRequestActor } from "./request-context.js";
 import { classifyWorkModulePermissions, emptyWorkModulePermissions, fullWorkModulePermissions, storedWorkModulePermissions } from "./work-permissions.js";
 import { countWords, documentShortSearchTerms, id, json, normalizeDocumentSearchText, normalizeParagraphSpacing, now, splitDocumentParagraphs } from "./utils.js";
 const defaultPlatformPageSizes = {
+    drafts: 30,
     settings: 30,
     characters: 30,
     races: 30,
@@ -29,6 +30,7 @@ function platformPageSizes(value) {
             : defaultPlatformPageSizes[key];
     };
     return {
+        drafts: pageSize("drafts"),
         settings: pageSize("settings"),
         characters: pageSize("characters"),
         races: pageSize("races"),
@@ -110,6 +112,7 @@ function settingsFromKnowledgeSections(sections) {
 export const versionedEntityTypes = [
     "work",
     "volume",
+    "draft",
     "setting",
     "race",
     "organization",
@@ -237,6 +240,8 @@ export class Store {
             return this.getWork(entityId);
         if (type === "volume")
             return this.getVolume(entityId);
+        if (type === "draft")
+            return this.getDraft(entityId);
         if (type === "setting")
             return this.getSetting(entityId);
         if (type === "race")
@@ -287,6 +292,12 @@ export class Store {
                 keywords: entity.keywords,
                 sortOrder: entity.sortOrder
             };
+        if (type === "draft")
+            return {
+                draftType: entity.draftType,
+                title: entity.title,
+                content: entity.content
+            };
         if (type === "setting")
             return {
                 title: entity.title,
@@ -393,6 +404,7 @@ export class Store {
         const entities = [
             ...this.db.all("SELECT id, updated_at FROM works").map((row) => ["work", requiredString(row, "id"), requiredString(row, "updated_at")]),
             ...this.db.all("SELECT id, updated_at FROM volumes").map((row) => ["volume", requiredString(row, "id"), requiredString(row, "updated_at")]),
+            ...this.db.all("SELECT id, updated_at FROM drafts").map((row) => ["draft", requiredString(row, "id"), requiredString(row, "updated_at")]),
             ...this.db.all("SELECT id, updated_at FROM settings").map((row) => ["setting", requiredString(row, "id"), requiredString(row, "updated_at")]),
             ...this.db.all("SELECT id, updated_at FROM races").map((row) => ["race", requiredString(row, "id"), requiredString(row, "updated_at")]),
             ...this.db.all("SELECT id, updated_at FROM organizations").map((row) => ["organization", requiredString(row, "id"), requiredString(row, "updated_at")]),
@@ -474,6 +486,8 @@ export class Store {
             restored = this.updateWork(entityId, snapshot, expectedVersionNo, "restore", sourceRef, changeNote);
         else if (type === "volume")
             restored = this.updateVolume(entityId, snapshot, expectedVersionNo, "restore", sourceRef, changeNote);
+        else if (type === "draft")
+            restored = this.updateDraft(entityId, snapshot, "restore", sourceRef, changeNote, expectedVersionNo);
         else if (type === "setting")
             restored = this.updateSetting(entityId, snapshot, "restore", sourceRef, changeNote, expectedVersionNo);
         else if (type === "race")
@@ -513,6 +527,9 @@ export class Store {
         if (type === "volume") {
             return this.db.transaction(() => this.insertVolumeWithId(workId, entityId, snapshot, "restore", sourceRef, changeNote));
         }
+        if (type === "draft") {
+            return this.insertDraftWithId(workId, entityId, snapshot, "restore", sourceRef, changeNote);
+        }
         if (type === "setting") {
             return this.insertSettingWithId(workId, entityId, snapshot, "restore", sourceRef, changeNote);
         }
@@ -668,7 +685,7 @@ export class Store {
             autoRunConsecutiveFailures: Math.max(0, Number(row?.auto_run_consecutive_failures ?? 0) || 0),
             bookSummaryContextPercent: Math.min(90, Math.max(1, Number(row?.book_summary_context_percent ?? 50) || 50)),
             contextCompactThreshold: Math.min(90, Math.max(50, Number(row?.context_compact_threshold ?? 85) || 85)),
-            agentTools: json(String(row?.agent_tools_json ?? '["story_index","read_chapters","search_story_entities","grep","read_character_sections"]'), ["story_index", "read_chapters", "search_story_entities", "grep", "read_character_sections"])
+            agentTools: json(String(row?.agent_tools_json ?? '["story_index","read_chapters","search_story_entities","grep","read_character_sections","search_drafts"]'), ["story_index", "read_chapters", "search_story_entities", "grep", "read_character_sections", "search_drafts"])
                 .map((tool) => tool === "query_story_knowledge" ? "search_story_entities" : tool)
                 .filter((tool, index, tools) => tools.indexOf(tool) === index),
             updatedAt: String(row?.updated_at ?? "")
@@ -2163,6 +2180,94 @@ export class Store {
        FROM chapters c JOIN volumes v ON v.id = c.volume_id WHERE c.id = ? AND c.work_id = ?`, chapterId, workId);
         return row ? numberValue(row, "sequence") : Number.MAX_SAFE_INTEGER;
     }
+    createDraft(workId, input, source = "create", sourceRef = null) {
+        this.getWork(workId);
+        return this.insertDraftWithId(workId, id("draft"), input, source, sourceRef);
+    }
+    insertDraftWithId(workId, draftId, input, source = "create", sourceRef = null, changeNote = "") {
+        const timestamp = now();
+        this.db.transaction(() => {
+            this.db.run(`INSERT INTO drafts (id, work_id, draft_type, title, content, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?)`, draftId, workId, input.draftType, input.title, input.content, timestamp, timestamp);
+            this.syncMarkdownAttachmentReferences(workId, "draft", draftId, input.content);
+            this.recordEntityVersion("draft", draftId, source, sourceRef, changeNote || "建立创作草稿", timestamp);
+            this.audit(workId, source === "restore" ? "draft.restored" : "draft.created", "draft", draftId, {
+                draftType: input.draftType,
+                source,
+                sourceRef
+            });
+        });
+        return this.getDraft(draftId);
+    }
+    listDrafts(workId, draftType, includeContent = false) {
+        this.getWork(workId);
+        return this.db.all(`SELECT * FROM drafts WHERE work_id = ? AND (? IS NULL OR draft_type = ?)
+       ORDER BY updated_at DESC, title`, workId, draftType ?? null, draftType ?? null).map((row) => this.mapDraft(row, includeContent));
+    }
+    listDraftsPage(workId, pagination, draftType, includeContent = false) {
+        this.getWork(workId);
+        const page = paginationSql(pagination);
+        const rows = this.db.all(`SELECT * FROM drafts WHERE work_id = ? AND (? IS NULL OR draft_type = ?)
+       ORDER BY updated_at DESC, title${page.sql}`, workId, draftType ?? null, draftType ?? null, ...page.params);
+        return paginated(rows.map((row) => this.mapDraft(row, includeContent)), pagination);
+    }
+    searchDrafts(workId, query, draftType, limit = 20) {
+        this.getWork(workId);
+        const safeLimit = Math.min(30, Math.max(1, Math.trunc(limit)));
+        const normalizedQuery = query.normalize("NFKC").trim();
+        const escapedQuery = normalizedQuery.replace(/[\\%_]/gu, "\\$&");
+        const pattern = `%${escapedQuery}%`;
+        const rows = normalizedQuery
+            ? this.db.all(`SELECT * FROM drafts
+           WHERE work_id = ? AND (? IS NULL OR draft_type = ?)
+             AND (title LIKE ? ESCAPE '\\' COLLATE NOCASE OR content LIKE ? ESCAPE '\\' COLLATE NOCASE)
+           ORDER BY CASE WHEN title LIKE ? ESCAPE '\\' COLLATE NOCASE THEN 0 ELSE 1 END, updated_at DESC
+           LIMIT ?`, workId, draftType ?? null, draftType ?? null, pattern, pattern, pattern, safeLimit)
+            : this.db.all(`SELECT * FROM drafts WHERE work_id = ? AND (? IS NULL OR draft_type = ?)
+           ORDER BY updated_at DESC, title LIMIT ?`, workId, draftType ?? null, draftType ?? null, safeLimit);
+        return rows.map((row) => this.mapDraft(row, true));
+    }
+    getDraft(draftId) {
+        const row = this.db.get("SELECT * FROM drafts WHERE id = ?", draftId);
+        if (!row)
+            throw notFound("草稿");
+        return this.mapDraft(row, true);
+    }
+    updateDraft(draftId, input, source = "manual", sourceRef = null, changeNote = "", expectedVersionNo) {
+        const current = this.getDraft(draftId);
+        const content = input.content ?? String(current.content);
+        this.db.transaction(() => {
+            this.assertExpectedVersion("draft", draftId, expectedVersionNo, "草稿");
+            this.db.run("UPDATE drafts SET draft_type = ?, title = ?, content = ?, updated_at = ? WHERE id = ?", input.draftType ?? String(current.draftType), input.title ?? String(current.title), content, now(), draftId);
+            this.syncMarkdownAttachmentReferences(String(current.workId), "draft", draftId, content);
+            this.recordEntityVersion("draft", draftId, source, sourceRef, changeNote || "更新创作草稿");
+            this.audit(String(current.workId), "draft.updated", "draft", draftId, { fields: Object.keys(input), source, sourceRef });
+        });
+        return this.getDraft(draftId);
+    }
+    deleteDraft(draftId, expectedVersionNo) {
+        const current = this.getDraft(draftId);
+        this.db.transaction(() => {
+            this.assertExpectedVersion("draft", draftId, expectedVersionNo, "草稿");
+            this.recordEntityVersion("draft", draftId, "delete", null, "删除创作草稿");
+            this.clearMarkdownAttachmentReferences("draft", draftId);
+            this.db.run("DELETE FROM drafts WHERE id = ?", draftId);
+            this.audit(String(current.workId), "draft.deleted", "draft", draftId);
+        });
+    }
+    mapDraft(row, includeContent) {
+        const content = requiredString(row, "content");
+        return {
+            id: requiredString(row, "id"),
+            workId: requiredString(row, "work_id"),
+            draftType: requiredString(row, "draft_type"),
+            title: requiredString(row, "title"),
+            ...(includeContent ? { content } : { contentPreview: content.replace(/\s+/gu, " ").trim().slice(0, 320) }),
+            versionNo: this.currentEntityVersionNo("draft", requiredString(row, "id")),
+            createdAt: requiredString(row, "created_at"),
+            updatedAt: requiredString(row, "updated_at")
+        };
+    }
     createSetting(workId, input, source = "create", sourceRef = null) {
         this.getWork(workId);
         return this.insertSettingWithId(workId, id("setting"), input, source, sourceRef);
@@ -5445,9 +5550,10 @@ export class Store {
     exportWork(workId) {
         const tree = this.getWorkTree(workId);
         return {
-            schemaVersion: 7,
+            schemaVersion: 8,
             exportedAt: now(),
             work: tree,
+            drafts: this.listDrafts(workId, undefined, true),
             settings: this.listSettings(workId),
             characters: this.listCharacters(workId, true, true),
             races: this.listRaces(workId),