omniai-web/src/features/ecommerce/ecommerceVideoService.ts

import {
  analyzeProductImages,
  buildProductSummary,
  extractSellingPoints,
  generateCreativeOptions,
  generateStoryboard,
  generateVideoPrompts,
  checkCompliance,
  type AdVideoUserConfig,
} from "../../api/adVideoPlanClient";
import { aiGenerationClient } from "../../api/aiGenerationClient";
import { waitForTask } from "../../api/taskSubscription";
import { resolveVideoRequestModel } from "../../utils/resolveVideoModel";
import { normalizeEcommerceImageMime } from "./ecommerceImageValidation";
import type {
  EcommerceVideoPlanProgress,
  EcommerceVideoPlanResult,
  EcommerceVideoSceneTask,
  PlanStep,
} from "./ecommerceVideoTypes";

export interface PlanCallbacks {
  onStepStart: (step: PlanStep) => void;
  onStepDone: (step: PlanStep) => void;
  onImagesUploaded?: (urls: string[]) => void;
  onUploadRejected?: (messages: string[]) => void;
  onPartialProgress?: (progress: EcommerceVideoPlanProgress) => void;
  signal?: AbortSignal;
  /** Partial state from a previous run; steps with existing data are skipped. */
  resumeFrom?: EcommerceVideoPlanProgress;
}

/**
 * Run the full ad video planning pipeline.
 * Supports resumption: if `resumeFrom` contains data for a step, that step is skipped.
 * After each step, `onPartialProgress` fires so callers can persist intermediate state.
 */
export async function runVideoPlan(
  imageDataUrls: string[],
  manualText: string,
  config: AdVideoUserConfig,
  callbacks: PlanCallbacks,
): Promise<EcommerceVideoPlanResult> {
  const { onStepStart, onStepDone, signal, resumeFrom = {} } = callbacks;
  const progress: EcommerceVideoPlanProgress = { ...resumeFrom };
  const emit = () => callbacks.onPartialProgress?.({ ...progress });

  // ── Step: upload ──────────────────────────────────────
  if (!progress.imageUrls?.length) {
    onStepStart("upload");
    const imageUrls: string[] = [];
    const rejected: string[] = [];
    for (const srcUrl of imageDataUrls) {
      try {
        const resp = await fetch(srcUrl);
        const rawBlob = await resp.blob();
        const mimeType = normalizeEcommerceImageMime(rawBlob.type);
        const blob = rawBlob.type === mimeType ? rawBlob : new Blob([rawBlob], { type: mimeType });
        const dataUrl = await new Promise<string>((resolve, reject) => {
          const reader = new FileReader();
          reader.onload = () => resolve(String(reader.result || ""));
          reader.onerror = () => reject(reader.error || new Error("文件读取失败"));
          reader.readAsDataURL(blob);
        });
        const result = await aiGenerationClient.uploadAsset({ dataUrl, mimeType, scope: "ecommerce-product" });
        imageUrls.push(result.url);
      } catch (err) {
        rejected.push(err instanceof Error ? err.message : "图片上传失败");
      }
    }
    if (rejected.length) {
      progress.uploadWarnings = rejected;
      callbacks.onUploadRejected?.(rejected);
    }
    if (!imageUrls.length) throw new Error("图片上传失败，请检查图片格式或网络后重试");
    progress.imageUrls = imageUrls;
    onStepDone("upload");
    callbacks.onImagesUploaded?.(imageUrls);
    emit();
  }

  // ── Step: analyze ─────────────────────────────────────
  if (progress.imageDescription === undefined) {
    onStepStart("analyze");
    progress.imageDescription = await analyzeProductImages(progress.imageUrls!, signal);
    onStepDone("analyze");
    emit();
  }

  // ── Step: summary ─────────────────────────────────────
  if (!progress.summary) {
    onStepStart("summary");
    progress.summary = await buildProductSummary(progress.imageDescription || "", manualText, signal);
    onStepDone("summary");
    emit();
  }

  // ── Step: selling ─────────────────────────────────────
  if (!progress.selling) {
    onStepStart("selling");
    progress.selling = await extractSellingPoints(progress.summary, signal);
    onStepDone("selling");
    emit();
  }

  // ── Step: creative ────────────────────────────────────
  if (!progress.creatives?.length) {
    onStepStart("creative");
    progress.creatives = await generateCreativeOptions(progress.selling, config, signal);
    if (!progress.creatives.length) throw new Error("未能生成有效的广告创意");
    onStepDone("creative");
    emit();
  }

  // ── Step: storyboard ──────────────────────────────────
  if (!progress.storyboard) {
    onStepStart("storyboard");
    progress.storyboard = await generateStoryboard(progress.creatives[0], progress.summary, config, signal);
    onStepDone("storyboard");
    emit();
  }

  // ── Step: prompts ─────────────────────────────────────
  if (!progress.videoPrompts) {
    onStepStart("prompts");
    progress.videoPrompts = await generateVideoPrompts(progress.storyboard, progress.summary, signal);
    onStepDone("prompts");
    emit();
  }

  // ── Step: compliance ──────────────────────────────────
  if (!progress.compliance) {
    onStepStart("compliance");
    progress.compliance = await checkCompliance(progress.summary, progress.selling, progress.storyboard, signal);
    onStepDone("compliance");
    emit();
  }

  return {
    imageUrls: progress.imageUrls!,
    imageDescription: progress.imageDescription,
    summary: progress.summary!,
    selling: progress.selling!,
    creatives: progress.creatives!,
    storyboard: progress.storyboard!,
    videoPrompts: progress.videoPrompts!,
    compliance: progress.compliance!,
  };
}

