feat checkout wechat

Signed-off-by: jingyang <3161362058@qq.com>
labring · Oct 30, 2023 · be8ab21 · be8ab21
1 parent 290f36f
commit be8ab21
Show file tree

Hide file tree

Showing 10 changed files with 193 additions and 104 deletions.
diff --git a/service/license/src/api/payment.ts b/service/license/src/api/payment.ts
@@ -1,10 +1,17 @@
 import { GET, POST } from '@/services/request';
-import { PaymentData, PaymentParams, PaymentResult, PaymentResultParams } from '@/types';
+import {
+  CheckWeChatType,
+  PaymentData,
+  PaymentParams,
+  PaymentResult,
+  PaymentResultParams
+} from '@/types';
 
 export const createPayment = (payload: PaymentParams) =>
   POST<PaymentData>('/api/payment/create', payload);
 
 export const handlePaymentResult = (payload: PaymentResultParams) =>
   POST<PaymentResult>('/api/payment/result', payload);
 
-export const checkWechatPay = () => GET<PaymentResult>('/api/payment/checkWechat');
+export const checkWechatPay = (type: CheckWeChatType) =>
+  GET<PaymentResult>('/api/payment/checkWechat', { type });
diff --git a/service/license/src/components/Pagination/index.tsx b/service/license/src/components/Pagination/index.tsx
@@ -3,7 +3,7 @@ import { useState } from 'react';
 type PaginationProps = {
   totalItems: number;
   itemsPerPage: number;
-  onPageChange: any;
+  onPageChange: (page: number) => void;
 };
 
 export default function Pagination({ totalItems, itemsPerPage, onPageChange }: PaginationProps) {

diff --git a/service/license/src/pages/api/payment/checkWechat.ts b/service/license/src/pages/api/payment/checkWechat.ts
@@ -1,12 +1,21 @@
 import { authSession } from '@/services/backend/auth';
-import { findRecentNopayOrder, updatePaymentStatus } from '@/services/backend/db/payment';
+import { createClusterAndLicense } from '@/services/backend/db/cluster';
+import { generateLicenseToken } from '@/services/backend/db/license';
+import { findRecentNopayOrder, updatePaymentAndIssueLicense } from '@/services/backend/db/payment';
 import { jsonRes } from '@/services/backend/response';
 import { getSealosPay } from '@/services/pay';
-import { PaymentResult, PaymentStatus } from '@/types';
+import {
+  CheckWeChatType,
+  ClusterType,
+  LicenseRecordPayload,
+  PaymentResult,
+  PaymentStatus
+} from '@/types';
 import type { NextApiRequest, NextApiResponse } from 'next';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
+    const { type } = req.query as { type: CheckWeChatType };
     const userInfo = await authSession(req.headers);
     if (!userInfo) {
       return jsonRes(res, { code: 401, message: 'token verify error' });
@@ -40,14 +49,42 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
       })
     }).then((res) => res.json());
 
-    const updateStatusResult = await updatePaymentStatus({
-      orderID: payment?.orderID,
-      status: result.status,
-      uid: userInfo.uid
-    });
+    if (type === 'license' && result.status === PaymentStatus.PaymentSuccess) {
+      await updatePaymentAndIssueLicense({
+        uid: userInfo.uid,
+        status: result.status,
+        amount: payment.amount,
+        quota: payment.amount,
+        orderID: payment.orderID,
+        payMethod: payment.payMethod,
+        type: 'Account'
+      });
+    }
+
+    if (type === 'cluster' && result.status === PaymentStatus.PaymentSuccess) {
+      const _token = generateLicenseToken({ type: 'Account', data: { amount: payment.amount } });
+      const record: LicenseRecordPayload = {
+        uid: userInfo.uid,
+        amount: payment.amount,
+        token: _token,
+        orderID: payment.orderID,
+        quota: payment.amount,
+        payMethod: payment.payMethod,
+        type: 'Account'
+      };
+      await createClusterAndLicense({
+        licensePayload: record,
+        clusterPayload: {
+          uid: userInfo.uid,
+          orderID: payment.orderID,
+          type: ClusterType.Enterprise
+        }
+      });
+    }
 
