@uzooai/anime-ai-cli 0.0.0-stage → 0.6.0

package/src/repaint-settings.js

@@ -0,0 +1,576 @@
+import fs from 'node:fs/promises'
+import path from 'node:path'
+import { isDeepStrictEqual } from 'node:util'
+import { atomic, json, root, safeName } from './store.js'
+import { repaintGet, recordPath } from './repaint.js'
+import { repaintPromptContext } from './repaint-reverse.js'
+import { repaintOf, jobPath } from './repaint-graph.js'
+import { requireServiceReady } from './service.js'
+
+const text = (value) => String(value ?? '')
+const plain = (value) =>
+  text(value)
+    .replace(/<[^>]*>/g, '')
+    .trim()
+const object = (v) => v && typeof v === 'object' && !Array.isArray(v)
+const fail = (code, message, cause) =>
+  Object.assign(new Error(message), {
+    code,
+    ...(cause
+      ? {
+          cause,
+          diagnostic: {
+            ...(cause.status ? { http_status: cause.status } : {}),
+            ...(cause.apiCode ? { api_code: cause.apiCode } : {}),
+            ...(cause.requestId ? { request_id: cause.requestId } : {})
+          }
+        }
+      : {})
+  })
+const api = (client, endpoint, options = {}, controls) =>
+  client.raw(
+    endpoint,
+    {
+      ...options,
+      headers: { ...options.headers, 'X-Client-Source': 'canvas' }
+    },
+    controls
+  )
+export const repaintSettingsPath = (profile, operation) =>
+  path.join(
+    root(),
+    'repaint-settings',
+    safeName(profile),
+    safeName(operation) + '.json'
+  )
+const persist = (r) => atomic(repaintSettingsPath(r.profile, r.operation), r)
+
+export function repaintSettingsInput(value) {
+  if (
+    !object(value) ||
+    Object.keys(value).some((k) => k !== 'custom_prompt') ||
+    typeof value.custom_prompt !== 'string'
+  )
+    throw new Error(
+      '全局规则输入仅接受custom_prompt字符串；附件新增/删除/替换用独立待接入流程'
+    )
+  if (
+    value.custom_prompt.length > 20000 ||
+    /<[^>]*>|\u0000/.test(value.custom_prompt)
+  )
+    throw new Error(
+      '全局规则须为不含HTML/空字符的纯文本，最长20000字符；空字符串表示明确清空'
+    )
+  return { custom_prompt: value.custom_prompt.trim() }
+}
+const fields = (n) =>
+  Object.fromEntries(
+    ['input', 'prompt', 'output'].map((k) => [k, text(n.data?.[k])])
+  )
+const nodeText = (n) => plain(n.data?.output || n.data?.prompt || n.data?.input)
+export function repaintAttachmentView(list, base) {
+  if (list == null) return []
+  if (
+    !Array.isArray(list) ||
+    list.length > 10 ||
+    list.some(
+      (a) =>
+        !object(a) ||
+        typeof a.name !== 'string' ||
+        !a.name ||
+        a.name.length > 255 ||
+        typeof a.media_type !== 'string' ||
+        !/^(?:image\/[^\s]+|application\/pdf)$/i.test(a.media_type.trim()) ||
+        typeof a.url !== 'string' ||
+        !a.url.trim()
+    )
+  )
+    throw new Error('已有全局附件结构无效；不清空或静默裁减')
+  return list.map((a) => {
+    const url = new URL(a.url.trim(), new URL(base).origin)
+    if (
+      !['http:', 'https:', 'data:'].includes(url.protocol) ||
+      url.username ||
+      url.password
+    )
+      throw new Error('已有附件URL不合规，停止')
+    return { name: a.name, media_type: a.media_type.trim(), url: url.href }
+  })
+}
+const attachmentView = repaintAttachmentView
+export async function repaintSettingsContext(client, profile, parentOperation) {
+  const parent = await json(recordPath(profile, parentOperation), null)
+  if (!parent) throw new Error('未找到原转绘operation')
+  const initial = await repaintGet(client, profile, parentOperation)
+  const index = initial.job.plan?.segments?.[0]?.index
+  const ctx = await repaintPromptContext(
+    client,
+    profile,
+    parentOperation,
+    index,
+    {
+      allowUnsyncedGlobal: true
+    }
+  )
+  const marker = ctx.doc.meta.cli_repaint_workflows[parentOperation]
+  const finals = ctx.doc.nodes.filter(
+    (n) =>
+      marker.node_ids.includes(n.id) && repaintOf(n).role === 'final-prompt'
+  )
+  if (
+    ctx.global.type !== 'text' ||
+    finals.length !== ctx.job.plan.segments.length ||
+    finals.some(
+      (n) =>
+        !ctx.doc.edges.some(
+          (e) =>
+            e.source === ctx.global.id &&
+            e.target === n.id &&
+            e.data?.role === 'global-prompt-link'
+        )
+    )
+  )
+    throw new Error('全局节点/各段系统线不完整，不编辑或重建')
+  const stored =
+    ctx.global.data?.repaint || ctx.global.data?.extra?.repaint || {}
+  const nodeAttachments = attachmentView(stored.attachments, client.baseUrl)
+  const serverAttachments = attachmentView(
+    ctx.job.delivery_prompt_attachments,
+    client.baseUrl
+  )
+  const binding = {
+    job_id: ctx.job.id,
+    plan_version: ctx.job.plan_version,
+    group_id: ctx.binding.group_id,
+    node_id: ctx.global.id,
+    segment_indices: ctx.job.plan.segments.map((s) => s.index),
+    node_ids: [...marker.node_ids].sort()
+  }
+  return {
+    ...ctx,
+    binding,
+    nodeAttachments,
+    serverAttachments,
+    nodeValue: nodeText(ctx.global),
+    serverValue: text(ctx.job.delivery_prompt_custom_prompt).trim()
+  }
+}
+const context = repaintSettingsContext
+export async function requireRepaintSettingsIdle(client, ctx, profile) {
+  if (
+    ctx.job.outputs?.some(
+      (o) =>
+        o.reverse_prompt_status === 'running' ||
+        o.delivery_prompt_status === 'running'
+    )
+  )
+    throw new Error('原任务仍有提示词正在处理，待完成后再修改全局规则')
+  const tasks = await api(client, jobPath(ctx.job.id) + '/redraw-tasks')
+  if (
+    !Array.isArray(tasks.items) ||
+    tasks.items.some((t) =>
+      ['queued', 'processing', 'retrying'].includes(t.status)
+    )
+  )
+    throw new Error('原任务有进行中视频或任务列表不完整，停止修改全局规则')
+  for (const folder of [
+    'repaint-reverse',
+    'repaint-delivery',
+    'repaint-video'
+  ]) {
+    const dir = path.join(root(), folder, safeName(profile))
+    const files = await fs.readdir(dir).catch((e) => {
+      if (e.code === 'ENOENT') return []
+      throw e
+    })
+    for (const name of files.filter((n) => n.endsWith('.json'))) {
+      const r = await json(path.join(dir, name))
+      if (
+        r.baseUrl === client.baseUrl &&
+        r.owner === ctx.parent.owner &&
+        r.binding?.job_id === ctx.job.id &&
+        (r.attempted || r.settingsAttempted) &&
+        !['completed', 'completed_fallback', 'failed'].includes(r.state)
+      )
+        throw new Error(
+          '该job存在未结束或未知提示词/视频operation，先查询原操作，不改变其全局输入'
+        )
+    }
+  }
+}
+const idle = requireRepaintSettingsIdle
+async function preventDuplicate(client, profile, operation, ctx) {
+  const dir = path.dirname(repaintSettingsPath(profile, operation))
+  const files = await fs.readdir(dir).catch((e) => {
+    if (e.code === 'ENOENT') return []
+    throw e
+  })
+  for (const name of files.filter((n) => n.endsWith('.json'))) {
+    const r = await json(path.join(dir, name))
+    if (
+      r.operation !== operation &&
+      r.baseUrl === client.baseUrl &&
+      r.owner === ctx.parent.owner &&
+      r.binding?.job_id === ctx.job.id &&
+      (r.serverAttempted || r.canvasAttempted) &&
+      r.state !== 'completed'
+    )
+      throw new Error(
+        '同job已有未确认的全局规则保存；先恢复原operation，不能换名覆盖'
+      )
+  }
+}
+function check(record, ctx) {
+  if (
+    !isDeepStrictEqual(record.binding, ctx.binding) ||
+    record.owner !== ctx.parent.owner
+  )
+    throw new Error('原全局规则job/方案/节点关联已变化，停止')
+  if (
+    ![record.attachments, record.targetAttachments].some(
+      (a) => a && isDeepStrictEqual(a, ctx.nodeAttachments)
+    ) ||
+    ![record.attachments, record.targetAttachments].some(
+      (a) => a && isDeepStrictEqual(a, ctx.serverAttachments)
+    )
+  )
+    throw new Error('全局附件已变化，停止原文本保存；不清空或覆盖附件')
+  const current = fields(ctx.global)
+  const sameBefore = Object.entries(record.nodeBefore).every(
+    ([k, v]) => current[k] === v
+  )
+  const sameTarget = Object.values(current).every(
+    (v) => v === record.input.custom_prompt
+  )
+  // Web codec may drop the redundant input field; output/prompt still bind the edit.
+  const webTarget =
+    current.output === record.input.custom_prompt &&
+    current.prompt === record.input.custom_prompt &&
+    !current.input
+  if (!(sameBefore || sameTarget || webTarget))
+    throw fail(
+      'REPAINT_SETTINGS_SYNC_PENDING',
+      '全局节点已被用户编辑，不覆盖；原保存记录保留'
+    )
+  if (
+    ![record.serverBefore, record.input.custom_prompt].includes(ctx.serverValue)
+  )
+    throw fail(
+      'REPAINT_SETTINGS_SYNC_PENDING',
+      '服务端全局规则已被其他编辑修改，不覆盖'
+    )
+  return {
+    nodeTarget:
+      ctx.nodeValue === record.input.custom_prompt &&
+      isDeepStrictEqual(
+        ctx.nodeAttachments,
+        record.targetAttachments || record.attachments
+      ),
+    serverTarget:
+      ctx.serverValue === record.input.custom_prompt &&
+      isDeepStrictEqual(
+        ctx.serverAttachments,
+        record.targetAttachments || record.attachments
+      )
+  }
+}
+async function bound(client, profile, operation) {
+  const record = await json(repaintSettingsPath(profile, operation), null)
+  if (
+    !record ||
+    record.schema !== 1 ||
+    record.profile !== profile ||
+    record.baseUrl !== client.baseUrl
+  )
+    throw new Error('全局规则operation不存在或服务不匹配')
+  const ctx = await context(client, profile, record.parentOperation)
+  check(record, ctx)
+  return { record, ctx }
+}
+function summary(record, ctx) {
+  return {
+    operation: record.operation,
+    repaint_operation: record.parentOperation,
+    state: record.state,
+    job_id: record.binding.job_id,
+    plan_version: record.binding.plan_version,
+    node_id: record.binding.node_id,
+    custom_prompt: record.input.custom_prompt,
+    server_value: ctx.serverValue,
+    canvas_value: ctx.nodeValue,
+    server_synced: ctx.serverValue === record.input.custom_prompt,
+    canvas_synced: ctx.nodeValue === record.input.custom_prompt,
+    attachment_count: (record.targetAttachments || record.attachments).length,
+    attachments_preserved: true,
+    generation_submitted: false,
+    processing_submitted: false,
+    existing_results_regenerated: false,
+    settings_save_billable: false
+  }
+}
+export async function getRepaintSettings(client, profile, parentOperation) {
+  const ctx = await context(client, profile, parentOperation)
+  return {
+    repaint_operation: parentOperation,
+    job_id: ctx.job.id,
+    plan_version: ctx.job.plan_version,
+    node_id: ctx.global.id,
+    server_value: ctx.serverValue,
+    canvas_value: ctx.nodeValue,
+    values_match: ctx.serverValue === ctx.nodeValue,
+    attachments_match: isDeepStrictEqual(
+      ctx.nodeAttachments,
+      ctx.serverAttachments
+    ),
+    attachment_count: ctx.serverAttachments.length,
+    settings_save_billable: false,
+    generation_submitted: false,
+    processing_submitted: false
+  }
+}
+export async function startRepaintSettings(
+  client,
+  profile,
+  parentOperation,
+  operation,
+  value,
+  confirmed = false
+) {
+  return startSettings(
+    client,
+    profile,
+    parentOperation,
+    operation,
+    value,
+    confirmed
+  )
+}
+// Only a registered upload receipt reaches this adapter; CLI does not accept
+// arbitrary attachment URLs or silently replace the existing ordered list.
+export async function appendRepaintAttachment(
+  client,
+  profile,
+  parentOperation,
+  operation,
+  receipt
+) {
+  const existing = await json(repaintSettingsPath(profile, operation), null)
+  if (existing) {
+    if (
+      existing.parentOperation !== parentOperation ||
+      !isDeepStrictEqual(existing.appendReceipt, receipt)
+    )
+      throw new Error('附件设置operation已绑定其他回执')
+    return recoverRepaintSettings(client, profile, operation)
+  }
+  const ctx = await context(client, profile, parentOperation)
+  if (
+    !receipt?.file_ref ||
+    !receipt?.url ||
+    !receipt?.name ||
+    !receipt?.mime_type
+  )
+    throw new Error('缺少真实附件注册回执')
+  const target = attachmentView(
+    [
+      ...(ctx.job.delivery_prompt_attachments || []),
+      { name: receipt.name, media_type: receipt.mime_type, url: receipt.url }
+    ],
+    client.baseUrl
+  )
+  return startSettings(
+    client,
+    profile,
+    parentOperation,
+    operation,
+    { custom_prompt: ctx.serverValue },
+    true,
+    { target, receipt }
+  )
+}
+async function startSettings(
+  client,
+  profile,
+  parentOperation,
+  operation,
+  value,
+  confirmed = false,
+  append
+) {
+  safeName(operation)
+  const input = repaintSettingsInput(value)
+  await requireServiceReady(client)
+  let ctx = await context(client, profile, parentOperation),
+    record = await json(repaintSettingsPath(profile, operation), null)
+  await preventDuplicate(client, profile, operation, ctx)
+  if (record) {
+    if (
+      record.parentOperation !== parentOperation ||
+      !isDeepStrictEqual(record.input, input)
+    )
+      throw new Error('operation已绑定其他全局文本，不修改原记录')
+    ;({ record, ctx } = await bound(client, profile, operation))
+    if (record.serverAttempted || record.canvasAttempted)
+      return confirmed
+        ? recoverRepaintSettings(client, profile, operation)
+        : summary(record, ctx)
+  } else {
+    if (
+      ctx.nodeValue !== ctx.serverValue ||
+      !isDeepStrictEqual(ctx.nodeAttachments, ctx.serverAttachments)
+    )
+      throw new Error(
+        '画布与后端已有不同全局规则/附件，先明确原编辑来源；不自动选边覆盖'
+      )
+    record = {
+      schema: 1,
+      profile,
+      operation,
+      parentOperation,
+      input,
+      owner: ctx.parent.owner,
+      baseUrl: client.baseUrl,
+      binding: ctx.binding,
+      nodeBefore: fields(ctx.global),
+      serverBefore: ctx.serverValue,
+      attachments: ctx.serverAttachments,
+      bodyAttachments: structuredClone(
+        ctx.job.delivery_prompt_attachments || []
+      ),
+      ...(append
+        ? { targetAttachments: append.target, appendReceipt: append.receipt }
+        : {}),
+      state: 'prepared',
+      createdAt: new Date().toISOString()
+    }
+  }
+  if (!confirmed) {
+    await persist(record)
+    return {
+      ...summary(record, ctx),
+      next: '展示原值与新全局要求；已授权的同次免费编辑无需重复费用确认，加--yes保存。保存不重处理任何已有结果。'
+    }
+  }
+  await idle(client, ctx, profile)
+  ctx = await context(client, profile, parentOperation)
+  check(record, ctx)
+  if (!check(record, ctx).serverTarget) {
+    record.serverAttempted = true
+    record.state = 'saving_server'
+    await persist(record)
+    try {
+      await api(
+        client,
+        jobPath(ctx.job.id) + '/delivery-prompt-settings',
+        {
+          method: 'POST',
+          body: {
+            custom_prompt: input.custom_prompt,
+            attachments: record.targetAttachments || record.bodyAttachments
+          }
+        },
+        { retry: false }
+      )
+    } catch (error) {
+      record.state = 'server_unknown'
+      await persist(record)
+      throw fail(
+        'REPAINT_SETTINGS_SAVE_UNKNOWN',
+        '全局规则服务端保存响应未确认；仅查询原operation，不重发设置或生成',
+        error
+      )
+    }
+    record.serverConfirmed = true
+    await persist(record)
+  }
+  return recoverRepaintSettings(client, profile, operation)
+}
+export async function repaintSettingsOperation(client, profile, operation) {
+  const { record, ctx } = await bound(client, profile, operation)
+  return summary(record, ctx)
+}
+export async function recoverRepaintSettings(client, profile, operation) {
+  await requireServiceReady(client)
+  const { record, ctx } = await bound(client, profile, operation)
+  const status = check(record, ctx)
+  if (
+    !record.serverAttempted &&
+    !record.canvasAttempted &&
+    !record.serverConfirmed &&
+    !status.serverTarget
+  )
+    return summary(record, ctx)
+  if (!status.serverTarget)
+    throw fail(
+      'REPAINT_SETTINGS_SAVE_UNKNOWN',
+      '原设置POST尚未查到目标文本；保留未知状态，不自动重发或覆盖后端'
+    )
+  record.serverConfirmed = true
+  await persist(record)
+  await idle(client, ctx, profile)
+  if (!status.nodeTarget) {
+    record.canvasAttempted = true
+    record.state = 'saving_canvas'
+    await persist(record)
+    ctx.global.data = {
+      ...ctx.global.data,
+      ...Object.fromEntries(
+        ['input', 'prompt', 'output'].map((k) => [
+          k,
+          record.input.custom_prompt
+        ])
+      )
+    }
+    if (record.targetAttachments) {
+      const d = ctx.global.data
+      if (d.repaint)
+        d.repaint = { ...d.repaint, attachments: record.targetAttachments }
+      d.extra = {
+        ...d.extra,
+        repaint: { ...d.extra?.repaint, attachments: record.targetAttachments }
+      }
+    }
+    try {
+      const saved = await api(
+        client,
+        '/canvas/docs/' + encodeURIComponent(ctx.doc.id),
+        {
+          method: 'PUT',
+          body: {
+            version: ctx.doc.version,
+            name: ctx.doc.name,
+            viewport: ctx.doc.viewport,
+            nodes: ctx.doc.nodes,
+            edges: ctx.doc.edges,
+            meta: ctx.doc.meta
+          }
+        },
+        { retry: false }
+      )
+      if (
+        saved.version !== ctx.doc.version + 1 ||
+        nodeText(saved.nodes?.find((n) => n.id === ctx.global.id) || {}) !==
+          record.input.custom_prompt
+      )
+        throw new Error('保存响应缺少对应全局节点或版本')
+    } catch (error) {
+      record.state = 'sync_pending'
+      await persist(record)
+      throw fail(
+        'REPAINT_SETTINGS_SYNC_PENDING',
+        '后端全局规则已保存，画布同步未确认；恢复只同步原文本，不重发设置或处理任务',
+        error
+      )
+    }
+  }
+  const fresh = await context(client, profile, record.parentOperation)
+  const final = check(record, fresh)
+  if (!final.nodeTarget || !final.serverTarget)
+    throw fail(
+      'REPAINT_SETTINGS_SYNC_PENDING',
+      '全局文本最终读回不一致，停止；不反复覆盖'
+    )
+  record.state = 'completed'
+  await persist(record)
+  return summary(record, fresh)
+}