@waterwx/dsh-novel-forge 1.7.3 → 1.9.0

package/src/routes.ts

@@ -13,6 +13,7 @@ import { spawn } from 'node:child_process'
 import { copyFileSync, existsSync, mkdirSync, readFileSync, rmSync, writeFileSync } from 'node:fs'
 import { join, basename, extname } from 'node:path'
 import { homedir } from 'node:os'
+import { randomBytes } from 'node:crypto'
 import { fileURLToPath } from 'node:url'
 import type { WebRoute } from '@deepseek-ai/dsh-host-webserver'
 import type { Context } from '@deepseek-ai/cordis'
@@ -21,6 +22,13 @@ import {
   NOVEL_API,
   type AssetsPatch,
   type AssetsResponse,
+  type AddModelRequest,
+  type AddModelResponse,
+  type LlmModelsResponse,
+  type LlmProvidersResponse,
+  type LlmVendorsResponse,
+  type RemoveProviderRequest,
+  type RemoveProviderResponse,
   type AssistantFrame,
   type AssistantHistoryResponse,
   type AssistantRequest,
@@ -100,15 +108,27 @@ import {
   type StoryboardPromptsRequest,
   type ImageTestRequest,
   type ImageTestResponse,
+  type LlmTestRequest,
+  type LlmTestResponse,
   type StoryboardPromptsResponse,
   type StyleEngineRequest,
   type SummaryRequest,
   type VolumesRequest,
   type VolumesResponse,
+  type AuthorAssetsResponse,
+  type AuthorAssetUpsertRequest,
+  type AuthorAssetRemoveRequest,
+  type AdaptAnalyzeRequest,
+  type AdaptAnalyzeResponse,
+  type AdaptProposeRequest,
+  type AdaptProposeResponse,
+  type AdaptExecuteRequest,
+  type AdaptExecuteResponse,
 } from './protocol.ts'
 import { readOutlineFromDocx } from './docx.ts'
 import { clearAssistantHistory, loadAssistantHistory, runAssistantTurn } from './assistant.ts'
 import { activateBook, bookshelfSnapshot, createBook, defaultOutputDirFor, importDir, loadBookshelf, removeBook, renameBook, seedBookshelfFromOutputDir } from './bookshelf.ts'
+import { loadAuthorAssets, upsertAuthorAsset, removeAuthorAsset, importDefaultAuthorAssets } from './author-assets.ts'
 import { BUILTIN_ANTI_AI_RULES, BUILTIN_GENRE_LIBRARY, BUILTIN_PROGRESSION_MODES, BUILTIN_STYLE_TEMPLATES, emptyProjectAssets } from './assets.ts'
 import {
   chapterFileName,
@@ -153,9 +173,18 @@ import {
   suggestOutlines,
   reverseOutlineFromChapters,
   breakdownBook,
+  analyzeAdaptation,
+  proposeAdaptation,
+  applyAdaptationReplacements,
   generateRoleReferenceImage,
   generateMangaRoleReferenceImage,
   testImageEndpoint,
+  testLlmModel,
+  registerLlmModel,
+  listLlmVendors,
+  listLlmModels,
+  listLlmProviders,
+  removeLlmProvider,
   generateStoryboardSkeleton,
   generateStoryboardTable,
   generateStoryboardPrompts,
@@ -2424,6 +2453,117 @@ export function makeRoutes(deps: NovelRoutesDeps): WebRoute[] {
     },
   }
 
+  // ------------------------------------------------------------- llm-test
+  /** LLM 连通性测试：对选中的提供商/模型发一次最小真实调用。 */
+  const llmTestRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.llmTest,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<LlmTestRequest>(req)
+      const provider = (body?.provider ?? '').trim()
+      const model = (body?.model ?? '').trim()
+      if (provider === '' || model === '') {
+        writeJson(res, 400, { error: 'provider 与 model 必填' })
+        return
+      }
+      const result = await testLlmModel(ctx, provider, model)
+      writeJson(res, 200, result satisfies LlmTestResponse)
+    },
+  }
+
+  // ------------------------------------------------------------- llm-add
+  /** 添加模型：厂商直填 API Key，或自定义 OpenAI 兼容路由（写 DSH 凭据 + router）。 */
+  const addModelRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.addModel,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<AddModelRequest>(req)
+      if (body === undefined) {
+        writeJson(res, 400, { error: '缺少请求体' })
+        return
+      }
+      try {
+        const result = await registerLlmModel(ctx, body)
+        writeJson(res, 200, result satisfies AddModelResponse)
+      } catch (error) {
+        writeJson(res, 500, { error: `添加模型失败：${(error as Error).message}` })
+      }
+    },
+  }
+
+  // ---------------------------------------------------------- llm-vendors
+  /** 运行时厂商目录（DSH pi-ai 可配置提供方 + 内置适配器）。 */
+  const llmVendorsRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.llmVendors,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'GET')) return
+      try {
+        const result = await listLlmVendors(ctx)
+        writeJson(res, 200, result satisfies LlmVendorsResponse)
+      } catch (error) {
+        writeJson(res, 500, { error: `读取厂商目录失败：${(error as Error).message}` })
+      }
+    },
+  }
+
+  // ---------------------------------------------------------- llm-models
+  /** 查询某个 provider 当前可用的模型（添加成功后可即时刷新下拉）。 */
+  const llmModelsRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.llmModels,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'GET')) return
+      const url = new URL(req.url ?? '', 'http://localhost')
+      const provider = decodeURIComponent((url.searchParams.get('provider') ?? '').trim())
+      try {
+        const result = await listLlmModels(ctx, provider)
+        writeJson(res, 200, result satisfies LlmModelsResponse)
+      } catch (error) {
+        writeJson(res, 500, { error: `读取模型列表失败：${(error as Error).message}` })
+      }
+    },
+  }
+
+  // ---------------------------------------------------------- llm-providers
+  /** 已注册的提供方路由列表（提供方管理卡片）。 */
+  const llmProvidersRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.llmProviders,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'GET')) return
+      try {
+        const result = await listLlmProviders(ctx)
+        writeJson(res, 200, result satisfies LlmProvidersResponse)
+      } catch (error) {
+        writeJson(res, 500, { error: `读取提供方列表失败：${(error as Error).message}` })
+      }
+    },
+  }
+
+  // ------------------------------------------------------------- llm-remove
+  /** 移除一个提供方（unset key + 移除 llm-pi-ai 路由）。 */
+  const llmRemoveRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.llmRemove,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<RemoveProviderRequest>(req)
+      if (body === undefined) {
+        writeJson(res, 400, { error: '缺少请求体' })
+        return
+      }
+      try {
+        const result = await removeLlmProvider(ctx, body)
+        writeJson(res, 200, result satisfies RemoveProviderResponse)
+      } catch (error) {
+        writeJson(res, 500, { error: `移除提供方失败：${(error as Error).message}` })
+      }
+    },
+  }
+
   // ---------------------------------------------------- storyboard table
   /** 分镜·导演级：骨架 → 分镜表（镜头级）。 */
   const storyboardTableRoute: WebRoute = {
@@ -2754,6 +2894,174 @@ export function makeRoutes(deps: NovelRoutesDeps): WebRoute[] {
     },
   }
 
