@waterwx/dsh-novel-forge 2.0.0-alpha.1 → 2.0.0

package/src/client/chip-prefs.ts

@@ -0,0 +1,121 @@
+/**
+ * 右上角「小说工坊生成状态」芯片的显示偏好（纯客户端、localStorage）。
+ *
+ * 为什么是 localStorage 而不是插件配置：这是纯界面 chrome，与显示模式 / 界面密度 /
+ * 编辑器字号 / 芯片自身的位置记忆同类；哪个浏览器想清屏就哪个浏览器清，不需要跨端同步。
+ * 将来若要跨端，把键名沿用成 config 字段 `showRunChip` 即可平滑升级。
+ *
+ * 三态：
+ *   off    关闭（同时停止 4 秒轮询，不白跑）
+ *   active 只在生产单**运行中或已暂停**时显示（不想错过生成，也不想留残影）
+ *   always 只要该目录有生产单状态就显示（默认，保持历史行为）
+ */
+
+import type { RunState } from '../protocol.ts'
+
+export type RunChipMode = 'off' | 'active' | 'always'
+
+/** localStorage 键（与芯片位置记忆同一命名空间）。 */
+export const RUN_CHIP_MODE_KEY = 'dsh-novel-forge.runChip.mode'
+
+/** 同页变更通知用的自定义事件名（跨标签页走原生 storage 事件）。 */
+export const RUN_CHIP_MODE_EVENT = 'dsh-novel-forge:runchip-mode'
+
+/** 默认值与历史行为一致：有生产单状态就显示。 */
+export const DEFAULT_RUN_CHIP_MODE: RunChipMode = 'always'
+
+/** 设置页下拉用的三态选项（顺序即用户看到的顺序）。 */
+export const RUN_CHIP_MODE_OPTIONS: ReadonlyArray<{ value: RunChipMode; label: string }> = [
+  { value: 'always', label: '总是显示' },
+  { value: 'active', label: '仅生成中 / 暂停时显示' },
+  { value: 'off', label: '关闭' },
+]
+
+const ALL_MODES: readonly RunChipMode[] = ['off', 'active', 'always']
+
+function isMode(value: unknown): value is RunChipMode {
+  return typeof value === 'string' && (ALL_MODES as readonly string[]).includes(value)
+}
+
+/** 安全拿 localStorage：隐私模式 / 非浏览器环境下为 undefined。 */
+function storage(): Storage | undefined {
+  try {
+    return typeof localStorage === 'undefined' ? undefined : localStorage
+  } catch {
+    return undefined
+  }
+}
+
+/** 读取偏好（缺失 / 非法 / 存储不可用 → 默认值）。 */
+export function readRunChipMode(): RunChipMode {
+  try {
+    const raw = storage()?.getItem(RUN_CHIP_MODE_KEY)
+    return isMode(raw) ? raw : DEFAULT_RUN_CHIP_MODE
+  } catch {
+    return DEFAULT_RUN_CHIP_MODE
+  }
+}
+
+/** 已注册的同页监听者（storage 事件只覆盖跨标签页，同页要靠这里）。 */
+const listeners = new Set<(mode: RunChipMode) => void>()
+
+/** 写入偏好并通知所有监听者；存储不可用时仍通知（本次会话内生效）。 */
+export function writeRunChipMode(mode: RunChipMode): void {
+  try {
+    storage()?.setItem(RUN_CHIP_MODE_KEY, mode)
+  } catch {
+    /* 存储不可用：本次会话内仍然生效 */
+  }
+  for (const listener of [...listeners]) {
+    try {
+      listener(mode)
+    } catch {
+      /* 单个监听者出错不影响其他人 */
+    }
+  }
+  try {
+    if (typeof window !== 'undefined' && typeof CustomEvent === 'function') {
+      window.dispatchEvent(new CustomEvent(RUN_CHIP_MODE_EVENT, { detail: mode }))
+    }
+  } catch {
+    /* 事件派发失败无关紧要 */
+  }
+}
+
+/** 订阅偏好变化（同页 + 跨标签页 storage）；返回退订函数。 */
+export function subscribeRunChipMode(listener: (mode: RunChipMode) => void): () => void {
+  listeners.add(listener)
+  const onStorage = (event: StorageEvent): void => {
+    if (event.key !== null && event.key !== RUN_CHIP_MODE_KEY) return
+    listener(readRunChipMode())
+  }
+  let attached = false
+  try {
+    if (typeof window !== 'undefined') {
+      window.addEventListener('storage', onStorage)
+      attached = true
+    }
+  } catch {
+    attached = false
+  }
+  return () => {
+    listeners.delete(listener)
+    if (!attached) return
+    try {
+      window.removeEventListener('storage', onStorage)
+    } catch {
+      /* ignore */
+    }
+  }
+}
+
+/**
+ * 给定偏好与当前生产单状态，判断芯片是否应可见。
+ * null（没有生产单状态）永远不显示——这是渲染层与测试共用的唯一判据。
+ */
+export function isRunChipVisible(mode: RunChipMode, state: RunState | null | undefined): boolean {
+  if (state === null || state === undefined) return false
+  if (mode === 'off') return false
+  if (mode === 'active') return state.status === 'running' || state.status === 'paused'
+  return true
+}