+    console.log('Handle wechat shutdown situation');
     return jsonRes(res, {
-      data: updateStatusResult
+      data: result
     });
   } catch (error) {
     console.error(error, '===payment error===\n');

diff --git a/service/license/src/pages/cluster/components/Record.tsx b/service/license/src/pages/cluster/components/Record.tsx
@@ -104,7 +104,13 @@ export default function ClusterRecord({ changeClusterId, clusterId }: TClusterRe
         </Flex>
       )}
       <Flex ml="auto" mt={'auto'} pr="20px">
-        <Pagination totalItems={data?.total || 0} itemsPerPage={pageSize} onPageChange={() => {}} />
+        <Pagination
+          totalItems={data?.total || 0}
+          itemsPerPage={pageSize}
+          onPageChange={(page) => {
+            setPage(page);
+          }}
+        />
       </Flex>
     </Flex>
   );

diff --git a/service/license/src/pages/license/components/Recharge.tsx b/service/license/src/pages/license/components/Recharge.tsx
@@ -43,7 +43,7 @@ export default function RechargeComponent() {
   const [wechatPaymentData, setWechatPaymentData] = useState<WechatPaymentData>();
   const { data: platformEnv } = useQuery(['getPlatformEnv'], getSystemEnv);
   // 用于避免微信支付，窗口关闭后感知不到的问题
-  // const { paymentData, setPaymentData, deletePaymentData } = usePaymentDataStore();
+  const { paymentData, setPaymentData, deletePaymentData, isExpired } = usePaymentDataStore();
 
   const onClosePayment = useCallback(() => {
     setOrderID('');
@@ -94,6 +94,7 @@ export default function RechargeComponent() {
           setOrderID(data.orderID);
           setWechatPaymentData({ tradeNO: data?.tradeNO, codeURL: data?.codeURL });
           setComplete(2);
+          setPaymentData(data.orderID);
         }
       },
       onError(err: any) {
@@ -121,6 +122,7 @@ export default function RechargeComponent() {
           position: 'top'
         });
         queryClient.invalidateQueries(['getLicenseActive']);
+        deletePaymentData();
       },
       onError(err: any) {
         toast({
@@ -156,22 +158,24 @@ export default function RechargeComponent() {
     }
   });
 
-  // checkWechatPay
-  // useQuery(['checkWechatPay'], () => checkWechatPay(), {
-  //   enabled: !!paymentData?.orderId,
-  //   onSuccess(data) {
-  //     console.log(data, '------');
-  //     if (data.status === PaymentStatus.PaymentSuccess) {
-  //       toast({
-  //         status: 'success',
-  //         title: t('Payment Successful'), // 这里改为license 签发成功
-  //         isClosable: true,
-  //         duration: 9000,
-  //         position: 'top'
-  //       });
-  //     }
-  //   }
-  // });
+  // checkWechatPay;
+  useQuery(['checkWechatPay'], () => checkWechatPay('license'), {
+    enabled: !isExpired() && !!paymentData?.orderId,
+    onSuccess(data) {
+      console.log(data, 'Handle wechat shutdown situation');
+      if (data.status === PaymentStatus.PaymentSuccess) {
+        toast({
+          status: 'success',
+          title: t('Payment Successful'), // 这里改为license 签发成功
+          isClosable: true,
+          duration: 9000,
+          position: 'top'
+        });
+        queryClient.invalidateQueries(['getLicenseActive']);
+        deletePaymentData();
+      }
+    }
+  });
 
   useEffect(() => {
     const { stripeState, orderID } = router.query;

diff --git a/service/license/src/pages/license/components/Record.tsx b/service/license/src/pages/license/components/Record.tsx
@@ -98,7 +98,11 @@ export default function History() {
         </Flex>
       )}
       <Flex ml="auto" mt={'auto'} pr="20px">
-        <Pagination totalItems={data?.total || 0} itemsPerPage={pageSize} onPageChange={() => {}} />
+        <Pagination
+          totalItems={data?.total || 0}
+          itemsPerPage={pageSize}
+          onPageChange={(page) => setPage(page)}
+        />
       </Flex>
     </Flex>
   );

diff --git a/service/license/src/pages/pricing/components/Product.tsx b/service/license/src/pages/pricing/components/Product.tsx
@@ -30,6 +30,7 @@ import { useCallback, useEffect, useState } from 'react';
 import ServicePackage from './ServicePackage';
 import useRouteParamsStore from '@/stores/routeParams';
 import useSessionStore from '@/stores/session';
+import usePaymentDataStore from '@/stores/payment';
 
 export default function Product() {
   const { t } = useTranslation();
@@ -48,6 +49,7 @@ export default function Product() {
   const [remainingSeconds, setRemainingSeconds] = useState(2); // 初始值为2秒
   const { data: routeParams, setRouteParams, clearRouteParams } = useRouteParamsStore();
   const { isUserLogin } = useSessionStore();
+  const { paymentData, setPaymentData, deletePaymentData, isExpired } = usePaymentDataStore();
 
   const onClosePayment = useCallback(() => {
     setOrderID('');
@@ -58,7 +60,7 @@ export default function Product() {
   const openPayModal = () => {
     setComplete(1);
     setPayType('wechat');
-    paymentMutation.mutate((599 * 100).toString());
+    paymentMutation.mutate((1 * 100).toString());
     onOpen();
   };
 
@@ -105,6 +107,7 @@ export default function Product() {
           setOrderID(data.orderID);
           setWechatData({ tradeNO: data?.tradeNO, codeURL: data?.codeURL });
           setComplete(2);
+          setPaymentData(data.orderID);
         }
       },
       onError(err: any) {
@@ -126,7 +129,7 @@ export default function Product() {
         console.log(data, 'clusterMutation');
         setComplete(3);
         queryClient.invalidateQueries(['getClusterRecord']);
-        // onClosePayment();
+        deletePaymentData();
       },
       onError(err: any) {
         toast({
@@ -148,6 +151,7 @@ export default function Product() {
         console.log(data, 'clusterAndLicenseMutation');
         setComplete(3);
         queryClient.invalidateQueries(['getClusterRecord']);
+        deletePaymentData();
       },
       onError(err: any) {
         toast({
@@ -184,19 +188,24 @@ export default function Product() {
   });
 
   // checkWechatPay
-  // useQuery(['checkWechatPay'], () => checkWechatPay(), {
-  //   onSuccess(data) {
-  //     if (data.status === PaymentStatus.PaymentSuccess) {
-  //       toast({
-  //         status: 'success',
-  //         title: t('License issued successfully'), // 这里改为license 签发成功
-  //         isClosable: true,
-  //         duration: 9000,
-  //         position: 'top'
-  //       });
-  //     }
-  //   }
-  // });
+  useQuery(['checkWechatPay'], () => checkWechatPay('cluster'), {
+    enabled: !isExpired() && !!paymentData?.orderId,
+    onSuccess(data) {
+      console.log(data, 'Handle wechat shutdown situation');
+      if (data.status === PaymentStatus.PaymentSuccess) {
+        toast({
+          status: 'success',
+          title: t('Payment Successful'), // 这里改为license 签发成功
+          isClosable: true,
+          duration: 9000,
+          position: 'top'
+        });
+        deletePaymentData();
+        queryClient.invalidateQueries(['getClusterRecord']);
+        setComplete(3);
+      }
+    }
+  });
 
   // handle stripe
   useEffect(() => {
@@ -273,6 +282,7 @@ export default function Product() {
     <Box flex={1} overflow={'scroll'} backgroundColor="#f2f5f7">
       <Flex
         minW={'1280px'}
+        // flexWrap={'wrap'}
         h="100%"
         pt="30px"
         pb="15px"