@musnows/scriverse 0.5.10 → 0.5.12

package/dist/store.js

@@ -6,7 +6,9 @@ import { paginated, paginationSql } from "./pagination.js";
 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";
+import { buildWritingCalendar, writingDateKey } from "./writing-progress-time.js";
 const defaultPlatformPageSizes = {
+    drafts: 30,
     settings: 30,
     characters: 30,
     races: 30,
@@ -29,6 +31,7 @@ function platformPageSizes(value) {
             : defaultPlatformPageSizes[key];
     };
     return {
+        drafts: pageSize("drafts"),
         settings: pageSize("settings"),
         characters: pageSize("characters"),
         races: pageSize("races"),
@@ -110,6 +113,7 @@ function settingsFromKnowledgeSections(sections) {
 export const versionedEntityTypes = [
     "work",
     "volume",
+    "draft",
     "setting",
     "race",
     "organization",
@@ -237,6 +241,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 +293,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 +405,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 +487,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 +528,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 +686,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 ?? "")
@@ -874,6 +892,41 @@ export class Store {
         }));
         return { ...work, volumes };
     }
+    getWorkVolumeDirectory(workId) {
+        const work = this.getWork(workId);
+        const permissions = work.modulePermissions;
+        if (permissions.prose === "none")
+            return { ...work, volumes: [] };
+        const volumeRows = this.db.all(`SELECT volume.*,
+        (SELECT COUNT(*) FROM chapters chapter WHERE chapter.volume_id = volume.id AND chapter.deleted_at IS NULL) AS chapter_count
+       FROM volumes volume WHERE volume.work_id = ? ORDER BY volume.sort_order, volume.created_at`, workId);
+        const volumes = volumeRows.map((row) => ({
+            ...this.mapVolume(row),
+            chapterCount: numberValue(row, "chapter_count"),
+            chapters: []
+        }));
+        return { ...work, volumes };
+    }
+    listVolumeChapters(volumeId) {
+        const volume = this.getVolume(volumeId);
+        const work = this.getWork(String(volume.workId));
+        if (work.modulePermissions.prose === "none")
+            return [];
+        return this.db.all(`SELECT id, work_id, volume_id, title, chapter_type, sort_order, word_count, version_no,
+        analysis_status, excluded_from_analysis, created_at, updated_at
+       FROM chapters WHERE volume_id = ? AND deleted_at IS NULL ORDER BY sort_order, created_at`, volumeId).map((row) => this.mapChapterDirectoryEntry(row));
+    }
+    listVolumeChaptersPage(volumeId, pagination) {
+        const volume = this.getVolume(volumeId);
+        const work = this.getWork(String(volume.workId));
+        if (work.modulePermissions.prose === "none")
+            return paginated([], pagination);
+        const page = paginationSql(pagination);
+        const rows = this.db.all(`SELECT id, work_id, volume_id, title, chapter_type, sort_order, word_count, version_no,
+        analysis_status, excluded_from_analysis, created_at, updated_at
+       FROM chapters WHERE volume_id = ? AND deleted_at IS NULL ORDER BY sort_order, created_at${page.sql}`, volumeId, ...page.params);
+        return paginated(rows.map((row) => this.mapChapterDirectoryEntry(row)), pagination);
+    }
     getWorkDirectoryPage(workId, pagination) {
         const work = this.getWork(workId);
         const permissions = work.modulePermissions;
@@ -2163,6 +2216,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);
@@ -4020,16 +4161,26 @@ export class Store {
         const conversation = this.db.get("SELECT * FROM ai_conversations WHERE id = ?", conversationId);
         if (!conversation)
             throw notFound("AI 对话");
+        const requestId = input.requestId?.trim() || null;
+        if (requestId) {
+            const existing = this.db.get("SELECT * FROM ai_conversation_messages WHERE conversation_id = ? AND request_id = ?", conversationId, requestId);
+            if (existing)
+                return this.mapAiConversationMessage(existing);
+        }
         const messageId = id("message");
         const timestamp = now();
         const title = requiredString(conversation, "title") === "新对话" && input.role === "user"
             ? input.content.replace(/\s+/gu, " ").trim().slice(0, 36) || "新对话"
             : requiredString(conversation, "title");
         this.db.transaction(() => {
-            this.db.run("INSERT INTO ai_conversation_messages (id, conversation_id, role, content, citations_json, metadata_json, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", messageId, conversationId, input.role, input.content, JSON.stringify(input.citations ?? []), JSON.stringify(input.metadata ?? {}), timestamp, currentRequestActor()?.userId ?? null);
-            this.db.run("UPDATE ai_conversations SET title = ?, updated_at = ? WHERE id = ?", title, timestamp, conversationId);
+            this.db.run("INSERT INTO ai_conversation_messages (id, conversation_id, role, content, citations_json, metadata_json, request_id, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) ON CONFLICT(conversation_id, request_id) WHERE request_id IS NOT NULL DO NOTHING", messageId, conversationId, input.role, input.content, JSON.stringify(input.citations ?? []), JSON.stringify(input.metadata ?? {}), requestId, timestamp, currentRequestActor()?.userId ?? null);
+            const inserted = this.db.get("SELECT id FROM ai_conversation_messages WHERE id = ?", messageId);
+            if (inserted)
+                this.db.run("UPDATE ai_conversations SET title = ?, updated_at = ? WHERE id = ?", title, timestamp, conversationId);
         });
-        const message = this.db.get("SELECT * FROM ai_conversation_messages WHERE id = ?", messageId);
+        const message = requestId
+            ? this.db.get("SELECT * FROM ai_conversation_messages WHERE conversation_id = ? AND request_id = ?", conversationId, requestId)
+            : this.db.get("SELECT * FROM ai_conversation_messages WHERE id = ?", messageId);
         if (!message)
             throw notFound("AI 对话消息");
         return this.mapAiConversationMessage(message);
@@ -4052,7 +4203,7 @@ export class Store {
         this.db.transaction(() => {
             this.db.run("INSERT INTO ai_conversations (id, work_id, title, compacted_summary, compacted_message_count, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", forkId, requiredString(conversation, "work_id"), title.slice(0, 200), forkSummary, forkCompactedCount, timestamp, timestamp, currentRequestActor()?.userId ?? null);
             for (const message of messages.slice(0, targetIndex + 1)) {
-                this.db.run("INSERT INTO ai_conversation_messages (id, conversation_id, role, content, citations_json, metadata_json, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?)", id("message"), forkId, requiredString(message, "role"), requiredString(message, "content"), requiredString(message, "citations_json"), requiredString(message, "metadata_json"), requiredString(message, "created_at"), currentRequestActor()?.userId ?? null);
+                this.db.run("INSERT INTO ai_conversation_messages (id, conversation_id, role, content, citations_json, metadata_json, request_id, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", id("message"), forkId, requiredString(message, "role"), requiredString(message, "content"), requiredString(message, "citations_json"), requiredString(message, "metadata_json"), optionalString(message, "request_id"), requiredString(message, "created_at"), currentRequestActor()?.userId ?? null);
             }
         });
         return this.getAiConversation(forkId);
@@ -4079,6 +4230,7 @@ export class Store {
             content: requiredString(row, "content"),
             citations: json(requiredString(row, "citations_json"), []),
             metadata: json(requiredString(row, "metadata_json"), {}),
+            requestId: optionalString(row, "request_id"),
             createdAt: requiredString(row, "created_at")
         };
     }
@@ -5445,9 +5597,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),
@@ -5509,18 +5662,14 @@ export class Store {
         const goal = this.db.get("SELECT * FROM writing_goals WHERE work_id = ?", workId);
         const dailyGoal = goal ? numberValue(goal, "daily_goal") : 1000;
         const targetTotal = goal ? numberValue(goal, "target_total") : 100000;
-        const today = new Date();
-        today.setUTCHours(0, 0, 0, 0);
-        const start = new Date(today);
-        start.setUTCDate(start.getUTCDate() - days + 1);
-        const startKey = start.toISOString().slice(0, 10);
+        const calendar = buildWritingCalendar(new Date(), days);
         const versions = this.db.all(`SELECT chapter_id, content, source, created_at FROM chapter_versions
-       WHERE work_id = ? AND created_at <= ? ORDER BY created_at, version_no, id`, workId, `${today.toISOString().slice(0, 10)}T23:59:59.999Z`);
+       WHERE work_id = ? AND created_at < ? ORDER BY created_at, version_no, id`, workId, calendar.endExclusive);
         const chapterWords = new Map();
         const events = new Map();
         for (const version of versions) {
-            const day = requiredString(version, "created_at").slice(0, 10);
-            if (day < startKey) {
+            const day = writingDateKey(new Date(requiredString(version, "created_at")), calendar.timeZone);
+            if (day < calendar.startKey) {
                 chapterWords.set(requiredString(version, "chapter_id"), requiredString(version, "source") === "delete" ? 0 : countWords(requiredString(version, "content")));
             }
             else {
@@ -5531,10 +5680,7 @@ export class Store {
         }
         let previousTotal = [...chapterWords.values()].reduce((sum, value) => sum + value, 0);
         const trend = [];
-        for (let index = 0; index < days; index += 1) {
-            const date = new Date(start);
-            date.setUTCDate(start.getUTCDate() + index);
-            const day = date.toISOString().slice(0, 10);
+        for (const day of calendar.dateKeys) {
             for (const version of events.get(day) ?? []) {
                 chapterWords.set(requiredString(version, "chapter_id"), requiredString(version, "source") === "delete" ? 0 : countWords(requiredString(version, "content")));
             }