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

package/src/draft-finalize.js

@@ -0,0 +1,229 @@
+import crypto from 'node:crypto'
+import path from 'node:path'
+import { isDeepStrictEqual as same } from 'node:util'
+import { atomic, json, root, safeName } from './store.js'
+import { scopedCanvas, materialScope } from './materials.js'
+import { save } from './graph.js'
+import { segment, wait } from './workflows.js'
+import { readProtocol, agreementError } from './agreement.js'
+import { requireServiceReady } from './service.js'
+
+const hash = value => crypto.createHash('sha256').update(JSON.stringify(value)).digest('hex')
+const fail = (code, message, cause) => Object.assign(new Error(message, { cause }), { code })
+export const draftFinalizePath = (profile, operation) => path.join(root(), 'draft-finalize', safeName(profile), safeName(operation) + '.json')
+const persist = record => atomic(draftFinalizePath(record.profile, record.operation), record)
+function inputOf(value) {
+  if (!value || Array.isArray(value) || typeof value !== 'object' ||
+      Object.keys(value).some(k => !['doc_id', 'node_id', 'space', 'team_uid'].includes(k)) ||
+      !['doc_id', 'node_id'].every(k => typeof value[k] === 'string' && value[k].trim()))
+    throw new Error('样片转正片输入仅支持 doc_id/node_id/space/team_uid；参数继承原样片')
+  materialScope(value.space, value.team_uid)
+  return structuredClone(value)
+}
+async function owner(client) {
+  const me = await client.raw('/auth/me')
+  if (!me.uid) throw new Error('账号缺少 UID')
+  return String(me.uid)
+}
+function snapshot(doc) {
+  if (!Number.isSafeInteger(doc.version) || doc.version < 1 || !Array.isArray(doc.nodes) ||
+      !Array.isArray(doc.edges) || !doc.meta || !doc.viewport || typeof doc.name !== 'string' ||
+      new Set(doc.nodes.map(n => n.id)).size !== doc.nodes.length || new Set(doc.edges.map(e => e.id)).size !== doc.edges.length)
+    throw new Error('画布快照不完整或 ID 重复')
+}
+const taskId = node => Number(node?.data?.last_task_id || node?.data?.lastTaskId || node?.data?.runtime?.taskId)
+function parentIdentity(task, history) {
+  // Task GET omits prompt/references/duration on the deployed service. Read the
+  // original history row instead of showing the node's possibly edited inputs.
+  return JSON.parse(JSON.stringify({ task_id: task.task_id ?? task.id, doc_id: task.doc_id, node_id: task.node_id,
+    model_name: task.model_name, prompt: history.prompt, duration: history.duration,
+    aspect_ratio: history.aspect_ratio || history.extra?.aspect_ratio || task.result?.aspect_ratio,
+    source_resolution: history.resolution || history.extra?.resolution, final_resolution: '1080p',
+    generation_mode: history.generation_mode || history.extra?.generation_mode,
+    reference_image_urls: history.reference_image_urls || [],
+    reference_video_urls: history.reference_video_urls || history.extra?.reference_video_urls || [],
+    reference_audio_urls: history.reference_audio_urls || history.extra?.reference_audio_urls || [],
+    reference_asset_ids: history.reference_asset_ids || [],
+    draft: { step: task.draft.step, expires_at: task.draft.expires_at }, file_ref: task.result.file_ref }))
+}
+async function previewHistory(client, doc, node, id) {
+  for (let offset = 0; offset < 10000; offset += 100) {
+    const result = await client.raw('/canvas/docs/' + segment(doc.id) + '/nodes/' + segment(node.id) +
+      '/history?' + new URLSearchParams({ status: 'completed', limit: 100, offset }))
+    if (result.doc_id !== doc.id || result.node_id !== node.id || !Array.isArray(result.items) || typeof result.has_more !== 'boolean')
+      throw new Error('样片原历史响应绑定不完整')
+    const matches = result.items.filter(t => Number(t.task_id) === id && !t.source)
+    if (matches.length > 1) throw new Error('样片原历史任务不唯一')
+    if (matches.length) {
+      const item = matches[0]
+      if (item.doc_id !== doc.id || item.node_id !== node.id || item.segment_type !== 'video' || item.status !== 'completed' ||
+          item.draft?.step !== 'preview' || typeof item.prompt !== 'string' || !Number.isFinite(item.duration) || item.duration <= 0)
+        throw new Error('样片原历史缺少可核验的继承参数')
+      return item
+    }
+    if (!result.has_more) break
+  }
+  throw new Error('未找到样片原历史，不用当前节点提示词冒充原任务输入')
+}
+async function context(client, input, expectedParent) {
+  const doc = await scopedCanvas(client, input.doc_id, input.space, input.team_uid)
+  snapshot(doc)
+  const node = doc.nodes.find(n => n.id === input.node_id), data = node?.data || {}, extra = data.extra || {}
+  const id = taskId(node)
+  if (!node || node.type !== 'video' || extra.node_kind === 'asset' || extra.repaint || extra.video_remake ||
+      extra.clip_redraw || extra.draft_finalize || data.repaint || data.videoRemake || data.draftFinalizeGeneration ||
+      !Number.isSafeInteger(id) || id < 1 || ['preparing','queued','processing','retrying','submitting'].includes(data.runtime?.status || data.status))
+    throw new Error('来源须为已有已完成普通视频样片节点')
+  if (expectedParent && id !== Number(expectedParent.task_id)) throw new Error('样片节点已切换任务，停止原操作')
+  const task = await client.raw('/canvas/generations/' + segment(id))
+  if (Number(task.task_id ?? task.id) !== id || task.doc_id !== doc.id || task.node_id !== node.id ||
+      task.segment_type !== 'video' || task.status !== 'completed' || task.billing_status !== 'charged' ||
+      task.draft?.step !== 'preview' || !['preview_ready','finalizing','consumed'].includes(task.draft.status) ||
+      !task.result?.file_ref || task.result.file_ref !== data.result?.file_ref)
+    throw new Error('原任务不是已结算且可用的真实样片；不凭模型名称认定样片')
+  if (task.draft.expires_at && (!Number.isFinite(Date.parse(task.draft.expires_at)) || Date.parse(task.draft.expires_at) <= Date.now()))
+    throw new Error('样片已过期或有效期不可确认')
+  const history = await previewHistory(client, doc, node, id)
+  const parent = parentIdentity(task, history)
+  // The estimate endpoint is authoritative for entitlement, expiry and inherited price.
+  const estimate = await client.raw('/canvas/generations/estimate', { method: 'POST', body: { segment_type: 'video', draft_task_id: id } }, { retry: false })
+  if (!Number.isFinite(estimate.estimated_credit_cost) || estimate.estimated_credit_cost < 0 || estimate.will_intercept)
+    throw new Error('正片报价无效或被拦截，未提交')
+  return { doc, node, task, parent, estimate }
+}
+const summary = r => ({ operation: r.operation, state: r.state, doc_id: r.input.doc_id,
+  parent_node_id: r.input.node_id, parent_task_id: r.parent.task_id, node_id: r.output.id,
+  task_id: r.taskId || null, estimate: r.estimate, wallet: r.wallet,
+  inherited: r.parent, generation_submitted: r.taskId ? true : ['submitting','unknown'].includes(r.state) ? null : false })
+async function bound(client, profile, operation) {
+  const r = await json(draftFinalizePath(profile, operation), null)
+  if (!r || r.baseUrl !== client.baseUrl || r.owner !== await owner(client)) throw new Error('样片 operation 的服务或账号不匹配')
+  if (r.hash !== hash({ input: r.input, parent: r.parent, output: r.output, edge: r.edge })) throw new Error('样片 operation 摘要不匹配')
+  return r
+}
+export async function prepareDraftFinalize(client, profile, value, operation) {
+  const input = inputOf(value), file = draftFinalizePath(profile, operation)
+  if (await json(file, null)) {
+    const r = await bound(client, profile, operation)
+    if (!same(r.input, input)) throw new Error('operation 已绑定其他输入')
+    return summary(r)
+  }
+  await requireServiceReady(client)
+  const uid = await owner(client), { doc, node, task, parent, estimate } = await context(client, input)
+  const output = { id: 'draft-final-' + crypto.randomUUID(), type: 'video',
+    position: { x: Math.round(Number(node.position?.x || 0) + Number(node.width || 0) + 80), y: Math.round(Number(node.position?.y || 0)) },
+    data: { title: '正式视频', mediaType: 'video', status: 'preparing', last_task_id: null,
+      extra: { node_kind: 'asset', aspect_ratio: task.result.aspect_ratio || '16:9',
+        draft_finalize: { role: 'generation-result', parent_node_id: node.id, parent_task_id: Number(task.task_id ?? task.id) } } } }
+  const edge = { id: 'draft-edge-' + crypto.randomUUID(), source: node.id, target: output.id, data: { draftFinalize: true } }
+  const wallet = { space: input.space, team_uid: doc.team_uid || null, source: doc.team_uid ? 'team' : 'personal' }
+  const r = { profile, operation, baseUrl: client.baseUrl, owner: uid, input, parent, output, edge,
+    hash: hash({ input, parent, output, edge }), estimate, wallet, key: crypto.randomUUID(), state: 'prepared', createdAt: new Date().toISOString() }
+  await persist(r)
+  return { ...summary(r), canvas_written: false }
+}
+function verifyOutput(doc, r) {
+  const node = doc.nodes.find(n => n.id === r.output.id)
+  if (!node || node.type !== 'video' || node.data?.extra?.node_kind !== 'asset' ||
+      !same(node.data.extra.draft_finalize, r.output.data.extra.draft_finalize) ||
+      !doc.edges.some(e => e.id === r.edge.id && e.source === r.edge.source && e.target === r.edge.target && e.data?.draftFinalize === true))
+    throw fail('DRAFT_SYNC_PENDING', '派生节点或来源线已变化；不重建、不提交、不覆盖')
+  return node
+}
+function verifyTask(task, r) {
+  if (!Number.isSafeInteger(Number(task.task_id ?? task.id)) || Number(task.task_id ?? task.id) < 1 ||
+      (r.taskId && Number(task.task_id ?? task.id) !== Number(r.taskId)) ||
+      task.doc_id !== r.input.doc_id || task.node_id !== r.output.id || task.segment_type !== 'video' ||
+      task.draft?.step !== 'finalize' || Number(task.draft.parent_task_id) !== Number(r.parent.task_id))
+    throw fail('DRAFT_SUBMISSION_UNKNOWN', '正片任务绑定未确认；保留原 operation，不换键重提')
+}
+export async function submitDraftFinalize(client, profile, operation, confirmed, expectedCost) {
+  const r = await bound(client, profile, operation)
+  if (r.taskId || ['submitting','unknown','saving','save_unknown'].includes(r.state)) return getDraftFinalize(client, profile, operation)
+  await requireServiceReady(client)
+  const { doc, parent, estimate } = await context(client, r.input, r.parent)
+  if (!same(parent, r.parent)) throw new Error('原样片参数或引用已变化；未提交')
+  if (!confirmed) return { ...summary(r), estimate, generation_submitted: false }
+  if (!Number.isFinite(expectedCost) || expectedCost < 0 || expectedCost !== estimate.estimated_credit_cost)
+    throw new Error('请展示实际钱包及继承参数、最新报价并取得当次确认；--expected-cost 必须匹配')
+  const protocol = await readProtocol(client)
+  if (protocol.required && !protocol.agreed) {
+    r.state = 'blocked_agreement'; await persist(r)
+    throw agreementError('AGREEMENT_REQUIRED', '由用户本人签署当前协议后，沿用原 operation 与原费用确认继续', protocol)
+  }
+  r.estimate = estimate; r.approvedCost = expectedCost
+  if (!r.nodeSaved) {
+    if (doc.nodes.some(n => n.id === r.output.id) || doc.edges.some(e => e.id === r.edge.id)) throw new Error('派生 ID 已存在；先查询原 operation')
+    const next = structuredClone(doc); next.nodes.push(r.output); next.edges.push(r.edge)
+    r.state = 'saving'; r.expectedVersion = doc.version; await persist(r)
+    try {
+      const saved = await save(client, doc, { nodes: next.nodes, edges: next.edges })
+      const actual = await scopedCanvas(client, r.input.doc_id, r.input.space, r.input.team_uid)
+      if (actual.version < saved.version || !same(actual.nodes, next.nodes) || !same(actual.edges, next.edges) ||
+          !same(actual.meta, doc.meta) || !same(actual.viewport, doc.viewport) || actual.name !== doc.name) throw new Error('正片节点读回不一致')
+      r.nodeSaved = true; r.state = 'node_saved'; await persist(r)
+    } catch (error) { r.state = 'save_unknown'; await persist(r); throw fail('DRAFT_SYNC_PENDING', '正片节点保存未确认；查询原 operation，不重放画布保存', error) }
+  } else verifyOutput(doc, r)
+  r.state = 'submitting'; await persist(r)
+  try {
+    const body = { doc_id: r.input.doc_id, node_id: r.output.id, draft_task_id: Number(r.parent.task_id), protocol_agreed: Boolean(protocol.agreed) }
+    const task = await client.raw('/canvas/generations/draft-finalize', { method: 'POST', body, headers: { 'Idempotency-Key': r.key } }, { retry: false })
+    verifyTask(task, r); r.taskId = task.task_id ?? task.id; r.state = 'submitted'; await persist(r)
+    return { ...summary(r), task }
+  } catch (error) {
+    r.state = error.apiCode === 4401 ? 'blocked_agreement' : 'unknown'; await persist(r)
+    if (error.apiCode === 4401) throw agreementError('AGREEMENT_REQUIRED', '当前协议需要本人签署；保留原 operation 和派生节点', protocol)
+    throw fail('DRAFT_SUBMISSION_UNKNOWN', '正片提交未知；只查询原 operation，不更换键、不删除派生节点', error)
+  }
+}
+export async function getDraftFinalize(client, profile, operation, sync = false) {
+  const r = await bound(client, profile, operation)
+  if (sync) await requireServiceReady(client)
+  const doc = await scopedCanvas(client, r.input.doc_id, r.input.space, r.input.team_uid); snapshot(doc)
+  if (['saving','save_unknown'].includes(r.state)) {
+    verifyOutput(doc, r)
+    const node = doc.nodes.find(n => n.id === r.output.id)
+    if (!same(node, r.output)) throw fail('DRAFT_SYNC_PENDING', '保存未知且派生内容变化，保留现场')
+    r.nodeSaved = true; r.state = 'node_saved'; await persist(r)
+  }
+  if (!r.taskId && ['submitting','unknown'].includes(r.state)) {
+    // Query the dedicated output node. No POST replay: parent expiry/entitlement
+    // checks can precede the backend's idempotency lookup.
+    const result = await client.raw('/canvas/generations?' + new URLSearchParams({ doc_id: r.input.doc_id, node_id: r.output.id, limit: 100 }))
+    if (!Array.isArray(result.items)) throw new Error('正片恢复查询响应无效')
+    const candidates = result.items.filter(t => t.node_id === r.output.id)
+    if (candidates.length !== 1) return { ...summary(r), next: '原任务尚未唯一可见；继续查询，不重提', replayed: false }
+    const task = await client.raw('/canvas/generations/' + segment(candidates[0].task_id ?? candidates[0].id))
+    verifyTask(task, { ...r, taskId: candidates[0].task_id ?? candidates[0].id }); r.taskId = task.task_id ?? task.id; r.state = 'submitted'; await persist(r)
+  }
+  if (!r.taskId) return summary(r)
+  const task = await client.raw('/canvas/generations/' + segment(r.taskId)); verifyTask(task, r)
+  if (!sync) return { ...summary(r), task }
+  const node = verifyOutput(doc, r), current = taskId(node)
+  if (current && current !== Number(r.taskId)) throw fail('DRAFT_SYNC_PENDING', '正片节点已绑定其他任务，不覆盖')
+  const next = structuredClone(doc), target = next.nodes.find(n => n.id === node.id)
+  target.data = { ...target.data, last_task_id: r.taskId, status: task.status,
+    runtime: { status: task.status, progress: task.progress || 0, error: task.error_message || '' },
+    extra: { ...target.data.extra, draft: task.draft } }
+  if (task.status === 'completed') {
+    if (!task.result?.url || !task.result.file_ref || !String(task.result.mime_type).startsWith('video/') || task.delivery_blocked) throw fail('DRAFT_SYNC_PENDING', '正片尚无可交付视频，未回填')
+    target.data.result = { ...task.result, task_id: r.taskId, status: 'completed' }
+    target.data.extra.file_ref = task.result.file_ref
+  }
+  if (!same(node, target)) {
+    try {
+      const saved = await save(client, doc, { nodes: next.nodes })
+      const actual = await scopedCanvas(client, r.input.doc_id, r.input.space, r.input.team_uid)
+      if (actual.version < saved.version || !same(actual.nodes, next.nodes) || !same(actual.edges, doc.edges) ||
+          !same(actual.meta, doc.meta) || !same(actual.viewport, doc.viewport) || actual.name !== doc.name) throw new Error('正片回填读回不一致')
+    } catch (error) { throw fail('DRAFT_SYNC_PENDING', '正片回填未确认；重新查询原 operation，不重提任务', error) }
+  }
+  r.state = task.status; await persist(r)
+  return { ...summary(r), task, canvas_synced: true }
+}
+export async function waitDraftFinalize(client, profile, operation, interval = 3000, timeout = 600000) {
+  const current = await getDraftFinalize(client, profile, operation)
+  if (!current.task_id) return current
+  await wait(client, current.task_id, interval, timeout)
+  return getDraftFinalize(client, profile, operation, true)
+}