package/src/client/docx.ts

@@ -1,69 +1,32 @@
-/**
- * Browser-side docx outline extraction: a .docx is a zip whose
- * word/document.xml holds the body text in <w:t> runs inside <w:p> paragraphs.
- * Uses fflate (inlined into the client bundle) so the user can pick or drag a
- * docx without any server upload.
- *
- * Import from 'fflate/browser' (not 'fflate'): the default entry resolves to
- * the Node build (esm/index.mjs), which calls module.createRequire() for the
- * optional worker_threads path — inlining that into the browser bundle leaves
- * a bare require("module") the client-modules table cannot answer.
- */
-
-import { unzipSync, strFromU8 } from 'fflate/browser'
-
-/** Decode the handful of XML entities docx bodies actually use. */
-function decodeEntities(text: string): string {
-  return text
-    .replace(/&lt;/g, '<')
-    .replace(/&gt;/g, '>')
-    .replace(/&quot;/g, '"')
-    .replace(/&apos;/g, "'")
-    .replace(/&amp;/g, '&')
-    .replace(/&nbsp;/g, ' ')
-}
-
-/** Extract plain text from a docx buffer: one line per <w:p> paragraph. */
-export function extractDocxTextFromBuffer(buffer: ArrayBuffer | Uint8Array): string {
-  const bytes = buffer instanceof Uint8Array ? buffer : new Uint8Array(buffer)
-  let files: ReturnType<typeof unzipSync>
-  try {
-    files = unzipSync(bytes)
-  } catch (error) {
-    throw new Error(`不是有效的 docx（zip 解压失败）：${(error as Error).message}`)
-  }
-  const document = files['word/document.xml']
-  if (document === undefined) {
-    throw new Error('不是有效的 docx（缺少 word/document.xml）')
-  }
-  const xml = strFromU8(document)
-
-  const paragraphs: string[] = []
-  const parts = xml.split(/<w:p\b[^>]*>/)
-  for (let i = 1; i < parts.length; i++) {
-    const segment = parts[i]!
-    const runs: string[] = []
-    const runRe = /<w:t(?:\s[^>]*)?>([\s\S]*?)<\/w:t>|<w:tab\b[^>]*\/>|<w:br\b[^>]*\/>/g
-    for (const match of segment.matchAll(runRe)) {
-      if (match[0].startsWith('<w:tab')) {
-        runs.push('\t')
-      } else if (match[0].startsWith('<w:br')) {
-        runs.push('\n')
-      } else {
-        runs.push(decodeEntities(match[1] ?? ''))
-      }
-    }
-    paragraphs.push(runs.join('').replace(/\u00a0/g, ' ').trimEnd())
-  }
-
-  const text = paragraphs.join('\n').replace(/\n{3,}/g, '\n\n').trim()
-  if (text.length === 0) {
-    throw new Error('docx 中没有可提取的文本')
-  }
-  return text
-}
-
-/** Read a File as ArrayBuffer. */
-export function readFileAsArrayBuffer(file: File): Promise<ArrayBuffer> {
-  return file.arrayBuffer()
-}
+/**
+ * Browser-side docx outline extraction: a .docx is a zip whose
+ * word/document.xml holds the body text in <w:t> runs inside <w:p> paragraphs.
+ * Uses fflate (inlined into the client bundle) so the user can pick or drag a
+ * docx without any server upload.
+ *
+ * Import from 'fflate/browser' (not 'fflate'): the default entry resolves to
+ * the Node build (esm/index.mjs), which calls module.createRequire() for the
+ * optional worker_threads path — inlining that into the browser bundle leaves
+ * a bare require("module") the client-modules table cannot answer.
+ *
+ * XML 解析共享核心在 ../docx-core.ts（与服务端同源）。
+ */
+
+import { unzipSync, strFromU8 } from 'fflate/browser'
+import { extractDocxTextFromXml } from '../docx-core.ts'
+
+/** Extract plain text from a docx buffer: one line per <w:p> paragraph. */
+export function extractDocxTextFromBuffer(buffer: ArrayBuffer | Uint8Array): string {
+  const bytes = buffer instanceof Uint8Array ? buffer : new Uint8Array(buffer)
+  let files: ReturnType<typeof unzipSync>
+  try {
+    files = unzipSync(bytes)
+  } catch (error) {
+    throw new Error(`不是有效的 docx（zip 解压失败）：${(error as Error).message}`)
+  }
+  const document = files['word/document.xml']
+  if (document === undefined) {
+    throw new Error('不是有效的 docx（缺少 word/document.xml）')
+  }
+  return extractDocxTextFromXml(strFromU8(document))
+}

