mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-07-11 03:21:55 +08:00
46 lines
1.4 KiB
TypeScript
46 lines
1.4 KiB
TypeScript
import { useState, useCallback } from 'react'
|
|
import debounce from 'lodash/debounce'
|
|
import { DebouncedFunc } from 'lodash'
|
|
import { validateProviderKey } from '@/service/common'
|
|
|
|
export enum ValidatedStatus {
|
|
Success = 'success',
|
|
Error = 'error',
|
|
Exceed = 'exceed'
|
|
}
|
|
|
|
const useValidateToken = (providerName: string): [boolean, ValidatedStatus | undefined, DebouncedFunc<(token: string) => Promise<void>>] => {
|
|
const [validating, setValidating] = useState(false)
|
|
const [validatedStatus, setValidatedStatus] = useState<ValidatedStatus | undefined>()
|
|
const validate = useCallback(debounce(async (token: string) => {
|
|
if (!token) {
|
|
setValidatedStatus(undefined)
|
|
return
|
|
}
|
|
setValidating(true)
|
|
try {
|
|
const res = await validateProviderKey({ url: `/workspaces/current/providers/${providerName}/token-validate`, body: { token } })
|
|
setValidatedStatus(res.result === 'success' ? ValidatedStatus.Success : ValidatedStatus.Error)
|
|
} catch (e: any) {
|
|
if (e.status === 400) {
|
|
e.json().then(({ code }: any) => {
|
|
if (code === 'provider_request_failed') {
|
|
setValidatedStatus(ValidatedStatus.Exceed)
|
|
}
|
|
})
|
|
} else {
|
|
setValidatedStatus(ValidatedStatus.Error)
|
|
}
|
|
} finally {
|
|
setValidating(false)
|
|
}
|
|
}, 500), [])
|
|
|
|
return [
|
|
validating,
|
|
validatedStatus,
|
|
validate,
|
|
]
|
|
}
|
|
|
|
export default useValidateToken |