@waterwx/dsh-novel-forge 1.1.2 → 1.1.3

package/lib/index.js

@@ -63,6 +63,10 @@ const NOVEL_API = {
 	world: "/api/dsh-novel-forge/world",
 	/** 封面：GET 读取（dataUrl）/ POST 上传或移除。 */
 	cover: "/api/dsh-novel-forge/blurb/cover",
+	/** 剧情线管理：增删改 + 关联章节。 */
+	plotlines: "/api/dsh-novel-forge/plotlines",
+	/** 敏感词检查：全书已写章节或指定文本。 */
+	sensitiveCheck: "/api/dsh-novel-forge/sensitive-check",
 	config: "/api/dsh-novel-forge/config",
 	openFolder: "/api/dsh-novel-forge/open-folder"
 };
@@ -769,6 +773,7 @@ function loadProject(outputDir) {
 		if (!Array.isArray(raw.assets.auxiliaryProgressions)) raw.assets.auxiliaryProgressions = [];
 		if (!Array.isArray(raw.assets.styleAssets)) raw.assets.styleAssets = [];
 		if (!Array.isArray(raw.facts)) raw.facts = [];
+		if (!Array.isArray(raw.plotlines)) raw.plotlines = [];
 		return raw;
 	} catch {
 		return;
@@ -779,6 +784,354 @@ function saveProject(outputDir, project) {
 	mkdirSync(outputDir, { recursive: true });
 	writeFileSync(join(outputDir, PROJECT_FILE), JSON.stringify(project, null, 2), "utf8");
 }
+/**
+* 内置违禁词库（网文平台常见审查类别）。只做硬匹配提示，不代替人工判断。
+* 词语刻意保持常见写法；作者可自行判断是否修改。
+*/
+const SENSITIVE_WORDS = [
+	{
+		word: "共匪",
+		category: "政治"
+	},
+	{
+		word: "独裁",
+		category: "政治"
+	},
+	{
+		word: "法轮",
+		category: "政治"
+	},
+	{
+		word: "六四",
+		category: "政治"
+	},
+	{
+		word: "天安门事件",
+		category: "政治"
+	},
+	{
+		word: "翻墙",
+		category: "政治"
+	},
+	{
+		word: "政治敏感",
+		category: "政治"
+	},
+	{
+		word: "乳沟",
+		category: "擦边"
+	},
+	{
+		word: "酥胸",
+		category: "擦边"
+	},
+	{
+		word: "淫荡",
+		category: "擦边"
+	},
+	{
+		word: "做爱",
+		category: "擦边"
+	},
+	{
+		word: "上床",
+		category: "擦边"
+	},
+	{
+		word: "裸体",
+		category: "擦边"
+	},
+	{
+		word: "一丝不挂",
+		category: "擦边"
+	},
+	{
+		word: "胴体",
+		category: "擦边"
+	},
+	{
+		word: "春药",
+		category: "擦边"
+	},
+	{
+		word: "催情",
+		category: "擦边"
+	},
+	{
+		word: "迷奸",
+		category: "擦边"
+	},
+	{
+		word: "强暴",
+		category: "擦边"
+	},
+	{
+		word: "轮奸",
+		category: "擦边"
+	},
+	{
+		word: "援交",
+		category: "擦边"
+	},
+	{
+		word: "嫖娼",
+		category: "擦边"
+	},
+	{
+		word: "卖淫",
+		category: "擦边"
+	},
+	{
+		word: "色情",
+		category: "擦边"
+	},
+	{
+		word: "情色",
+		category: "擦边"
+	},
+	{
+		word: "撸管",
+		category: "擦边"
+	},
+	{
+		word: "自慰",
+		category: "擦边"
+	},
+	{
+		word: "口交",
+		category: "擦边"
+	},
+	{
+		word: "打炮",
+		category: "擦边"
+	},
+	{
+		word: "约炮",
+		category: "擦边"
+	},
+	{
+		word: "一夜情",
+		category: "擦边"
+	},
+	{
+		word: "碎尸",
+		category: "暴力"
+	},
+	{
+		word: "分尸",
+		category: "暴力"
+	},
+	{
+		word: "凌迟",
+		category: "暴力"
+	},
+	{
+		word: "剥皮",
+		category: "暴力"
+	},
+	{
+		word: "开膛",
+		category: "暴力"
+	},
+	{
+		word: "剖腹",
+		category: "暴力"
+	},
+	{
+		word: "挖心",
+		category: "暴力"
+	},
+	{
+		word: "虐杀",
+		category: "暴力"
+	},
+	{
+		word: "凌辱",
+		category: "暴力"
+	},
+	{
+		word: "血腥",
+		category: "暴力"
+	},
+	{
+		word: "大屠杀",
+		category: "暴力"
+	},
+	{
+		word: "灭门",
+		category: "暴力"
+	},
+	{
+		word: "满门抄斩",
+		category: "暴力"
+	},
+	{
+		word: "腰斩",
+		category: "暴力"
+	},
+	{
+		word: "活埋",
+		category: "暴力"
+	},
+	{
+		word: "点天灯",
+		category: "暴力"
+	},
+	{
+		word: "傻逼",
+		category: "辱骂"
+	},
+	{
+		word: "傻B",
+		category: "辱骂"
+	},
+	{
+		word: "草泥马",
+		category: "辱骂"
+	},
+	{
+		word: "妈的",
+		category: "辱骂"
+	},
+	{
+		word: "尼玛",
+		category: "辱骂"
+	},
+	{
+		word: "去死",
+		category: "辱骂"
+	},
+	{
+		word: "废物",
+		category: "辱骂"
+	},
+	{
+		word: "垃圾",
+		category: "辱骂"
+	},
+	{
+		word: "人渣",
+		category: "辱骂"
+	},
+	{
+		word: "贱人",
+		category: "辱骂"
+	},
+	{
+		word: "婊子",
+		category: "辱骂"
+	},
+	{
+		word: "狗日的",
+		category: "辱骂"
+	},
+	{
+		word: "加微信",
+		category: "广告"
+	},
+	{
+		word: "加QQ",
+		category: "广告"
+	},
+	{
+		word: "微信公众号",
+		category: "广告"
+	},
+	{
+		word: "淘宝",
+		category: "广告"
+	},
+	{
+		word: "拼多多",
+		category: "广告"
+	},
+	{
+		word: "刷单",
+		category: "广告"
+	},
+	{
+		word: "充值返利",
+		category: "广告"
+	},
+	{
+		word: "扫码领",
+		category: "广告"
+	},
+	{
+		word: "加群领",
+		category: "广告"
+	},
+	{
+		word: "vx",
+		category: "广告"
+	},
+	{
+		word: "扣扣",
+		category: "广告"
+	},
+	{
+		word: "赌博",
+		category: "其他"
+	},
+	{
+		word: "赌场",
+		category: "其他"
+	},
+	{
+		word: "毒品",
+		category: "其他"
+	},
+	{
+		word: "冰毒",
+		category: "其他"
+	},
+	{
+		word: "摇头丸",
+		category: "其他"
+	},
+	{
+		word: "自杀方法",
+		category: "其他"
+	},
+	{
+		word: "邪教",
+		category: "其他"
+	},
+	{
+		word: "传销",
+		category: "其他"
+	},
+	{
+		word: "军火",
+		category: "其他"
+	},
+	{
+		word: "枪支",
+		category: "其他"
+	},
+	{
+		word: "管制刀具",
+		category: "其他"
+	}
+];
+/** 对一段文本做违禁词硬匹配，返回命中（词/类别/次数）。 */
+function checkSensitiveText(text) {
+	const hits = [];
+	for (const entry of SENSITIVE_WORDS) {
+		let count = 0;
+		let idx = text.indexOf(entry.word);
+		while (idx !== -1) {
+			count++;
+			idx = text.indexOf(entry.word, idx + entry.word.length);
+		}
+		if (count > 0) hits.push({
+			word: entry.word,
+			category: entry.category,
+			count
+		});
+	}
+	return hits;
+}
 /** List generated chapter files in the output dir (sorted). */
 function listChapterFiles(outputDir) {
 	if (!existsSync(outputDir)) return [];
@@ -971,7 +1324,8 @@ async function extractBible(ctx, config, outline) {
 		].includes(entry.role) ? entry.role : "other",
 		traits: strArray(entry.traits),
 		goals: typeof entry.goals === "string" ? entry.goals : "",
-		relations: typeof entry.relations === "string" ? entry.relations : ""
+		relations: typeof entry.relations === "string" ? entry.relations : "",
+		knowledge: strArray(entry.knowledge)
 	})).filter((card) => card.name !== "") : [];
 	const bible = {
 		genre: typeof raw.genre === "string" ? raw.genre : "",
@@ -1067,6 +1421,7 @@ function writeSystemPrompt(project) {
 					other: "其他"
 				}[card.role];
 				sections.push(`- ${card.name}（${roleName}）：${card.traits.join("、")}${card.goals !== "" ? `；目标：${card.goals}` : ""}${card.relations !== "" ? `；关系：${card.relations}` : ""}`);
+				if (Array.isArray(card.knowledge) && card.knowledge.length > 0) sections.push(`  已知信息（该角色知道的：${card.knowledge.join("；")}；未列出的信息该角色一律不知道，不得写其知晓或提及）`);
 			}
 		}
 		if (bible.redLines.length > 0) sections.push("写作红线（违反即失败）：\n" + bible.redLines.map((r) => `- ${r}`).join("\n"));
