@waterwx/dsh-novel-forge 1.2.0 → 1.3.0-alpha

package/src/novel-context.ts

@@ -0,0 +1,286 @@
+/**
+ * 统一上下文组装与相关设定检索
+ *
+ * 取代各阶段各自拼装 prompt，消除「生成看得到、审稿看不到」的断层。
+ * 所有阶段通过 buildChapterContext 获取同一来源的事实与设定。
+ */
+
+import type { ProjectState, ChapterPlan, StoryBible, RoleRecord, ChapterFact, Foreshadow, Plotline } from './protocol'
+import { readFileSync, existsSync } from 'node:fs'
+import { join } from 'node:path'
+
+export interface ChapterContext {
+  /** 平台合规红线（永远全量） */
+  complianceRedLines: string[]
+  /** 本书写作红线（永远全量） */
+  bookRedLines: string[]
+  /** 当前章相关角色卡（全量信息） */
+  relevantRoles: Array<{ name: string; role: string; traits: string[]; goals?: string; knowledge?: string[] }>
+  /** 当前章相关道藏规则 */
+  relevantWorldRules: string[]
+  /** 活跃剧情线 */
+  activePlotlines: Plotline[]
+  /** 未回收伏笔 */
+  activeForeshadows: Foreshadow[]
+  /** 上一章结尾原文 */
+  prevChapterTail: string
+  /** 上一章摘要 */
+  prevChapterSummary?: string
+  /** 最近事实（近因记忆，top20） */
+  recentFacts: string[]
+  /** 相关旧事实（按本章 beats 检索，top15） */
+  relatedFacts: string[]
+  /** 当前卷大纲 */
+  currentVolumeOutline: string
+  /** 本章 beats */
+  beats: string
+  /** 本章标题 */
+  chapterTitle: string
+}
+
+export interface ContextBuildOptions {
+  /** 阶段：writing/review/revise/polish/plan */
+  stage?: 'writing' | 'review' | 'revise' | 'polish' | 'plan'
+  /** 是否注入完整角色卡（false 时只注入角色名+定位） */
+  fullRoleCards?: boolean
+  /** 相关事实检索上限 */
+  relatedFactsLimit?: number
+  /** 最近事实数量 */
+  recentFactsLimit?: number
+}
+
+/**
+ * 相关事实检索：trigram 重合度 + 角色名命中加权 + 近因加权
+ * 从 generateChapterStream 抽取，扩展为通用函数。
+ */
+export function retrieveRelatedFacts(
+  facts: ChapterFact[],
+  beatsText: string,
+  roleNames: string[],
+  limit = 15,
+): string[] {
+  if (facts.length === 0) return []
+
+  const trigrams = (s: string): Set<string> => {
+    const out = new Set<string>()
+    for (let i = 0; i + 3 <= s.length; i++) {
+      const tri = s.slice(i, i + 3)
+      if (tri.trim() !== '') out.add(tri)
+    }
+    return out
+  }
+
+  const beatsTri = trigrams(beatsText)
+  const beatRoles = roleNames.filter(n => beatsText.includes(n))
+
+  const recentTexts = new Set(facts.slice(-20).map(f => f.text))
+
+  return facts
+    .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
+      }
+      // 近因加权：越新越优先（封顶 40 章）
+      score += Math.min(idx, 40) / 10
+      return { f, score }
+    })
+    .filter(x => x.score >= 3)
+    .sort((a, b) => b.score - a.score)
+    .slice(0, limit)
+    .map(x => `[第${x.f.chapterNo}章] ${x.f.text}`)
+    .filter(t => !recentTexts.has(t.slice(t.indexOf(']') + 2)))
+}
+
+/**
+ * 相关角色检索：beats 中出现的角色 + 主角/主要反派永远全量
+ */
+export function retrieveRelevantRoles(
+  bible: StoryBible | undefined,
+  roles: RoleRecord[] | undefined,
+  beatsText: string,
+): ChapterContext['relevantRoles'] {
+  const result: ChapterContext['relevantRoles'] = []
+  const seen = new Set<string>()
+
+  // 主角/主要反派永远全量
+  const alwaysRoles = (bible?.characters ?? []).filter(c =>
+    c.role === 'protagonist' || c.role === 'antagonist'
+  )
+  for (const card of alwaysRoles) {
+    if (seen.has(card.name)) continue
+    seen.add(card.name)
+    result.push({
+      name: card.name,
+      role: card.role,
+      traits: card.traits,
+      goals: card.goals !== '' ? card.goals : undefined,
+      knowledge: Array.isArray(card.knowledge) && card.knowledge.length > 0 ? card.knowledge : undefined,
+    })
+  }
+
+  // beats 中出现的角色
+  const beatRoles = (bible?.characters ?? []).filter(c => beatsText.includes(c.name))
+  for (const card of beatRoles) {
+    if (seen.has(card.name)) continue
+    seen.add(card.name)
+    result.push({
+      name: card.name,
+      role: card.role,
+      traits: card.traits,
+      goals: card.goals !== '' ? card.goals : undefined,
+      knowledge: Array.isArray(card.knowledge) && card.knowledge.length > 0 ? card.knowledge : undefined,
+    })
+  }
+
+  // 角色库补充（不在道藏里的）
+  for (const r of roles ?? []) {
+    if (seen.has(r.name)) continue
+    if (!beatsText.includes(r.name) && r.roleLabel !== 'protagonist' && r.roleLabel !== 'antagonist') continue
+    seen.add(r.name)
+    result.push({
+      name: r.name,
+      role: r.roleLabel,
+      traits: Array.isArray(r.traits) ? r.traits : [],
+      goals: r.goals !== undefined && r.goals !== '' ? r.goals : undefined,
+    })
+  }
+
+  return result
+}
+
+/**
+ * 构建章节统一上下文
+ *
+ * 各阶段调用此函数获取同一来源的事实与设定，消除上下文断层。
+ * 检索结果不足时自动回退全量，保证不丢失关键信息。
+ */
+export function buildChapterContext(
+  project: ProjectState,
+  chapter: ChapterPlan,
+  outputDir: string,
+  options: ContextBuildOptions = {},
+): ChapterContext {
+  const {
+    fullRoleCards = true,
+    relatedFactsLimit = 15,
+    recentFactsLimit = 20,
+  } = options
+
+  const bible = project.bible
+  const allFacts = project.facts ?? []
+  const roleNames = (project.roles ?? []).map(r => r.name).filter((n): n is string => typeof n === 'string' && n !== '')
+
+  // 上一章结尾 + 摘要
+  let prevChapterTail = ''
+  let prevChapterSummary: string | undefined
+  const prev = project.chapters.find(c => c.no === chapter.no - 1)
+  if (prev?.file !== undefined) {
+    try {
+      const prevPath = join(outputDir, prev.file)
+      if (existsSync(prevPath)) {
+        const text = readFileSync(prevPath, 'utf8')
+        prevChapterTail = text.replace(/^#\s+.*$/m, '').trim().slice(-900)
+      }
+    } catch { /* 文件缺失时忽略 */ }
+  }
+  prevChapterSummary = prev?.summary
+
+  // 事实：最近 + 相关
+  const recentFacts = allFacts.slice(-recentFactsLimit).map(f => `[第${f.chapterNo}章] ${f.text}`)
+  const relatedFacts = retrieveRelatedFacts(allFacts, chapter.beats, roleNames, relatedFactsLimit)
+
+  // 角色：相关角色
+  const relevantRoles = retrieveRelevantRoles(bible, project.roles, chapter.beats)
+
+  // 道藏规则：全部（规则数量通常不多，全量注入更安全）
+  const relevantWorldRules = bible?.worldRules ?? []
+
+  // 活跃剧情线
+  const activePlotlines = (project.plotlines ?? []).filter(l =>
+    l.status === 'active' || l.status === 'paused'
+  )
+
+  // 未回收伏笔
+  const activeForeshadows = (project.foreshadows ?? []).filter(f =>
+    f.status === 'planted' || f.status === 'progressing'
+  )
+
+  // 当前卷大纲
+  const volumeNo = chapter.volume ?? 0
+  const volume = project.volumes?.find(v => v.no === volumeNo)
+  const currentVolumeOutline = volume !== undefined
+    ? `第${volume.no}卷《${volume.title}》：${volume.summary}`
+    : ''
+
+  return {
+    complianceRedLines: [], // 由调用方注入 COMPLIANCE_REDLINES
+    bookRedLines: bible?.redLines ?? [],
+    relevantRoles: fullRoleCards ? relevantRoles : relevantRoles.map(r => ({ name: r.name, role: r.role, traits: [] })),
+    relevantWorldRules,
+    activePlotlines,
+    activeForeshadows,
+    prevChapterTail,
+    prevChapterSummary,
+    recentFacts,
+    relatedFacts,
+    currentVolumeOutline,
+    beats: chapter.beats,
+    chapterTitle: chapter.title,
+  }
+}
+
+/**
+ * 将上下文渲染为 prompt 文本块
+ * 各阶段可按需选择注入哪些块。
+ */
+export function renderContextBlocks(ctx: ChapterContext): {
+  rolesBlock: string
+  worldRulesBlock: string
+  factsBlock: string
+  plotlinesBlock: string
+  foreshadowsBlock: string
+  continuityBlock: string
+  redLinesBlock: string
+} {
+  const roleName = { protagonist: '主角', supporting: '配角', antagonist: '反派', other: '其他', female_lead: '女主', female_support: '女配', support: '配角', extra: '路人' } as Record<string, string>
+
+  const rolesBlock = ctx.relevantRoles.length > 0
+    ? '==================== 相关角色卡 ====================\n' +
+      ctx.relevantRoles.map(r =>
+        `- ${r.name}（${roleName[r.role] ?? r.role}）：${r.traits.join('、')}${r.goals ? `；目标：${r.goals}` : ''}${r.knowledge ? `\n  已知信息：${r.knowledge.join('；')}（未列出的该角色不知道）` : ''}`
+      ).join('\n')
+    : ''
+
+  const worldRulesBlock = ctx.relevantWorldRules.length > 0
+    ? '==================== 世界规则 ====================\n' + ctx.relevantWorldRules.map(r => `- ${r}`).join('\n')
+    : ''
+
+  const factsBlock = (ctx.recentFacts.length > 0 || ctx.relatedFacts.length > 0)
+    ? '==================== 事实库（最近 + 相关） ====================\n' +
+      (ctx.recentFacts.length > 0 ? `【最近 ${ctx.recentFacts.length} 条】\n${ctx.recentFacts.join('\n')}` : '') +
+      (ctx.relatedFacts.length > 0 ? `\n【相关旧事实 ${ctx.relatedFacts.length} 条】\n${ctx.relatedFacts.join('\n')}` : '')
+    : ''
+
+  const plotlinesBlock = ctx.activePlotlines.length > 0
+    ? '==================== 活跃剧情线 ====================\n' +
+      ctx.activePlotlines.map(l => `- [${l.kind}${l.status === 'paused' ? '·暂停' : ''}] ${l.name}：${l.goal}${l.progress ? `（进度：${l.progress}）` : ''}`).join('\n')
+    : ''
+
+  const foreshadowsBlock = ctx.activeForeshadows.length > 0
+    ? '==================== 活跃伏笔 ====================\n' +
+      ctx.activeForeshadows.map(f => `- [${f.status === 'planted' ? '已埋设' : '推进中'}] ${f.description}${f.targetChapter ? `（预计第${f.targetChapter}章回收）` : ''}`).join('\n')
+    : ''
+
+  const continuityBlock = ctx.prevChapterTail !== ''
+    ? `==================== 上一章结尾（紧接此状态继续） ====================\n${ctx.prevChapterTail}`
+    : ''
+
+  const redLinesBlock = ctx.bookRedLines.length > 0
+    ? '==================== 本书红线 ====================\n' + ctx.bookRedLines.map(r => `- ${r}`).join('\n')
+    : ''
+
+  return { rolesBlock, worldRulesBlock, factsBlock, plotlinesBlock, foreshadowsBlock, continuityBlock, redLinesBlock }
+}

package/src/prompt-slots.ts

@@ -0,0 +1,125 @@
+/**
+ * 提示词槽位：给作者留几个**安全可改**的位置，用来微调本书的写作偏好。
+ *
+ * 设计边界（很重要）：
+ *   * 槽位只追加"作者偏好"，**不允许覆盖**规则工程与合规红线——道藏、写作红线、
+ *     九条内容合规红线、反 AI 规则、阶段契约都由宿主锁定，槽位碰不到；
+ *   * 槽位内容随项目保存（`novel-project.json`），换书即换槽位；
+ *   * 每个槽位有长度上限，超长直接截断并标记（沿用「截断显式化」的规则）。
+ */
+
+import type { ProjectState, PromptSlotId } from './protocol.ts'
+
+/** 单个槽位的元信息。 */
+export interface PromptSlotSpec {
+  id: PromptSlotId
+  /** 面板显示名。 */
+  label: string
+  /** 一句话说明它影响什么。 */
+  hint: string
+  /** 示例（面板占位符）。 */
+  placeholder: string
+  /** 字符上限。 */
+  maxChars: number
+}
+
+/** 允许作者编辑的槽位（白名单；不在此列表的提示词一律不可改）。 */
+export const PROMPT_SLOTS: readonly PromptSlotSpec[] = [
+  {
+    id: 'styleExtra',
+    label: '文风补充',
+    hint: '追加到写作提示词末尾的风格偏好（句式、叙述习惯、用词倾向）。',
+    placeholder: '例如：多用短句和对话推进，少用成语；环境描写只写能影响动作的部分。',
+    maxChars: 1200,
+  },
+  {
+    id: 'antiAiExtra',
+    label: '反 AI 补充',
+    hint: '本书额外的"别这么写"清单，与内置反 AI 规则叠加（不能覆盖内置规则）。',
+    placeholder: '例如：不要每段都用比喻；不要用"空气中弥漫着"这类句式。',
+    maxChars: 1200,
+  },
+  {
+    id: 'dialogueExtra',
+    label: '对话补充',
+    hint: '角色说话方式的额外约束，写入时会与角色卡一起注入。',
+    placeholder: '例如：长辈说话带方言词；主角对熟人用绰号称呼。',
+    maxChars: 800,
+  },
+  {
+    id: 'endingHookExtra',
+    label: '章末钩子偏好',
+    hint: '对结尾钩子的额外要求（与章节计划里的 endingHook 叠加）。',
+    placeholder: '例如：结尾尽量落在一个未闭合的选择上，而不是新出现的敌人。',
+    maxChars: 600,
+  },
+]
+
+/** 槽位 id → 规格。 */
+const SLOT_BY_ID = new Map(PROMPT_SLOTS.map(s => [s.id, s] as const))
+
+/** 单个槽位的渲染结果。 */
+export interface RenderedSlot {
+  id: PromptSlotId
+  label: string
+  text: string
+  /** 是否因超长被截断。 */
+  truncated: boolean
+  /** 原文长度（截断前）。 */
+  chars: number
+}
+
+/**
+ * 读取并规整项目里的槽位内容（去空白、按上限截断）。
+ *
+ * @param project 项目状态。
+ * @returns 非空槽位列表（保持 PROMPT_SLOTS 的声明顺序）。
+ */
+export function readPromptSlots(project: ProjectState): RenderedSlot[] {
+  const stored = project.promptSlots ?? {}
+  const out: RenderedSlot[] = []
+  for (const spec of PROMPT_SLOTS) {
+    const raw = (stored[spec.id] ?? '').trim()
+    if (raw === '') continue
+    const truncated = raw.length > spec.maxChars
+    out.push({
+      id: spec.id,
+      label: spec.label,
+      text: truncated ? raw.slice(0, spec.maxChars) : raw,
+      truncated,
+      chars: raw.length,
+    })
+  }
+  return out
+}
+
+/**
+ * 把槽位渲染成注入提示词的段落。
+ *
+ * @param project 项目状态。
+ * @returns 提示词片段；无槽位时返回空串。
+ */
+export function renderPromptSlots(project: ProjectState): string {
+  const slots = readPromptSlots(project)
+  if (slots.length === 0) return ''
+  const lines = ['==================== 作者偏好（本书自定义，优先级低于道藏/红线/合规） ====================']
+  for (const slot of slots) {
+    lines.push(`【${slot.label}】${slot.text}${slot.truncated ? `（已按 ${SLOT_BY_ID.get(slot.id)?.maxChars ?? 0} 字上限截断，原 ${slot.chars} 字）` : ''}`)
+  }
+  lines.push('以上偏好只影响表达方式；与道藏、写作红线、内容合规红线冲突时，一律以后者为准。')
+  return lines.join('\n')
+}
+
+/**
+ * 校验并规整一次槽位写入。
+ *
+ * @param id 槽位 id。
+ * @param value 内容（空串表示清空该槽位）。
+ * @returns 归一化后的内容。
+ * @throws 槽位 id 不在白名单时抛错。
+ */
+export function normalizeSlotValue(id: string, value: string): string {
+  const spec = SLOT_BY_ID.get(id as PromptSlotId)
+  if (spec === undefined) throw new Error(`未知提示词槽位：${id}`)
+  return value.replace(/\r\n/g, '\n').trim()
+}

package/src/prompting.ts

@@ -0,0 +1,26 @@
+/** 章节写作约束：把写作骨架渲染成可附加到系统提示词末尾的必达/禁止/自查块。 */
+export function renderChapterWriterSkeleton(meta: {
+  targetChars: number
+  minChars: number
+  maxChars: number
+  pov?: string
+  tonePreference?: string
+  endingHookPreference?: string
+  antiAiRules?: string
+}): string {
+  const pov = meta.pov ?? '第三人称有限视角，严格跟随主角所见所知。'
+  const tone = meta.tonePreference ?? '文风贴合本书设定，用具体细节与动作推进。'
+  const hook = meta.endingHookPreference ?? '章末留一个明确的钩子（新信息、新风险或未闭合的选择）。'
+  const antiAi = meta.antiAiRules ?? ''
+  return [
+    '==================== 章节生成骨架（必达 / 禁止 / 输出前自查） ====================',
+    '【叙事视角】' + pov,
+    '【本章必达】本章必须发生实质变化——局面、关系、信息、风险、决策至少动一项；开篇直接进入情境，不复述上一章。',
+    '【禁止事项】不写以总结、复盘、解释为主的段落；不新增核心角色，不写入与上下文冲突的设定；不为凑字数注水。',
+    '【篇幅】目标 ' + meta.targetChars + ' 字，区间 ' + meta.minChars + '-' + meta.maxChars + ' 字；不够就继续推进有效情节，不许草草收尾。',
+    '【结尾】' + hook,
+    '【表达】用具体动作、对话与可感知细节往前推；' + tone,
+    '【反 AI】' + antiAi,
+    '【输出前自查】逐项确认：读者的回报、关键转折、章末净变化是否看得见？上一章留下的钩子有没有交代？人物的硬事实有没有被写反？确认无误再输出，正文里不要出现核查过程。',
+  ].join('\n')
+}