video-flow-b/app/service/usecase/ScriptEditUseCase.ts

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接口生成剧本
        await generateScriptStream({
          text: prompt,
        },(content)=>{
          stream_callback?.(content)
          this.scriptValueObject.parseFromString(content)
        });
    } 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();
  }

}