@@ -1085,6 +1440,17 @@ function writeSystemPrompt(project) {
 		sections.push("==================== 活跃伏笔（近期需推进或回收的线索） ====================");
 		for (const f of active) sections.push(`- [${f.status === "planted" ? "已埋设" : "推进中"}] ${f.description}${f.targetChapter !== void 0 ? `（预计 ${f.targetChapter} 章回收）` : ""}`);
 	}
+	const lines = (project.plotlines ?? []).filter((l) => l.status === "active" || l.status === "paused");
+	if (lines.length > 0) {
+		const kindName = {
+			main: "主线",
+			branch: "支线",
+			character: "人物线",
+			mystery: "悬念线"
+		};
+		sections.push("==================== 剧情线（本章应推进至少一条活跃线） ====================");
+		for (const l of lines) sections.push(`- [${kindName[l.kind]}${l.status === "paused" ? "·暂停中" : ""}] ${l.name}：${l.goal}${l.progress !== "" ? `（当前进度：${l.progress}）` : ""}`);
+	}
 	sections.push("");
 	sections.push("写作硬性要求：");
 	sections.push("1. 每章 3000-4000 字（按中文字符计），只输出章节正文，不要输出标题、章回名、作者的话或任何 Markdown 标记。");
@@ -1187,7 +1553,10 @@ function reviewSystemPrompt(project) {
 		if (bible.worldRules.length > 0) sections.push("世界规则：\n" + bible.worldRules.map((r) => `- ${r}`).join("\n"));
 		if (bible.characters.length > 0) {
 			sections.push("角色卡：");
-			for (const card of bible.characters) sections.push(`- ${card.name}（${card.role}）：${card.traits.join("、")}`);
+			for (const card of bible.characters) {
+				sections.push(`- ${card.name}（${card.role}）：${card.traits.join("、")}`);
+				if (Array.isArray(card.knowledge) && card.knowledge.length > 0) sections.push(`  该角色知道：${card.knowledge.join("；")}（未列出的信息该角色不知道）`);
+			}
 		}
 		if (bible.redLines.length > 0) sections.push("红线：\n" + bible.redLines.map((r) => `- ${r}`).join("\n"));
 	}
