@exccoder/dsh-media-workbench 0.0.0-stage → 0.2.2

package/src/llm.js

@@ -0,0 +1,60 @@
+// 用 DSH 里已经配好的默认模型写作，不额外存密钥
+export const modelStatus = { ok: null, error: null }
+
+export function modelInfo(ctx) {
+  try {
+    const selection = ctx.get('agentDefaultModel')?.currentSelection()
+    if (!ctx.get('llm') || !selection?.provider || !selection?.model) return null
+    return { provider: selection.provider, model: selection.model, label: selection.model }
+  } catch {
+    return null
+  }
+}
+
+/**
+ * 流式调用当前模型。
+ * @param {object} ctx
+ * @param {{system: string, user: string, maxTokens?: number, signal?: AbortSignal, onDelta?: (t: string) => void}} request
+ * @returns {Promise<{text: string, model: string}>}
+ */
+export async function streamText(ctx, { system, user, maxTokens = 4000, signal, onDelta }) {
+  const info = modelInfo(ctx)
+  if (!info) throw Object.assign(new Error('当前没有可用的写作模型。先在 DSH 里选择模型（或配置 DeepSeek 账号），再回来试。'), { status: 503 })
+  const llm = ctx.get('llm')
+  let text = ''
+  try {
+    const call = await llm.prepareCall({ provider: info.provider, model: info.model, maxTokens }, signal, (controls, model) => {
+      const effort = model?.reasoning?.efforts?.[0]?.id
+      return effort === undefined ? controls : { ...controls, reasoningEffort: effort }
+    })
+    for await (const chunk of call.stream({
+      ...call.config,
+      system,
+      messages: [{ role: 'user', content: [{ type: 'text', text: user }], source: { kind: '@exccoder/dsh-media-workbench' } }],
+      signal,
+    })) {
+      if (chunk.type === 'text-delta') {
+        text += chunk.text
+        onDelta?.(chunk.text)
+      } else if (chunk.type === 'finish' && ['error', 'aborted'].includes(chunk.reason?.kind)) {
+        throw new Error(chunk.reason.failure?.message || '模型调用中断')
+      }
+    }
+    modelStatus.ok = true
+    modelStatus.error = null
+    return { text, model: info.label }
+  } catch (err) {
+    modelStatus.ok = false
+    modelStatus.error = err.message
+    throw err
+  }
+}
+
+export function extractJson(text) {
+  const fenced = /```(?:json)?\s*([\s\S]*?)```/.exec(text)
+  const body = fenced ? fenced[1] : text
+  const start = body.search(/[{[]/)
+  const end = Math.max(body.lastIndexOf('}'), body.lastIndexOf(']'))
+  if (start < 0 || end < start) throw new Error('模型没有返回 JSON')
+  return JSON.parse(body.slice(start, end + 1))
+}

package/src/settings.js

@@ -0,0 +1,233 @@
+// 插件设置：接口地址、模型、密钥。密钥只留本机，返回浏览器时永远脱敏。
+import { access, chmod, mkdir, readFile, rename, unlink, writeFile } from 'node:fs/promises'
+import { constants } from 'node:fs'
+import { join } from 'node:path'
+
+export const IMAGE_PRESETS = [
+  { id: 'openai', name: 'OpenAI', baseUrl: 'https://api.openai.com/v1', model: 'gpt-image-1', note: 'gpt-image-1 / dall-e-3' },
+  { id: 'siliconflow', name: '硅基流动', baseUrl: 'https://api.siliconflow.cn/v1', model: 'Kwai-Kolors/Kolors', note: 'Kolors 等图像模型' },
+  { id: 'zhipu', name: '智谱 BigModel', baseUrl: 'https://open.bigmodel.cn/api/paas/v4', model: 'cogview-3-flash', note: 'CogView 系列' },
+  { id: 'ark', name: '火山方舟', baseUrl: 'https://ark.cn-beijing.volces.com/api/v3', model: 'doubao-seedream-4-0-250828', note: 'Seedream 系列' },
+  { id: 'custom', name: '自定义', baseUrl: '', model: '', note: '任何兼容 /images/generations 的服务' },
+]
+
+export const VIDEO_PRESETS = [
+  { id: 'openai', name: 'OpenAI Sora', baseUrl: 'https://api.openai.com/v1', model: 'sora-2', note: 'POST /videos，轮询 GET /videos/{id}' },
+  { id: 'custom', name: '自定义', baseUrl: '', model: '', note: '兼容的异步视频接口，可在高级设置里改路径' },
+]
+
+const DEFAULTS = {
+  image: {
+    preset: 'openai',
+    baseUrl: '',
+    model: '',
+    apiKey: '',
+    size: '1024x1024',
+    quality: '',
+    extra: '',
+  },
+  video: {
+    preset: 'openai',
+    baseUrl: '',
+    model: '',
+    apiKey: '',
+    createPath: '/videos',
+    statusPath: '/videos/{id}',
+    contentPath: '/videos/{id}/content',
+    imageField: 'image',
+    extra: '',
+  },
+  language: 'auto',
+  storage: {
+    outputDir: '',
+  },
+}
+
+function trim(value, max = 300) {
+  return typeof value === 'string' ? value.trim().slice(0, max) : undefined
+}
+
+function normalizeBaseUrl(value) {
+  if (typeof value !== 'string') return undefined
+  const url = value.trim().replace(/\/+$/, '').replace(/\/(images\/generations|images\/edits|videos)$/, '')
+  if (url && !/^https?:\/\/[^\s]+$/i.test(url)) {
+    const err = new Error('接口地址要以 http:// 或 https:// 开头')
+    err.status = 400
+    throw err
+  }
+  return url
+}
+
+function normalizePath(value, fallback, max = 200) {
+  const text = trim(value, max)
+  if (text === undefined) return undefined
+  if (!text) return fallback
+  if (!text.startsWith('/')) {
+    const err = new Error('接口路径要以 / 开头')
+    err.status = 400
+    throw err
+  }
+  return text
+}
+
+function normalizeOutputDir(value) {
+  if (value === undefined) return undefined
+  if (typeof value !== 'string') {
+    const err = new Error('产物目录要是文本路径')
+    err.status = 400
+    throw err
+  }
+  let dir = value.trim()
+  if (/^[a-zA-Z]:[\\/]?$/.test(dir)) {
+    dir = `${dir.slice(0, 2)}\\`
+  } else {
+    dir = dir.replace(/[\\/]+$/, '')
+  }
+  if (!dir) return ''
+  if (!/^(?:[a-zA-Z]:[\\/]|\/)/.test(dir)) {
+    const err = new Error('产物目录要填绝对路径，例如 D:\\AI作品')
+    err.status = 400
+    throw err
+  }
+  return dir
+}
+
+function normalizeExtra(value) {
+  if (value === undefined) return undefined
+  if (typeof value !== 'string') {
+    const err = new Error('附加参数要是 JSON 文本')
+    err.status = 400
+    throw err
+  }
+  const text = value.trim()
+  if (!text) return ''
+  let parsed
+  try {
+    parsed = JSON.parse(text)
+  } catch {
+    const err = new Error('附加参数不是有效的 JSON')
+    err.status = 400
+    throw err
+  }
+  if (!parsed || typeof parsed !== 'object' || Array.isArray(parsed)) {
+    const err = new Error('附加参数要是一个 JSON 对象')
+    err.status = 400
+    throw err
+  }
+  return text
+}
+
+export class Settings {
+  constructor(root) {
+    this.file = join(root, 'settings.json')
+  }
+
+  async read() {
+    try {
+      const saved = JSON.parse(await readFile(this.file, 'utf8'))
+      return {
+        ...DEFAULTS,
+        ...saved,
+        image: { ...DEFAULTS.image, ...(saved.image ?? {}) },
+        video: { ...DEFAULTS.video, ...(saved.video ?? {}) },
+        storage: { ...DEFAULTS.storage, ...(saved.storage ?? {}) },
+      }
+    } catch (err) {
+      if (err.code !== 'ENOENT') throw err
+      return structuredClone(DEFAULTS)
+    }
+  }
+
+  async update(patch = {}) {
+    const current = await this.read()
+    const next = structuredClone(current)
+    const image = patch.image ?? {}
+    for (const key of ['preset', 'model', 'size', 'quality']) {
+      const value = trim(image[key], 200)
+      if (value !== undefined) next.image[key] = value
+    }
+    const imageUrl = normalizeBaseUrl(image.baseUrl)
+    if (imageUrl !== undefined) next.image.baseUrl = imageUrl
+    const imageExtra = normalizeExtra(image.extra)
+    if (imageExtra !== undefined) next.image.extra = imageExtra
+    if (typeof image.apiKey === 'string' && image.apiKey.trim()) next.image.apiKey = image.apiKey.trim().slice(0, 500)
+    if (image.clearKey === true) next.image.apiKey = ''
+
+    const video = patch.video ?? {}
+    for (const key of ['preset', 'model', 'imageField']) {
+      const value = trim(video[key], 200)
+      if (value !== undefined) next.video[key] = value
+    }
+    const videoUrl = normalizeBaseUrl(video.baseUrl)
+    if (videoUrl !== undefined) next.video.baseUrl = videoUrl
+    const createPath = normalizePath(video.createPath, '/videos')
+    if (createPath !== undefined) next.video.createPath = createPath
+    const statusPath = normalizePath(video.statusPath, '/videos/{id}')
+    if (statusPath !== undefined) next.video.statusPath = statusPath
+    const contentPath = normalizePath(video.contentPath, '/videos/{id}/content')
+    if (contentPath !== undefined) next.video.contentPath = contentPath
+    const videoExtra = normalizeExtra(video.extra)
+    if (videoExtra !== undefined) next.video.extra = videoExtra
+    if (typeof video.apiKey === 'string' && video.apiKey.trim()) next.video.apiKey = video.apiKey.trim().slice(0, 500)
+    if (video.clearKey === true) next.video.apiKey = ''
+
+    const storage = patch.storage ?? {}
+    const outputDir = normalizeOutputDir(storage.outputDir)
+    if (outputDir !== undefined) {
+      if (outputDir) {
+        try {
+          await mkdir(outputDir, { recursive: true })
+          await access(outputDir, constants.W_OK)
+          const probe = join(outputDir, `.media-workbench-write-test-${process.pid}`)
+          await writeFile(probe, '')
+          await unlink(probe).catch(() => {})
+        } catch {
+          const err = new Error('产物目录不可写，换一个目录')
+          err.status = 400
+          throw err
+        }
+      }
+      next.storage.outputDir = outputDir
+    }
+
+    if (['auto', 'zh', 'en'].includes(patch.language)) next.language = patch.language
+
+    const tmp = `${this.file}.${process.pid}.tmp`
+    await writeFile(tmp, JSON.stringify(next, null, 2), { mode: 0o600 })
+    await chmod(tmp, 0o600)
+    await rename(tmp, this.file)
+    return next
+  }
+}
+
+export function imageKey(settings) {
+  return settings.image.apiKey || process.env.MEDIA_WORKBENCH_IMAGE_API_KEY || ''
+}
+
+export function videoKey(settings) {
+  return settings.video.apiKey || process.env.MEDIA_WORKBENCH_VIDEO_API_KEY || ''
+}
+
+export function imageReady(settings) {
+  return Boolean(settings.image.baseUrl && settings.image.model && imageKey(settings))
+}
+
+export function videoReady(settings) {
+  return Boolean(settings.video.baseUrl && settings.video.model && videoKey(settings))
+}
+
+export function publicSettings(settings) {
+  const image = { ...settings.image }
+  const video = { ...settings.video }
+  const imageHint = image.apiKey ? image.apiKey.slice(-4) : ''
+  const videoHint = video.apiKey ? video.apiKey.slice(-4) : ''
+  delete image.apiKey
+  delete video.apiKey
+  return {
+    ...settings,
+    image: { ...image, hasKey: Boolean(imageKey(settings)), keyHint: imageHint },
+    video: { ...video, hasKey: Boolean(videoKey(settings)), keyHint: videoHint },
+    imageReady: imageReady(settings),
+    videoReady: videoReady(settings),
+  }
+}

package/src/store.js

@@ -0,0 +1,241 @@
+// 本机持久化：$DSH_HOME/media-workbench-modern/state.json
+import { mkdir, readFile, rename, stat, writeFile } from 'node:fs/promises'
+import { homedir } from 'node:os'
+import { join } from 'node:path'
+import { templateGraph } from './graph.js'
+
+export const STATE_VERSION = 1
+
+export function dshHome(env = process.env) {
+  return env.DSH_HOME || join(homedir(), '.dsh')
+}
+
+const WELCOME = `写作不是一次成形的。
+
+先把脑子里的画面写下来，不用管顺序；等它有了轮廓，再回头删掉多余的解释。留下动作、声音、光线，和人物没有说出口的那句话。
+
+这一段可以随便改。选中文字，用右侧的「润色」或「改写」；想往下写，就点「续写」。`
+
+function seed() {
+  const now = new Date().toISOString()
+  return {
+    version: STATE_VERSION,
+    images: [],
+    videos: [],
+    docs: [
+      {
+        id: 'doc_welcome',
+        title: '欢迎来到写作工作台',
+        content: WELCOME,
+        outline: '',
+        wordGoal: 1200,
+        createdAt: now,
+        updatedAt: now,
+      },
+    ],
+    graphs: [templateGraph('inspiration')],
+    blueprints: [],
+    preferences: {
+      imageStyle: '电影感',
+      imageRatio: '16:9',
+      imageCount: 1,
+      videoMode: 'text',
+      videoRatio: '16:9',
+      videoDuration: 5,
+      writingTone: '自然叙述',
+      writingPov: '第三人称',
+      writingLength: 'medium',
+    },
+  }
+}
+
+function obj(value) {
+  return value && typeof value === 'object' && !Array.isArray(value) ? value : {}
+}
+
+function list(value, max) {
+  return Array.isArray(value) ? value.filter(row => row && obj(row) && typeof row.id === 'string').slice(0, max) : []
+}
+
+export function normalizeState(input) {
+  const state = obj(input)
+  const base = seed()
+  const preferences = { ...base.preferences, ...obj(state.preferences) }
+  const graphs = list(state.graphs, 40).map(g => ({
+    ...g,
+    name: typeof g.name === 'string' ? g.name : '未命名星图',
+    nodes: Array.isArray(g.nodes) ? g.nodes : [],
+    edges: Array.isArray(g.edges) ? g.edges : [],
+    updatedAt: g.updatedAt ?? new Date().toISOString(),
+  }))
+  return {
+    version: STATE_VERSION,
+    images: list(state.images, 300).map(normalizeImage),
+    videos: list(state.videos, 150).map(normalizeVideo),
+    docs: list(state.docs, 200).map(normalizeDoc),
+    graphs: graphs.length ? graphs : base.graphs,
+    blueprints: Array.isArray(state.blueprints) ? state.blueprints.slice(0, 60) : [],
+    preferences,
+  }
+}
+
+function normalizeImage(row) {
+  const status = ['queued', 'running', 'done', 'error'].includes(row.status) ? row.status : 'done'
+  const references = (Array.isArray(row.references) ? row.references : row.reference ? [row.reference] : [])
+    .filter(name => typeof name === 'string' && name)
+    .slice(0, 8)
+  return {
+    id: row.id,
+    prompt: String(row.prompt ?? ''),
+    negative: String(row.negative ?? ''),
+    style: String(row.style ?? ''),
+    model: String(row.model ?? ''),
+    provider: String(row.provider ?? ''),
+    size: String(row.size ?? ''),
+    reference: String(row.reference ?? references[0] ?? ''),
+    references,
+    file: String(row.file ?? ''),
+    width: Number(row.width) || 0,
+    height: Number(row.height) || 0,
+    status,
+    error: String(row.error ?? ''),
+    favorite: Boolean(row.favorite),
+    createdAt: row.createdAt ?? new Date().toISOString(),
+  }
+}
+
+function normalizeVideo(row) {
+  const status = ['queued', 'running', 'done', 'error'].includes(row.status) ? row.status : 'done'
+  return {
+    id: row.id,
+    prompt: String(row.prompt ?? ''),
+    mode: ['text', 'image', 'storyboard'].includes(row.mode) ? row.mode : 'text',
+    model: String(row.model ?? ''),
+    aspect: String(row.aspect ?? '16:9'),
+    duration: Number(row.duration) || 5,
+    reference: String(row.reference ?? ''),
+    file: String(row.file ?? ''),
+    poster: String(row.poster ?? ''),
+    frames: Array.isArray(row.frames) ? row.frames.filter(f => typeof f === 'string').slice(0, 8) : [],
+    progress: Math.max(0, Math.min(100, Number(row.progress) || 0)),
+    status,
+    error: String(row.error ?? ''),
+    createdAt: row.createdAt ?? new Date().toISOString(),
+  }
+}
+
+function normalizeDoc(row) {
+  return {
+    id: row.id,
+    title: String(row.title ?? '未命名作品'),
+    content: String(row.content ?? ''),
+    outline: String(row.outline ?? ''),
+    wordGoal: Math.max(0, Number(row.wordGoal) || 0),
+    createdAt: row.createdAt ?? new Date().toISOString(),
+    updatedAt: row.updatedAt ?? new Date().toISOString(),
+  }
+}
+
+export class Store {
+  constructor(root = join(dshHome(), 'media-workbench-modern')) {
+    this.root = root
+    this.file = join(root, 'state.json')
+    this.filesDir = join(root, 'files')
+    this.state = null
+    this.mtimeMs = 0
+    this.queue = Promise.resolve()
+  }
+
+  async useFilesDir(dir) {
+    this.filesDir = dir
+    await mkdir(dir, { recursive: true })
+  }
+
+  async refresh() {
+    await mkdir(this.filesDir, { recursive: true })
+    let info = null
+    try {
+      info = await stat(this.file)
+    } catch (err) {
+      if (err.code !== 'ENOENT') throw err
+    }
+    if (!info) {
+      if (!this.state) {
+        this.state = seed()
+        await this.persist()
+      }
+      return
+    }
+    if (this.state && info.mtimeMs === this.mtimeMs) return
+    try {
+      this.state = normalizeState(JSON.parse(await readFile(this.file, 'utf8')))
+    } catch {
+      // 文件坏了不覆盖用户数据：另存一份，再从空状态开始
+      const broken = `${this.file}.broken-${Date.now()}`
+      await rename(this.file, broken).catch(() => {})
+      this.state = seed()
+      await this.persist()
+      return
+    }
+    this.mtimeMs = info.mtimeMs
+  }
+
+  async persist() {
+    const tmp = `${this.file}.${process.pid}.tmp`
+    await writeFile(tmp, JSON.stringify(this.state, null, 2))
+    await rename(tmp, this.file)
+    this.mtimeMs = (await stat(this.file)).mtimeMs
+  }
+
+  run(task) {
+    const next = this.queue.then(task, task)
+    this.queue = next.catch(() => {})
+    return next
+  }
+
+  read() {
+    return this.run(async () => {
+      await this.refresh()
+      return this.state
+    })
+  }
+
+  mutate(fn) {
+    return this.run(async () => {
+      await this.refresh()
+      const result = await fn(this.state)
+      await this.persist()
+      return result
+    })
+  }
+
+  /** 进程重启后，把没跑完的记录标成中断，避免界面一直转圈 */
+  settle() {
+    return this.mutate(state => {
+      let changed = 0
+      for (const row of state.images) {
+        if (row.status === 'queued' || row.status === 'running') {
+          row.status = 'error'
+          row.error = '上次生成被中断，可以重试'
+          changed++
+        }
+      }
+      for (const row of state.videos) {
+        if (row.status === 'queued' || row.status === 'running') {
+          row.status = 'error'
+          row.error = '上次生成被中断，可以重试'
+          changed++
+        }
+      }
+      for (const graph of state.graphs) {
+        if (graph.run?.status === 'running' || graph.run?.status === 'queued') {
+          graph.run.status = 'error'
+          graph.run.error = '上次运行被中断'
+          for (const node of graph.nodes) if (node.status === 'running') { node.status = 'error'; node.error = '上次运行被中断' }
+          changed++
+        }
+      }
+      return changed
+    })
+  }
+}