@waterwx/dsh-novel-forge 1.3.2 → 1.4.0

package/lib/index.js

@@ -1,10 +1,10 @@
+import { homedir } from "node:os";
+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";
 import { appendFileSync, copyFileSync, existsSync, mkdirSync, readFileSync, readdirSync, rmSync, writeFileSync } from "node:fs";
-import { join } from "node:path";
 import { BlockAssembler, ReasoningEffortId, createAssistantMessage, createUserMessage } from "@deepseek-ai/dsh-llm";
-import { homedir } from "node:os";
 import { randomBytes } from "node:crypto";
 import { strFromU8, unzipSync } from "fflate";
 //#region src/assets.ts
@@ -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();
@@ -3525,9 +3632,10 @@ function removeBook(id) {
 function activeBookOutputDir() {
 	return activeBook(loadBookshelf())?.outputDir;
 }
-/** 默认输出目录推断：小说目录/书名。 */
+/** 默认输出目录推断：~/.dsh/novels/书名。 */
 function defaultOutputDirFor(bookName) {
-	return join("D:\\ryan work\\harness\\novels", bookName.replace(/[\\/:*?"<>|]/g, "").trim().slice(0, 40) || "未命名小说");
+	const clean = bookName.replace(/[\\/:*?"<>|]/g, "").trim().slice(0, 40) || "未命名小说";
+	return join(homedir(), ".dsh", "novels", clean);
 }
 //#endregion
 //#region src/run.ts
@@ -3877,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 扫描已生成章节，输出矛盾问题清单。 */
@@ -5728,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",
@@ -6651,6 +6815,8 @@ function makeRoutes(deps) {
 		bookshelfRoute,
 		bookshelfActivateRoute,
 		bookshelfRemoveRoute,
+		bookshelfImportDirRoute,
+		bookshelfImportTextRoute,
 		resetRoute,
 		auditRoute,
 		charactersRefreshRoute,
@@ -6730,6 +6896,14 @@ function makeRoutes(deps) {
 }
 //#endregion
 //#region src/index.ts
+/**
+* dsh-novel-forge — host half. Mounts the AI novel-forge workbench: docx
+* outline import, LLM chapter planning, chapter-by-chapter generation
+* (3000-4000 chars each), Markdown output into your chosen folder, and the
+* /api/dsh-novel-forge route family. The browser half (./client) renders the
+* workbench panel. Everything rides official NPM SDK packages — no dsh source
+* changes.
+*/
 /** Stable cordis plugin name. */
 const name = "novel-forge";
 /** Services required before the novel-forge surfaces can mount. */
@@ -6747,8 +6921,8 @@ const NOVEL_SETTINGS_NAMESPACE = settingsNamespace("dsh-novel-forge");
 const Config = z.object({
 	announceToAgent: z.boolean().default(true),
 	enabled: z.boolean().default(true),
-	outlinePath: z.string().default("D:\\ryan work\\harness\\novels\\归墟玉主\\《归墟玉主》全书大纲_重新排版版.docx"),
-	outputDir: z.string().default("D:\\ryan work\\harness\\novels\\归墟玉主"),
+	outlinePath: z.string().default(""),
+	outputDir: z.string().default(join(homedir(), ".dsh", "novels")),
 	provider: z.string().default("deepseek-official"),
 	model: z.string().default("deepseek-v4-flash"),
 	reasoningEffort: z.union([
@@ -6769,8 +6943,8 @@ const Config = z.object({
 });
 /** Schema defaults, re-read for hand-built test contexts. */
 const DEFAULT_ANNOUNCE = true;
-const DEFAULT_OUTLINE_PATH = "D:\\ryan work\\harness\\novels\\归墟玉主\\《归墟玉主》全书大纲_重新排版版.docx";
-const DEFAULT_OUTPUT_DIR = "D:\\ryan work\\harness\\novels\\归墟玉主";
+const DEFAULT_OUTLINE_PATH = "";
+const DEFAULT_OUTPUT_DIR = join(homedir(), ".dsh", "novels");
 const DEFAULT_PROVIDER = "deepseek-official";
 const DEFAULT_MODEL = "deepseek-v4-flash";
 const DEFAULT_REASONING_EFFORT = "off";
@@ -6783,7 +6957,7 @@ const DEFAULT_AUTO_REVIEW_AFTER_REVISE = true;
 /** Order of the announcement section within the tool-guidance band. */
 const SECTION_ORDER = 160;
 /** Model-facing announcement: plugin presence, capabilities, and limits. */
-const NOVEL_GUIDANCE = "本机已安装 dsh-novel-forge 插件（AI 编译小说工作台）：侧边栏「小说工坊」入口。能力：读取 docx 大纲（默认 D:\\ryan work\\harness\\novels\\归墟玉主 大纲）或粘贴大纲文本；用 LLM 提炼道藏（人设/世界观/金手指规则/写作红线，即设定圣经）；生成卷计划与章节计划；逐章调用 LLM 生成 3000-4000 字正文并保存为 Markdown（默认输出到 D:\\ryan work\\harness\\novels\\归墟玉主）；每章自动生成摘要（叙事记忆）、自动 AI 审稿（人设/设定/红线/文笔/爽点/逻辑），支持按审稿意见重写、去 AI 味润色、暗线（伏笔）管理、批量连写与全本导出（txt/md）。限制：生成消耗 LLM API 额度；输出目录与模型可在插件设置中修改；章节正文质量取决于大纲完整度。用户提到「小说 / 大纲 / 写小说 / 章节 / 审稿 / 润色 / 归墟玉主」时即指本插件，请据此协作。";
+const NOVEL_GUIDANCE = "本机已安装 dsh-novel-forge 插件（AI 编译小说工作台）：侧边栏「小说工坊」入口。能力：读取 docx 大纲或粘贴大纲文本；用 LLM 提炼道藏（人设/世界观/金手指规则/写作红线，即设定圣经）；生成卷计划与章节计划；逐章调用 LLM 生成 3000-4000 字正文并保存为 Markdown（默认输出到用户主目录 ~/.dsh/novels）；每章自动生成摘要（叙事记忆）、自动 AI 审稿（人设/设定/红线/文笔/爽点/逻辑），支持按审稿意见重写、去 AI 味润色、暗线（伏笔）管理、批量连写与全本导出（txt/md）。限制：生成消耗 LLM API 额度；输出目录与模型可在插件设置中修改；章节正文质量取决于大纲完整度。用户提到「小说 / 大纲 / 写小说 / 章节 / 审稿 / 润色」时即指本插件，请据此协作。";
 /** Resolve a config-like value into the full runtime config. */
 function resolveConfig(value) {
 	return {