fix: login api toast

This commit is contained in:
zerosoul
2022-04-07 22:11:39 +08:00
parent 9b0bece22c
commit 6c022b0652
+96 -88
View File
@@ -1,100 +1,108 @@
import { fetchBaseQuery } from '@reduxjs/toolkit/query'; import { fetchBaseQuery } from "@reduxjs/toolkit/query";
import toast from 'react-hot-toast'; import toast from "react-hot-toast";
import dayjs from 'dayjs'; import dayjs from "dayjs";
import { updateToken, resetAuthData } from '../slices/auth.data'; import { updateToken, resetAuthData } from "../slices/auth.data";
import BASE_URL, { tokenHeader } from '../config'; import BASE_URL, { tokenHeader } from "../config";
const whiteList = [ const whiteList = [
'login', "login",
'register', "register",
'checkInviteTokenValid', "checkInviteTokenValid",
'getLoginConfig', "getLoginConfig",
'getServer', "getServer",
'getOpenid', "getOpenid",
'getMetamaskNonce', "getMetamaskNonce",
'renew' "renew",
]; ];
const baseQuery = fetchBaseQuery({ const baseQuery = fetchBaseQuery({
baseUrl: BASE_URL, baseUrl: BASE_URL,
prepareHeaders: (headers, { getState, endpoint }) => { prepareHeaders: (headers, { getState, endpoint }) => {
console.log('req', endpoint); console.log("req", endpoint);
const { token } = getState().authData; const { token } = getState().authData;
if (token && !whiteList.includes(endpoint)) { if (token && !whiteList.includes(endpoint)) {
headers.set(tokenHeader, token); headers.set(tokenHeader, token);
} }
return headers; return headers;
} },
}); });
let waitingForRenew = null; let waitingForRenew = null;
const baseQueryWithTokenCheck = async (args, api, extraOptions) => { const baseQueryWithTokenCheck = async (args, api, extraOptions) => {
if (waitingForRenew) { if (waitingForRenew) {
await 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);
} }
} else { // 先检查token是否过期,过期则renew
result = await baseQuery(args, api, extraOptions); const {
} token,
if (result?.error) { refreshToken,
console.log('api error', result.error, api.endpoint); expireTime = new Date().getTime(),
switch (result.error.originalStatus || result.error.status) { } = api.getState().authData;
case 'FETCH_ERROR': let result = null;
{ if (
toast.error(`${api.endpoint}: Failed to fetch`); !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; } else {
case 404: result = await baseQuery(args, api, extraOptions);
{
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;
} }
} if (result?.error) {
return result; 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; export default baseQueryWithTokenCheck;