+  // -------------------------------------------------- author assets (总数据)
+  const authorAssetsRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.authorAssets,
+    handler: (req, res) => {
+      if (!guard(req, res, 'GET')) return
+      const response: AuthorAssetsResponse = { assets: loadAuthorAssets() }
+      writeJson(res, 200, response)
+    },
+  }
+
+  const authorAssetsUpsertRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.authorAssetsUpsert,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<AuthorAssetUpsertRequest>(req)
+      const asset = body?.asset
+      if (asset === undefined || asset.name.trim() === '' || asset.content.trim() === '') {
+        writeJson(res, 400, { error: 'asset 须含 name / content' })
+        return
+      }
+      const response: AuthorAssetsResponse = { assets: upsertAuthorAsset(asset) }
+      writeJson(res, 200, response)
+    },
+  }
+
+  const authorAssetsRemoveRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.authorAssetsRemove,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<AuthorAssetRemoveRequest>(req)
+      const id = body?.id ?? ''
+      if (id === '') { writeJson(res, 400, { error: 'id 必填' }); return }
+      const response: AuthorAssetsResponse = { assets: removeAuthorAsset(id) }
+      writeJson(res, 200, response)
+    },
+  }
+
+  const authorAssetsImportDefaultRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.authorAssetsImportDefault,
+    handler: (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const response: AuthorAssetsResponse = { assets: importDefaultAuthorAssets() }
+      writeJson(res, 200, response)
+    },
+  }
+
+  // --------------------------------------------------- adaptation (改编 P0 分析)
+  const adaptAnalyzeRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.adaptAnalyze,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      if (!getConfig().enableAdaptMode) { writeJson(res, 404, { error: '改编模式未开启' }); return }
+      const body = await readJsonBody<AdaptAnalyzeRequest>(req)
+      const config = getConfig()
+      let text = body?.text?.trim() ?? ''
+      if (text === '' && body?.filePath !== undefined && body.filePath !== '') {
+        try {
+          text = readFileSync(body.filePath, 'utf8')
+        } catch (err) {
+          writeJson(res, 400, { error: '读取文件失败：' + (err as Error).message })
+          return
+        }
+      }
+      if (text.length < 200) {
+        writeJson(res, 400, { error: '全文内容过短（<200 字符），请上传完整小说文本' })
+        return
+      }
+      try {
+        const response: AdaptAnalyzeResponse = await analyzeAdaptation(ctx, config, text)
+        writeJson(res, 200, response)
+      } catch (err) {
+        writeJson(res, 400, { error: (err as Error).message })
+      }
+    },
+  }
+
+  const adaptProposeRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.adaptPropose,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      if (!getConfig().enableAdaptMode) { writeJson(res, 404, { error: '改编模式未开启' }); return }
+      const body = await readJsonBody<AdaptProposeRequest>(req)
+      const config = getConfig()
+      const text = (body?.text ?? '').trim()
+      const selections = body?.selections ?? []
+      if (text.length < 200 || selections.length === 0) {
+        writeJson(res, 400, { error: '请提供全文与至少一条要改的维度' })
+        return
+      }
+      try {
+        const response: AdaptProposeResponse = await proposeAdaptation(ctx, config, text, selections, body?.dimensions)
+        writeJson(res, 200, response)
+      } catch (err) {
+        writeJson(res, 400, { error: (err as Error).message })
+      }
+    },
+  }
+
+  const adaptExecuteRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.adaptExecute,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      if (!getConfig().enableAdaptMode) { writeJson(res, 404, { error: '改编模式未开启' }); return }
+      const body = await readJsonBody<AdaptExecuteRequest>(req)
+      const text = body?.text ?? ''
+      const mappings = body?.mappings ?? []
+      if (text.length < 200 || mappings.length === 0) {
+        writeJson(res, 400, { error: '请提供全文与映射表' })
+        return
+      }
+      if (body?.mode === 'rewrite') {
+        writeJson(res, 400, { error: 'rewrite 模式尚未实现，请使用 replace（术语替换）' })
+        return
+      }
+      const { adaptedText, hits } = applyAdaptationReplacements(text, mappings)
+      const response: AdaptExecuteResponse = { adaptedText, mappings: mappings.length, hits }
+      writeJson(res, 200, response)
+    },
+  }
+
+  // ------------------------------------------ theme custom background (upload + serve)
+  const THEME_BG_DIR = join(homedir(), '.dsh', 'dsh-novel-forge-assets')
+  const MIME_BY_EXT: Record<string, string> = { png: 'image/png', jpg: 'image/jpeg', jpeg: 'image/jpeg', webp: 'image/webp', gif: 'image/gif' }
+
+  const themeBackgroundUploadRoute: WebRoute = {
+    kind: 'exact',
+    path: NOVEL_API.themeBackgroundUpload,
+    handler: async (req, res) => {
+      if (!guard(req, res, 'POST')) return
+      const body = await readJsonBody<{ dataUrl?: string }>(req)
+      const dataUrl = body?.dataUrl ?? ''
+      const m = /^data:image\/(png|jpeg|jpg|webp|gif);base64,(.+)$/i.exec(dataUrl)
+      if (m === null) { writeJson(res, 400, { error: '仅支持 PNG/JPG/WebP/GIF 图片' }); return }
+      const ext = m[1]!.toLowerCase() === 'jpg' ? 'jpeg' : m[1]!.toLowerCase()
+      const buf = Buffer.from(m[2]!, 'base64')
+      if (buf.length === 0) { writeJson(res, 400, { error: '图片数据为空' }); return }
+      mkdirSync(THEME_BG_DIR, { recursive: true })
+      const name = 'theme-bg-' + Date.now().toString(36) + '-' + randomBytes(3).toString('hex') + '.' + ext
+      writeFileSync(join(THEME_BG_DIR, name), buf)
+      writeJson(res, 200, { url: NOVEL_API.themeBackgroundGet + '/' + name })
+    },
+  }
+
+  const themeBackgroundGetRoute: WebRoute = {
+    kind: 'prefix',
+    path: NOVEL_API.themeBackgroundGet,
+    handler: (req, res) => {
+      if (!guard(req, res, 'GET')) return
+      let pathName = ''
+      try { pathName = new URL(req.url ?? '', 'http://localhost').pathname } catch { /* ignore */ }
+      const name = basename(pathName)
+      if (!name.startsWith('theme-bg-')) { writeJson(res, 404, { error: 'not found' }); return }
+      const file = join(THEME_BG_DIR, name)
+      if (!existsSync(file)) { writeJson(res, 404, { error: 'not found' }); return }
+      const ext = extname(file).slice(1)
+      const type = MIME_BY_EXT[ext] ?? 'application/octet-stream'
+      res.writeHead(200, { 'content-type': type, 'cache-control': 'public, max-age=31536000' })
+      res.end(readFileSync(file))
+    },
+  }
+
   return [
     statusRoute,
     loadOutlineRoute,
@@ -2810,12 +3118,27 @@ export function makeRoutes(deps: NovelRoutesDeps): WebRoute[] {
     styleImageRoute,
     storyboardSkeletonRoute,
     imageTestRoute,
+    llmTestRoute,
+    addModelRoute,
+    llmVendorsRoute,
+    llmModelsRoute,
+    llmProvidersRoute,
+    llmRemoveRoute,
     storyboardTableRoute,
     storyboardPromptsRoute,
     breakdownRoute,
     runStartRoute,
     runControlRoute,
     runStatusRoute,
+    authorAssetsRoute,
+    authorAssetsUpsertRoute,
+    authorAssetsRemoveRoute,
+    authorAssetsImportDefaultRoute,
+    adaptAnalyzeRoute,
+    adaptProposeRoute,
+    adaptExecuteRoute,
+    themeBackgroundUploadRoute,
+    themeBackgroundGetRoute,
   ]
 }