@@ -3867,6 +4236,19 @@ function makeRoutes(deps) {
 				if (Array.isArray(body?.worldRules)) project.bible.worldRules = body.worldRules.filter((r) => r.trim() !== "");
 				if (Array.isArray(body?.redLines)) project.bible.redLines = body.redLines.filter((r) => r.trim() !== "");
 				if (Array.isArray(body?.style)) project.bible.style = body.style.filter((r) => r.trim() !== "");
+				if (Array.isArray(body?.characters)) project.bible.characters = body.characters.filter((c) => c !== void 0 && c !== null && typeof c.name === "string" && c.name.trim() !== "").map((c) => ({
+					name: c.name.trim(),
+					role: [
+						"protagonist",
+						"supporting",
+						"antagonist",
+						"other"
+					].includes(c.role) ? c.role : "other",
+					traits: Array.isArray(c.traits) ? c.traits.filter((t) => typeof t === "string" && t.trim() !== "") : [],
+					goals: typeof c.goals === "string" ? c.goals : "",
+					relations: typeof c.relations === "string" ? c.relations : "",
+					knowledge: Array.isArray(c.knowledge) ? c.knowledge.filter((k) => typeof k === "string" && k.trim() !== "") : void 0
+				}));
 				project.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
 				saveProject(config.outputDir, project);
 				writeJson(res, 200, { bible: project.bible });
@@ -4029,6 +4411,108 @@ function makeRoutes(deps) {
 				writeJson(res, 200, { bookName: project.bookName });
 			}
 		},