package/src/generation-graph.js

@@ -0,0 +1,124 @@
+import { writeNode } from './canvas.js'
+import { segment } from './workflows.js'
+import fs from 'node:fs/promises'
+import path from 'node:path'
+import { root, safeName, json } from './store.js'
+import { isDeepStrictEqual } from 'node:util'
+
+async function submissionBaseline(client,taskId) {
+  if(!client.profile)return null
+  const directory=path.join(root(),'operations',safeName(client.profile))
+  const files=await fs.readdir(directory).catch(error=>{if(error.code==='ENOENT')return [];throw error})
+  const candidates=[]
+  for(const file of files.filter(f=>f.endsWith('.json'))){
+    const record=await json(path.join(directory,file),null)
+    if(String(record?.taskId)===String(taskId)&&record.baseUrl===client.baseUrl&&record.syncBaseline)candidates.push(record)
+  }
+  if(candidates.length!==1)return null
+  const me=await client.raw('/auth/me')
+  return String(me.uid??me.id)===candidates[0].owner?candidates[0].syncBaseline:null
+}
+async function matchingDraftModel(client,doc,node,task){
+  if(task.segment_type!=='video'||task.draft?.step!=='preview'||node.data?.extra?.draft_preview!==true)return false
+  const query=doc.team_uid?'?team_uid='+segment(doc.team_uid):''
+  const catalog=await client.raw('/canvas/models'+query)
+  const sample=catalog.items?.find(m=>m.label===node.data.model_name&&m.is_sample_mode===true)
+  if(!sample)return false
+  // The public deployment omits model_code/base_model_code. Only the verified
+  // Seedance 2.5 Draft alias is accepted in that reduced catalog contract.
+  if(!sample.base_model_code)return sample.label==='Seedance 2.5 Draft'&&task.model_name==='Seedance 2.5'&&
+    sample.sample_mode?.preview_resolution==='480p'&&catalog.items.some(m=>m.label===task.model_name&&m.is_sample_mode!==true)
+  const base=catalog.items.find(m=>m.model_code===sample.base_model_code&&m.is_sample_mode!==true)
+  return task.model_name===sample.base_model_code||!!base&&task.model_name===base.label
+}
+
+// Task completion and canvas persistence are separate contracts. Reconcile the
+// original real task; never create or re-submit a generation here.
+export async function syncGeneration(client, taskId, { baseline } = {}) {
+  const task = await client.raw(`/canvas/generations/${segment(taskId)}`)
+  if (
+    String(task.task_id ?? task.id) !== String(taskId) ||
+    !task.doc_id ||
+    !task.node_id ||
+    !['text', 'image', 'video'].includes(task.segment_type)
+  )
+    throw new Error('生成任务绑定不完整')
+  if (!['completed', 'failed', 'cancelled', 'canceled'].includes(task.status))
+    return { synced: false, task_id: taskId, status: task.status, next: '继续查询原任务' }
+  const doc = await client.raw(`/canvas/docs/${segment(task.doc_id)}`)
+  const node = doc.nodes?.find((n) => n.id === task.node_id)
+  if (
+    !node ||
+    node.type !== task.segment_type ||
+    node.data?.extra?.node_kind === 'asset' ||
+    node.data?.extra?.repaint ||
+    node.data?.extra?.video_remake ||
+    node.data?.extra?.clip_redraw || node.data?.extra?.draft_finalize || node.data?.draftFinalizeGeneration
+  )
+    throw new Error('任务目标不是普通生成节点；派生流程不能通用回填')
+  const current = String(node.data?.last_task_id || node.data?.result?.task_id || '')
+  if (current && current !== String(taskId)) {
+    const original=baseline || await submissionBaseline(client,taskId)
+    if(!original || original.doc_id!==doc.id || (original.team_uid||null)!==(doc.team_uid||null) ||
+       !isDeepStrictEqual(original.node,node) || !/^\d+$/.test(current) || BigInt(current)>=BigInt(taskId))
+      throw new Error('节点已关联其他生成任务；提交前快照不匹配，停止以保留后续结果')
+    const query=new URLSearchParams({doc_id:task.doc_id,node_id:task.node_id,limit:'1'})
+    const latest=await client.raw('/canvas/generations?'+query)
+    if(String(latest.items?.[0]?.task_id??latest.items?.[0]?.id)!==String(taskId))
+      throw new Error('节点已关联其他生成任务；存在后续任务或最新任务无法核验，停止回填')
+  }
+  if (node.data?.model_name && node.data.model_name !== task.model_name && !await matchingDraftModel(client,doc,node,task))
+    throw new Error('节点模型已变化；先核对任务与节点')
+  const patch = {
+    status: task.status,
+    last_task_id: task.task_id ?? task.id,
+    error_message: task.error_message || ''
+  }
+  if (task.status === 'completed') {
+    const result = task.result
+    if (
+      !result ||
+      (task.segment_type === 'text'
+        ? typeof result.text !== 'string' || !result.text.trim()
+        : !result.url || !result.file_ref)
+    )
+      throw new Error('已完成任务缺少真实输出；停止回填')
+    patch.result = { ...result, task_id: task.task_id ?? task.id, status: 'completed' }
+    if (task.segment_type === 'text') patch.output = result.text
+    else patch.extra = { file_ref: result.file_ref,
+      ...(task.segment_type==='video'&&task.draft?.step==='preview'?{draft:structuredClone(task.draft)}:{}) }
+    if(task.segment_type==='video'&&task.draft?.step==='preview'&&Object.hasOwn(node.data||{},'draft'))patch.draft=structuredClone(task.draft)
+  }
+  const same = Object.entries(patch).every(([key, value]) =>
+    key === 'extra'
+      ? Object.entries(value).every(([k, v]) => isDeepStrictEqual(node.data?.extra?.[k],v))
+      : JSON.stringify(node.data?.[key]) === JSON.stringify(value)
+  )
+  if (same)
+    return {
+      synced: true,
+      unchanged: true,
+      task_id: taskId,
+      doc_id: doc.id,
+      node_id: node.id,
+      version: doc.version,
+      status: task.status
+    }
+  const bound = await writeNode(client, {
+    docId: doc.id,
+    nodeId: node.id,
+    patch: { data: patch },
+    update: true,
+    expectedSpace: doc.team_uid ? 'team' : 'personal',
+    expectedTeamUid: doc.team_uid,
+    expectedVersion: doc.version
+  })
+  return {
+    synced: true,
+    task_id: taskId,
+    doc_id: doc.id,
+    node_id: node.id,
+    version: bound.version,
+    status: task.status
+  }
+}