Feat/provider add zhipuai (#1192)
Co-authored-by: Joel <iamjoel007@gmail.com>
This commit is contained in:
zxhlyh
@@ -1,71 +0,0 @@
|
||||
'use client'
|
||||
|
||||
import { useEffect, useState } from 'react'
|
||||
import { useTranslation } from 'react-i18next'
|
||||
import useSWR from 'swr'
|
||||
import { useSearchParams } from 'next/navigation'
|
||||
import { useContext } from 'use-context-selector'
|
||||
import I18n from '@/context/i18n'
|
||||
import { ProviderEnum } from '@/app/components/header/account-setting/model-page/declarations'
|
||||
import { fetchSparkFreeQuotaVerify } from '@/service/common'
|
||||
import type { ConfirmCommonProps } from '@/app/components/base/confirm/common'
|
||||
|
||||
export type ConfirmType = Pick<ConfirmCommonProps, 'type' | 'title' | 'desc'>
|
||||
|
||||
export const useAnthropicCheckPay = () => {
|
||||
const { t } = useTranslation()
|
||||
const [confirm, setConfirm] = useState<ConfirmType | null>(null)
|
||||
const searchParams = useSearchParams()
|
||||
const providerName = searchParams.get('provider_name')
|
||||
const paymentResult = searchParams.get('payment_result')
|
||||
|
||||
useEffect(() => {
|
||||
if (providerName === ProviderEnum.anthropic && (paymentResult === 'succeeded' || paymentResult === 'cancelled')) {
|
||||
setConfirm({
|
||||
type: paymentResult === 'succeeded' ? 'success' : 'danger',
|
||||
title: paymentResult === 'succeeded' ? t('common.actionMsg.paySucceeded') : t('common.actionMsg.payCancelled'),
|
||||
})
|
||||
}
|
||||
}, [providerName, paymentResult, t])
|
||||
|
||||
return confirm
|
||||
}
|
||||
|
||||
const QUOTA_RECEIVE_STATUS = {
|
||||
success: {
|
||||
'en': 'Anthropic',
|
||||
'zh-Hans': '领取成功，将在 5 分钟后自动增加配额',
|
||||
},
|
||||
fail: {
|
||||
'en': 'Anthropic',
|
||||
'zh-Hans': '领取失败',
|
||||
},
|
||||
}
|
||||
|
||||
export const useSparkCheckQuota = () => {
|
||||
const { locale } = useContext(I18n)
|
||||
const [shouldVerify, setShouldVerify] = useState(false)
|
||||
const { data } = useSWR(
|
||||
shouldVerify
|
||||
? `/workspaces/current/model-providers/${ProviderEnum.spark}/free-quota-qualification-verify`
|
||||
: null,
|
||||
fetchSparkFreeQuotaVerify,
|
||||
)
|
||||
const searchParams = useSearchParams()
|
||||
const type = searchParams.get('type')
|
||||
const provider = searchParams.get('provider')
|
||||
const result = searchParams.get('result')
|
||||
|
||||
useEffect(() => {
|
||||
if (type === 'provider_apply_callback' && provider === ProviderEnum.spark && result === 'success')
|
||||
setShouldVerify(true)
|
||||
}, [type, provider, result])
|
||||
|
||||
return data
|
||||
? {
|
||||
type: data.flag ? 'success' : 'danger',
|
||||
title: data.flag ? QUOTA_RECEIVE_STATUS.success[locale] : QUOTA_RECEIVE_STATUS.fail[locale],
|
||||
desc: !data.flag ? data.reason : undefined,
|
||||
}
|
||||
: null
|
||||
}
|
||||
124
web/hooks/use-pay.tsx
Normal file
124
web/hooks/use-pay.tsx
Normal file
@@ -0,0 +1,124 @@
|
||||
'use client'
|
||||
|
||||
import { useCallback, useEffect, useState } from 'react'
|
||||
import { useRouter, useSearchParams } from 'next/navigation'
|
||||
import { useTranslation } from 'react-i18next'
|
||||
import useSWR from 'swr'
|
||||
import { useContext } from 'use-context-selector'
|
||||
import I18n from '@/context/i18n'
|
||||
import { ProviderEnum } from '@/app/components/header/account-setting/model-page/declarations'
|
||||
import { fetchFreeQuotaVerify } from '@/service/common'
|
||||
import type { ConfirmCommonProps } from '@/app/components/base/confirm/common'
|
||||
import Confirm from '@/app/components/base/confirm/common'
|
||||
|
||||
export type ConfirmType = Pick<ConfirmCommonProps, 'type' | 'title' | 'desc'>
|
||||
|
||||
export const useAnthropicCheckPay = () => {
|
||||
const { t } = useTranslation()
|
||||
const [confirm, setConfirm] = useState<ConfirmType | null>(null)
|
||||
const searchParams = useSearchParams()
|
||||
const providerName = searchParams.get('provider_name')
|
||||
const paymentResult = searchParams.get('payment_result')
|
||||
|
||||
useEffect(() => {
|
||||
if (providerName === ProviderEnum.anthropic && (paymentResult === 'succeeded' || paymentResult === 'cancelled')) {
|
||||
setConfirm({
|
||||
type: paymentResult === 'succeeded' ? 'success' : 'danger',
|
||||
title: paymentResult === 'succeeded' ? t('common.actionMsg.paySucceeded') : t('common.actionMsg.payCancelled'),
|
||||
})
|
||||
}
|
||||
}, [providerName, paymentResult, t])
|
||||
|
||||
return confirm
|
||||
}
|
||||
|
||||
const QUOTA_RECEIVE_STATUS = {
|
||||
[ProviderEnum.spark]: {
|
||||
success: {
|
||||
'en': 'Successful collection, the quota will be automatically increased after 5 minutes.',
|
||||
'zh-Hans': '领取成功，将在 5 分钟后自动增加配额',
|
||||
},
|
||||
fail: {
|
||||
'en': 'Failure to collect',
|
||||
'zh-Hans': '领取失败',
|
||||
},
|
||||
},
|
||||
[ProviderEnum.zhipuai]: {
|
||||
success: {
|
||||
'en': 'Successful collection',
|
||||
'zh-Hans': '领取成功',
|
||||
},
|
||||
fail: {
|
||||
'en': 'Failure to collect',
|
||||
'zh-Hans': '领取失败',
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
const FREE_CHECK_PROVIDER = [ProviderEnum.spark, ProviderEnum.zhipuai]
|
||||
export const useCheckFreeQuota = () => {
|
||||
const { locale } = useContext(I18n)
|
||||
const router = useRouter()
|
||||
const [shouldVerify, setShouldVerify] = useState(false)
|
||||
const searchParams = useSearchParams()
|
||||
const type = searchParams.get('type')
|
||||
const provider = searchParams.get('provider') as (ProviderEnum.spark | ProviderEnum.zhipuai)
|
||||
const result = searchParams.get('result')
|
||||
const token = searchParams.get('token')
|
||||
|
||||
const { data, error } = useSWR(
|
||||
shouldVerify
|
||||
? `/workspaces/current/model-providers/${provider}/free-quota-qualification-verify?token=${token}`
|
||||
: null,
|
||||
fetchFreeQuotaVerify,
|
||||
)
|
||||
|
||||
useEffect(() => {
|
||||
if (error)
|
||||
router.replace('/', { forceOptimisticNavigation: false })
|
||||
}, [error, router])
|
||||
|
||||
useEffect(() => {
|
||||
if (type === 'provider_apply_callback' && FREE_CHECK_PROVIDER.includes(provider) && result === 'success')
|
||||
setShouldVerify(true)
|
||||
}, [type, provider, result])
|
||||
|
||||
return (data && provider)
|
||||
? {
|
||||
type: data.flag ? 'success' : 'danger',
|
||||
title: data.flag ? QUOTA_RECEIVE_STATUS[provider].success[locale] : QUOTA_RECEIVE_STATUS[provider].fail[locale],
|
||||
desc: !data.flag ? data.reason : undefined,
|
||||
}
|
||||
: null
|
||||
}
|
||||
|
||||
export const CheckModal = () => {
|
||||
const router = useRouter()
|
||||
const { t } = useTranslation()
|
||||
const [showPayStatusModal, setShowPayStatusModal] = useState(true)
|
||||
const anthropicConfirmInfo = useAnthropicCheckPay()
|
||||
const freeQuotaConfirmInfo = useCheckFreeQuota()
|
||||
|
||||
const handleCancelShowPayStatusModal = useCallback(() => {
|
||||
setShowPayStatusModal(false)
|
||||
router.replace('/', { forceOptimisticNavigation: false })
|
||||
}, [router])
|
||||
|
||||
const confirmInfo = anthropicConfirmInfo || freeQuotaConfirmInfo
|
||||
|
||||
if (!confirmInfo || !showPayStatusModal)
|
||||
return null
|
||||
|
||||
return (
|
||||
<Confirm
|
||||
isShow
|
||||
onCancel={handleCancelShowPayStatusModal}
|
||||
onConfirm={handleCancelShowPayStatusModal}
|
||||
type={confirmInfo.type}
|
||||
title={confirmInfo.title}
|
||||
desc={confirmInfo.desc}
|
||||
showOperateCancel={false}
|
||||
confirmText={(confirmInfo.type === 'danger' && t('common.operation.ok')) || ''}
|
||||
/>
|
||||
)
|
||||
}
|
||||
Reference in New Issue
Block a user