package/src/client/index.ts

@@ -9,6 +9,7 @@ import { NovelApi } from './api.ts'
 import { mountPanel } from './mount.tsx'
 import { PanelController } from './panel/controller.ts'
 import { mountSidebarEntry } from './sidebar-entry.ts'
+import { mountRunStatus } from './run-status.ts'
 
 /** Required services (fiber inject waiting). */
 export const inject = ['slots', 'locale']
@@ -24,6 +25,7 @@ export function apply(ctx: ClientContext): void {
   try {
     disposers.push(mountSidebarEntry(controller))
     disposers.push(mountPanel(controller, api))
+    disposers.push(mountRunStatus(controller, api))
   } catch (error) {
     // DOM failures degrade the panel, never the GUI.
     console.warn('[dsh-novel-forge] mount failed:', error)

package/src/client/llmLive.ts

@@ -0,0 +1,111 @@
+/**
+ * 浏览器端「AI 创作实况」钩子：连接 /llm-live/stream (SSE)，按 LLM 调用分组会话。
+ * 轻量实现，不持久化缓存（对齐上游 useLlmLiveFeed 的核心行为）。
+ */
+
+import { useEffect, useMemo, useState } from 'react'
+import { NOVEL_API, type LlmLiveFrame, type LlmTokenUsage } from '../protocol.ts'
+
+// 帧结构定义在 protocol.ts（宿主与浏览器共用同一份线上形状），此处只再导出。
+export type { LlmLiveFrame }
+
+export interface LlmLiveSession {
+  sessionId: string
+  label: string
+  model?: string
+  phase: 'requesting' | 'streaming' | 'completed' | 'failed'
+  phaseMessage: string
+  preview: string
+  totalChars: number
+  startedAt: string
+  updatedAt: string
+  /** 本次调用的 token 用量（供应商未上报时缺省）。 */
+  usage?: LlmTokenUsage
+  /** 本次调用耗时（毫秒）。 */
+  elapsedMs?: number
+  /** 首字耗时（毫秒）。 */
+  firstTokenMs?: number
+  /** 是否保存了可查看的 Prompt。 */
+  hasPrompt?: boolean
+  /** Prompt 字符数（system + user）。 */
+  promptChars?: number
+}
+
+const MAX_PREVIEW_CHARS = 20_000
+
+function applyFrame(current: Record<string, LlmLiveSession>, frame: LlmLiveFrame): Record<string, LlmLiveSession> {
+  const next = { ...current }
+  const id = frame.sessionId
+  let session = next[id]
+  if (frame.type === 'session_started') {
+    session = {
+      sessionId: id,
+      label: frame.label ?? 'LLM 调用',
+      model: frame.model,
+      phase: 'requesting',
+      phaseMessage: '正在连接模型',
+      preview: '',
+      totalChars: 0,
+      startedAt: frame.at,
+      updatedAt: frame.at,
+      hasPrompt: frame.hasPrompt,
+      promptChars: frame.promptChars,
+    }
+    next[id] = session
+    return next
+  }
+  if (session === undefined) return next
+  if (frame.type === 'output_delta') {
+    const preview = (session.preview + (frame.content ?? '')).slice(-MAX_PREVIEW_CHARS)
+    session = { ...session, phase: session.phase === 'requesting' ? 'streaming' : session.phase, phaseMessage: session.phase === 'requesting' ? '模型正在返回内容' : session.phaseMessage, preview, totalChars: frame.totalChars ?? session.totalChars + (frame.content?.length ?? 0), updatedAt: frame.at }
+    next[id] = session
+  } else if (frame.type === 'reasoning_delta') {
+    session = { ...session, phase: session.phase === 'requesting' ? 'streaming' : session.phase, updatedAt: frame.at }
+    next[id] = session
+  } else if (frame.type === 'phase_changed') {
+    session = { ...session, phase: frame.phase ?? session.phase, phaseMessage: frame.phaseMessage ?? session.phaseMessage, updatedAt: frame.at }
+    next[id] = session
+  } else if (frame.type === 'session_completed') {
+    session = {
+      ...session,
+      phase: frame.phase ?? 'completed',
+      phaseMessage: frame.phase === 'failed' ? (frame.error ?? '调用失败') : '模型结果已准备完成',
+      totalChars: frame.totalChars ?? session.totalChars,
+      preview: session.preview || (frame.preview ?? ''),
+      updatedAt: frame.at,
+      usage: frame.usage ?? session.usage,
+      elapsedMs: frame.elapsedMs ?? session.elapsedMs,
+      firstTokenMs: frame.firstTokenMs ?? session.firstTokenMs,
+    }
+    next[id] = session
+  }
+  return next
+}
+
+export function useLlmLiveFeed(enabled = true): { connected: boolean; sessions: LlmLiveSession[] } {
+  const [sessions, setSessions] = useState<Record<string, LlmLiveSession>>({})
+  const [connected, setConnected] = useState(false)
+  useEffect(() => {
+    if (!enabled) {
+      setSessions({})
+      setConnected(false)
+      return
+    }
+    const es = new EventSource(NOVEL_API.llmLive)
+    es.onopen = () => setConnected(true)
+    es.onmessage = (ev) => {
+      try {
+        const frame = JSON.parse(ev.data as string) as LlmLiveFrame
+        setSessions(prev => applyFrame(prev, frame))
+      } catch { /* ignore malformed frame */ }
+    }
+    // EventSource 断线会自动重连；这里只在断线瞬间标记未连接，重连成功后会再次 onopen。
+    es.onerror = () => setConnected(false)
+    return () => {
+      es.close()
+      setConnected(false)
+    }
+  }, [enabled])
+  const list = useMemo(() => Object.values(sessions).sort((a, b) => a.startedAt.localeCompare(b.startedAt)), [sessions])
+  return { connected, sessions: list }
+}