import { VideoSegmentEntity } from "../domain/Entities";
import { LensType } from "../domain/valueObject";
import {
  getShotList,
  regenerateShot,
  optimizeShotContent,
} from "@/api/video_flow";

/**
 * 视频片段编辑用例
 * 负责视频片段内容的初始化、修改和优化
 */
export class VideoSegmentEditUseCase {
  private loading: boolean = false;
  private abortController: AbortController | null = null;

  /**
   * @description 获取视频片段列表
   * @param projectId 项目ID
   * @returns Promise<VideoSegmentEntity[]> 视频片段列表
   */
  async getVideoSegmentList(projectId: string): Promise<VideoSegmentEntity[]> {
    try {
      this.loading = true;

      const response = await getShotList({ projectId });

      if (!response.successful) {
        throw new Error(response.message || "获取视频片段列表失败");
      }

      return response.data || [];
    } catch (error) {
      console.error("获取视频片段列表失败:", error);
      throw error;
    } finally {
      this.loading = false;
    }
  }

  /**
   * @description 通过视频镜头描述数据重新生成视频
   * @param shotPrompt 镜头描述数据
   * @param shotId 视频片段ID（可选，如果重新生成现有片段）
   * @param roleReplaceParams 角色替换参数（可选）
   * @param sceneReplaceParams 场景替换参数（可选）
   * @returns Promise<VideoSegmentEntity> 重新生成的视频片段
   */
  async regenerateVideoSegment(
    shotPrompt: LensType[],
    shotId?: string,
    roleReplaceParams?: { oldId: string; newId: string }[],
    sceneReplaceParams?: { oldId: string; newId: string }[]
  ): Promise<VideoSegmentEntity> {
    try {
      this.loading = true;

      // 创建新的中断控制器
      this.abortController = new AbortController();

      const response = await regenerateShot({
        shotId,
        shotPrompt,
        roleReplaceParams,
        sceneReplaceParams,
      });

      if (!response.successful) {
        throw new Error(response.message || "重新生成视频片段失败");
      }

      return response.data;
    } catch (error) {
      if (this.abortController?.signal.aborted) {
        console.log("视频片段重新生成被中断");
        throw new Error("操作被中断");
      }
      console.error("重新生成视频片段失败:", error);
      throw error;
    } finally {
      this.loading = false;
      this.abortController = null;
    }
  }

  /**
   * @description 通过AI优化镜头数据（包含对话内容）
   * @param shotId 视频片段ID
   * @param userRequirement 用户优化需求
   * @param lensData 镜头数据数组
   * @returns Promise<LensType[]> 优化后的镜头数据
   */
  async optimizeVideoContent(
    shotId: string,
    userRequirement: string,
    lensData: LensType[]
  ): Promise<LensType[]> {
    try {
      this.loading = true;

      // 调用AI优化接口
      const response = await optimizeShotContent({
        shotId,
        userRequirement,
        lensData,
      });

      if (!response.successful) {
        throw new Error(response.message || "AI优化视频内容失败");
      }

      return response.data;
    } catch (error) {
      console.error("AI优化视频内容失败:", error);
      throw error;
    } finally {
      this.loading = false;
    }
  }
  /**
   * @description 中断当前操作
   */
  abortOperation(): void {
    if (this.abortController) {
      this.abortController.abort();
      this.loading = false;
    }
  }

  /**
   * @description 获取加载状态
   * @returns boolean 是否正在加载
   */
  isLoading(): boolean {
    return this.loading;
  }
}