@musnows/scriverse 0.7.1 → 0.7.3

package/dist/store.js

@@ -1,6 +1,6 @@
 import { DRAFT_SETTING_MODULES } from "./domain.js";
 import { createHash } from "node:crypto";
-import { PLATFORM_AI_WORK_ID } from "./database.js";
+import { ENTITY_VERSION_BASELINE_MIGRATION_VERSION, 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";
@@ -10,6 +10,7 @@ import { canWriteWorkModule, classifyWorkModulePermissions, emptyWorkModulePermi
 import { countWords, documentShortSearchTerms, id, json, normalizeDocumentSearchText, normalizeParagraphSpacing, now, splitDocumentParagraphs } from "./utils.js";
 import { buildWritingCalendar, writingDateKey } from "./writing-progress-time.js";
 import { resolveMaxAgentToolCallLimit } from "./ai-tool-results.js";
+const WORK_LIST_BATCH_SIZE = 500;
 export const attachmentPermissionModules = ["prose", "drafts", "settings", "characters", "races", "organizations"];
 export const WORK_AGENT_TOOL_IDS = [
     "story_index",
@@ -37,6 +38,7 @@ export function normalizeWorkAgentTools(value) {
     }
     return WORK_AGENT_TOOL_IDS.filter((toolId) => enabled.has(toolId));
 }
+const galaxyFrameRates = [24, 30, 60, 90, 120, 144, 165, 240];
 const defaultPlatformPageSizes = {
     drafts: 30,
     settings: 30,
@@ -78,7 +80,7 @@ function platformPageSizes(value) {
 }
 function galaxyFrameRate(value) {
     const candidate = Number(value);
-    return candidate === 24 || candidate === 60 ? candidate : 30;
+    return galaxyFrameRates.includes(candidate) ? candidate : 30;
 }
 function settingsFromInput(settingsMarkdown, settings, fallback = []) {
     if (settingsMarkdown !== undefined)
@@ -265,7 +267,7 @@ export class Store {
     db;
     constructor(db) {
         this.db = db;
-        this.backfillEntityVersionBaselines();
+        this.migrateEntityVersionBaselines();
     }
     currentEntityVersionNo(type, entityId) {
         const row = this.db.get("SELECT MAX(version_no) AS version_no FROM entity_versions WHERE entity_type = ? AND entity_id = ?", type, entityId);
@@ -489,6 +491,26 @@ export class Store {
             }
         });
     }
+    migrateEntityVersionBaselines() {
+        const migrationSql = "SELECT 1 AS present FROM schema_migrations WHERE version = ?";
+        if (this.db.get(migrationSql, ENTITY_VERSION_BASELINE_MIGRATION_VERSION))
+            return;
+        this.db.transaction(() => {
+            if (this.db.get(migrationSql, ENTITY_VERSION_BASELINE_MIGRATION_VERSION))
+                return;
+            if (!this.db.get("SELECT 1 AS present FROM entity_versions LIMIT 1")) {
+                this.backfillEntityVersionBaselines();
+            }
+            const integrity = this.db.all("PRAGMA integrity_check");
+            if (integrity.some((row) => row.integrity_check !== "ok")) {
+                throw new Error(`数据库完整性检查失败：${integrity.map((row) => row.integrity_check).join("；")}`);
+            }
+            const foreignKeys = this.db.all("PRAGMA foreign_key_check");
+            if (foreignKeys.length > 0)
+                throw new Error(`数据库外键检查失败：发现 ${foreignKeys.length} 条异常记录`);
+            this.db.run("INSERT INTO schema_migrations (version, applied_at) VALUES (?, ?)", ENTITY_VERSION_BASELINE_MIGRATION_VERSION, now());
+        });
+    }
     listEntityVersions(type, entityId) {
         const rows = this.db.all(`SELECT version.*, user.display_name AS actor_display_name, user.username AS actor_username
        FROM entity_versions version LEFT JOIN users user ON user.id = version.created_by_user_id
@@ -661,11 +683,11 @@ export class Store {
     listWorks() {
         const actor = currentRequestActor();
         if (!actor || (actor.role === "admin" && actor.authentication !== "api-key")) {
-            return this.db.all("SELECT * FROM works WHERE COALESCE(is_internal, 0) = 0 ORDER BY updated_at DESC").map((row) => this.mapWork(row));
+            return this.mapWorks(this.db.all("SELECT * FROM works WHERE COALESCE(is_internal, 0) = 0 ORDER BY updated_at DESC"));
         }
-        return this.db.all(`SELECT DISTINCT work.* FROM works work LEFT JOIN work_memberships membership ON membership.work_id = work.id
+        return this.mapWorks(this.db.all(`SELECT DISTINCT work.* FROM works work LEFT JOIN work_memberships membership ON membership.work_id = work.id
        WHERE COALESCE(work.is_internal, 0) = 0 AND (work.owner_user_id = ? OR membership.user_id = ?)
-       ORDER BY work.updated_at DESC`, actor.userId, actor.userId).map((row) => this.mapWork(row));
+       ORDER BY work.updated_at DESC`, actor.userId, actor.userId));
     }
     listWorksPage(pagination) {
         const actor = currentRequestActor();
@@ -675,7 +697,7 @@ export class Store {
             : this.db.all(`SELECT DISTINCT work.* FROM works work LEFT JOIN work_memberships membership ON membership.work_id = work.id
          WHERE COALESCE(work.is_internal, 0) = 0 AND (work.owner_user_id = ? OR membership.user_id = ?)
          ORDER BY work.updated_at DESC${page.sql}`, actor.userId, actor.userId, ...page.params);
-        return paginated(rows.map((row) => this.mapWork(row)), pagination);
+        return paginated(this.mapWorks(rows), pagination);
     }
     getWork(workId) {
         const row = this.db.get("SELECT * FROM works WHERE id = ?", workId);
@@ -966,7 +988,7 @@ export class Store {
         if (!row)
             return null;
         const mimeType = requiredString(row, "mime_type");
-        if (mimeType !== "image/jpeg" && mimeType !== "image/png" && mimeType !== "image/webp") {
+        if (mimeType !== "image/jpeg" && mimeType !== "image/png" && mimeType !== "image/webp" && mimeType !== "image/gif") {
             throw new AppError(500, "INVALID_COVER_MIME", "作品封面类型无效");
         }
         return {
@@ -2124,12 +2146,45 @@ export class Store {
         }
         this.notifyAnalysisTaskQueued(workId);
     }
-    mapWork(row) {
+    mapWorks(rows) {
+        if (rows.length === 0)
+            return [];
+        const actor = currentRequestActor();
+        const workIds = [...new Set(rows.map((row) => requiredString(row, "id")))];
+        const batch = {
+            memberships: new Map(),
+            counts: new Map(),
+            covers: new Map()
+        };
+        for (let offset = 0; offset < workIds.length; offset += WORK_LIST_BATCH_SIZE) {
+            const batchIds = workIds.slice(offset, offset + WORK_LIST_BATCH_SIZE);
+            const placeholders = batchIds.map(() => "?").join(", ");
+            if (actor) {
+                const memberships = this.db.all(`SELECT work_id, role, permissions_json FROM work_memberships
+           WHERE user_id = ? AND work_id IN (${placeholders})`, actor.userId, ...batchIds);
+                for (const membership of memberships) {
+                    batch.memberships.set(requiredString(membership, "work_id"), membership);
+                }
+            }
+            const counts = this.db.all(`SELECT work_id, COUNT(*) AS chapter_count, COALESCE(SUM(word_count), 0) AS word_count
+         FROM chapters WHERE work_id IN (${placeholders}) AND deleted_at IS NULL GROUP BY work_id`, ...batchIds);
+            for (const count of counts)
+                batch.counts.set(requiredString(count, "work_id"), count);
+            const covers = this.db.all(`SELECT work_id, updated_at FROM work_covers WHERE work_id IN (${placeholders})`, ...batchIds);
+            for (const cover of covers)
+                batch.covers.set(requiredString(cover, "work_id"), cover);
+        }
+        return rows.map((row) => this.mapWork(row, batch));
+    }
+    mapWork(row, batch) {
         const actor = currentRequestActor();
+        const workId = requiredString(row, "id");
         const ownerUserId = optionalString(row, "owner_user_id");
-        const membership = actor
-            ? this.db.get("SELECT role, permissions_json FROM work_memberships WHERE work_id = ? AND user_id = ?", requiredString(row, "id"), actor.userId)
-            : undefined;
+        const membership = batch
+            ? batch.memberships.get(workId)
+            : actor
+                ? this.db.get("SELECT role, permissions_json FROM work_memberships WHERE work_id = ? AND user_id = ?", workId, actor.userId)
+                : undefined;
         const membershipRole = String(membership?.role ?? "");
         const ownerAccess = ownerUserId === actor?.userId;
         const adminAccess = actor?.role === "admin" && actor.authentication !== "api-key";
@@ -2143,19 +2198,23 @@ export class Store {
             : adminAccess
                 ? "admin"
                 : membershipRole ? classifyWorkModulePermissions(modulePermissions) : null;
-        const count = this.db.get("SELECT COUNT(*) AS chapter_count, COALESCE(SUM(word_count), 0) AS word_count FROM chapters WHERE work_id = ? AND deleted_at IS NULL", requiredString(row, "id"));
-        const cover = this.db.get("SELECT updated_at FROM work_covers WHERE work_id = ?", requiredString(row, "id"));
+        const count = batch
+            ? batch.counts.get(workId)
+            : this.db.get("SELECT COUNT(*) AS chapter_count, COALESCE(SUM(word_count), 0) AS word_count FROM chapters WHERE work_id = ? AND deleted_at IS NULL", workId);
+        const cover = batch
+            ? batch.covers.get(workId)
+            : this.db.get("SELECT updated_at FROM work_covers WHERE work_id = ?", workId);
         return {
-            id: requiredString(row, "id"),
+            id: workId,
             title: requiredString(row, "title"),
             author: requiredString(row, "author"),
             description: requiredString(row, "description"),
             language: requiredString(row, "language"),
             coverUrl: cover
-                ? `/api/works/${encodeURIComponent(requiredString(row, "id"))}/cover?v=${encodeURIComponent(requiredString(cover, "updated_at"))}`
+                ? `/api/works/${encodeURIComponent(workId)}/cover?v=${encodeURIComponent(requiredString(cover, "updated_at"))}`
                 : optionalString(row, "cover_url"),
             tags: json(requiredString(row, "tags_json"), []),
-            versionNo: numberValue(row, "version_no") || this.currentEntityVersionNo("work", requiredString(row, "id")),
+            versionNo: numberValue(row, "version_no") || this.currentEntityVersionNo("work", workId),
             ownerUserId,
             accessRole,
             modulePermissions,
@@ -5202,7 +5261,7 @@ export class Store {
     }
     updateTask(taskId, input) {
         const current = this.getTask(taskId);
-        const terminal = ["completed", "partial", "review", "expired", "cancelled"];
+        const terminal = ["completed", "partial", "failed", "review", "expired", "cancelled"];
         if (terminal.includes(String(current.status)) && input.status !== current.status) {
             throw new AppError(409, "INVALID_TASK_TRANSITION", "终态任务不能再变更状态");
         }