第三方回调的状态映射完整性

// ✅ 正确：保存完整的映射关系
const { taskId, resultUrls } = callbackData.data;

// 提取 vendor_task_id
const vendorTaskId = typeof taskId === 'string' ? taskId : taskId?.id;

await supabase
  .from('ai_generations')
  .update({
    vendor_task_id: vendorTaskId,  // 关键：保存映射
    generated_video_url: resultUrls[0],
    status: 'completed',
    updated_at: new Date().toISOString()
  })
  .eq('id', jobId);

console.log(`[Job ${jobId}] Mapped to vendor task: ${vendorTaskId}`);

// 获取任务类型
const { data: job } = await supabase
  .from('ai_generations')
  .select('type')
  .eq('id', jobId)
  .single();

const taskType = job.type; // 'video' 或 'image'

// 根据类型选择字段
const updateData = {
  vendor_task_id: vendorTaskId,
  status: 'completed',
  updated_at: new Date().toISOString()
};

if (taskType === 'video') {
  updateData.generated_video_url = finalUrl;
} else if (taskType === 'image') {
  updateData.generated_image_url = finalUrl;
}

await supabase
  .from('ai_generations')
  .update(updateData)
  .eq('id', jobId);

console.log(`[Job ${jobId}] Type: ${taskType}, URL: ${finalUrl}`);

// 1. 从 KIE.AI 下载文件
const response = await fetch(kieUrl);
const arrayBuffer = await response.arrayBuffer();
const fileSize = (arrayBuffer.byteLength / 1024 / 1024).toFixed(2);

console.log(`[Job ${jobId}] Downloaded file. Size: ${fileSize} MB`);

// 2. 生成 R2 存储路径
const fileExtension = taskType === 'video' ? 'mp4' : 'png';
const uniqueId = crypto.randomUUID();
const r2Key = `${taskType}s/${jobId}/${uniqueId}.${fileExtension}`;

// 3. 上传到 R2
await env.CREATIONS_BUCKET.put(r2Key, arrayBuffer, {
  httpMetadata: {
    contentType: taskType === 'video' ? 'video/mp4' : 'image/png'
  }
});

// 4. 生成公开访问 URL
const publicUrl = `https://s.purikura.io/${r2Key}`;

console.log(`[Job ${jobId}] Uploaded to R2: ${publicUrl}`);

// 5. 更新数据库
const updateData = {
  vendor_task_id: vendorTaskId,
  status: 'completed',
  [taskType === 'video' ? 'generated_video_url' : 'generated_image_url']: publicUrl
};

await supabase
  .from('ai_generations')
  .update(updateData)
  .eq('id', jobId);

// 新增 RESTful 路由
app.post('/api/callbacks/kie/:jobId', async (c) => {
  const { jobId } = c.req.param();
  const callbackData = await c.req.json();

  console.log(`[Callback Job ${jobId}] Received via RESTful route.`);

  // 处理逻辑与旧路由相同
  return await handleKieCallback(c, jobId, callbackData);
});

// 保留旧路由以兼容现有集成
app.post('/kie-callback', async (c) => {
  const jobId = c.req.query('job_id');

  if (!jobId) {
    return c.json({ error: 'Missing job_id parameter' }, 400);
  }

  const callbackData = await c.req.json();
  console.log(`[Callback Job ${jobId}] Received via legacy route.`);

  return await handleKieCallback(c, jobId, callbackData);
});

const callbackUrl = `${env.CALLBACK_BASE_URL}/api/callbacks/kie/${jobId}`;