export interface RenderSceneImageInput {
  sceneId: number;
  prompt: string;
  aspectRatio: string;
  productImageUrls: string[];
}

export interface RenderImageCallbacks {
  onSceneImageSubmitted: (sceneId: number, taskId: string) => void;
  onSceneImageProgress: (sceneId: number, progress: number) => void;
  onSceneImageCompleted: (sceneId: number, resultUrl: string) => void;
  onSceneImageFailed: (sceneId: number, error: string) => void;
}

export async function renderSceneImage(
  input: RenderSceneImageInput,
  callbacks: RenderImageCallbacks,
  abortRef: { current: boolean },
): Promise<void> {
  const { taskId } = await aiGenerationClient.createImageTask({
    model: "gpt-image-2",
    prompt: input.prompt,
    ratio: input.aspectRatio,
    quality: "2K",
    referenceUrls: input.productImageUrls,
  });

  callbacks.onSceneImageSubmitted(input.sceneId, taskId);

  const resultUrl = await waitForTask(taskId, {
    abortRef,
    onProgress: (e) => callbacks.onSceneImageProgress(input.sceneId, e.progress),
  });

  if (resultUrl) {
    callbacks.onSceneImageCompleted(input.sceneId, resultUrl);
  } else {
    callbacks.onSceneImageFailed(input.sceneId, "图片生成未返回结果");
  }
}

export interface RenderSceneInput {
  sceneId: number;
  prompt: string;
  durationSeconds: number;
  imageUrl: string;
  productImageUrls: string[];
  aspectRatio: string;
  resolution: string;
  model?: string;
}

export interface RenderCallbacks {
  onSceneSubmitted: (sceneId: number, taskId: string) => void;
  onSceneProgress: (sceneId: number, progress: number) => void;
  onSceneCompleted: (sceneId: number, resultUrl: string) => void;
  onSceneFailed: (sceneId: number, error: string) => void;
}

export async function renderScene(
  input: RenderSceneInput,
  callbacks: RenderCallbacks,
  abortRef: { current: boolean },
): Promise<void> {
  const allReferenceUrls = [...input.productImageUrls, input.imageUrl];
  const model = resolveVideoRequestModel({
    model: input.model || "happyhorse-1.0",
    referenceUrls: allReferenceUrls,
  });

  const { taskId } = await aiGenerationClient.createVideoTask({
    model,
    prompt: input.prompt,
    ratio: input.aspectRatio,
    duration: input.durationSeconds,
    quality: input.resolution,
    resolution: input.resolution,
    frameMode: "start-end",
    referenceUrls: allReferenceUrls,
    hasReferenceVideo: false,
  });

  callbacks.onSceneSubmitted(input.sceneId, taskId);

  const resultUrl = await waitForTask(taskId, {
    abortRef,
    onProgress: (e) => callbacks.onSceneProgress(input.sceneId, e.progress),
  });

  if (resultUrl) {
    callbacks.onSceneCompleted(input.sceneId, resultUrl);
  } else {
    callbacks.onSceneFailed(input.sceneId, "任务未返回结果");
  }
}

export function buildSceneTasks(
  plan: EcommerceVideoPlanResult,
): EcommerceVideoSceneTask[] {
  return plan.storyboard.scenes.map((scene) => {
    const matchedPrompt = plan.videoPrompts.find((p) => p.scene_id === scene.scene_id);
    return {
      sceneId: scene.scene_id,
      prompt: matchedPrompt?.positive_prompt || scene.visual_description,
      durationSeconds: Number.parseInt(scene.duration, 10) || 5,
      status: "idle" as const,
      progress: 0,
    };
  });
}