48 lines
1.5 KiB
TypeScript
48 lines
1.5 KiB
TypeScript
import { root } from '@repo/core'
|
|
import { type RunTemplateInput, TemplateController } from '@repo/sdk'
|
|
import { useCallback, useRef, useState } from 'react'
|
|
|
|
import { type ApiError } from '@/lib/types'
|
|
import { handleError } from './use-error'
|
|
|
|
export const useTemplateActions = () => {
|
|
const [loading, setLoading] = useState(false)
|
|
const [error, setError] = useState<ApiError | null>(null)
|
|
const lastParamsRef = useRef<RunTemplateInput | null>(null)
|
|
|
|
const runTemplate = useCallback(async (params: RunTemplateInput): Promise<{ generationId?: string; error?: ApiError }> => {
|
|
setLoading(true)
|
|
setError(null)
|
|
lastParamsRef.current = params
|
|
|
|
const template = root.get(TemplateController)
|
|
const { data, error } = await handleError(async () => await template.run({
|
|
templateId: params.templateId,
|
|
data: params.data || {},
|
|
}))
|
|
|
|
setLoading(false)
|
|
|
|
if (error) {
|
|
setError(error)
|
|
return { error }
|
|
}
|
|
|
|
return { generationId: data?.generationId }
|
|
}, [])
|
|
|
|
const retry = useCallback(async (): Promise<{ generationId?: string; error?: ApiError }> => {
|
|
if (!lastParamsRef.current) {
|
|
return { error: { message: 'No previous operation to retry' } as ApiError }
|
|
}
|
|
return runTemplate(lastParamsRef.current)
|
|
}, [runTemplate])
|
|
|
|
return {
|
|
loading,
|
|
error,
|
|
runTemplate,
|
|
retry,
|
|
}
|
|
}
|