@waterwx/dsh-novel-forge 1.3.1 → 1.3.2-alpha

package/src/author-assets.ts

@@ -0,0 +1,149 @@
+/**
+ * 作者资产库/总数据（跨书）— 按作者维度聚合的可复用资产。
+ * 与书架同惯例持久化到 ~/.dsh/dsh-novel-forge-author-assets.json。
+ * 提供读取/持久化/新增或更新/删除，以及「导入默认」：把书架书的写作资产/角色 + 
+ * 内置全局库（题材/反AI规则/风格模板/推进模式）批量沉淀成资产条目（默认库）。
+ */
+
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs'
+import { homedir } from 'node:os'
+import { join } from 'node:path'
+import { randomBytes } from 'node:crypto'
+import type { AuthorAssetLibrary, AuthorStyleAsset } from './protocol.ts'
+
+import { loadBookshelf } from './bookshelf.ts'
+import { loadProject } from './engine.ts'
+
+/** 作者资产库配置文件路径。 */
+export function authorAssetsFile(): string {
+  return join(homedir(), '.dsh', 'dsh-novel-forge-author-assets.json')
+}
+
+/** 默认空资产库。 */
+function defaultLibrary(): AuthorAssetLibrary {
+  return { version: 1, items: [] }
+}
+
+/** 读取作者资产库（不存在/损坏则返回空库，不抛错）。 */
+export function loadAuthorAssets(): AuthorAssetLibrary {
+  const file = authorAssetsFile()
+  if (!existsSync(file)) return defaultLibrary()
+  try {
+    let raw = readFileSync(file, 'utf8')
+    if (raw.charCodeAt(0) === 0xFEFF) raw = raw.slice(1)
+    const parsed = JSON.parse(raw) as Partial<AuthorAssetLibrary>
+    if (!Array.isArray(parsed.items)) return defaultLibrary()
+    return { version: 1, items: parsed.items }
+  } catch (error) {
+    // 资产库损坏回退默认值会遮蔽数据丢失，留痕到服务端日志。
+    console.error(`[novel-forge] author assets file unreadable/corrupt: ${file} — ${(error as Error).message}`)
+    return defaultLibrary()
+  }
+}
+
+/** 持久化作者资产库。 */
+export function saveAuthorAssets(library: AuthorAssetLibrary): void {
+  const file = authorAssetsFile()
+  mkdirSync(join(homedir(), '.dsh'), { recursive: true })
+  writeFileSync(file, JSON.stringify(library, null, 2), 'utf8')
+}
+
+/** 新建一条资产（自动生成 id 与时间戳）。 */
+export function createAuthorAsset(input: Omit<AuthorStyleAsset, 'id' | 'createdAt' | 'updatedAt'>): AuthorStyleAsset {
+  const now = new Date().toISOString()
+  const asset: AuthorStyleAsset = {
+    ...input,
+    id: 'aa-' + Date.now().toString(36) + '-' + randomBytes(3).toString('hex'),
+    sourceBooks: input.sourceBooks ?? [],
+    tags: input.tags ?? [],
+    createdAt: now,
+    updatedAt: now,
+  }
+  const library = loadAuthorAssets()
+  library.items.push(asset)
+  saveAuthorAssets(library)
+  return asset
+}
+
+/** 按 id 更新一条资产；不存在则追加。 */
+export function upsertAuthorAsset(input: AuthorStyleAsset): AuthorAssetLibrary {
+  const library = loadAuthorAssets()
+  const idx = library.items.findIndex(a => a.id === input.id)
+  const now = new Date().toISOString()
+  if (idx >= 0) {
+    const prev = library.items[idx]
+    if (prev !== undefined) {
+      library.items[idx] = {
+        ...prev,
+        ...input,
+        id: prev.id,
+        createdAt: prev.createdAt,
+        updatedAt: now,
+      }
+    }
+  } else {
+    library.items.push({ ...input, createdAt: input.createdAt !== '' ? input.createdAt : now, updatedAt: now })
+  }
+  saveAuthorAssets(library)
+  return library
+}
+
+/** 删除一条资产；返回更新后的资产库。 */
+export function removeAuthorAsset(id: string): AuthorAssetLibrary {
+  const library = loadAuthorAssets()
+  library.items = library.items.filter(a => a.id !== id)
+  saveAuthorAssets(library)
+  return library
+}
+
+/**
+ * 导入默认：把书架所有书的写作资产/角色 + 内置全局库批量沉淀到作者资产库。
+ * 按 `kind:name` 去重（已存在则跳过），不覆盖用户已有条目。
+ * @returns 更新后的资产库。
+ */
+export function importDefaultAuthorAssets(): AuthorAssetLibrary {
+  const library = loadAuthorAssets()
+  const seen = new Set(library.items.map(a => a.kind + ':' + a.name))
+  const now = new Date().toISOString()
+  const push = (kind: AuthorStyleAsset['kind'], name: string, summary: string, content: string, tags: string[], sourceBooks: string[], structured?: Record<string, unknown>): void => {
+    const key = kind + ':' + name.trim()
+    if (name.trim() === '' || seen.has(key)) return
+    seen.add(key)
+    library.items.push({
+      id: 'aa-' + Date.now().toString(36) + '-' + randomBytes(3).toString('hex'),
+      kind, name: name.trim(), summary: summary.trim(), content: content.trim(),
+      sourceBooks, tags, structured, createdAt: now, updatedAt: now,
+    });
+  };
+
+  // 1) 清理内置公共库重复沉淀：作者库只保留个人沉淀，不再收录内置公共资产。
+  const builtinKinds = new Set<AuthorStyleAsset['kind']>(['genre', 'antiAi', 'progression', 'style', 'plotBeat'])
+  library.items = library.items.filter(a => !(a.tags.includes('内置') && builtinKinds.has(a.kind) && (a.sourceBooks ?? []).length === 0))
+
+  // 2) 书架里的书：写作资产 + 角色模板
+  const shelf = loadBookshelf()
+  for (const book of shelf.books) {
+    const project = loadProject(book.outputDir)
+    if (project === undefined) continue
+    const src = [project.bookName]
+    const assets = project.assets
+    if (assets !== undefined) {
+      if (assets.genre !== undefined) push('genre', assets.genre.name, assets.genre.description, assets.genre.description, ['书', '题材'], src, { description: assets.genre.description });
+      if (assets.primaryProgression !== undefined) push('progression', assets.primaryProgression.name, assets.primaryProgression.driver, '驱动：' + assets.primaryProgression.driver + '\n期待：' + assets.primaryProgression.readerExpectation, src, [], { driver: assets.primaryProgression.driver, primary: true });
+      for (const p of assets.auxiliaryProgressions ?? []) push('progression', p.name, p.driver, '驱动：' + p.driver + '\n期待：' + p.readerExpectation, src, [], { driver: p.driver, primary: false });
+      for (const r of assets.antiAiRules ?? []) push('antiAi', r.name, r.avoid, '避免：' + r.avoid + '\n修正：' + r.fix, src, [], { avoid: r.avoid, fix: r.fix, detectPatterns: r.detectPatterns });
+      for (const s of assets.styleAssets ?? []) push('style', s.name, s.proseRules.join('；'), 
+        '叙述：' + s.proseRules.join('；') + '\n台词：' + s.dialogueRules.join('；') + '\n描写：' + s.descriptionRules.join('；') + '\n边界：' + s.boundaries.join('；'), src, [], { sourceText: s.sourceText });
+    }
+    // 作者库角色模板只收男主/女主（可复用原型），其余角色/路人不入库，避免刷屏。
+    for (const role of project.roles ?? []) {
+      if (role.roleLabel !== 'protagonist' && role.roleLabel !== 'female_lead') continue
+      const persona = ('定位：' + role.roleLabel + '\n身份：' + role.identity + '\n性格：' + role.traits.join('、') + '\n目标：' + role.goals + '\n关系：' + role.relations.join('、') + '\n成长线：' + (role.arc.length > 0 ? role.arc.join(' > ') : '—') + '\n知情度：' + role.knowledge.join('、')).trim();
+      push('roleTemplate', role.name, role.identity, persona, src, [], { roleLabel: role.roleLabel, traits: role.traits, relations: role.relations, arc: role.arc, knowledge: role.knowledge });
+    }
+  }
+
+  saveAuthorAssets(library)
+  return library
+}
+

