@musnows/scriverse 0.7.8 → 0.7.9

package/dist/store.js

@@ -4,7 +4,7 @@ import { ENTITY_VERSION_BASELINE_MIGRATION_VERSION, PLATFORM_AI_WORK_ID } from "
 import { exportWorkDocx } from "./docx-export.js";
 import { createEpubArchive } from "./epub-export.js";
 import { AppError, notFound } from "./errors.js";
-import { normalizeWorkSearchQuery } from "./hybrid-search.js";
+import { documentParagraphLineRanges, normalizeWorkSearchQuery } from "./hybrid-search.js";
 import { accountReference, logger } from "./logger.js";
 import { paginated, paginationSql } from "./pagination.js";
 import { currentRequestActor } from "./request-context.js";
@@ -1240,6 +1240,51 @@ export class Store {
         }));
         return { ...work, volumes, directoryPage: pageResult };
     }
+    getStoryIndexChapterPage(workId, offset, limit) {
+        const work = this.getWork(workId);
+        const permissions = work.modulePermissions;
+        if (permissions.prose === "none")
+            return { totalChapters: 0, chapters: [] };
+        const countRow = this.db.get(`SELECT COUNT(*) AS count FROM chapters chapter
+       JOIN volumes volume ON volume.id = chapter.volume_id
+       WHERE chapter.work_id = ? AND chapter.deleted_at IS NULL AND volume.deleted_at IS NULL`, workId);
+        const chapterRows = this.db.all(`SELECT chapter.id, chapter.title, chapter.version_no, volume.title AS volume_title
+       FROM chapters chapter
+       JOIN volumes volume ON volume.id = chapter.volume_id
+       WHERE chapter.work_id = ? AND chapter.deleted_at IS NULL AND volume.deleted_at IS NULL
+       ORDER BY volume.sort_order, volume.created_at, chapter.sort_order, chapter.created_at
+       LIMIT ? OFFSET ?`, workId, limit, offset);
+        const chapterIds = chapterRows.map((row) => requiredString(row, "id"));
+        const summaries = new Map();
+        if (chapterIds.length > 0) {
+            const placeholders = chapterIds.map(() => "?").join(", ");
+            const insightRows = this.db.all(`SELECT insight.chapter_id, insight.summary
+         FROM chapter_insights insight
+         JOIN chapters chapter ON chapter.id = insight.chapter_id AND chapter.version_no = insight.chapter_version
+         WHERE chapter.work_id = ? AND insight.chapter_id IN (${placeholders})
+           AND NOT EXISTS (
+             SELECT 1 FROM chapter_insights newer
+             WHERE newer.chapter_id = insight.chapter_id
+               AND newer.chapter_version = insight.chapter_version
+               AND (newer.created_at > insight.created_at OR (newer.created_at = insight.created_at AND newer.id > insight.id))
+           )`, workId, ...chapterIds);
+            for (const row of insightRows)
+                summaries.set(requiredString(row, "chapter_id"), requiredString(row, "summary"));
+        }
+        return {
+            totalChapters: numberValue(countRow ?? {}, "count"),
+            chapters: chapterRows.map((row) => {
+                const chapterId = requiredString(row, "id");
+                return {
+                    id: chapterId,
+                    volumeTitle: requiredString(row, "volume_title"),
+                    title: requiredString(row, "title"),
+                    versionNo: numberValue(row, "version_no"),
+                    summary: summaries.get(chapterId) ?? ""
+                };
+            })
+        };
+    }
     listFileVersions(workId) {
         this.getWork(workId);
         return this.db
@@ -1989,6 +2034,7 @@ export class Store {
          AND json_extract(scope_json, '$.type') = 'volume'
          AND json_extract(scope_json, '$.volumeId') IN (?, ?)`, timestamp, String(lockedChapter.workId), sourceVolumeId, targetVolumeId);
             this.db.run("UPDATE chapters SET version_no = ?, analysis_status = 'expired', updated_at = ? WHERE id = ?", versionNo, timestamp, chapterId);
+            this.syncChapterParagraphSearchVersion(chapterId, versionNo);
             this.insertChapterVersionRow({
                 workId: String(lockedChapter.workId),
                 chapterId,
@@ -2181,6 +2227,7 @@ export class Store {
                     const versionNo = Number(chapter.versionNo) + 1;
                     const sortOrder = orderedByVolume.get(action.volumeId)?.indexOf(chapterId) ?? 0;
                     this.db.run("UPDATE chapters SET version_no = ?, analysis_status = 'expired', updated_at = ? WHERE id = ?", versionNo, timestamp, chapterId);
+                    this.syncChapterParagraphSearchVersion(chapterId, versionNo);
                     this.insertChapterVersionRow({
                         workId,
                         chapterId,
@@ -2344,16 +2391,27 @@ export class Store {
         return chapterId;
     }
     syncChapterParagraphSearch(workId, chapterId, content) {
+        const chapterVersion = numberValue(this.db.get("SELECT version_no FROM chapters WHERE id = ? AND work_id = ?", chapterId, workId) ?? {}, "version_no");
+        const ranges = documentParagraphLineRanges(content);
         this.db.run("DELETE FROM chapter_paragraph_search WHERE chapter_id = ?", chapterId);
         for (const [paragraphOrder, paragraph] of splitDocumentParagraphs(content).entries()) {
             const searchContent = normalizeDocumentSearchText(paragraph);
             const inserted = this.db.run(`INSERT INTO chapter_paragraph_search (work_id, chapter_id, paragraph_order, content, search_content)
          VALUES (?, ?, ?, ?, ?)`, workId, chapterId, paragraphOrder, paragraph, searchContent);
+            const range = ranges[paragraphOrder];
+            if (range) {
+                this.db.run(`INSERT INTO chapter_paragraph_line_ranges (paragraph_id, chapter_version, start_line, end_line)
+           VALUES (?, ?, ?, ?)`, inserted.lastInsertRowid, chapterVersion, range.startLine, range.endLine);
+            }
             for (const term of documentShortSearchTerms(searchContent)) {
                 this.db.run("INSERT INTO chapter_paragraph_short_terms (paragraph_id, term) VALUES (?, ?)", inserted.lastInsertRowid, term);
             }
         }
     }
+    syncChapterParagraphSearchVersion(chapterId, versionNo) {
+        this.db.run(`UPDATE chapter_paragraph_line_ranges SET chapter_version = ?
+       WHERE paragraph_id IN (SELECT id FROM chapter_paragraph_search WHERE chapter_id = ?)`, versionNo, chapterId);
+    }
     searchChapterParagraphs(workId, keyword, limit = 20) {
         this.getWork(workId);
         const normalizedKeyword = normalizeDocumentSearchText(keyword.trim());
@@ -4119,7 +4177,7 @@ export class Store {
     searchCharacterProfileSections(workId, query, limit = 20) {
         this.getWork(workId);
         const normalized = normalizeDocumentSearchText(query);
-        const columns = `SELECT section.*, character.name AS character_name
+        const columns = `SELECT section.*, character.name AS character_name, character.is_dead AS character_is_dead
       FROM character_profile_section_search search
       JOIN character_profile_sections section ON section.id = search.section_id
       JOIN characters character ON character.id = search.character_id`;
@@ -4131,7 +4189,11 @@ export class Store {
          WHERE search.work_id = ? AND character.merged_into_character_id IS NULL
            AND character_profile_section_search_fts MATCH ?
          ORDER BY bm25(character_profile_section_search_fts), character.name, section.sort_order LIMIT ?`, workId, `"${normalized.replaceAll('"', '""')}"`, limit);
-        return rows.map((row) => ({ ...this.mapCharacterProfileSection(row), characterName: requiredString(row, "character_name") }));
+        return rows.map((row) => ({
+            ...this.mapCharacterProfileSection(row),
+            characterName: requiredString(row, "character_name"),
+            isDead: booleanValue(row, "character_is_dead")
+        }));
     }
     mapAttachment(row) {
         const attachmentId = requiredString(row, "id");
@@ -5264,18 +5326,34 @@ export class Store {
             throw notFound("AI 对话");
         if (requiredString(conversation, "work_id") !== workId)
             throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
-        const rows = this.db.all("SELECT id, role, content, metadata_json FROM ai_conversation_messages WHERE conversation_id = ? ORDER BY created_at, rowid", conversationId);
-        const compactedMessageCount = Math.min(rows.length, Math.max(0, numberValue(conversation, "compacted_message_count")));
+        const countRow = this.db.get("SELECT COUNT(*) AS count FROM ai_conversation_messages WHERE conversation_id = ?", conversationId);
+        const totalMessageCount = numberValue(countRow ?? {}, "count");
+        const compactedMessageCount = Math.min(totalMessageCount, Math.max(0, numberValue(conversation, "compacted_message_count")));
+        const tailMessageCount = totalMessageCount - compactedMessageCount;
+        let rows = [];
+        if (tailMessageCount > 0 && compactedMessageCount === 0) {
+            rows = this.db.all(`SELECT id, role, content, metadata_json FROM ai_conversation_messages
+         WHERE conversation_id = ? ORDER BY created_at, rowid LIMIT ?`, conversationId, tailMessageCount);
+        }
+        else if (tailMessageCount > 0) {
+            const boundary = this.db.get(`SELECT created_at, rowid FROM ai_conversation_messages
+         WHERE conversation_id = ? ORDER BY created_at, rowid LIMIT 1 OFFSET ?`, conversationId, compactedMessageCount - 1);
+            if (boundary) {
+                rows = this.db.all(`SELECT id, role, content, metadata_json FROM ai_conversation_messages
+           WHERE conversation_id = ?
+             AND (created_at > ? OR (created_at = ? AND rowid > ?))
+           ORDER BY created_at, rowid LIMIT ?`, conversationId, requiredString(boundary, "created_at"), requiredString(boundary, "created_at"), numberValue(boundary, "rowid"), tailMessageCount);
+            }
+        }
         return {
             workId,
             roleplayCharacterId: optionalString(conversation, "roleplay_character_id"),
             summary: requiredString(conversation, "compacted_summary"),
             compactedMessageCount,
-            totalMessageCount: rows.length,
+            totalMessageCount,
             warningPending: Boolean(optionalString(conversation, "context_warning_at")),
             injectedEntities: parseAiInjectedEntities(optionalString(conversation, "injected_entities_json") ?? EMPTY_AI_INJECTED_ENTITIES),
-            messages: rows.slice(compactedMessageCount)
-                .filter((message) => requiredString(message, "id") !== excludeMessageId)
+            messages: rows.filter((message) => requiredString(message, "id") !== excludeMessageId)
                 .map((message) => ({
                 id: requiredString(message, "id"),
                 role: requiredString(message, "role") === "assistant" ? "assistant" : "user",
@@ -7236,26 +7314,33 @@ export class Store {
             return [{ type: "none" }];
         return [{ type: "unknown", scope }];
     }
-    search(workId, query) {
+    search(workId, query, requestedTypes) {
         this.getWork(workId);
         const normalizedQuery = normalizeWorkSearchQuery(query);
         if (!normalizedQuery)
             return [];
         const pattern = `%${escapeSqlLikePattern(normalizedQuery)}%`;
-        const chapters = this.db.all("SELECT id, title, content, volume_id FROM chapters WHERE work_id = ? AND deleted_at IS NULL AND (title LIKE ? ESCAPE '\\' OR content LIKE ? ESCAPE '\\') LIMIT 50", workId, pattern, pattern);
-        const races = this.listRaces(workId).filter((race) => {
-            const lineage = race.lineage;
-            const effectiveSettings = race.effectiveSettings;
-            return [
-                race.name,
-                race.description,
-                ...race.settings,
-                ...lineage.map((item) => item.name),
-                ...effectiveSettings.flatMap((item) => [item.value, item.sourceRaceName])
-            ].join("\n").toLocaleLowerCase("zh-CN").includes(normalizedQuery);
-        }).slice(0, 50);
-        const settings = this.db.all("SELECT id, title, content, category FROM settings WHERE work_id = ? AND (title LIKE ? ESCAPE '\\' OR content LIKE ? ESCAPE '\\') LIMIT 50", workId, pattern, pattern);
-        const characters = this.db.all(`WITH RECURSIVE character_race_lineage(character_id, race_id, parent_race_id, name, path) AS (
+        const accepts = (type) => !requestedTypes || requestedTypes.has(type);
+        const chapters = accepts("chapter")
+            ? this.db.all("SELECT id, title, content, volume_id FROM chapters WHERE work_id = ? AND deleted_at IS NULL AND (title LIKE ? ESCAPE '\\' OR content LIKE ? ESCAPE '\\') LIMIT 50", workId, pattern, pattern)
+            : [];
+        const races = accepts("race")
+            ? this.listRaces(workId).filter((race) => {
+                const lineage = race.lineage;
+                const effectiveSettings = race.effectiveSettings;
+                return [
+                    race.name,
+                    race.description,
+                    ...race.settings,
+                    ...lineage.map((item) => item.name),
+                    ...effectiveSettings.flatMap((item) => [item.value, item.sourceRaceName])
+                ].join("\n").toLocaleLowerCase("zh-CN").includes(normalizedQuery);
+            }).slice(0, 50)
+            : [];
+        const settings = accepts("setting")
+            ? this.db.all("SELECT id, title, content, category FROM settings WHERE work_id = ? AND (title LIKE ? ESCAPE '\\' OR content LIKE ? ESCAPE '\\') LIMIT 50", workId, pattern, pattern)
+            : [];
+        const characters = accepts("character") ? this.db.all(`WITH RECURSIVE character_race_lineage(character_id, race_id, parent_race_id, name, path) AS (
          SELECT character.id, race.id, race.parent_race_id, race.name, race.name
          FROM characters character JOIN races race ON race.id = character.race_id
          WHERE character.work_id = ?
@@ -7271,9 +7356,11 @@ export class Store {
        WHERE character.work_id = ? AND character.merged_into_character_id IS NULL AND (
          character.name LIKE ? ESCAPE '\\' OR character.aliases_json LIKE ? ESCAPE '\\' OR character.species LIKE ? ESCAPE '\\'
          OR EXISTS (SELECT 1 FROM character_race_lineage lineage WHERE lineage.character_id = character.id AND lineage.name LIKE ? ESCAPE '\\')
-       ) LIMIT 50`, workId, workId, pattern, pattern, pattern, pattern);
-        const organizations = this.db.all("SELECT id, name, description, is_dissolved, settings_json FROM organizations WHERE work_id = ? AND (name LIKE ? ESCAPE '\\' OR description LIKE ? ESCAPE '\\' OR settings_json LIKE ? ESCAPE '\\') LIMIT 50", workId, pattern, pattern, pattern);
-        const characterSections = this.searchCharacterProfileSections(workId, normalizedQuery, 30);
+       ) LIMIT 50`, workId, workId, pattern, pattern, pattern, pattern) : [];
+        const organizations = accepts("organization")
+            ? this.db.all("SELECT id, name, description, is_dissolved, settings_json FROM organizations WHERE work_id = ? AND (name LIKE ? ESCAPE '\\' OR description LIKE ? ESCAPE '\\' OR settings_json LIKE ? ESCAPE '\\') LIMIT 50", workId, pattern, pattern, pattern)
+            : [];
+        const characterSections = accepts("character") ? this.searchCharacterProfileSections(workId, normalizedQuery, 30) : [];
         const snippet = (content) => {
             const index = content.toLocaleLowerCase().indexOf(normalizedQuery);
             const start = Math.max(0, index - 40);
@@ -7295,7 +7382,7 @@ export class Store {
                 title: `${String(section.characterName)} / ${String(section.title)}`,
                 snippet: snippet(String(section.contentMarkdown)),
                 sectionType: String(section.sectionType),
-                isDead: Boolean(this.getCharacter(String(section.characterId)).isDead)
+                isDead: Boolean(section.isDead)
             })),
             ...settings.map((row) => ({ type: "setting", id: requiredString(row, "id"), title: requiredString(row, "title"), snippet: snippet(requiredString(row, "content")), category: requiredString(row, "category") })),
             ...races.map((race) => {