@waterwx/dsh-novel-forge 1.3.3 → 1.4.0

package/lib/index.js

@@ -1,5 +1,5 @@
 import { homedir } from "node:os";
-import { join } from "node:path";
+import { basename, extname, join } from "node:path";
 import { installSettingsSection, settingsNamespace } from "@deepseek-ai/dsh-settings";
 import z from "schemastery";
 import { exec } from "node:child_process";
@@ -1948,6 +1948,94 @@ async function extractScenes(ctx, config, project) {
 	}
 	return scenes;
 }
+/** 从 txt/md 全本拆分章节并建立项目：正文落盘为章节文件，status=written（待审稿）。 */
+function importBookText(filePath, outputDir) {
+	const lines = readFileSync(filePath, "utf8").split(/\r?\n/);
+	const chapterHead = /^\s*(?:#\s*)?第\s*(\d+|[一二三四五六七八九十百千]+)\s*[章回节卷]\s*(.*?)\s*$/;
+	const cnNum = {
+		一: 1,
+		二: 2,
+		三: 3,
+		四: 4,
+		五: 5,
+		六: 6,
+		七: 7,
+		八: 8,
+		九: 9,
+		十: 10,
+		百: 100,
+		千: 1e3
+	};
+	const parseCn = (s) => {
+		if (/^\d+$/.test(s)) return Number(s);
+		let total = 0;
+		let section = 0;
+		for (const ch of s) {
+			const v = cnNum[ch];
+			if (v === void 0) return 0;
+			if (v >= 10) {
+				total += (section > 0 ? section : 1) * v;
+				section = 0;
+			} else section = v;
+		}
+		return total + section;
+	};
+	const chunks = [];
+	let current = null;
+	for (const line of lines) {
+		const m = chapterHead.exec(line);
+		if (m !== null) {
+			if (current !== null) chunks.push(current);
+			const no = parseCn(m[1]);
+			const title = (m[2] ?? "").trim() || "";
+			current = {
+				no: no > 0 ? no : chunks.length + 1,
+				title,
+				body: []
+			};
+		} else if (current !== null) current.body.push(line);
+	}
+	if (current !== null) chunks.push(current);
+	if (chunks.length === 0) throw new Error("未识别到章节（需要\"第X章\"格式，或带 # 的章节标题）");
+	const seen = /* @__PURE__ */ new Set();
+	const ordered = chunks.filter((c) => {
+		if (seen.has(c.no)) return false;
+		seen.add(c.no);
+		return true;
+	}).sort((a, b) => a.no - b.no);
+	const bookName = basename(filePath, extname(filePath)).slice(0, 40) || "导入小说";
+	const project = createProject(bookName);
+	mkdirSync(outputDir, { recursive: true });
+	const skipped = [];
+	for (const c of ordered) {
+		const body = c.body.join("\n").trim();
+		if (body.length < 50) {
+			skipped.push("第" + c.no + "章" + (c.title !== "" ? "「" + c.title + "」" : "") + "（内容过短，已跳过）");
+			continue;
+		}
+		const chapter = {
+			no: c.no,
+			volume: 0,
+			title: c.title !== "" ? c.title : "第" + c.no + "章",
+			beats: "",
+			targetChars: 0,
+			status: "written",
+			file: "",
+			chars: 0
+		};
+		chapter.file = chapterFileName(chapter);
+		writeFileSync(join(outputDir, chapter.file), "# 第" + c.no + "章 " + chapter.title + "\n\n" + body + "\n", "utf8");
+		chapter.chars = body.length;
+		project.chapters.push(chapter);
+	}
+	project.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+	saveProject(outputDir, project);
+	return {
+		bookName,
+		chapters: project.chapters.length,
+		skipped
+	};
+}
 /**
 * 为单个角色提炼「动漫形象描述词」：扫描该角色出场的已写章节正文，
 * 截取含外貌描写的段落，交给 LLM 提炼中文描述 + 英文绘图标签。
@@ -3501,6 +3589,25 @@ function seedBookshelfFromOutputDir(outputDir) {
 	createBook(loadProject(outputDir)?.bookName ?? outputDir.split(/[\\/]/).pop() ?? "未命名小说", outputDir);
 	return true;
 }
+/** 导入已有项目目录到书架：校验 novel-project.json，已存在则直接激活。 */
+function importDir(outputDir) {
+	const project = loadProject(outputDir);
+	if (project === void 0) throw new Error(`目录中未找到有效的 novel-project.json：${outputDir}`);
+	const store = loadBookshelf();
+	const existed = store.books.find((b) => b.outputDir === outputDir);
+	if (existed !== void 0) {
+		store.activeBookId = existed.id;
+		saveBookshelf(store);
+		return {
+			book: existed,
+			existed: true
+		};
+	}
+	return {
+		book: createBook(project.bookName !== "" ? project.bookName : outputDir.split(/[\\/]/).pop() ?? "未命名小说", outputDir),
+		existed: false
+	};
+}
 /** 激活一本书。 */
 function activateBook(id) {
 	const store = loadBookshelf();
@@ -3878,6 +3985,10 @@ const NOVEL_API = {
 	/** 清空助手对话记录。 */
 	assistantClear: "/api/dsh-novel-forge/assistant/clear",
 	bookshelf: "/api/dsh-novel-forge/bookshelf",
+	/** 导入已有项目目录（含 novel-project.json）到书架。 */
+	bookshelfImportDir: "/api/dsh-novel-forge/bookshelf/import-dir",
+	/** 导入 txt/md 全本：拆章建项目并登记书架。 */
+	bookshelfImportText: "/api/dsh-novel-forge/bookshelf/import-text",
 	/** 重置项目（可选携带新大纲）：清空设定/卷/章节/伏笔/资产/事实库。 */
 	reset: "/api/dsh-novel-forge/reset",
 	/** 全书一致性质检：LLM 扫描已生成章节，输出矛盾问题清单。 */
@@ -5729,6 +5840,58 @@ function makeRoutes(deps) {
 			writeJson(res, 200, bookshelfSnapshot(loadBookshelf()));
 		}
 	};
+	/** 导入已有项目目录（Mode A）：校验 novel-project.json，登记/激活书架。 */
+	const bookshelfImportDirRoute = {
+		kind: "exact",
+		path: NOVEL_API.bookshelfImportDir,
+		handler: async (req, res) => {
+			if (!guard(req, res, "POST")) return;
+			const outputDir = (await readJsonBody(req))?.outputDir?.trim();
+			if (outputDir === void 0 || outputDir === "") {
+				writeJson(res, 400, { error: "outputDir 不能为空" });
+				return;
+			}
+			try {
+				const { book, existed } = importDir(outputDir);
+				writeJson(res, 200, {
+					book,
+					existed
+				});
+			} catch (err) {
+				writeJson(res, 400, { error: err instanceof Error ? err.message : String(err) });
+			}
+		}
+	};
+	/** 导入 txt/md 全本（Mode B）：拆章落盘建项目，登记书架。 */
+	const bookshelfImportTextRoute = {
+		kind: "exact",
+		path: NOVEL_API.bookshelfImportText,
+		handler: async (req, res) => {
+			if (!guard(req, res, "POST")) return;
+			const body = await readJsonBody(req);
+			const filePath = body?.filePath?.trim();
+			if (filePath === void 0 || filePath === "") {
+				writeJson(res, 400, { error: "filePath 不能为空" });
+				return;
+			}
+			if (!existsSync(filePath)) {
+				writeJson(res, 400, { error: `文件不存在：${filePath}` });
+				return;
+			}
+			try {
+				const bookName = basename(filePath, extname(filePath)).slice(0, 40) || "导入小说";
+				const outDir = body?.outputDir?.trim() !== void 0 && body.outputDir.trim() !== "" ? body.outputDir.trim() : defaultOutputDirFor(bookName);
+				const result = importBookText(filePath, outDir);
+				const { book } = importDir(outDir);
+				writeJson(res, 200, {
+					...result,
+					book
+				});
+			} catch (err) {
+				writeJson(res, 400, { error: err instanceof Error ? err.message : String(err) });
+			}
+		}
+	};
 	/** 全书一致性质检。 */
 	const auditRoute = {
 		kind: "exact",
@@ -6652,6 +6815,8 @@ function makeRoutes(deps) {
 		bookshelfRoute,
 		bookshelfActivateRoute,
 		bookshelfRemoveRoute,
+		bookshelfImportDirRoute,
+		bookshelfImportTextRoute,
 		resetRoute,
 		auditRoute,
 		charactersRefreshRoute,