xiaoxia-saas/apps/web/src/pages/generate/hooks/useStepNavigation.ts

/**
 * GeneratePage 步骤导航（#1970 流程重构）
 * 步骤：选择模式(1) → 选择素材(2) → 选择标题(3) → 确认生成(4) → 选择封面(5)
 *
 * - 步骤1（选择模式）：下一步分支由外层弹窗处理（VoiceSelectModal / ScriptSelectModal），
 *   本 hook 的 goNext 仅在未选模式时拦截；外层 Modal onConfirm 里主动 setCurrentStep(2)。
 * - 步骤2（选择素材）：弹数量选择弹窗（PreviewCountModal），确认后跳步骤3。
 * - 步骤3 底部按钮是「确认生成视频」（由 GenerateStepActions 调 onConfirmGenerate），
 *   创建成功后跳步骤4；本 hook 的 goNext 只负责 2→3 和 4→5 的「下一步」。
 * - 步骤4（确认生成进度页）：全部渲染完成后「下一步」解锁进封面。
 */
import { message } from "antd"
import type { TitleSettings } from "../types"
import type { EditMode } from "../components/Step1EditMode"

export interface UseStepNavigationOptions {
  currentStep: number
  setCurrentStep: (step: number | ((prev: number) => number)) => void
  editMode: EditMode
  materialMode: "manual" | "auto"
  selectedMaterials: string[]
  smartSelectedIds: string[]
  titleSettings: TitleSettings
  /** 是否已完成视频生成（步骤4全部渲染完成后才能进入封面） */
  generated: boolean
  /** 点素材下一步时弹出数量选择弹窗 */
  onOpenCountModal: () => void
  /** 步骤1下一步：根据 editMode 打开对应弹窗（随机→配音 / 叙事→文案） */
  onOpenStep1Modal: () => void
}

export interface UseStepNavigationReturn {
  goNext: () => void
  goPrev: () => void
}

export const useStepNavigation = (options: UseStepNavigationOptions): UseStepNavigationReturn => {
  const {
    currentStep,
    setCurrentStep,
    materialMode,
    selectedMaterials,
    smartSelectedIds,
    generated,
    onOpenCountModal,
    onOpenStep1Modal,
  } = options

  const goNext = () => {
    if (currentStep === 1) {
      // 步骤1：先校验素材/配音等由弹窗负责，goNext 只负责触发弹窗
      onOpenStep1Modal()
      return
    }
    if (currentStep === 2) {
      // 素材校验
      if (materialMode === "manual" && selectedMaterials.length === 0) {
        message.warning("请至少选择一个素材")
        return
      }
      if (materialMode === "auto" && smartSelectedIds.length === 0) {
        message.warning("请先进行智能匹配并选择素材")
        return
      }
      // 弹数量选择弹窗
      onOpenCountModal()
      return
    }
    // 步骤4（确认生成）：全部渲染完成后才能下一步进封面
    if (currentStep === 4) {
      if (!generated) {
        message.warning("视频还在渲染中，请等待生成完成")
        return
      }
    }
    if (currentStep < 5) {
      setCurrentStep((s) => s + 1)
    }
  }

  const goPrev = () => {
    if (currentStep > 1) {
      setCurrentStep((s) => s - 1)
    }
  }

  return { goNext, goPrev }
}

export default useStepNavigation