+		{
+			kind: "exact",
+			path: NOVEL_API.plotlines,
+			handler: async (req, res) => {
+				if (!guard(req, res, "POST")) return;
+				const config = getConfig();
+				const project = requireProject(res);
+				if (project === void 0) return;
+				const body = await readJsonBody(req);
+				if (project.plotlines === void 0) project.plotlines = [];
+				const op = body?.op;
+				if (op === "add" && body?.line !== void 0) {
+					const line = body.line;
+					project.plotlines.push({
+						id: line.id !== "" ? line.id : `pl-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`,
+						name: line.name.slice(0, 40),
+						kind: line.kind,
+						goal: line.goal.slice(0, 300),
+						progress: line.progress.slice(0, 300),
+						status: line.status,
+						chapters: Array.isArray(line.chapters) ? line.chapters.filter((n) => typeof n === "number") : [],
+						createdAt: line.createdAt !== "" ? line.createdAt : (/* @__PURE__ */ new Date()).toISOString()
+					});
+				} else if (op === "update" && body?.line !== void 0 && body.line.id !== "") {
+					const idx = project.plotlines.findIndex((l) => l.id === body.line.id);
+					if (idx !== -1) {
+						const line = body.line;
+						project.plotlines[idx] = {
+							...project.plotlines[idx],
+							name: line.name.slice(0, 40),
+							kind: line.kind,
+							goal: line.goal.slice(0, 300),
+							progress: line.progress.slice(0, 300),
+							status: line.status
+						};
+					}
+				} else if (op === "remove" && body?.id !== void 0) project.plotlines = project.plotlines.filter((l) => l.id !== body.id);
+				else if (op === "link" && body?.id !== void 0 && typeof body.chapterNo === "number" && body.chapterNo > 0) {
+					const line = project.plotlines.find((l) => l.id === body.id);
+					if (line !== void 0 && !line.chapters.includes(body.chapterNo)) {
+						line.chapters.push(body.chapterNo);
+						if (line.status === "active" && line.progress === "") line.progress = `推进至第 ${body.chapterNo} 章`;
+					}
+				}
+				project.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+				saveProject(config.outputDir, project);
+				writeJson(res, 200, { plotlines: project.plotlines });
+			}
+		},
+		{
+			kind: "exact",
+			path: NOVEL_API.sensitiveCheck,
+			handler: async (req, res) => {
+				if (!guard(req, res, "POST")) return;
+				const config = getConfig();
+				const project = requireProject(res);
+				if (project === void 0) return;
+				const body = await readJsonBody(req);
+				const hits = [];
+				let scanned = 0;
+				if (body?.text !== void 0) for (const hit of checkSensitiveText(body.text)) hits.push({
+					chapterNo: 0,
+					word: hit.word,
+					category: hit.category,
+					count: hit.count
+				});
+				else if (typeof body?.chapterNo === "number") {
+					const chapter = project.chapters.find((c) => c.no === body.chapterNo);
+					if (chapter !== void 0) {
+						const text = readChapterFile(config.outputDir, chapter);
+						if (text !== void 0) {
+							scanned = 1;
+							for (const hit of checkSensitiveText(text)) hits.push({
+								chapterNo: chapter.no,
+								word: hit.word,
+								category: hit.category,
+								count: hit.count
+							});
+						}
+					}
+				} else if (body?.all === true) for (const chapter of project.chapters) {
+					if (chapter.status === "pending" || chapter.status === "generating") continue;
+					const text = readChapterFile(config.outputDir, chapter);
+					if (text === void 0) continue;
+					scanned++;
+					for (const hit of checkSensitiveText(text)) hits.push({
+						chapterNo: chapter.no,
+						word: hit.word,
+						category: hit.category,
+						count: hit.count
+					});
+				}
+				else {
+					writeJson(res, 400, { error: "请提供 chapterNo / text / all 之一" });
+					return;
+				}
+				writeJson(res, 200, {
+					hits,
+					scannedChapters: scanned
+				});
+			}
+		},
 		{
 			kind: "exact",
 			path: NOVEL_API.config,