import { ApiError } from "@/lib/types"
import { useState, useCallback } from "react"
import { root } from '@repo/core'
import { TemplateGenerationController, ListTemplateGenerationsInput, ListTemplateGenerationsResult } from "@repo/sdk"
import { useError } from "./use-error"

export const useTemplateGenerations = () => {
  const [loading, setLoading] = useState(false)
  const [error, setError] = useState<ApiError | null>(null)
  const [data, setData] = useState<ListTemplateGenerationsResult | undefined>()

  const load = useCallback(async (params?: ListTemplateGenerationsInput) => {
    setLoading(true)
    setError(null)

    const templateGeneration = root.get(TemplateGenerationController)
    const { data, error } = await useError(async () => await templateGeneration.list({
      page: params?.page || 1,
      limit: params?.limit || 20,
      ...params
    }))

    if (error) {
      setError(error)
      setLoading(false)
      return { data: undefined, error }
    }

    setData(data)
    setLoading(false)
    return { data, error: null }
  }, [])

  const refetch = useCallback((params?: ListTemplateGenerationsInput) => {
    return load(params)
  }, [load])

  return {
    data,
    loading,
    error,
    load,
    refetch,
  }
}