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

/**
 * GeneratePage 步骤导航
 * 步骤顺序（6步）：模板(1) → 素材(2) → 配音(3) → 标题(4) → 确认生成(5) → 封面(6)
 */
import { message } from "antd"
import type { TitleSettings } from "../types"

export interface UseStepNavigationOptions {
  currentStep: number
  setCurrentStep: (step: number | ((prev: number) => number)) => void
  selectedTemplate: string
  materialMode: "manual" | "auto"
  selectedMaterials: string[]
  smartSelectedIds: string[]
  titleSettings: TitleSettings
  /** 预览是否已就绪（素材已加载，可播放） */
  previewReady: boolean
  /** 是否已完成视频生成（步骤5确认生成后才能进入封面） */
  generated: boolean
}

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

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

  const goNext = () => {
    if (currentStep === 1 && !selectedTemplate) {
      message.warning("请先选择一个模板")
      return
    }
    if (currentStep === 2 && materialMode === "manual" && selectedMaterials.length === 0) {
      message.warning("请至少选择一个素材")
      return
    }
    if (currentStep === 2 && materialMode === "auto" && smartSelectedIds.length === 0) {
      message.warning("请先进行智能匹配并选择素材")
      return
    }
    // Step4（标题+预览）：标题必填 + 预览必须已加载
    if (currentStep === 4) {
      if (!titleSettings.title.trim()) {
        message.warning("请选择或输入标题")
        return
      }
      if (!previewReady) {
        message.warning("预览视频正在加载，请稍候")
        return
      }
    }
    // Step5（确认生成）：必须已完成生成才能进入封面
    if (currentStep === 5 && !generated) {
      message.warning("请先生成视频")
      return
    }
    if (currentStep < 6) {
      setCurrentStep((s) => s + 1)
    }
  }

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

  return { goNext, goPrev }
}

export default useStepNavigation