@musnows/scriverse 0.6.1 → 0.6.2

package/dist/store.js

@@ -1,3 +1,4 @@
+import { DRAFT_SETTING_MODULES } from "./domain.js";
 import { createHash } from "node:crypto";
 import { PLATFORM_AI_WORK_ID } from "./database.js";
 import { AppError, notFound } from "./errors.js";
@@ -7,6 +8,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";
 import { buildWritingCalendar, writingDateKey } from "./writing-progress-time.js";
+export const attachmentPermissionModules = ["prose", "drafts", "settings", "characters", "races", "organizations"];
 const defaultPlatformPageSizes = {
     drafts: 30,
     settings: 30,
@@ -302,6 +304,8 @@ export class Store {
         if (type === "draft")
             return {
                 draftType: entity.draftType,
+                volumeId: entity.volumeId,
+                settingModule: entity.settingModule,
                 title: entity.title,
                 content: entity.content
             };
@@ -817,13 +821,23 @@ export class Store {
         const storageKeys = this.db.all("SELECT DISTINCT storage_key FROM attachments WHERE work_id = ?", workId)
             .map((row) => requiredString(row, "storage_key"));
         this.db.transaction(() => {
+            this.db.raw.exec("PRAGMA defer_foreign_keys = ON");
             const current = this.getWork(workId);
             this.assertExpectedVersion("work", workId, expectedVersionNo, "作品", Number(current.versionNo));
             this.recordEntityVersion("work", workId, "delete", null, "删除作品");
             this.audit(null, "work.deleted", "work", workId, { title: work.title });
+            this.db.run("DELETE FROM characters WHERE work_id = ?", workId);
+            this.db.run("DELETE FROM organizations WHERE work_id = ?", workId);
+            this.db.run("UPDATE races SET parent_race_id = NULL WHERE work_id = ? AND parent_race_id IS NOT NULL", workId);
+            this.db.run("DELETE FROM races WHERE work_id = ?", workId);
             this.db.run("DELETE FROM works WHERE id = ?", workId);
+            this.db.run("DELETE FROM relationship_source_index_queue WHERE work_id = ?", workId);
+            for (const storageKey of storageKeys) {
+                if (!this.attachmentStorageKeyInUse(storageKey))
+                    this.enqueueAttachmentCleanup(storageKey);
+            }
         });
-        return storageKeys.filter((storageKey) => Number(this.db.get("SELECT COUNT(*) AS count FROM attachments WHERE storage_key = ?", storageKey)?.count ?? 0) === 0);
+        return storageKeys.filter((storageKey) => !this.attachmentStorageKeyInUse(storageKey));
     }
     setWorkCover(workId, mimeType, content, expectedVersionNo) {
         const sha256 = createHash("sha256").update(content).digest("hex");
@@ -1026,6 +1040,7 @@ export class Store {
             for (const row of this.db.all("SELECT id FROM volumes WHERE work_id = ?", workId)) {
                 this.recordEntityVersion("volume", requiredString(row, "id"), "delete", fileVersionId, "替换作品树前保存分卷历史");
             }
+            this.clearDraftVolumeBindings(workId, null, fileVersionId, "恢复文件版本时原分卷已被替换");
             this.db.run("DELETE FROM volumes WHERE work_id = ?", workId);
             for (const volume of volumes) {
                 const volumeId = id("volume");
@@ -1077,6 +1092,7 @@ export class Store {
             for (const row of this.db.all("SELECT id FROM volumes WHERE work_id = ?", workId)) {
                 this.recordEntityVersion("volume", requiredString(row, "id"), "delete", fileVersionId, "导入前保存分卷历史");
             }
+            this.clearDraftVolumeBindings(workId, null, fileVersionId, "覆盖导入时原分卷已被替换");
             this.db.run("DELETE FROM volumes WHERE work_id = ?", workId);
         }
         else {
@@ -1163,6 +1179,7 @@ export class Store {
                 throw new AppError(409, "VOLUME_HAS_DELETED_CHAPTERS", "分卷回收站中仍有章节，请先彻底删除或恢复并移动这些章节");
             }
             this.recordEntityVersion("volume", volumeId, "delete", null, "删除分卷");
+            this.clearDraftVolumeBindings(String(current.workId), [volumeId], null, "绑定的分卷已删除");
             this.db.run("DELETE FROM volumes WHERE id = ?", volumeId);
             this.audit(String(current.workId), "volume.deleted", "volume", volumeId, { versionNo: Number(current.versionNo) });
         });
@@ -2276,9 +2293,10 @@ export class Store {
     }
     insertDraftWithId(workId, draftId, input, source = "create", sourceRef = null, changeNote = "") {
         const timestamp = now();
+        const binding = this.normalizeDraftBinding(workId, input.draftType, input.volumeId ?? null, input.settingModule ?? null);
         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.db.run(`INSERT INTO drafts (id, work_id, draft_type, volume_id, setting_module, title, content, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, draftId, workId, input.draftType, binding.volumeId, binding.settingModule, 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, {
@@ -2291,14 +2309,18 @@ export class Store {
     }
     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));
+        return this.db.all(`SELECT draft.*, volume.title AS volume_title FROM drafts draft
+       LEFT JOIN volumes volume ON volume.id = draft.volume_id
+       WHERE draft.work_id = ? AND (? IS NULL OR draft.draft_type = ?)
+       ORDER BY draft.updated_at DESC, draft.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);
+        const rows = this.db.all(`SELECT draft.*, volume.title AS volume_title FROM drafts draft
+       LEFT JOIN volumes volume ON volume.id = draft.volume_id
+       WHERE draft.work_id = ? AND (? IS NULL OR draft.draft_type = ?)
+       ORDER BY draft.updated_at DESC, draft.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) {
@@ -2308,17 +2330,21 @@ export class Store {
         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
+            ? this.db.all(`SELECT draft.*, volume.title AS volume_title FROM drafts draft
+           LEFT JOIN volumes volume ON volume.id = draft.volume_id
+           WHERE draft.work_id = ? AND (? IS NULL OR draft.draft_type = ?)
+             AND (draft.title LIKE ? ESCAPE '\\' COLLATE NOCASE OR draft.content LIKE ? ESCAPE '\\' COLLATE NOCASE)
+           ORDER BY CASE WHEN draft.title LIKE ? ESCAPE '\\' COLLATE NOCASE THEN 0 ELSE 1 END, draft.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);
+            : this.db.all(`SELECT draft.*, volume.title AS volume_title FROM drafts draft
+           LEFT JOIN volumes volume ON volume.id = draft.volume_id
+           WHERE draft.work_id = ? AND (? IS NULL OR draft.draft_type = ?)
+           ORDER BY draft.updated_at DESC, draft.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);
+        const row = this.db.get(`SELECT draft.*, volume.title AS volume_title FROM drafts draft
+       LEFT JOIN volumes volume ON volume.id = draft.volume_id WHERE draft.id = ?`, draftId);
         if (!row)
             throw notFound("想法");
         return this.mapDraft(row, true);
@@ -2326,9 +2352,15 @@ export class Store {
     updateDraft(draftId, input, source = "manual", sourceRef = null, changeNote = "", expectedVersionNo) {
         const current = this.getDraft(draftId);
         const content = input.content ?? String(current.content);
+        const draftType = input.draftType ?? current.draftType;
+        const typeChanged = input.draftType !== undefined && input.draftType !== current.draftType;
+        const restoreMissingBinding = source === "restore";
+        const volumeId = Object.hasOwn(input, "volumeId") ? input.volumeId ?? null : typeChanged || restoreMissingBinding ? null : current.volumeId;
+        const settingModule = Object.hasOwn(input, "settingModule") ? input.settingModule ?? null : typeChanged || restoreMissingBinding ? null : current.settingModule;
+        const binding = this.normalizeDraftBinding(String(current.workId), draftType, volumeId, settingModule);
         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.db.run("UPDATE drafts SET draft_type = ?, volume_id = ?, setting_module = ?, title = ?, content = ?, updated_at = ? WHERE id = ?", draftType, binding.volumeId, binding.settingModule, 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 });
@@ -2351,6 +2383,9 @@ export class Store {
             id: requiredString(row, "id"),
             workId: requiredString(row, "work_id"),
             draftType: requiredString(row, "draft_type"),
+            volumeId: optionalString(row, "volume_id"),
+            volumeTitle: optionalString(row, "volume_title"),
+            settingModule: optionalString(row, "setting_module"),
             title: requiredString(row, "title"),
             ...(includeContent ? { content } : { contentPreview: content.replace(/\s+/gu, " ").trim().slice(0, 320) }),
             versionNo: this.currentEntityVersionNo("draft", requiredString(row, "id")),
@@ -2358,6 +2393,41 @@ export class Store {
             updatedAt: requiredString(row, "updated_at")
         };
     }
+    normalizeDraftBinding(workId, draftType, volumeId, settingModule) {
+        if (draftType === "prose") {
+            if (settingModule !== null) {
+                throw new AppError(400, "DRAFT_BINDING_TYPE_MISMATCH", "正文想法不能绑定设定模块");
+            }
+            if (volumeId !== null) {
+                const volume = this.getVolume(volumeId);
+                if (volume.workId !== workId) {
+                    throw new AppError(400, "DRAFT_VOLUME_WORK_MISMATCH", "分卷不属于当前作品");
+                }
+            }
+            return { volumeId, settingModule: null };
+        }
+        if (volumeId !== null) {
+            throw new AppError(400, "DRAFT_BINDING_TYPE_MISMATCH", "设定想法不能绑定分卷");
+        }
+        if (settingModule !== null && !DRAFT_SETTING_MODULES.includes(settingModule)) {
+            throw new AppError(400, "DRAFT_SETTING_MODULE_INVALID", "设定想法绑定模块无效");
+        }
+        return { volumeId: null, settingModule };
+    }
+    clearDraftVolumeBindings(workId, volumeIds, sourceRef, changeNote) {
+        const drafts = volumeIds === null
+            ? this.db.all("SELECT id FROM drafts WHERE work_id = ? AND volume_id IS NOT NULL", workId)
+            : volumeIds.length
+                ? this.db.all(`SELECT id FROM drafts WHERE work_id = ? AND volume_id IN (${volumeIds.map(() => "?").join(", ")})`, workId, ...volumeIds)
+                : [];
+        const timestamp = now();
+        for (const draft of drafts) {
+            const draftId = requiredString(draft, "id");
+            this.db.run("UPDATE drafts SET volume_id = NULL, updated_at = ? WHERE id = ?", timestamp, draftId);
+            this.recordEntityVersion("draft", draftId, "manual", sourceRef, changeNote, timestamp);
+            this.audit(workId, "draft.updated", "draft", draftId, { fields: ["volumeId"], source: "volume-deleted", sourceRef });
+        }
+    }
     createSetting(workId, input, source = "create", sourceRef = null) {
         this.getWork(workId);
         return this.insertSettingWithId(workId, id("setting"), input, source, sourceRef);
@@ -3224,11 +3294,16 @@ export class Store {
             createdAt: requiredString(row, "created_at")
         };
     }
-    createAttachment(workId, input) {
+    createAttachment(workId, input, accessModule = "settings") {
         this.getWork(workId);
         const existing = this.db.get("SELECT * FROM attachments WHERE work_id = ? AND stored_sha256 = ?", workId, input.storedSha256);
-        if (existing)
+        if (existing) {
+            this.db.transaction(() => {
+                this.db.run("INSERT OR IGNORE INTO attachment_access_modules (attachment_id, module, created_at) VALUES (?, ?, ?)", requiredString(existing, "id"), accessModule, now());
+                this.db.run("DELETE FROM attachment_cleanup_queue WHERE storage_key = ?", requiredString(existing, "storage_key"));
+            });
             return { attachment: this.mapAttachment(existing), created: false };
+        }
         const attachmentId = id("attachment");
         const timestamp = now();
         this.db.transaction(() => {
@@ -3236,6 +3311,8 @@ export class Store {
          (id, work_id, original_name, original_mime_type, stored_mime_type, original_byte_length, stored_byte_length,
           original_sha256, stored_sha256, storage_key, width, height, page_count, animated, created_at, created_by_user_id)
          VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`, attachmentId, workId, input.originalName, input.originalMimeType, input.storedMimeType, input.originalByteLength, input.storedByteLength, input.originalSha256, input.storedSha256, input.storageKey, input.width, input.height, input.pageCount, input.animated ? 1 : 0, timestamp, currentRequestActor()?.userId ?? null);
+            this.db.run("INSERT INTO attachment_access_modules (attachment_id, module, created_at) VALUES (?, ?, ?)", attachmentId, accessModule, timestamp);
+            this.db.run("DELETE FROM attachment_cleanup_queue WHERE storage_key = ?", input.storageKey);
             this.audit(workId, "attachment.created", "attachment", attachmentId, {
                 originalMimeType: input.originalMimeType,
                 storedMimeType: input.storedMimeType,
@@ -3262,18 +3339,100 @@ export class Store {
             throw notFound("附件");
         return this.mapAttachment(row);
     }
+    attachmentModules(attachmentId) {
+        this.getAttachment(attachmentId);
+        const modules = new Set();
+        for (const row of this.db.all("SELECT module FROM attachment_access_modules WHERE attachment_id = ?", attachmentId)) {
+            const module = String(row.module);
+            if (attachmentPermissionModules.includes(module))
+                modules.add(module);
+        }
+        const referenceModules = {
+            chapter: "prose",
+            draft: "drafts",
+            setting: "settings",
+            "character-section": "characters",
+            race: "races",
+            organization: "organizations"
+        };
+        for (const row of this.db.all("SELECT DISTINCT entity_type FROM attachment_references WHERE attachment_id = ?", attachmentId)) {
+            const module = referenceModules[String(row.entity_type)];
+            if (module)
+                modules.add(module);
+        }
+        return [...modules];
+    }
+    attachmentStorageKeyInUse(storageKey) {
+        return Number(this.db.get("SELECT COUNT(*) AS count FROM attachments WHERE storage_key = ?", storageKey)?.count ?? 0) > 0;
+    }
+    enqueueAttachmentCleanup(storageKey) {
+        const timestamp = now();
+        this.db.run(`INSERT INTO attachment_cleanup_queue (storage_key, attempts, last_error, created_at, updated_at)
+       VALUES (?, 0, NULL, ?, ?) ON CONFLICT(storage_key) DO UPDATE SET updated_at = excluded.updated_at`, storageKey, timestamp, timestamp);
+    }
+    listAttachmentCleanupQueue(limit = 100) {
+        return this.db.all("SELECT storage_key, attempts FROM attachment_cleanup_queue ORDER BY updated_at, storage_key LIMIT ?", Math.max(1, Math.min(1_000, Math.trunc(limit)))).map((row) => ({ storageKey: requiredString(row, "storage_key"), attempts: numberValue(row, "attempts") }));
+    }
+    attachmentCleanupStillRequired(storageKey) {
+        return !this.attachmentStorageKeyInUse(storageKey);
+    }
+    completeAttachmentCleanup(storageKey) {
+        this.db.run("DELETE FROM attachment_cleanup_queue WHERE storage_key = ?", storageKey);
+    }
+    failAttachmentCleanup(storageKey, message) {
+        this.db.run("UPDATE attachment_cleanup_queue SET attempts = attempts + 1, last_error = ?, updated_at = ? WHERE storage_key = ?", message.slice(0, 500), now(), storageKey);
+    }
+    attachmentHistoricalReferenceCount(attachmentId) {
+        const needle = `attachment://${attachmentId}`;
+        const sources = [
+            ["entity_versions", "snapshot_json"],
+            ["character_profile_section_versions", "snapshot_json"],
+            ["character_versions", "snapshot_json"],
+            ["chapter_versions", "content"],
+            ["file_versions", "snapshot_json"]
+        ];
+        return sources.reduce((count, [table, column]) => count + Number(this.db.get(`SELECT COUNT(*) AS count FROM ${table} WHERE instr(${column}, ?) > 0`, needle)?.count ?? 0), 0);
+    }
+    queueUnreferencedAttachments(retentionMs = 24 * 60 * 60_000, limit = 100) {
+        const cutoff = new Date(Date.now() - Math.max(0, retentionMs)).toISOString();
+        const candidates = this.db.all(`SELECT attachment.* FROM attachments attachment
+       WHERE attachment.created_at <= ?
+         AND NOT EXISTS (
+           SELECT 1 FROM attachment_references reference WHERE reference.attachment_id = attachment.id
+         )
+       ORDER BY attachment.created_at, attachment.id LIMIT ?`, cutoff, Math.max(1, Math.min(1_000, Math.trunc(limit))));
+        let queued = 0;
+        for (const candidate of candidates) {
+            const attachmentId = requiredString(candidate, "id");
+            if (this.attachmentHistoricalReferenceCount(attachmentId) > 0)
+                continue;
+            const storageKey = requiredString(candidate, "storage_key");
+            this.db.transaction(() => {
+                this.db.run("DELETE FROM attachments WHERE id = ?", attachmentId);
+                this.audit(requiredString(candidate, "work_id"), "attachment.garbage-collected", "attachment", attachmentId, { storageKey });
+                if (!this.attachmentStorageKeyInUse(storageKey))
+                    this.enqueueAttachmentCleanup(storageKey);
+            });
+            queued += 1;
+        }
+        return queued;
+    }
     deleteAttachment(attachmentId) {
         const attachment = this.getAttachment(attachmentId);
         const references = Number(this.db.get("SELECT COUNT(*) AS count FROM attachment_references WHERE attachment_id = ?", attachmentId)?.count ?? 0);
         if (references > 0)
             throw new AppError(409, "ATTACHMENT_IN_USE", "附件仍被资料引用，无法删除");
+        if (this.attachmentHistoricalReferenceCount(attachmentId) > 0) {
+            throw new AppError(409, "ATTACHMENT_IN_VERSION_HISTORY", "附件仍被历史版本引用，无法删除");
+        }
         const storageKey = String(attachment.storageKey);
         this.db.transaction(() => {
             this.db.run("DELETE FROM attachments WHERE id = ?", attachmentId);
             this.audit(String(attachment.workId), "attachment.deleted", "attachment", attachmentId, { storageKey });
+            if (!this.attachmentStorageKeyInUse(storageKey))
+                this.enqueueAttachmentCleanup(storageKey);
         });
-        const remaining = Number(this.db.get("SELECT COUNT(*) AS count FROM attachments WHERE storage_key = ?", storageKey)?.count ?? 0);
-        return { storageKey, removeStoredFile: remaining === 0 };
+        return { storageKey, cleanupQueued: !this.attachmentStorageKeyInUse(storageKey) };
     }
     getCharacter(characterId) {
         const row = this.db.get("SELECT * FROM characters WHERE id = ?", characterId);