@musnows/scriverse 0.7.4 → 0.7.6

package/dist/store.js

@@ -155,6 +155,9 @@ function settingsFromKnowledgeSections(sections) {
     return sections.map((section) => section.contentMarkdown).filter((content) => content.trim());
 }
 const CHAPTER_OUTLINE_BOARD_PREVIEW_LENGTH = 600;
+function chapterOutlineBoardLikePattern(value) {
+    return `%${value.normalize("NFKC").toLocaleLowerCase("zh-CN").replaceAll("\\", "\\\\").replaceAll("%", "\\%").replaceAll("_", "\\_")}%`;
+}
 export const versionedEntityTypes = [
     "work",
     "volume",
@@ -1013,20 +1016,24 @@ export class Store {
         return this.getWork(workId);
     }
     deleteWork(workId, expectedVersionNo) {
-        this.db.transaction(() => {
+        return this.db.transaction(() => {
             const current = this.getWork(workId);
             this.assertExpectedVersion("work", workId, expectedVersionNo, "作品", Number(current.versionNo));
             const timestamp = now();
+            const activeTaskIds = this.db.all("SELECT id FROM analysis_tasks WHERE work_id = ? AND status IN ('pending', 'running') ORDER BY id", workId).map((row) => requiredString(row, "id"));
             const versionNo = this.recordEntityVersion("work", workId, "delete", null, "删除作品（可恢复）", timestamp);
             this.db.run("UPDATE works SET version_no = ?, deleted_at = ?, updated_at = ? WHERE id = ?", versionNo, timestamp, timestamp, workId);
+            this.db.run(`UPDATE analysis_tasks SET status = 'expired', next_attempt_at = NULL, updated_at = ?
+         WHERE work_id = ? AND status IN ('pending', 'running')`, timestamp, workId);
+            this.db.run("DELETE FROM relationship_source_index_queue WHERE work_id = ?", workId);
             this.audit(workId, "work.deleted", "work", workId, {
                 title: current.title,
                 versionNo,
                 recoverable: true,
                 expiresAt: recycleBinExpiresAt(timestamp)
             });
+            return activeTaskIds;
         });
-        return [];
     }
     restoreWork(workId, expectedVersionNo) {
         const deleted = this.db.get("SELECT * FROM works WHERE id = ? AND deleted_at IS NOT NULL", workId);
@@ -2520,10 +2527,101 @@ export class Store {
             return null;
         return this.mapChapterOutline(row, chapter);
     }
-    getChapterOutlineBoard(workId) {
+    getChapterOutlineBoard(workId, filters, pagination) {
         this.getWork(workId);
         const previewLength = CHAPTER_OUTLINE_BOARD_PREVIEW_LENGTH;
-        const rows = this.db.all(`SELECT volume.id AS volume_id, volume.title AS volume_title, volume.sort_order AS volume_order,
+        const where = [
+            "chapter.work_id = ?",
+            "chapter.deleted_at IS NULL",
+            "volume.deleted_at IS NULL"
+        ];
+        const whereParams = [workId];
+        if (filters.volumeId) {
+            where.push("chapter.volume_id = ?");
+            whereParams.push(filters.volumeId);
+        }
+        if (filters.outlineStatus === "empty") {
+            where.push("outline.chapter_id IS NULL");
+        }
+        else if (filters.outlineStatus !== "all") {
+            where.push("outline.status = ?");
+            whereParams.push(filters.outlineStatus);
+        }
+        const associatedForeshadow = (statusSql = "") => `(
+      EXISTS (
+        SELECT 1 FROM foreshadow_occurrences occurrence
+        JOIN foreshadows foreshadow ON foreshadow.id = occurrence.foreshadow_id
+        WHERE occurrence.chapter_id = chapter.id AND foreshadow.work_id = chapter.work_id${statusSql}
+      ) OR EXISTS (
+        SELECT 1 FROM foreshadows foreshadow
+        WHERE foreshadow.work_id = chapter.work_id
+          AND foreshadow.planned_payoff_chapter_id = chapter.id${statusSql}
+      )
+    )`;
+        if (filters.foreshadowStatus === "none") {
+            where.push(`NOT ${associatedForeshadow()}`);
+        }
+        else if (filters.foreshadowStatus !== "all") {
+            const statusSql = filters.foreshadowStatus === "unresolved"
+                ? " AND foreshadow.status IN ('planned', 'planted')"
+                : filters.foreshadowStatus === "resolved"
+                    ? " AND foreshadow.status = 'resolved'"
+                    : " AND foreshadow.status = 'abandoned'";
+            where.push(associatedForeshadow(statusSql));
+        }
+        const trimmedQuery = filters.query.trim();
+        if (trimmedQuery) {
+            const pattern = chapterOutlineBoardLikePattern(trimmedQuery);
+            where.push(`(
+        lower(chapter.title) LIKE ? ESCAPE '\\'
+        OR lower(chapter.chapter_type) LIKE ? ESCAPE '\\'
+        OR lower(outline.goal) LIKE ? ESCAPE '\\'
+        OR lower(outline.conflict) LIKE ? ESCAPE '\\'
+        OR lower(outline.turning_point) LIKE ? ESCAPE '\\'
+        OR lower(outline.notes) LIKE ? ESCAPE '\\'
+        OR EXISTS (
+          SELECT 1 FROM foreshadow_occurrences occurrence
+          JOIN foreshadows foreshadow ON foreshadow.id = occurrence.foreshadow_id
+          WHERE occurrence.chapter_id = chapter.id AND foreshadow.work_id = chapter.work_id
+            AND lower(foreshadow.title) LIKE ? ESCAPE '\\'
+        )
+        OR EXISTS (
+          SELECT 1 FROM foreshadows foreshadow
+          WHERE foreshadow.work_id = chapter.work_id
+            AND foreshadow.planned_payoff_chapter_id = chapter.id
+            AND lower(foreshadow.title) LIKE ? ESCAPE '\\'
+        )
+      )`);
+            whereParams.push(...Array.from({ length: 8 }, () => pattern));
+        }
+        const whereSql = where.join(" AND ");
+        const associationCount = (statusSql = "") => `(
+      SELECT COUNT(*) FROM foreshadows sorted_foreshadow
+      WHERE sorted_foreshadow.work_id = chapter.work_id${statusSql}
+        AND (
+          sorted_foreshadow.planned_payoff_chapter_id = chapter.id
+          OR EXISTS (
+            SELECT 1 FROM foreshadow_occurrences sorted_occurrence
+            WHERE sorted_occurrence.foreshadow_id = sorted_foreshadow.id
+              AND sorted_occurrence.chapter_id = chapter.id
+          )
+        )
+    )`;
+        const chapterTreeOrder = "chapter.sort_order, chapter.created_at, chapter.id";
+        const chapterOrder = filters.sort === "status"
+            ? `CASE WHEN outline.chapter_id IS NULL THEN 0 WHEN outline.status = 'draft' THEN 1 WHEN outline.status = 'ready' THEN 2 ELSE 3 END, ${chapterTreeOrder}`
+            : filters.sort === "foreshadows"
+                ? `${associationCount(" AND sorted_foreshadow.status IN ('planned', 'planted')")} DESC, ${associationCount()} DESC, ${chapterTreeOrder}`
+                : filters.sort === "title"
+                    ? `chapter.title COLLATE NOCASE, ${chapterTreeOrder}`
+                    : chapterTreeOrder;
+        const orderSql = `volume.sort_order, volume.created_at, volume.id, ${chapterOrder}`;
+        const total = numberValue(this.db.get(`SELECT COUNT(*) AS count
+       FROM chapters chapter
+       JOIN volumes volume ON volume.id = chapter.volume_id AND volume.work_id = chapter.work_id
+       LEFT JOIN chapter_outlines outline ON outline.chapter_id = chapter.id
+       WHERE ${whereSql}`, ...whereParams) ?? {}, "count");
+        const pageRows = this.db.all(`SELECT volume.id AS volume_id, volume.title AS volume_title, volume.sort_order AS volume_order,
        chapter.id AS chapter_id, chapter.title AS chapter_title, chapter.chapter_type,
        chapter.sort_order AS chapter_order,
        outline.chapter_id AS outline_chapter_id,
@@ -2532,34 +2630,59 @@ export class Store {
        substr(outline.turning_point, 1, ?) AS turning_point, length(outline.turning_point) > ? AS turning_point_truncated,
        substr(outline.notes, 1, ?) AS notes, length(outline.notes) > ? AS notes_truncated,
        outline.status, outline.updated_at AS outline_updated_at
+       FROM chapters chapter
+       JOIN volumes volume ON volume.id = chapter.volume_id AND volume.work_id = chapter.work_id
+       LEFT JOIN chapter_outlines outline ON outline.chapter_id = chapter.id
+       WHERE ${whereSql}
+       ORDER BY ${orderSql}
+       LIMIT ? OFFSET ?`, previewLength, previewLength, previewLength, previewLength, previewLength, previewLength, previewLength, previewLength, ...whereParams, pagination.limit + 1, pagination.offset);
+        const chapterPage = paginated(pageRows, pagination, total);
+        const volumeRows = this.db.all(`SELECT volume.id, volume.title, volume.sort_order,
+       COUNT(chapter.id) AS chapter_count
        FROM volumes volume
        LEFT JOIN chapters chapter
          ON chapter.volume_id = volume.id AND chapter.work_id = volume.work_id AND chapter.deleted_at IS NULL
-       LEFT JOIN chapter_outlines outline ON outline.chapter_id = chapter.id
        WHERE volume.work_id = ? AND volume.deleted_at IS NULL
-       ORDER BY volume.sort_order, volume.created_at, volume.id,
-         chapter.sort_order, chapter.created_at, chapter.id`, previewLength, previewLength, previewLength, previewLength, previewLength, previewLength, previewLength, previewLength, workId);
+       GROUP BY volume.id
+       ORDER BY volume.sort_order, volume.created_at, volume.id`, workId);
+        const filteredVolumeRows = this.db.all(`SELECT chapter.volume_id, COUNT(*) AS chapter_count
+       FROM chapters chapter
+       JOIN volumes volume ON volume.id = chapter.volume_id AND volume.work_id = chapter.work_id
+       LEFT JOIN chapter_outlines outline ON outline.chapter_id = chapter.id
+       WHERE ${whereSql}
+       GROUP BY chapter.volume_id`, ...whereParams);
+        const filteredCountByVolume = new Map(filteredVolumeRows.map((row) => [
+            requiredString(row, "volume_id"),
+            numberValue(row, "chapter_count")
+        ]));
+        const volumeOptions = volumeRows.map((row) => ({
+            id: requiredString(row, "id"),
+            title: requiredString(row, "title"),
+            sortOrder: numberValue(row, "sort_order"),
+            chapterCount: numberValue(row, "chapter_count"),
+            filteredChapterCount: filteredCountByVolume.get(requiredString(row, "id")) ?? 0
+        }));
+        const volumeOptionById = new Map(volumeOptions.map((volume) => [volume.id, volume]));
+        const volumeOrderById = new Map(volumeOptions.map((volume, index) => [volume.id, index]));
         const volumeById = new Map();
         const chapterById = new Map();
-        let outlinedChapterCount = 0;
-        for (const row of rows) {
+        for (const row of chapterPage.items) {
             const volumeId = requiredString(row, "volume_id");
             let volume = volumeById.get(volumeId);
             if (!volume) {
+                const summary = volumeOptionById.get(volumeId);
                 volume = {
                     id: volumeId,
                     title: requiredString(row, "volume_title"),
                     sortOrder: numberValue(row, "volume_order"),
+                    chapterCount: summary?.chapterCount ?? 0,
+                    filteredChapterCount: summary?.filteredChapterCount ?? 0,
                     chapters: []
                 };
                 volumeById.set(volumeId, volume);
             }
-            const chapterId = optionalString(row, "chapter_id");
-            if (!chapterId)
-                continue;
+            const chapterId = requiredString(row, "chapter_id");
             const hasOutline = optionalString(row, "outline_chapter_id") !== null;
-            if (hasOutline)
-                outlinedChapterCount += 1;
             const chapter = {
                 id: chapterId,
                 title: requiredString(row, "chapter_title"),
@@ -2580,6 +2703,7 @@ export class Store {
             volume.chapters.push(chapter);
             chapterById.set(chapterId, chapter);
         }
+        const chapterIds = [...chapterById.keys()];
         const associations = new Map();
         const associate = (chapterId, source, role, plannedPayoff = false) => {
             if (!chapterId || !chapterById.has(chapterId))
@@ -2597,28 +2721,29 @@ export class Store {
             byForeshadow.set(source.id, summary);
             associations.set(chapterId, byForeshadow);
         };
-        const foreshadowRows = this.db.all(`SELECT foreshadow.id, foreshadow.title, foreshadow.status, foreshadow.importance,
-       occurrence_chapter.id AS occurrence_chapter_id, occurrence_volume.id AS occurrence_volume_id,
-       occurrence.role, payoff_chapter.id AS payoff_chapter_id, payoff_volume.id AS payoff_volume_id
-       FROM foreshadows foreshadow
-       LEFT JOIN foreshadow_occurrences occurrence ON occurrence.foreshadow_id = foreshadow.id
-       LEFT JOIN chapters occurrence_chapter
-         ON occurrence_chapter.id = occurrence.chapter_id
-         AND occurrence_chapter.work_id = foreshadow.work_id
-         AND occurrence_chapter.deleted_at IS NULL
-       LEFT JOIN volumes occurrence_volume
-         ON occurrence_volume.id = occurrence_chapter.volume_id AND occurrence_volume.deleted_at IS NULL
-       LEFT JOIN chapters payoff_chapter
-         ON payoff_chapter.id = foreshadow.planned_payoff_chapter_id
-         AND payoff_chapter.work_id = foreshadow.work_id
-         AND payoff_chapter.deleted_at IS NULL
-       LEFT JOIN volumes payoff_volume
-         ON payoff_volume.id = payoff_chapter.volume_id AND payoff_volume.deleted_at IS NULL
-       WHERE foreshadow.work_id = ?
-       ORDER BY CASE foreshadow.importance WHEN 'high' THEN 0 WHEN 'medium' THEN 1 ELSE 2 END,
-         foreshadow.created_at, foreshadow.id, occurrence.created_at, occurrence.id`, workId);
-        const foreshadowIds = new Set();
-        const unresolvedForeshadowIds = new Set();
+        const placeholders = chapterIds.map(() => "?").join(", ");
+        const foreshadowRows = chapterIds.length === 0 ? [] : this.db.all(`SELECT association.* FROM (
+         SELECT chapter.id AS chapter_id, foreshadow.id, foreshadow.title, foreshadow.status,
+           foreshadow.importance, foreshadow.created_at AS foreshadow_created_at,
+           occurrence.role, 0 AS planned_payoff, occurrence.created_at AS association_created_at,
+           occurrence.id AS association_id
+         FROM chapters chapter
+         JOIN foreshadow_occurrences occurrence ON occurrence.chapter_id = chapter.id
+         JOIN foreshadows foreshadow ON foreshadow.id = occurrence.foreshadow_id
+         WHERE chapter.id IN (${placeholders}) AND chapter.work_id = ? AND chapter.deleted_at IS NULL
+           AND foreshadow.work_id = ?
+         UNION ALL
+         SELECT chapter.id AS chapter_id, foreshadow.id, foreshadow.title, foreshadow.status,
+           foreshadow.importance, foreshadow.created_at AS foreshadow_created_at,
+           NULL AS role, 1 AS planned_payoff, foreshadow.created_at AS association_created_at,
+           foreshadow.id AS association_id
+         FROM chapters chapter
+         JOIN foreshadows foreshadow ON foreshadow.planned_payoff_chapter_id = chapter.id
+         WHERE chapter.id IN (${placeholders}) AND chapter.work_id = ? AND chapter.deleted_at IS NULL
+           AND foreshadow.work_id = ?
+       ) association
+       ORDER BY CASE association.importance WHEN 'high' THEN 0 WHEN 'medium' THEN 1 ELSE 2 END,
+         association.foreshadow_created_at, association.id, association.association_created_at, association.association_id`, ...chapterIds, workId, workId, ...chapterIds, workId, workId);
         for (const row of foreshadowRows) {
             const source = {
                 id: requiredString(row, "id"),
@@ -2626,15 +2751,8 @@ export class Store {
                 status: requiredString(row, "status"),
                 importance: requiredString(row, "importance")
             };
-            foreshadowIds.add(source.id);
-            if (source.status === "planned" || source.status === "planted")
-                unresolvedForeshadowIds.add(source.id);
-            const occurrenceChapterId = optionalString(row, "occurrence_volume_id")
-                ? optionalString(row, "occurrence_chapter_id")
-                : null;
             const role = optionalString(row, "role");
-            associate(occurrenceChapterId, source, role === "setup" || role === "reminder" || role === "payoff" ? role : undefined);
-            associate(optionalString(row, "payoff_volume_id") ? optionalString(row, "payoff_chapter_id") : null, source, undefined, true);
+            associate(requiredString(row, "chapter_id"), source, role === "setup" || role === "reminder" || role === "payoff" ? role : undefined, booleanValue(row, "planned_payoff"));
         }
         for (const [chapterId, byForeshadow] of associations) {
             const chapter = chapterById.get(chapterId);
@@ -2645,14 +2763,47 @@ export class Store {
                 roles: [...summary.roles]
             }));
         }
+        const chapterFiltersActive = Boolean(trimmedQuery || filters.outlineStatus !== "all" || filters.foreshadowStatus !== "all");
+        if (pagination.page === 1) {
+            for (const option of volumeOptions) {
+                if (option.chapterCount !== 0)
+                    continue;
+                if (filters.volumeId && option.id !== filters.volumeId)
+                    continue;
+                if (chapterFiltersActive && filters.volumeId !== option.id)
+                    continue;
+                volumeById.set(option.id, { ...option, chapters: [] });
+            }
+        }
+        const volumes = [...volumeById.values()].sort((left, right) => ((volumeOrderById.get(left.id) ?? Number.MAX_SAFE_INTEGER) - (volumeOrderById.get(right.id) ?? Number.MAX_SAFE_INTEGER)));
+        const statsRow = this.db.get(`SELECT
+       (SELECT COUNT(*) FROM chapters chapter
+        JOIN volumes volume ON volume.id = chapter.volume_id AND volume.work_id = chapter.work_id
+        WHERE chapter.work_id = ? AND chapter.deleted_at IS NULL AND volume.deleted_at IS NULL) AS chapter_count,
+       (SELECT COUNT(*) FROM chapter_outlines outline
+        JOIN chapters chapter ON chapter.id = outline.chapter_id
+        JOIN volumes volume ON volume.id = chapter.volume_id AND volume.work_id = chapter.work_id
+        WHERE chapter.work_id = ? AND chapter.deleted_at IS NULL AND volume.deleted_at IS NULL) AS outlined_chapter_count,
+       (SELECT COUNT(*) FROM foreshadows foreshadow WHERE foreshadow.work_id = ?) AS foreshadow_count,
+       (SELECT COUNT(*) FROM foreshadows foreshadow
+        WHERE foreshadow.work_id = ? AND foreshadow.status IN ('planned', 'planted')) AS unresolved_foreshadow_count`, workId, workId, workId, workId) ?? {};
         return {
             workId,
-            volumes: [...volumeById.values()],
+            volumes,
+            volumeOptions,
+            filters: { ...filters, query: trimmedQuery },
+            page: chapterPage.page,
+            limit: chapterPage.limit,
+            itemCount: chapterById.size,
+            total,
+            pageCount: Math.max(1, Math.ceil(total / pagination.limit)),
+            hasMore: chapterPage.hasMore,
+            nextPage: chapterPage.nextPage,
             stats: {
-                chapterCount: chapterById.size,
-                outlinedChapterCount,
-                foreshadowCount: foreshadowIds.size,
-                unresolvedForeshadowCount: unresolvedForeshadowIds.size
+                chapterCount: numberValue(statsRow, "chapter_count"),
+                outlinedChapterCount: numberValue(statsRow, "outlined_chapter_count"),
+                foreshadowCount: numberValue(statsRow, "foreshadow_count"),
+                unresolvedForeshadowCount: numberValue(statsRow, "unresolved_foreshadow_count")
             }
         };
     }
@@ -5812,7 +5963,10 @@ export class Store {
         return numberValue(row ?? {}, "value");
     }
     listAutoRunWorkIds() {
-        return this.db.all("SELECT work_id FROM work_ai_settings WHERE auto_run_enabled = 1 ORDER BY work_id").map((row) => requiredString(row, "work_id"));
+        return this.db.all(`SELECT settings.work_id FROM work_ai_settings settings
+       JOIN works work ON work.id = settings.work_id
+       WHERE settings.auto_run_enabled = 1 AND work.deleted_at IS NULL
+       ORDER BY settings.work_id`).map((row) => requiredString(row, "work_id"));
     }
     claimPendingTask(taskId, runningLimit) {
         return this.db.transaction(() => {
@@ -7185,24 +7339,46 @@ export class Store {
         });
     }
     async exportEpub(workId, volumeId) {
-        const tree = this.getWorkTree(workId);
-        const allVolumes = tree.volumes;
-        const selectedVolume = volumeId ? allVolumes.find((volume) => String(volume.id) === volumeId) : undefined;
+        const work = this.getWork(workId);
+        const allVolumeRows = this.db.all("SELECT id, title FROM volumes WHERE work_id = ? AND deleted_at IS NULL ORDER BY sort_order, created_at", workId);
+        const selectedVolume = volumeId ? allVolumeRows.find((volume) => requiredString(volume, "id") === volumeId) : undefined;
         if (volumeId && !selectedVolume)
             throw notFound("分卷");
-        const sourceVolumes = selectedVolume ? [selectedVolume] : allVolumes;
-        const title = selectedVolume ? `${String(tree.title)} - ${String(selectedVolume.title)}` : String(tree.title);
+        const sourceVolumeRows = selectedVolume ? [selectedVolume] : allVolumeRows;
+        const chapterRows = volumeId
+            ? this.db.all(`SELECT id, volume_id, title, version_no FROM chapters
+         WHERE work_id = ? AND volume_id = ? AND deleted_at IS NULL ORDER BY sort_order, created_at`, workId, volumeId)
+            : this.db.all(`SELECT id, volume_id, title, version_no FROM chapters
+         WHERE work_id = ? AND deleted_at IS NULL ORDER BY sort_order, created_at`, workId);
+        const chaptersByVolume = new Map();
+        for (const chapter of chapterRows) {
+            const chapterVolumeId = requiredString(chapter, "volume_id");
+            const chapters = chaptersByVolume.get(chapterVolumeId) ?? [];
+            chapters.push(chapter);
+            chaptersByVolume.set(chapterVolumeId, chapters);
+        }
+        const title = selectedVolume ? `${String(work.title)} - ${requiredString(selectedVolume, "title")}` : String(work.title);
         const cover = this.findWorkCover(workId);
         const archive = await createEpubArchive({
             title,
-            author: String(tree.author ?? ""),
-            description: String(tree.description ?? ""),
-            language: String(tree.language ?? "zh-CN"),
-            volumes: sourceVolumes.map((volume) => ({
-                title: String(volume.title),
-                chapters: volume.chapters.map((chapter) => ({
-                    title: String(chapter.title),
-                    content: String(chapter.content ?? "")
+            author: String(work.author ?? ""),
+            description: String(work.description ?? ""),
+            language: String(work.language ?? "zh-CN"),
+            volumes: sourceVolumeRows.map((volume) => ({
+                title: requiredString(volume, "title"),
+                chapters: (chaptersByVolume.get(requiredString(volume, "id")) ?? []).map((chapter) => ({
+                    title: requiredString(chapter, "title"),
+                    content: () => {
+                        const chapterId = requiredString(chapter, "id");
+                        const versionNo = numberValue(chapter, "version_no");
+                        const current = this.db.get("SELECT content FROM chapters WHERE id = ? AND work_id = ? AND version_no = ? AND deleted_at IS NULL", chapterId, workId, versionNo);
+                        if (current)
+                            return requiredString(current, "content");
+                        const historical = this.db.get("SELECT content FROM chapter_versions WHERE chapter_id = ? AND work_id = ? AND version_no = ?", chapterId, workId, versionNo);
+                        if (!historical)
+                            throw new AppError(409, "EPUB_EXPORT_SOURCE_CHANGED", "导出过程中章节版本已不可用，请重新导出");
+                        return requiredString(historical, "content");
+                    }
                 }))
             })),
             cover: cover ? { mimeType: cover.mimeType, content: cover.content } : null