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

package/src/repaint-sequence.js

@@ -0,0 +1,132 @@
+import path from 'node:path'
+import { isDeepStrictEqual as equal } from 'node:util'
+import { atomic, json, root, safeName } from './store.js'
+import { repaintPromptContext, reversePath, startReverse, getReverse, recoverReverse } from './repaint-reverse.js'
+import { deliveryInput, deliveryPath, startDelivery, getDelivery, recoverDelivery } from './repaint-delivery.js'
+import { repaintVideoInput, repaintVideoPath, startRepaintVideo, getRepaintVideo, recoverRepaintVideo } from './repaint-video.js'
+import { repaintAttachmentView } from './repaint-settings.js'
+import { requireServiceReady } from './service.js'
+
+const object = value => value !== null && typeof value === 'object' && !Array.isArray(value)
+const paths = { reverse: reversePath, delivery: deliveryPath, video: repaintVideoPath }
+const getters = { reverse: getReverse, delivery: getDelivery, video: getRepaintVideo }
+const recoverers = { reverse: recoverReverse, delivery: recoverDelivery, video: recoverRepaintVideo }
+export const repaintSequencePath = (profile, operation) =>
+  path.join(root(), 'repaint-sequence', safeName(profile), safeName(operation) + '.json')
+
+export function repaintSequenceInput(value) {
+  if (!object(value) || Object.keys(value).some(k => k !== 'steps') || !Array.isArray(value.steps) ||
+      !value.steps.length || value.steps.length > 100)
+    throw new Error('多段调度输入仅支持steps数组（1–100项），不是批量接口或合成任务')
+  const names = new Set(), actions = new Set()
+  return { steps: value.steps.map(step => {
+    if (!object(step) || Object.keys(step).some(k => !['type', 'segment_index', 'operation', 'input'].includes(k)) ||
+        !Object.hasOwn(paths, step.type) || !Number.isSafeInteger(step.segment_index) || step.segment_index < 1 ||
+        typeof step.operation !== 'string')
+      throw new Error('调度步骤需要type reverse/delivery/video、正整数segment_index及operation')
+    safeName(step.operation)
+    const action = step.type + ':' + step.segment_index
+    if (names.has(step.operation) || actions.has(action)) throw new Error('调度中operation或同片段同阶段重复')
+    names.add(step.operation); actions.add(action)
+    if (step.type === 'reverse' && Object.hasOwn(step, 'input')) throw new Error('反推步骤不接受额外input')
+    if (step.type !== 'reverse' && !Object.hasOwn(step, 'input'))
+      throw new Error('最终提示词须明确替换规则或空对象；视频须明确模型参数，不自动选择')
+    return { type: step.type, segment_index: step.segment_index, operation: step.operation,
+      ...(step.type === 'delivery' ? { input: deliveryInput(step.input) } : {}),
+      ...(step.type === 'video' ? { input: repaintVideoInput(step.input) } : {}) }
+  }) }
+}
+
+async function context(client, profile, parent, input) {
+  await requireServiceReady(client)
+  const bindings = [], indexes = [...new Set(input.steps.map(s => s.segment_index))]
+  let owner, attachments
+  for (const index of indexes) {
+    const ctx = await repaintPromptContext(client, profile, parent, index)
+    owner = ctx.parent.owner
+    const current = repaintAttachmentView(ctx.job.delivery_prompt_attachments, client.baseUrl)
+    if (attachments && !equal(attachments, current)) throw new Error('核对期间全局附件发生变化')
+    attachments = current
+    bindings.push(ctx.binding)
+  }
+  return { owner, bindings, attachments }
+}
+function summary(record, steps, next = null) {
+  return { operation: record.operation, repaint_operation: record.parentOperation,
+    workflow: 'canvas_single_step_sequence', product_batch_supported: false,
+    state: next ? 'pending' : 'completed', steps, next_step: next,
+    total_estimated_credit_cost: null,
+    estimate_note: '后续阶段依赖前序结果，逐项实时预估；当前报价不授权下一项',
+    generation_submitted: false, processing_submitted: false, canvas_written: false }
+}
+export async function prepareRepaintSequence(client, profile, parent, operation, value) {
+  safeName(parent); safeName(operation)
+  const input = repaintSequenceInput(value)
+  const binding = await context(client, profile, parent, input)
+  const file = repaintSequencePath(profile, operation), previous = await json(file, null)
+  const record = { schema: 1, profile, operation, parentOperation: parent,
+    baseUrl: client.baseUrl, input, ...binding }
+  if (previous && !equal(previous, record)) throw new Error('调度operation已绑定其他输入或配置；先查询原调度')
+  if (!previous) await atomic(file, record)
+  return inspectRepaintSequence(client, profile, operation)
+}
+async function bound(client, profile, operation) {
+  const record = await json(repaintSequencePath(profile, operation), null)
+  if (!record || record.schema !== 1 || record.operation !== operation || record.profile !== profile || record.baseUrl !== client.baseUrl)
+    throw new Error('调度operation不存在或服务不匹配')
+  const binding = await context(client, profile, record.parentOperation, repaintSequenceInput(record.input))
+  if (!equal(binding, { owner: record.owner, bindings: record.bindings, attachments: record.attachments }))
+    throw new Error('调度账号、方案、来源、节点或全局配置变化；停止原调度，不重做')
+  return record
+}
+async function inspect(client, profile, record) {
+  const steps = []
+  let next = null
+  for (const step of record.input.steps) {
+    const child = await json(paths[step.type](profile, step.operation), null)
+    if (child && (child.parentOperation !== record.parentOperation || child.index !== step.segment_index ||
+        child.owner !== record.owner || child.baseUrl !== record.baseUrl ||
+        (step.type !== 'reverse' && !equal(child.input, step.input))))
+      throw new Error('子operation已绑定其他任务/参数；不认领或换键重提')
+    const observed = child ? await getters[step.type](client, profile, step.operation) : null
+    const completed = observed?.state === 'completed' && observed.canvas_synced === true && !observed.fallback
+    const status = completed ? 'completed' : child?.attempted ? observed?.state || 'submission_unknown' : 'not_submitted'
+    const item = { ...step, state: status, ...(observed ? { result: observed } : {}) }
+    steps.push(item)
+    if (!completed && !next) next = item
+  }
+  return summary(record, steps, next)
+}
+export async function inspectRepaintSequence(client, profile, operation) {
+  const record = await bound(client, profile, operation)
+  return inspect(client, profile, record)
+}
+export async function nextRepaintSequence(client, profile, operation, { confirmed = false, expectedCost, step: expectedStep, recover = false, ...options } = {}) {
+  if (confirmed && recover) throw new Error('恢复与收费提交须分开执行')
+  const record = await bound(client, profile, operation)
+  const view = await inspect(client, profile, record), step = view.next_step
+  if ((confirmed || recover) && (!step || expectedStep !== step.operation))
+    throw new Error('当次确认的子operation不再是下一项；重新查看，不将旧确认用于后续任务')
+  if (!step) return view
+  const child = await json(paths[step.type](profile, step.operation), null)
+  if (recover) {
+    if (!child) throw new Error('原子operation尚未准备，不能恢复')
+    const result = await recoverers[step.type](client, profile, step.operation)
+    return { ...view, executed_step: step.operation, result, canvas_written: null, canvas_synced: result.canvas_synced === true,
+      next: '本次仅恢复原子操作；再次sequence-get查看下一项，不自动提交' }
+  }
+  if (child?.attempted) {
+    if (confirmed) throw new Error('原子操作已提交或结果未知；只查询/恢复，不重新提交')
+    return { ...view, next: '原子操作已提交或未知；sequence-recover --step 原子operation仅恢复原操作，失败不自动重做' }
+  }
+  const args = [client, profile, record.parentOperation, step.segment_index, step.operation]
+  let result
+  if (step.type === 'reverse') result = await startReverse(...args, confirmed, expectedCost)
+  else if (step.type === 'delivery') result = await startDelivery(...args, step.input, confirmed, expectedCost)
+  else result = await startRepaintVideo(...args, step.input, confirmed, expectedCost, options)
+  return { ...view, executed_step: confirmed ? step.operation : null, prepared_step: step.operation, result,
+    generation_submitted: confirmed && step.type === 'video' ? result.submitted : false,
+    processing_submitted: confirmed && step.type !== 'video' ? result.processing_submitted : false,
+    canvas_written: confirmed ? null : false, canvas_synced: result.canvas_synced === true,
+    next: '本次仅准备或执行这一项；展示result中的参数/钱包/实时费用并取得该项确认，后续另行准备与确认' }
+}