@musnows/scriverse 0.6.1 → 0.6.3

package/dist/store.js

@@ -1,5 +1,7 @@
+import { DRAFT_SETTING_MODULES } from "./domain.js";
 import { createHash } from "node:crypto";
 import { PLATFORM_AI_WORK_ID } from "./database.js";
+import { exportWorkDocx } from "./docx-export.js";
 import { AppError, notFound } from "./errors.js";
 import { accountReference, logger } from "./logger.js";
 import { paginated, paginationSql } from "./pagination.js";
@@ -7,6 +9,32 @@ 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"];
+export const WORK_AGENT_TOOL_IDS = [
+    "story_index",
+    "read_chapters",
+    "grep",
+    "search_story_entities",
+    "read_character_sections",
+    "search_drafts"
+];
+const DEFAULT_WORK_AGENT_TOOLS = [...WORK_AGENT_TOOL_IDS];
+export function normalizeWorkAgentTools(value) {
+    const source = Array.isArray(value)
+        ? value
+        : typeof value === "string"
+            ? json(value, DEFAULT_WORK_AGENT_TOOLS)
+            : DEFAULT_WORK_AGENT_TOOLS;
+    const enabled = new Set();
+    for (const item of source) {
+        if (typeof item !== "string")
+            continue;
+        const toolId = item === "query_story_knowledge" ? "search_story_entities" : item;
+        if (WORK_AGENT_TOOL_IDS.includes(toolId))
+            enabled.add(toolId);
+    }
+    return WORK_AGENT_TOOL_IDS.filter((toolId) => enabled.has(toolId));
+}
 const defaultPlatformPageSizes = {
     drafts: 30,
     settings: 30,
@@ -302,6 +330,8 @@ export class Store {
         if (type === "draft")
             return {
                 draftType: entity.draftType,
+                volumeId: entity.volumeId,
+                settingModule: entity.settingModule,
                 title: entity.title,
                 content: entity.content
             };
@@ -681,6 +711,9 @@ export class Store {
         return {
             workId,
             systemPrompt: String(row?.system_prompt ?? ""),
+            dailyTokenQuota: row?.daily_token_quota === null || row?.daily_token_quota === undefined
+                ? null
+                : Math.max(10_000, Number(row.daily_token_quota)),
             autoRunEnabled: Number(row?.auto_run_enabled ?? 0) === 1,
             autoRunConcurrency: Math.min(8, Math.max(1, Number(row?.auto_run_concurrency ?? 2) || 2)),
             autoRunBatchLimit: Math.min(200, Math.max(1, Number(row?.auto_run_batch_limit ?? 20) || 20)),
@@ -692,9 +725,9 @@ 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","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),
+            agentToolCallLimit: Math.min(48, Math.max(5, Number(row?.agent_tool_call_limit ?? 12) || 12)),
+            agentToolCallGlobalMultiplier: Math.min(6, Math.max(1, Number(row?.agent_tool_call_global_multiplier ?? 3) || 3)),
+            agentTools: normalizeWorkAgentTools(row?.agent_tools_json),
             titleGenerationModelId: row?.title_generation_model_id === null || row?.title_generation_model_id === undefined
                 ? null
                 : String(row.title_generation_model_id),
@@ -706,6 +739,9 @@ export class Store {
         const current = this.getWorkAiSettings(workId);
         const timestamp = now();
         const nextPrompt = input.systemPrompt ?? String(current.systemPrompt);
+        const nextDailyTokenQuota = input.dailyTokenQuota === undefined
+            ? (current.dailyTokenQuota === null ? null : Number(current.dailyTokenQuota))
+            : input.dailyTokenQuota;
         const nextEnabled = input.autoRunEnabled ?? Boolean(current.autoRunEnabled);
         const nextConcurrency = input.autoRunConcurrency ?? Number(current.autoRunConcurrency);
         const nextBatchLimit = input.autoRunBatchLimit ?? Number(current.autoRunBatchLimit);
@@ -713,18 +749,22 @@ export class Store {
         const nextFailureThreshold = input.autoRunFailureThreshold ?? Number(current.autoRunFailureThreshold);
         const nextBookSummaryContextPercent = input.bookSummaryContextPercent ?? Number(current.bookSummaryContextPercent);
         const nextContextCompactThreshold = input.contextCompactThreshold ?? Number(current.contextCompactThreshold);
-        const nextAgentTools = input.agentTools ?? current.agentTools;
+        const nextAgentToolCallLimit = input.agentToolCallLimit ?? Number(current.agentToolCallLimit);
+        const nextAgentToolCallGlobalMultiplier = input.agentToolCallGlobalMultiplier ?? Number(current.agentToolCallGlobalMultiplier);
+        const nextAgentTools = normalizeWorkAgentTools(input.agentTools ?? current.agentTools);
         const nextTitleGenerationModelId = input.titleGenerationModelId === undefined
             ? (current.titleGenerationModelId ? String(current.titleGenerationModelId) : null)
             : input.titleGenerationModelId?.trim() || null;
         this.db.run(`INSERT INTO work_ai_settings (
-         work_id, system_prompt, auto_run_enabled, auto_run_concurrency, auto_run_batch_limit,
+         work_id, system_prompt, daily_token_quota, auto_run_enabled, auto_run_concurrency, auto_run_batch_limit,
          auto_run_daily_task_limit, auto_run_failure_threshold, auto_run_paused, auto_run_pause_reason,
          auto_run_resume_at, auto_run_consecutive_failures, book_summary_context_percent,
-         context_compact_threshold, agent_tools_json, title_generation_model_id, updated_at
-       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+         context_compact_threshold, agent_tool_call_limit, agent_tool_call_global_multiplier,
+         agent_tools_json, title_generation_model_id, updated_at
+       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
        ON CONFLICT(work_id) DO UPDATE SET
          system_prompt = excluded.system_prompt,
+         daily_token_quota = excluded.daily_token_quota,
          auto_run_enabled = excluded.auto_run_enabled,
          auto_run_concurrency = excluded.auto_run_concurrency,
          auto_run_batch_limit = excluded.auto_run_batch_limit,
@@ -736,11 +776,14 @@ export class Store {
          auto_run_consecutive_failures = excluded.auto_run_consecutive_failures,
          book_summary_context_percent = excluded.book_summary_context_percent,
          context_compact_threshold = excluded.context_compact_threshold,
+         agent_tool_call_limit = excluded.agent_tool_call_limit,
+         agent_tool_call_global_multiplier = excluded.agent_tool_call_global_multiplier,
          agent_tools_json = excluded.agent_tools_json,
          title_generation_model_id = excluded.title_generation_model_id,
-         updated_at = excluded.updated_at`, workId, nextPrompt, nextEnabled ? 1 : 0, Math.min(8, Math.max(1, nextConcurrency)), Math.min(200, Math.max(1, nextBatchLimit)), Math.min(10_000, Math.max(0, nextDailyTaskLimit)), Math.min(10, Math.max(1, nextFailureThreshold)), current.autoRunPaused ? 1 : 0, String(current.autoRunPauseReason ?? ""), current.autoRunResumeAt === null ? null : String(current.autoRunResumeAt), Math.max(0, Number(current.autoRunConsecutiveFailures) || 0), Math.min(90, Math.max(1, nextBookSummaryContextPercent)), Math.min(90, Math.max(50, nextContextCompactThreshold)), JSON.stringify(nextAgentTools), nextTitleGenerationModelId, timestamp);
+         updated_at = excluded.updated_at`, workId, nextPrompt, nextDailyTokenQuota, nextEnabled ? 1 : 0, Math.min(8, Math.max(1, nextConcurrency)), Math.min(200, Math.max(1, nextBatchLimit)), Math.min(10_000, Math.max(0, nextDailyTaskLimit)), Math.min(10, Math.max(1, nextFailureThreshold)), current.autoRunPaused ? 1 : 0, String(current.autoRunPauseReason ?? ""), current.autoRunResumeAt === null ? null : String(current.autoRunResumeAt), Math.max(0, Number(current.autoRunConsecutiveFailures) || 0), Math.min(90, Math.max(1, nextBookSummaryContextPercent)), Math.min(90, Math.max(50, nextContextCompactThreshold)), Math.min(48, Math.max(5, nextAgentToolCallLimit)), Math.min(6, Math.max(1, nextAgentToolCallGlobalMultiplier)), JSON.stringify(nextAgentTools), nextTitleGenerationModelId, timestamp);
         this.audit(workId, "work.ai-settings.updated", "work-ai-settings", workId, {
             systemPromptChanged: input.systemPrompt !== undefined,
+            dailyTokenQuota: nextDailyTokenQuota,
             autoRunEnabled: nextEnabled,
             autoRunConcurrency: Math.min(8, Math.max(1, nextConcurrency)),
             autoRunBatchLimit: Math.min(200, Math.max(1, nextBatchLimit)),
@@ -748,6 +791,8 @@ export class Store {
             autoRunFailureThreshold: Math.min(10, Math.max(1, nextFailureThreshold)),
             bookSummaryContextPercent: Math.min(90, Math.max(1, nextBookSummaryContextPercent)),
             contextCompactThreshold: Math.min(90, Math.max(50, nextContextCompactThreshold)),
+            agentToolCallLimit: Math.min(48, Math.max(5, nextAgentToolCallLimit)),
+            agentToolCallGlobalMultiplier: Math.min(6, Math.max(1, nextAgentToolCallGlobalMultiplier)),
             agentTools: nextAgentTools,
             titleGenerationModelId: nextTitleGenerationModelId
         });
@@ -817,13 +862,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");
@@ -842,12 +897,22 @@ export class Store {
         return this.getWork(workId);
     }
     getWorkCover(workId) {
+        const cover = this.findWorkCover(workId);
+        if (!cover)
+            throw notFound("作品封面");
+        return cover;
+    }
+    findWorkCover(workId) {
         this.getWork(workId);
         const row = this.db.get("SELECT * FROM work_covers WHERE work_id = ?", workId);
         if (!row)
-            throw notFound("作品封面");
+            return null;
+        const mimeType = requiredString(row, "mime_type");
+        if (mimeType !== "image/jpeg" && mimeType !== "image/png" && mimeType !== "image/webp") {
+            throw new AppError(500, "INVALID_COVER_MIME", "作品封面类型无效");
+        }
         return {
-            mimeType: requiredString(row, "mime_type"),
+            mimeType,
             content: Buffer.from(row.content),
             byteLength: numberValue(row, "byte_length"),
             sha256: requiredString(row, "sha256"),
@@ -1026,6 +1091,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 +1143,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,19 +1230,22 @@ 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) });
         });
     }
     createChapter(workId, input) {
-        this.getWork(workId);
-        const volume = this.getVolume(input.volumeId);
-        if (volume.workId !== workId)
-            throw new AppError(400, "VOLUME_WORK_MISMATCH", "卷不属于当前作品");
-        const last = this.db.get("SELECT COALESCE(MAX(sort_order), -1) AS value FROM chapters WHERE volume_id = ? AND deleted_at IS NULL", input.volumeId);
-        const chapterId = this.insertChapter(workId, input.volumeId, input.title, input.content ?? "", numberValue(last ?? {}, "value") + 1, "manual", null, input.chapterType ?? "正文");
-        this.audit(workId, "chapter.created", "chapter", chapterId);
-        return this.getChapter(chapterId);
+        return this.db.transaction(() => {
+            this.getWork(workId);
+            const volume = this.getVolume(input.volumeId);
+            if (volume.workId !== workId)
+                throw new AppError(400, "VOLUME_WORK_MISMATCH", "卷不属于当前作品");
+            const last = this.db.get("SELECT COALESCE(MAX(sort_order), -1) AS value FROM chapters WHERE volume_id = ? AND deleted_at IS NULL", input.volumeId);
+            const chapterId = this.insertChapter(workId, input.volumeId, input.title, input.content ?? "", numberValue(last ?? {}, "value") + 1, "manual", null, input.chapterType ?? "正文");
+            this.audit(workId, "chapter.created", "chapter", chapterId);
+            return this.getChapter(chapterId);
+        });
     }
     getChapter(chapterId) {
         const row = this.db.get("SELECT * FROM chapters WHERE id = ? AND deleted_at IS NULL", chapterId);
@@ -2276,9 +2346,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 +2362,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 +2383,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 +2405,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 +2436,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 +2446,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 +3347,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 +3364,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 +3392,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);
@@ -4128,7 +4340,8 @@ export class Store {
         this.getWork(workId);
         const conversationId = id("conversation");
         const timestamp = now();
-        this.db.run("INSERT INTO ai_conversations (id, work_id, title, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?)", conversationId, workId, title.trim() || "新对话", timestamp, timestamp, currentRequestActor()?.userId ?? null);
+        const agentTools = normalizeWorkAgentTools(this.getWorkAiSettings(workId).agentTools);
+        this.db.run("INSERT INTO ai_conversations (id, work_id, title, agent_tools_json, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?)", conversationId, workId, title.trim() || "新对话", JSON.stringify(agentTools), timestamp, timestamp, currentRequestActor()?.userId ?? null);
         return this.getAiConversation(conversationId);
     }
     listAiConversations(workId) {
@@ -4286,7 +4499,9 @@ export class Store {
         const forkCompactedCount = targetIndex + 1 >= sourceCompactedCount ? Math.min(sourceCompactedCount, targetIndex + 1) : 0;
         const forkSummary = forkCompactedCount ? requiredString(conversation, "compacted_summary") : "";
         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);
+            this.db.run("INSERT INTO ai_conversations (id, work_id, title, compacted_summary, compacted_message_count, agent_tools_json, created_at, updated_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)", forkId, requiredString(conversation, "work_id"), title.slice(0, 200), forkSummary, forkCompactedCount, conversation.agent_tools_json == null
+                ? JSON.stringify(normalizeWorkAgentTools(this.getWorkAiSettings(requiredString(conversation, "work_id")).agentTools))
+                : String(conversation.agent_tools_json), 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, 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);
             }
@@ -4303,10 +4518,29 @@ export class Store {
             compactedMessageCount: numberValue(row, "compacted_message_count"),
             hasCompactedSummary: Boolean(requiredString(row, "compacted_summary")),
             contextWarningPending: Boolean(optionalString(row, "context_warning_at")),
+            agentTools: row.agent_tools_json == null || row.agent_tools_json === undefined
+                ? null
+                : normalizeWorkAgentTools(row.agent_tools_json),
             createdAt: requiredString(row, "created_at"),
             updatedAt: requiredString(row, "updated_at")
         };
     }
+    /** 锁定本对话可用工具集；已锁定则保持不变，避免中途改作品设置破坏 prompt cache。 */
+    ensureAiConversationAgentTools(conversationId, workId) {
+        const conversation = this.db.get("SELECT id, work_id, agent_tools_json FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        if (requiredString(conversation, "work_id") !== workId) {
+            throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
+        }
+        if (conversation.agent_tools_json != null && conversation.agent_tools_json !== undefined) {
+            return normalizeWorkAgentTools(conversation.agent_tools_json);
+        }
+        const agentTools = normalizeWorkAgentTools(this.getWorkAiSettings(workId).agentTools);
+        this.db.run("UPDATE ai_conversations SET agent_tools_json = ?, updated_at = ? WHERE id = ? AND agent_tools_json IS NULL", JSON.stringify(agentTools), now(), conversationId);
+        const locked = this.db.get("SELECT agent_tools_json FROM ai_conversations WHERE id = ?", conversationId);
+        return normalizeWorkAgentTools(locked?.agent_tools_json ?? agentTools);
+    }
     mapAiConversationMessage(row) {
         return {
             id: requiredString(row, "id"),
@@ -5710,6 +5944,22 @@ export class Store {
         }
         return lines.join("\n").trimEnd() + "\n";
     }
+    async exportDocx(workId) {
+        const tree = this.getWorkTree(workId);
+        const cover = this.findWorkCover(workId);
+        const volumes = tree.volumes.map((volume) => ({
+            title: String(volume.title),
+            chapters: volume.chapters.map((chapter) => ({
+                title: String(chapter.title),
+                content: String(chapter.content ?? "")
+            }))
+        }));
+        return exportWorkDocx({
+            title: String(tree.title),
+            volumes,
+            cover: cover ? { mimeType: cover.mimeType, content: cover.content } : null
+        });
+    }
     listAuditLogs(workId) {
         this.getWork(workId);
         return this.db.all(`SELECT log.*, user.display_name AS actor_display_name, user.username AS actor_username