@waterwx/dsh-novel-forge 1.1.5 → 1.1.7

package/lib/index.js

@@ -71,6 +71,8 @@ const NOVEL_API = {
 	reviewBackfill: "/api/dsh-novel-forge/review/backfill",
 	/** 章节复位：generating 卡死 → pending（可重新生成）。 */
 	chapterReset: "/api/dsh-novel-forge/chapter/reset",
+	/** 章节直接通过：作者对 rejected/written 章节行使最终决定权。 */
+	chapterApprove: "/api/dsh-novel-forge/chapter/approve",
 	/** 敏感词检查：全书已写章节或指定文本。 */
 	sensitiveCheck: "/api/dsh-novel-forge/sensitive-check",
 	config: "/api/dsh-novel-forge/config",
@@ -791,6 +793,28 @@ function saveProject(outputDir, project) {
 	writeFileSync(join(outputDir, PROJECT_FILE), JSON.stringify(project, null, 2), "utf8");
 }
 /**
+* 并发保护：长任务（章节计划生成/正文生成）在内存中持有旧快照，
+* 期间其他请求可能修改了「易变字段」（道藏/角色库/剧情线/人物志存档/简介/封面）。
+* 保存前用磁盘最新版本合并这些字段，避免旧快照覆盖新修改（曾导致角色卡丢失）。
+* 注意：调用方若自己修改了这些字段，不要使用本函数。
+*/
+function mergeVolatileFromDisk(outputDir, project) {
+	try {
+		const disk = loadProject(outputDir);
+		if (disk === void 0) return;
+		project.bible = disk.bible;
+		project.roles = disk.roles;
+		project.plotlines = disk.plotlines;
+		project.roleStatus = disk.roleStatus;
+		project.blurb = disk.blurb;
+		project.coverPath = disk.coverPath;
+		project.facts = disk.facts;
+		project.assets = disk.assets;
+		project.world = disk.world;
+		project.volumes = disk.volumes;
+	} catch {}
+}
+/**
 * 内置违禁词库（网文平台常见审查类别）。只做硬匹配提示，不代替人工判断。
 * 词语刻意保持常见写法；作者可自行判断是否修改。
 */
@@ -2131,15 +2155,32 @@ async function* generateChapterStream(ctx, config, project, outputDir, chapterNo
 	const prevSummary = prev?.summary;
 	const allFacts = project.facts ?? [];
 	const recentFacts = allFacts.slice(-20).map((f) => f.text);
+	const recentSet = new Set(recentFacts);
 	const beatsText = chapter.beats;
-	const relatedFacts = allFacts.slice(-120).filter((f) => {
-		const head = f.text.slice(0, 24);
-		for (let i = 0; i + 3 <= head.length; i++) {
-			const tri = head.slice(i, i + 3);
-			if (tri.trim() !== "" && beatsText.includes(tri)) return true;
+	const roleNames = (project.roles ?? []).map((r) => r.name).filter((n) => typeof n === "string" && n !== "");
+	const trigrams = (s) => {
+		const out = /* @__PURE__ */ new Set();
+		for (let i = 0; i + 3 <= s.length; i++) {
+			const tri = s.slice(i, i + 3);
+			if (tri.trim() !== "") out.add(tri);
 		}
-		return false;
-	}).slice(-15).map((f) => `[第${f.chapterNo}章] ${f.text}`);
+		return out;
+	};
+	const beatsTri = trigrams(beatsText);
+	const beatRoles = roleNames.filter((n) => beatsText.includes(n));
+	const relatedFacts = allFacts.map((f, idx) => {
+		const head = f.text.slice(0, 80);
+		let score = 0;
+		for (const tri of trigrams(head)) if (beatsTri.has(tri)) score += 1;
+		if (beatRoles.length > 0) {
+			for (const n of beatRoles) if (head.includes(n)) score += 8;
+		}
+		score += Math.min(idx, 40) / 10;
+		return {
+			f,
+			score
+		};
+	}).filter((x) => x.score >= 3).sort((a, b) => b.score - a.score).slice(0, 15).map((x) => `[第${x.f.chapterNo}章] ${x.f.text}`).filter((t) => !recentSet.has(t.slice(t.indexOf("]") + 2)));
 	const messages = [createUserMessage({
 		content: [{
 			type: "text",
@@ -2591,7 +2632,11 @@ async function backfillFacts(ctx, config, project, outputDir) {
 * 最近出现章节），LLM 只负责聚合「当前状态」一句话。
 */
 async function refreshCharacters(ctx, config, project, outputDir) {
-	const roster = project.bible?.characters ?? [];
+	const roster = (((project.roles ?? []).length > 0 ? project.roles : project.bible?.characters) ?? []).map((r) => ({
+		name: r.name,
+		traits: r.traits ?? [],
+		role: r.roleLabel !== void 0 ? r.roleLabel : r.role ?? "other"
+	}));
 	const facts = project.facts ?? [];
 	if (roster.length === 0 && facts.length === 0) return [];
 	const stat = /* @__PURE__ */ new Map();
@@ -2922,6 +2967,7 @@ function assistantSystemPrompt(project) {
 		"- 中文回复；文字量适中，别啰嗦。",
 		"",
 		"使用规则（非常重要）：",
+		"- 写操作（chapter_generate / chapter_rewrite / chapter_review / outline_replace / bible_set_* / foreshadow_* / assets_set_* / export_txt）只有在作者明确要求时才能调用——例如作者说\"生成第 120 章\"\"把第 105 章结尾改一下\"\"帮我审一下第 88 章\"。作者只是提问、闲聊、查信息时，一律用文字回答，禁止调用任何写操作，也不要先斩后奏（如\"为了回答你，我先把第 X 章生成了\"）。",
 		"- 当你想执行任何工具时，你的【整个回复】必须只包含动作指令标签，格式如下（不要有任何解释文字、不要用自然语言说\"我要去改\"，直接输出标签）：",
 		"  正确示例：<dsh-action name=\"outline_replace\">{\"old\":\"要替换的原文\",\"new\":\"新文本\"}</dsh-action>",
 		"  正确示例：<dsh-action name=\"chapter_text\">{\"no\":1}</dsh-action>",
@@ -3276,6 +3322,33 @@ async function* runAssistantTurn(ctx, config, project, outputDir, userMessage) {
 	const MAX_NUDGES = 6;
 	/** 连续收到 hex 乱码回复的次数（≥2 次判定 LLM 侧异常，放弃本轮避免死循环）。 */
 	let garbleCount = 0;
+	const WRITE_TOOL_KEYS = {
+		chapter_generate: /(生成|写第\s*\d+\s*章|写一[章篇]|新写|续写|接着写|继续写|开始写|写正文|写书|创作)/,
+		chapter_rewrite: /(重写|改写|修订|修改|改一下|调整|替换|润色|优化|修正|完善|回炉|换一种|从头)/,
+		chapter_review: /(审|检查|校验|点评|评估|把关|质量|怎么样|如何)/,
+		outline_replace: /(大纲|总纲|简介)/,
+		bible_set_rule: /(道藏|设定|规则|红线|世界|金手指)/,
+		bible_set_redline: /(道藏|设定|红线)/,
+		foreshadow_add: /(暗线|伏笔|埋)/,
+		foreshadow_update: /(暗线|伏笔)/,
+		export_txt: /(导出|打包|下载|txt)/,
+		assets_set_genre: /(题材)/,
+		assets_set_progression: /(推进)/,
+		assets_add_rule: /(规则|文戒|反AI)/
+	};
+	/** 本轮已放行过写操作：后续写操作（生成→审稿→修订闭环）不再逐个拦截。 */
+	let writeUnlocked = false;
+	const guardWrite = (name, userMessage) => {
+		if (writeUnlocked) return true;
+		const key = WRITE_TOOL_KEYS[name];
+		if (key === void 0) return true;
+		const recentUsers = history.filter((m) => m.role === "user").slice(-2).map((m) => m.content).join("\n");
+		if (key.test(recentUsers)) {
+			writeUnlocked = true;
+			return true;
+		}
+		return false;
+	};
 	for (;;) {
 		if (round++ > 20) break;
 		const reply = await chatOnce(ctx, config, system, history);
@@ -3303,7 +3376,8 @@ async function* runAssistantTurn(ctx, config, project, outputDir, userMessage) {
 			const mentionsTarget = /(编年录|道藏|暗线|总纲|卷首语|章节|正文|规则|红线|伏笔|简介|大纲|事实|设定|世界|角色|人物|第\s*\d+\s*章)/.test(reply);
 			const strayTag = /<[a-z_-]*action[^>]*>/.test(reply);
 			if ((intendsAction || mentionsTarget || strayTag) && nudged < MAX_NUDGES) {
-				const nudge = nudged === 0 ? "你的上一条回复表达了想操作项目的意图（或动作标签格式有误），因此没有执行任何操作。请直接输出动作标签来执行，格式必须为 <dsh-action name=\"工具名\">{\"参数\":值}</dsh-action>（注意拼写是 dsh-action，不是 dash-action；标签成对出现，参数为合法 JSON）。如果需要先看内容，先输出 outline_text 或 chapter_text 标签。" : `你第 ${nudged + 1} 次表达了操作意图但没有输出动作标签，因此仍未执行任何操作。铁律：你的【整个回复】现在必须只包含一个 <dsh-action> 标签（例如 <dsh-action name="chapter_text">{"no":1}</dsh-action>），禁止任何解释、铺垫或"我这就去"之类的文字。若你其实不打算执行任何操作，请明确回复「不执行」。`;
+				const userWriteIntent = Object.values(WRITE_TOOL_KEYS).some((re) => re.test(userMessage));
+				const nudge = nudged === 0 ? userWriteIntent ? "你的上一条回复表达了想操作项目的意图（或动作标签格式有误），因此没有执行任何操作。请直接输出动作标签来执行，格式必须为 <dsh-action name=\"工具名\">{\"参数\":值}</dsh-action>（注意拼写是 dsh-action，不是 dash-action；标签成对出现，参数为合法 JSON）。如果需要先看内容，先输出 outline_text 或 chapter_text 标签。" : "你刚才的回复看起来在讨论项目内容，但没有必要执行任何操作。如果用户只是在提问或闲聊，请直接以文字回答即可，不要输出动作标签，也不要自行调用任何写操作（生成/修订/删除等只有用户明确要求时才允许）。若确实需要先查看数据，最多使用只读工具（outline_text / chapter_text / book_overview / facts_query）。" : userWriteIntent ? `你第 ${nudged + 1} 次表达了操作意图但没有输出动作标签，因此仍未执行任何操作。铁律：你的【整个回复】现在必须只包含一个 <dsh-action> 标签（例如 <dsh-action name="chapter_text">{"no":1}</dsh-action>），禁止任何解释、铺垫或"我这就去"之类的文字。若你其实不打算执行任何操作，请明确回复「不执行」。` : `你第 ${nudged + 1} 次回复仍不需要执行操作。再次强调：用户没有要求修改，请直接给出文字回答（可以简短引用项目数据），不要输出动作标签。写操作只会在用户明确要求时被允许。`;
 				nudged++;
 				history.push({
 					role: "tool",
@@ -3334,6 +3408,39 @@ async function* runAssistantTurn(ctx, config, project, outputDir, userMessage) {
 		}
 		const { name, args, index } = action;
 		const prose = reply.slice(0, index).trim();
+		if (!guardWrite(name, userMessage)) {
+			const denied = `【操作被拒绝】${name} 是写操作（会修改正文/项目数据），但你当前的消息里没有明确要求执行该修改。如果需要，请明确说明（如「生成第 120 章」「把第 105 章结尾改一下」）。我不会擅自修改你的作品。`;
+			history.push({
+				role: "tool",
+				content: denied,
+				tool: name,
+				ts: (/* @__PURE__ */ new Date()).toISOString()
+			});
+			appendHistory(outputDir, {
+				role: "tool",
+				content: denied,
+				tool: name,
+				ts: (/* @__PURE__ */ new Date()).toISOString()
+			});
+			yield {
+				frame: "tool",
+				name,
+				status: "error",
+				detail: denied
+			};
+			const assistantEntry = {
+				role: "assistant",
+				content: denied,
+				ts: (/* @__PURE__ */ new Date()).toISOString()
+			};
+			history.push(assistantEntry);
+			appendHistory(outputDir, assistantEntry);
+			yield {
+				frame: "delta",
+				text: denied
+			};
+			return;
+		}
 		yield {
 			frame: "tool",
 			name,
@@ -3795,6 +3902,7 @@ function makeRoutes(deps) {
 			try {
 				const next = project ?? createProject(outline);
 				const chapters = await planChapters(ctx, config, next, count, body?.volume, config.outputDir);
+				mergeVolatileFromDisk(config.outputDir, next);
 				next.chapters.push(...chapters);
 				next.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
 				saveProject(config.outputDir, next);
@@ -3840,6 +3948,7 @@ function makeRoutes(deps) {
 			chapter.status = "generating";
 			chapter.error = void 0;
 			chapter.generatingAt = (/* @__PURE__ */ new Date()).toISOString();
+			mergeVolatileFromDisk(config.outputDir, project);
 			saveProject(config.outputDir, project);
 			const send = (frame) => {
 				res.write(JSON.stringify(frame) + "\n");
@@ -3873,6 +3982,7 @@ function makeRoutes(deps) {
 				});
 				else {
 					chapter.status = "approved";
+					mergeVolatileFromDisk(config.outputDir, project);
 					saveProject(config.outputDir, project);
 				}
 				if (config.autoAuthorReview ?? true) try {
@@ -3886,6 +3996,7 @@ function makeRoutes(deps) {
 						const review = await authorReviewChapter(ctx, config, project, no, currentBody, prevTail);
 						chapter.authorReview = review;
 						if (review.advancedLines !== void 0) autoLinkPlotlines(project, no, review.advancedLines);
+						mergeVolatileFromDisk(config.outputDir, project);
 						saveProject(config.outputDir, project);
 						send({
 							type: "author-review",
@@ -3900,6 +4011,7 @@ function makeRoutes(deps) {
 			} catch (error) {
 				chapter.status = "error";
 				chapter.error = error.message;
+				mergeVolatileFromDisk(config.outputDir, project);
 				saveProject(config.outputDir, project);
 				if (!res.writableEnded) {
 					send({
@@ -5010,6 +5122,34 @@ function makeRoutes(deps) {
 			});
 		}
 	};
+	/** 章节直接通过：作者行使最终决定权（不重审，保留审稿记录）。 */
+	const chapterApproveRoute = {
+		kind: "exact",
+		path: NOVEL_API.chapterApprove,
+		handler: async (req, res) => {
+			if (!guard(req, res, "POST")) return;
+			const config = getConfig();
+			const project = requireProject(res);
+			if (project === void 0) return;
+			const no = (await readJsonBody(req))?.chapterNo;
+			if (!Number.isInteger(no) || no === void 0 || no < 1) {
+				writeJson(res, 400, { error: "chapterNo 须为正整数" });
+				return;
+			}
+			const chapter = project.chapters.find((c) => c.no === no);
+			if (chapter === void 0) {
+				writeJson(res, 404, { error: `章节 ${no} 不在计划中` });
+				return;
+			}
+			chapter.status = "approved";
+			project.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+			saveProject(config.outputDir, project);
+			writeJson(res, 200, {
+				ok: true,
+				no
+			});
+		}
+	};
 	return [
 		statusRoute,
 		loadOutlineRoute,
@@ -5133,6 +5273,7 @@ function makeRoutes(deps) {
 			}
 		},
 		chapterResetRoute,
+		chapterApproveRoute,
 		{
 			kind: "exact",
 			path: NOVEL_API.config,