import { ScriptSlice, ScriptValueObject } from "../domain/valueObject";
import { generateScriptStream, applyScriptToShot } from "@/api/video_flow";

export class ScriptEditUseCase {
  loading: boolean = false;
  private scriptValueObject: ScriptValueObject;
  private abortController: AbortController | null = null;

  constructor(script: string) {
    this.scriptValueObject = new ScriptValueObject(script);
  }

  /**
   * @description: AI生成剧本方法
   * @param prompt 剧本提示词
   * @param stream_callback 流式数据回调函数
   * @returns Promise<void>
   */
  async generateScript(prompt: string, stream_callback?: (data: any) => void): Promise<void> {
    try {
      this.loading = true;

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

      // 使用API接口生成剧本
      const response = await generateScriptStream({
        text: prompt,
      });

      if (!response.successful) {
        throw new Error(response.message || "AI生成剧本失败");
      }

      // 使用for await处理流式数据
      for await (const data of response.data) {
        // 检查是否被中断
        if (this.abortController.signal.aborted) {
          console.log("剧本生成被中断");
          break;
        }

        // TODO: 根据流式数据更新剧本片段
        // 这里需要根据实际的流式数据格式来处理
        // 可能需要将流式数据转换为ScriptSlice并添加到scriptValueObject中
        stream_callback?.(data);
      }
    } catch (error) {
      if (this.abortController?.signal.aborted) {
        console.log("剧本生成被中断");
        return;
      }
      console.error("AI生成剧本出错:", error);
      throw error;
    } finally {
      this.loading = false;
      this.abortController = null;
    }
  }

  /**
   * @description: 中断剧本生成
   */
  abortGenerateScript(): void {
    if (this.abortController) {
      this.abortController.abort();
      this.loading = false;
    }
  }

  /**
   * @description: 应用剧本方法
   * @returns Promise<void>
   */
  async applyScript(projectId: string): Promise<void> {
    try {
      this.loading = true;

      // 调用应用剧本接口
      const response = await applyScriptToShot({
        projectId,
        scriptText: this.scriptValueObject.toString(),
      });

      if (!response.successful) {
        throw new Error(response.message || "应用剧本失败");
      }

      console.log("剧本应用成功");
    } catch (error) {
      console.error("应用剧本失败:", error);
      throw error;
    } finally {
      this.loading = false;
    }
  }

  /**
   * @description: 获取当前剧本片段列表
   * @returns ScriptSlice[]
   */
  getScriptSlices(): ScriptSlice[] {
    return this.scriptValueObject.scriptSlices;
  }

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

  /**
   * @description: 更新剧本
   * @param scriptText 剧本文本字符串
   */
  updateScript(scriptText: string): void {
    this.scriptValueObject = new ScriptValueObject(scriptText);
  }

  /**
   * @description: 将当前剧本片段转换为字符串
   * @returns string
   */
  toString(): string {
    return this.scriptValueObject.toString();
  }

  /**
   * @description: 更新指定ID的剧本片段
   * @param id 剧本片段唯一标识
   * @param text 新的文本内容
   * @param metaData 新的元数据
   * @returns boolean 是否更新成功
   */
  updateScriptSlice(id: string, text: string, metaData?: Record<string, any>): boolean {
    const index = this.scriptValueObject.scriptSlices.findIndex(slice => slice.id === id);
    if (index === -1) {
      return false;
    }

    // 只更新text和metaData字段，保持其他字段不变
    this.scriptValueObject.scriptSlices[index] = {
      ...this.scriptValueObject.scriptSlices[index],
      text,
      metaData: metaData || this.scriptValueObject.scriptSlices[index].metaData
    };

    return true;
  }
}