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

import { useState, useEffect } from "react"
import { useQuery } from "@tanstack/react-query"

export function useTemplateSelection() {
  const [selectedTemplate, setSelectedTemplate] = useState("")
  const { data: userTemplates = [] } = useQuery({
    queryKey: ["generate-templates"],
    staleTime: 60_000,
  })

  /* 模板加载完成后自动选中第一个 */
  useEffect(() => {
    if (userTemplates.length > 0 && !selectedTemplate) {
      setSelectedTemplate(userTemplates[0].id)
    }
  }, [userTemplates, selectedTemplate])

  return { selectedTemplate, setSelectedTemplate, userTemplates }
}