feat: spark free quota verify (#1152)

web/app/(commonLayout)/apps/Apps.tsx

@@ -1,7 +1,7 @@
 'use client'
 
 import { useCallback, useEffect, useRef, useState } from 'react'
-import { useRouter, useSearchParams } from 'next/navigation'
+import { useRouter } from 'next/navigation'
 import useSWRInfinite from 'swr/infinite'
 import { useTranslation } from 'react-i18next'
 import AppCard from './AppCard'
@@ -10,8 +10,11 @@ import type { AppListResponse } from '@/models/app'
 import { fetchAppList } from '@/service/apps'
 import { useAppContext } from '@/context/app-context'
 import { NEED_REFRESH_APP_LIST_KEY } from '@/config'
-import { ProviderEnum } from '@/app/components/header/account-setting/model-page/declarations'
 import Confirm from '@/app/components/base/confirm/common'
+import {
+  useAnthropicCheckPay,
+  useSparkCheckQuota,
+} from '@/hooks/use-pay'
 
 const getKey = (pageIndex: number, previousPageData: AppListResponse) => {
   if (!pageIndex || previousPageData.has_more)
@@ -24,11 +27,10 @@ const Apps = () => {
   const { isCurrentWorkspaceManager } = useAppContext()
   const { data, isLoading, setSize, mutate } = useSWRInfinite(getKey, fetchAppList, { revalidateFirstPage: false })
   const anchorRef = useRef<HTMLDivElement>(null)
-  const searchParams = useSearchParams()
   const router = useRouter()
-  const payProviderName = searchParams.get('provider_name')
-  const payStatus = searchParams.get('payment_result')
-  const [showPayStatusModal, setShowPayStatusModal] = useState(false)
+  const [showPayStatusModal, setShowPayStatusModal] = useState(true)
+  const anthropicConfirmInfo = useAnthropicCheckPay()
+  const sparkConfirmInfo = useSparkCheckQuota()
 
   const handleCancelShowPayStatusModal = useCallback(() => {
     setShowPayStatusModal(false)
@@ -41,9 +43,7 @@ const Apps = () => {
       localStorage.removeItem(NEED_REFRESH_APP_LIST_KEY)
       mutate()
     }
-    if (payProviderName === ProviderEnum.anthropic && (payStatus === 'succeeded' || payStatus === 'cancelled'))
-      setShowPayStatusModal(true)
-  }, [mutate, payProviderName, payStatus, t])
+  }, [mutate, t])
 
   useEffect(() => {
     let observer: IntersectionObserver | undefined
@@ -65,23 +65,29 @@ const Apps = () => {
         <AppCard key={app.id} app={app} onRefresh={mutate} />
       )))}
       {
-        showPayStatusModal && (
+        showPayStatusModal && anthropicConfirmInfo && (
           <Confirm
             isShow
             onCancel={handleCancelShowPayStatusModal}
             onConfirm={handleCancelShowPayStatusModal}
-            type={
-              payStatus === 'succeeded'
-                ? 'success'
-                : 'danger'
-            }
-            title={
-              payStatus === 'succeeded'
-                ? t('common.actionMsg.paySucceeded')
-                : t('common.actionMsg.payCancelled')
-            }
+            type={anthropicConfirmInfo.type}
+            title={anthropicConfirmInfo.title}
             showOperateCancel={false}
-            confirmText={(payStatus === 'cancelled' && t('common.operation.ok')) || ''}
+            confirmText={(anthropicConfirmInfo.type === 'danger' && t('common.operation.ok')) || ''}
+          />
+        )
+      }
+      {
+        showPayStatusModal && sparkConfirmInfo && (
+          <Confirm
+            isShow
+            onCancel={handleCancelShowPayStatusModal}
+            onConfirm={handleCancelShowPayStatusModal}
+            type={sparkConfirmInfo.type}
+            title={sparkConfirmInfo.title}
+            desc={sparkConfirmInfo.desc}
+            showOperateCancel={false}
+            confirmText={(sparkConfirmInfo.type === 'danger' && t('common.operation.ok')) || ''}
           />
         )
       }