88 lines
2.4 KiB
TypeScript
88 lines
2.4 KiB
TypeScript
import { ApiError } from "@/lib/types"
|
|
import { useState, useCallback } from "react"
|
|
import { root } from '@repo/core'
|
|
import { TemplateController, TemplateGenerationController, RunTemplateInput, CreateTemplateGenerationInput } from "@repo/sdk"
|
|
import { useError } from "../data/use-error"
|
|
|
|
export const useTemplateActions = () => {
|
|
const [loading, setLoading] = useState(false)
|
|
const [error, setError] = useState<ApiError | null>(null)
|
|
|
|
const runTemplate = useCallback(async (params: RunTemplateInput) => {
|
|
setLoading(true)
|
|
setError(null)
|
|
|
|
const template = root.get(TemplateController)
|
|
const { data, error } = await useError(async () => await template.run(params))
|
|
|
|
if (error) {
|
|
setError(error)
|
|
setLoading(false)
|
|
return { generationId: null, error }
|
|
}
|
|
|
|
setLoading(false)
|
|
return { generationId: data?.generationId, error: null }
|
|
}, [])
|
|
|
|
const createGeneration = useCallback(async (params: CreateTemplateGenerationInput) => {
|
|
setLoading(true)
|
|
setError(null)
|
|
|
|
const templateGeneration = root.get(TemplateGenerationController)
|
|
const { data, error } = await useError(async () => await templateGeneration.create(params))
|
|
|
|
if (error) {
|
|
setError(error)
|
|
setLoading(false)
|
|
return { generation: null, error }
|
|
}
|
|
|
|
setLoading(false)
|
|
return { generation: data, error: null }
|
|
}, [])
|
|
|
|
const deleteGeneration = useCallback(async (id: string) => {
|
|
setLoading(true)
|
|
setError(null)
|
|
|
|
const templateGeneration = root.get(TemplateGenerationController)
|
|
const { data, error } = await useError(async () => await templateGeneration.delete({ id }))
|
|
|
|
if (error) {
|
|
setError(error)
|
|
setLoading(false)
|
|
return { success: false, error }
|
|
}
|
|
|
|
setLoading(false)
|
|
return { success: true, message: data?.message, error: null }
|
|
}, [])
|
|
|
|
const batchDeleteGenerations = useCallback(async (ids: string[]) => {
|
|
setLoading(true)
|
|
setError(null)
|
|
|
|
const templateGeneration = root.get(TemplateGenerationController)
|
|
const { data, error } = await useError(async () => await templateGeneration.batchDelete({ ids }))
|
|
|
|
if (error) {
|
|
setError(error)
|
|
setLoading(false)
|
|
return { success: false, error }
|
|
}
|
|
|
|
setLoading(false)
|
|
return { success: true, message: data?.message, error: null }
|
|
}, [])
|
|
|
|
return {
|
|
loading,
|
|
error,
|
|
runTemplate,
|
|
createGeneration,
|
|
deleteGeneration,
|
|
batchDeleteGenerations,
|
|
}
|
|
}
|