package/src/bookshelf.ts

@@ -1,151 +1,187 @@
-/**
- * 书架（Bookshelf）— 多书管理：一本书记录一个独立输出目录。
- * 状态持久化到 ~/.dsh/dsh-novel-forge-bookshelf.json（跟随 dsh 配置惯例）。
- */
-
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs'
-import { homedir } from 'node:os'
-import { join } from 'node:path'
-import { randomBytes } from 'node:crypto'
-import type { BookEntry, BookshelfSnapshot } from './protocol.ts'
-import { loadProject } from './engine.ts'
-
-/** 书架配置文件路径。 */
-export function bookshelfFile(): string {
-  return join(homedir(), '.dsh', 'dsh-novel-forge-bookshelf.json')
-}
-
-interface BookshelfStore {
-  books: BookEntry[]
-  activeBookId: string | null
-}
-
-function defaultStore(): BookshelfStore {
-  return { books: [], activeBookId: null }
-}
-
-/** 读取书架（无则返回空）。 */
-export function loadBookshelf(): BookshelfStore {
-  const file = bookshelfFile()
-  if (!existsSync(file)) return defaultStore()
-  try {
-    let raw = readFileSync(file, 'utf8')
-    if (raw.charCodeAt(0) === 0xFEFF) raw = raw.slice(1)
-    const parsed = JSON.parse(raw) as BookshelfStore
-    if (!Array.isArray(parsed.books)) return defaultStore()
-    return { books: parsed.books, activeBookId: parsed.activeBookId ?? null }
-  } catch {
-    return defaultStore()
-  }
-}
-
-/** 持久化书架。 */
-function saveBookshelf(store: BookshelfStore): void {
-  const file = bookshelfFile()
-  mkdirSync(join(homedir(), '.dsh'), { recursive: true })
-  writeFileSync(file, JSON.stringify(store, null, 2), 'utf8')
-}
-
-/** 当前激活的书。 */
-export function activeBook(store: BookshelfStore): BookEntry | undefined {
-  return store.books.find(b => b.id === store.activeBookId)
-}
-
-/** 书架快照（含每本书的进度摘要）。 */
-export function bookshelfSnapshot(store: BookshelfStore): BookshelfSnapshot {
-  return {
-    books: store.books.map(book => {
-      const project = loadProject(book.outputDir)
-      const done = project === undefined ? 0 : project.chapters.filter(c => c.status === 'approved' || c.status === 'written' || c.status === 'rejected').length
-      const hasCover = project?.coverPath !== undefined && project.coverPath !== '' && existsSync(join(book.outputDir, project.coverPath))
-      return {
-        ...book,
-        done,
-        total: project?.chapters.length ?? 0,
-        hasProject: project !== undefined,
-        hasCover,
-        blurb: project?.blurb,
-      }
-    }),
-    activeBookId: store.activeBookId,
-  }
-}
-
-/** 新建一本书（自动成为当前书）。 */
-export function createBook(bookName: string, outputDir: string): BookEntry {
-  const store = loadBookshelf()
-  const id = `book-${Date.now().toString(36)}-${randomBytes(3).toString('hex')}`
-  const now = new Date().toISOString()
-  const book: BookEntry = { id, bookName, outputDir, createdAt: now, updatedAt: now }
-  store.books.push(book)
-  store.activeBookId = id
-  saveBookshelf(store)
-  return book
-}
-
-/** 更新某本书的书名（开书向导导入大纲后书名以大纲首行为准）。 */
-export function renameBook(id: string, bookName: string): boolean {
-  const store = loadBookshelf()
-  const book = store.books.find(b => b.id === id)
-  if (book === undefined) return false
-  book.bookName = bookName
-  book.updatedAt = new Date().toISOString()
-  saveBookshelf(store)
-  return true
-}
-
-/**
- * 播种：书架为空时，把指定输出目录下已有的项目自动登记为第一本书。
- * 兼容升级场景 —— 旧版插件直接在输出目录写项目，从未登记书架。
- * @param outputDir - 候选输出目录（通常为 settings 的默认输出目录）。
- * @returns 是否发生了播种。
- */
-export function seedBookshelfFromOutputDir(outputDir: string): boolean {
-  const store = loadBookshelf()
-  if (store.books.length > 0) return false
-  if (!existsSync(outputDir)) return false
-  // 有项目文件，或至少有章节文件，才视为"已有的书"。
-  const hasProject = existsSync(join(outputDir, 'novel-project.json'))
-  const hasChapters = existsSync(outputDir)
-  if (!hasProject && !hasChapters) return false
-  const project = loadProject(outputDir)
-  const bookName = project?.bookName ?? outputDir.split(/[\\/]/).pop() ?? '未命名小说'
-  createBook(bookName, outputDir)
-  return true
-}
-
-/** 激活一本书。 */
-export function activateBook(id: string): BookEntry | undefined {
-  const store = loadBookshelf()
-  const book = store.books.find(b => b.id === id)
-  if (book === undefined) return undefined
-  store.activeBookId = id
-  book.updatedAt = new Date().toISOString()
-  saveBookshelf(store)
-  return book
-}
-
-/** 移除一本书。 */
-export function removeBook(id: string): boolean {
-  const store = loadBookshelf()
-  const idx = store.books.findIndex(b => b.id === id)
-  if (idx === -1) return false
-  store.books.splice(idx, 1)
-  if (store.activeBookId === id) {
-    store.activeBookId = store.books[0]?.id ?? null
-  }
-  saveBookshelf(store)
-  return true
-}
-
-/** 当前书输出目录（无书架则 undefined，回退 settings）。 */
-export function activeBookOutputDir(): string | undefined {
-  const book = activeBook(loadBookshelf())
-  return book?.outputDir
-}
-
-/** 默认输出目录推断：小说目录/书名。 */
-export function defaultOutputDirFor(bookName: string): string {
-  const clean = bookName.replace(/[\\/:*?"<>|]/g, '').trim().slice(0, 40) || '未命名小说'
-  return join('D:\\ryan work\\harness\\novels', clean)
-}
+/**
+ * 书架（Bookshelf）— 多书管理：一本书记录一个独立输出目录。
+ * 状态持久化到 ~/.dsh/dsh-novel-forge-bookshelf.json（跟随 dsh 配置惯例）。
+ */
+
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs'
+import { homedir } from 'node:os'
+import { join } from 'node:path'
+import { randomBytes } from 'node:crypto'
+import type { BookEntry, BookshelfSnapshot } from './protocol.ts'
+import { loadProject } from './engine.ts'
+
+/** 书架配置文件路径。 */
+export function bookshelfFile(): string {
+  return join(homedir(), '.dsh', 'dsh-novel-forge-bookshelf.json')
+}
+
+interface BookshelfStore {
+  books: BookEntry[]
+  activeBookId: string | null
+}
+
+function defaultStore(): BookshelfStore {
+  return { books: [], activeBookId: null }
+}
+
+/** 读取书架（无则返回空）。 */
+export function loadBookshelf(): BookshelfStore {
+  const file = bookshelfFile()
+  if (!existsSync(file)) return defaultStore()
+  try {
+    let raw = readFileSync(file, 'utf8')
+    if (raw.charCodeAt(0) === 0xFEFF) raw = raw.slice(1)
+    const parsed = JSON.parse(raw) as BookshelfStore
+    if (!Array.isArray(parsed.books)) return defaultStore()
+    return { books: parsed.books, activeBookId: parsed.activeBookId ?? null }
+  } catch (error) {
+    // 书架文件损坏回退默认值会遮蔽数据丢失，留痕到服务端日志。
+    console.error(`[novel-forge] bookshelf file unreadable/corrupt: ${file} — ${(error as Error).message}`)
+    return defaultStore()
+  }
+}
+
+/** 持久化书架。 */
+function saveBookshelf(store: BookshelfStore): void {
+  const file = bookshelfFile()
+  mkdirSync(join(homedir(), '.dsh'), { recursive: true })
+  writeFileSync(file, JSON.stringify(store, null, 2), 'utf8')
+}
+
+/** 当前激活的书。 */
+export function activeBook(store: BookshelfStore): BookEntry | undefined {
+  return store.books.find(b => b.id === store.activeBookId)
+}
+
+/** 书架快照（含每本书的进度摘要）。 */
+export function bookshelfSnapshot(store: BookshelfStore): BookshelfSnapshot {
+  return {
+    books: store.books.map(book => {
+      const project = loadProject(book.outputDir)
+      const done = project === undefined ? 0 : project.chapters.filter(c => c.status === 'approved' || c.status === 'written' || c.status === 'rejected').length
+      const hasCover = project?.coverPath !== undefined && project.coverPath !== '' && existsSync(join(book.outputDir, project.coverPath))
+      return {
+        ...book,
+        done,
+        total: project?.chapters.length ?? 0,
+        hasProject: project !== undefined,
+        hasCover,
+        blurb: project?.blurb,
+      }
+    }),
+    activeBookId: store.activeBookId,
+  }
+}
+
+/** 新建一本书（自动成为当前书）。 */
+export function createBook(bookName: string, outputDir: string): BookEntry {
+  const store = loadBookshelf()
+  const id = `book-${Date.now().toString(36)}-${randomBytes(3).toString('hex')}`
+  const now = new Date().toISOString()
+  const book: BookEntry = { id, bookName, outputDir, createdAt: now, updatedAt: now }
+  store.books.push(book)
+  store.activeBookId = id
+  saveBookshelf(store)
+  return book
+}
+
+/** 更新某本书的书名（开书向导导入大纲后书名以大纲首行为准）。 */
+export function renameBook(id: string, bookName: string): boolean {
+  const store = loadBookshelf()
+  const book = store.books.find(b => b.id === id)
+  if (book === undefined) return false
+  book.bookName = bookName
+  book.updatedAt = new Date().toISOString()
+  saveBookshelf(store)
+  return true
+}
+
+/**
+ * 播种：书架为空时，把指定输出目录下已有的项目自动登记为第一本书。
+ * 兼容升级场景 —— 旧版插件直接在输出目录写项目，从未登记书架。
+ * @param outputDir - 候选输出目录（通常为 settings 的默认输出目录）。
+ * @returns 是否发生了播种。
+ */
+export function seedBookshelfFromOutputDir(outputDir: string): boolean {
+  const store = loadBookshelf()
+  if (store.books.length > 0) return false
+  if (!existsSync(outputDir)) return false
+  // 有项目文件，或至少有章节文件，才视为"已有的书"。
+  const hasProject = existsSync(join(outputDir, 'novel-project.json'))
+  const hasChapters = existsSync(outputDir)
+  if (!hasProject && !hasChapters) return false
+  const project = loadProject(outputDir)
+  const bookName = project?.bookName ?? outputDir.split(/[\\/]/).pop() ?? '未命名小说'
+  createBook(bookName, outputDir)
+  return true
+}
+
+/** 导入已有项目目录到书架：校验 novel-project.json，已存在则直接激活。 */
+export function importDir(outputDir: string): { book: BookEntry; existed: boolean } {
+  const project = loadProject(outputDir)
+  if (project === undefined) throw new Error(`目录中未找到有效的 novel-project.json：${outputDir}`)
+  const store = loadBookshelf()
+  const existed = store.books.find(b => b.outputDir === outputDir)
+  if (existed !== undefined) {
+    store.activeBookId = existed.id
+    saveBookshelf(store)
+    return { book: existed, existed: true }
+  }
+  const bookName = project.bookName !== '' ? project.bookName : (outputDir.split(/[\\/]/).pop() ?? '未命名小说')
+  const book = createBook(bookName, outputDir)
+  return { book, existed: false }
+}
+
+/** 激活一本书。 */
+export function activateBook(id: string): BookEntry | undefined {
+  const store = loadBookshelf()
+  const book = store.books.find(b => b.id === id)
+  if (book === undefined) return undefined
+  store.activeBookId = id
+  book.updatedAt = new Date().toISOString()
+  saveBookshelf(store)
+  return book
+}
+
+/** 移除一本书。 */
+export function removeBook(id: string): boolean {
+  const store = loadBookshelf()
+  const idx = store.books.findIndex(b => b.id === id)
+  if (idx === -1) return false
+  store.books.splice(idx, 1)
+  if (store.activeBookId === id) {
+    store.activeBookId = store.books[0]?.id ?? null
+  }
+  saveBookshelf(store)
+  return true
+}
+
+/** 当前书输出目录（无书架则 undefined，回退 settings）。 */
+export function activeBookOutputDir(): string | undefined {
+  const book = activeBook(loadBookshelf())
+  return book?.outputDir
+}
+
+/**
+ * 书目录重指向：把 outputDir 等于 from 的书条目全部改为 to（目录迁移联动）。
+ * @returns 重指向的条目数。
+ */
+export function repointBookDirs(from: string, to: string): number {
+  const store = loadBookshelf()
+  let count = 0
+  for (const book of store.books) {
+    if (book.outputDir === from) {
+      book.outputDir = to
+      book.updatedAt = new Date().toISOString()
+      count += 1
+    }
+  }
+  if (count > 0) saveBookshelf(store)
+  return count
+}
+
+/** 默认输出目录推断：~/.dsh/novels/书名。 */
+export function defaultOutputDirFor(bookName: string): string {
+  const clean = bookName.replace(/[\\/:*?"<>|]/g, '').trim().slice(0, 40) || '未命名小说'
+  return join(homedir(), '.dsh', 'novels', clean)
+}