68 lines
1.7 KiB
TypeScript
68 lines
1.7 KiB
TypeScript
import { useCallback, useState } from "react"
|
|
import { toast } from "sonner"
|
|
import type { useDataTable } from "@/components/data-table"
|
|
import type { ApiResponse } from "@/lib/api"
|
|
|
|
export function useStatus() {
|
|
return useState<"load" | "fail" | "done">("load")
|
|
}
|
|
|
|
export function useFetch<TArgs extends unknown[], TResult>(
|
|
table: ReturnType<typeof useDataTable>,
|
|
fetchData: (...args: TArgs) => Promise<ApiResponse<TResult>>,
|
|
messages: {
|
|
done?: string
|
|
fail?: string
|
|
},
|
|
) {
|
|
return useCallback(
|
|
async (...args: TArgs) => {
|
|
try {
|
|
table.setStatus?.("load")
|
|
const resp = await fetchData(...args)
|
|
if (!resp.success) {
|
|
throw new Error(resp.message)
|
|
}
|
|
|
|
table.setStatus?.("done")
|
|
table.refresh(table.pagination.page, table.pagination.size)
|
|
toast.success(messages.done || "获取数据成功")
|
|
} catch (e) {
|
|
table.setStatus?.("fail")
|
|
toast.error(messages.fail || "获取数据失败", {
|
|
description: (e as Error).message || "未知错误",
|
|
})
|
|
}
|
|
},
|
|
[
|
|
fetchData,
|
|
table.setStatus,
|
|
table.pagination.page,
|
|
table.pagination.size,
|
|
table.refresh,
|
|
messages,
|
|
],
|
|
)
|
|
}
|
|
|
|
type Action = <P extends unknown[], R>(...args: P) => Promise<R>
|
|
|
|
export function useAction(action: Action) {
|
|
const [status, setStatus] = useStatus()
|
|
const func = useCallback(
|
|
async (...args: Parameters<Action>) => {
|
|
try {
|
|
setStatus("load")
|
|
await action(...args)
|
|
setStatus("done")
|
|
} catch (e) {
|
|
setStatus("fail")
|
|
throw e
|
|
}
|
|
},
|
|
[action, setStatus],
|
|
)
|
|
|
|
return [func, status]
|
|
}
|