From 6c022b06526ee3e9b5b10e4905abee9cd24bb7c0 Mon Sep 17 00:00:00 2001
From: zerosoul <yanggc888@163.com>
Date: Thu, 7 Apr 2022 22:11:39 +0800
Subject: [PATCH] fix: login api toast

---
 src/app/services/base.query.js | 184 +++++++++++++++++----------------
 1 file changed, 96 insertions(+), 88 deletions(-)

diff --git a/src/app/services/base.query.js b/src/app/services/base.query.js
index f28b0a95..5ff5480f 100644
--- a/src/app/services/base.query.js
+++ b/src/app/services/base.query.js
@@ -1,100 +1,108 @@
-import { fetchBaseQuery } from '@reduxjs/toolkit/query';
-import toast from 'react-hot-toast';
-import dayjs from 'dayjs';
-import { updateToken, resetAuthData } from '../slices/auth.data';
-import BASE_URL, { tokenHeader } from '../config';
+import { fetchBaseQuery } from "@reduxjs/toolkit/query";
+import toast from "react-hot-toast";
+import dayjs from "dayjs";
+import { updateToken, resetAuthData } from "../slices/auth.data";
+import BASE_URL, { tokenHeader } from "../config";
 const whiteList = [
- 'login',
- 'register',
- 'checkInviteTokenValid',
- 'getLoginConfig',
- 'getServer',
- 'getOpenid',
- 'getMetamaskNonce',
- 'renew'
+  "login",
+  "register",
+  "checkInviteTokenValid",
+  "getLoginConfig",
+  "getServer",
+  "getOpenid",
+  "getMetamaskNonce",
+  "renew",
 ];
 const baseQuery = fetchBaseQuery({
- baseUrl: BASE_URL,
- prepareHeaders: (headers, { getState, endpoint }) => {
-  console.log('req', endpoint);
-  const { token } = getState().authData;
-  if (token && !whiteList.includes(endpoint)) {
-   headers.set(tokenHeader, token);
-  }
-  return headers;
- }
+  baseUrl: BASE_URL,
+  prepareHeaders: (headers, { getState, endpoint }) => {
+    console.log("req", endpoint);
+    const { token } = getState().authData;
+    if (token && !whiteList.includes(endpoint)) {
+      headers.set(tokenHeader, token);
+    }
+    return headers;
+  },
 });
 let waitingForRenew = null;
 const baseQueryWithTokenCheck = async (args, api, extraOptions) => {
- if (waitingForRenew) {
-  await waitingForRenew;
- }
- // 先检查token是否过期，过期则renew
- const { token, refreshToken, expireTime = new Date().getTime() } = api.getState().authData;
- let result = null;
- if (!whiteList.includes(api.endpoint) && dayjs().isAfter(new Date(expireTime - 20 * 1000))) {
-  // 快过期了，renew
-  waitingForRenew = baseQuery(
-   {
-    url: '/token/renew',
-    method: 'POST',
-    body: {
-     token,
-     refresh_token: refreshToken
-    }
-   },
-   api,
-   extraOptions
-  );
-  result = await waitingForRenew;
-  waitingForRenew = null;
-  if (result.data) {
-   // store the new token
-   api.dispatch(updateToken(result.data));
-   // retry the initial query
-   result = await baseQuery(args, api, extraOptions);
+  if (waitingForRenew) {
+    await waitingForRenew;
   }
- } else {
-  result = await baseQuery(args, api, extraOptions);
- }
- if (result?.error) {
-  console.log('api error', result.error, api.endpoint);
-  switch (result.error.originalStatus || result.error.status) {
-   case 'FETCH_ERROR':
-    {
-     toast.error(`${api.endpoint}: Failed to fetch`);
+  // 先检查token是否过期，过期则renew
+  const {
+    token,
+    refreshToken,
+    expireTime = new Date().getTime(),
+  } = api.getState().authData;
+  let result = null;
+  if (
+    !whiteList.includes(api.endpoint) &&
+    dayjs().isAfter(new Date(expireTime - 20 * 1000))
+  ) {
+    // 快过期了，renew
+    waitingForRenew = baseQuery(
+      {
+        url: "/token/renew",
+        method: "POST",
+        body: {
+          token,
+          refresh_token: refreshToken,
+        },
+      },
+      api,
+      extraOptions
+    );
+    result = await waitingForRenew;
+    waitingForRenew = null;
+    if (result.data) {
+      // store the new token
+      api.dispatch(updateToken(result.data));
+      // retry the initial query
+      result = await baseQuery(args, api, extraOptions);
     }
-    break;
-   case 404:
-    {
-     toast.error('Request Not Found');
-    }
-    break;
-   case 500:
-    {
-     toast.error(result.error.data || 'server error');
-    }
-    break;
-   case 401:
-    {
-     // if (api.endpoint === "renew") {
-     // toast.error("token expired, please login again");
-     api.dispatch(resetAuthData());
-     location.href = '/#/login';
-     // } else {
-     toast.error('API Not Authenticated');
-     // return;
-     // }
-    }
-    break;
-   case 403:
-    toast.error('Request Not Allowed');
-    break;
-   default:
-    break;
+  } else {
+    result = await baseQuery(args, api, extraOptions);
   }
- }
- return result;
+  if (result?.error) {
+    console.log("api error", result.error, api.endpoint);
+    switch (result.error.originalStatus || result.error.status) {
+      case "FETCH_ERROR":
+        {
+          toast.error(`${api.endpoint}: Failed to fetch`);
+        }
+        break;
+      case 404:
+        {
+          toast.error("Request Not Found");
+        }
+        break;
+      case 500:
+        {
+          toast.error(result.error.data || "server error");
+        }
+        break;
+      case 401:
+        {
+          if (api.endpoint !== "login") {
+            api.dispatch(resetAuthData());
+            location.href = "/#/login";
+            toast.error("API Not Authenticated");
+          }
+          // toast.error("token expired, please login again");
+          // } else {
+          // return;
+          // }
+        }
+        break;
+      case 403:
+        toast.error("Request Not Allowed");
+        break;
+      default:
+        break;
+    }
+  }
+  return result;
 };
 
 export default baseQueryWithTokenCheck;