ColdBreeze-chat-web/src/app/services/base.query.ts

import toast from "react-hot-toast";
import { fetchBaseQuery } from "@reduxjs/toolkit/query";
import dayjs from "dayjs";

import { getLocalAuthData } from "@/utils";
import BASE_URL, { tokenHeader, IS_OFFICIAL_DEMO } from "../config";
import { resetAuthData, updateToken } from "../slices/auth.data";

const whiteList = [
  "guestLogin",
  "login",
  "register",
  "sendLoginMagicLink",
  "sendRegMagicLink",
  "checkEmail",
  "checkMagicTokenValid",
  "getGoogleAuthPublicConfig",
  "getGithubAuthPublicConfig",
  "getSMTPStatus",
  "getLoginConfig",
  "getServerVersion",
  "getServer",
  "getOpenid",
  "getMetamaskNonce",
  "renew",
  "getInitialized",
  "createAdmin",
  "getBotRelatedChannels",
  "sendMessageByBot",
  "getAgoraVoicingList",
  "preCheckFileFromUrl",
  "passkeyLoginStart",
  "passkeyLoginFinish",
];
const whiteList401 = [
  "getAgoraVoicingList",
  "getAgoraChannels",
  "getGoogleAuthConfig",
  "getGithubAuthConfig",
];
const errorWhiteList = [
  "preCheckFileFromUrl",
  "getFavoriteDetails",
  "getOGInfo",
  "getArchiveMessage",
  "getGoogleAuthPublicConfig",
  "getGithubAuthPublicConfig",
  "getGoogleAuthConfig",
  "getGithubAuthConfig",
];
const whiteList404 = [
  "login",
  "getArchiveMessage",
  "preCheckFileFromUrl",
  "deleteMessage",
  "deleteMessages",
  "getWidgetExtCSS",
];
const baseQuery = fetchBaseQuery({
  baseUrl: BASE_URL,
  prepareHeaders: (headers, { endpoint }) => {
    const { token } = getLocalAuthData();
    if (IS_OFFICIAL_DEMO && "crypto" in window) {
      const uuid = window.crypto.randomUUID();
      headers.set("request_uuid", uuid);
    }
    // const { token } = (getState() as RootState).authData;
    if (token && !whiteList.includes(endpoint)) {
      headers.set(tokenHeader, token);
    }
    return headers;
  },
});

let waitingForRenew: null | any = null;
const baseQueryWithTokenCheck = async (args: any, api: any, extraOptions: any) => {
  if (waitingForRenew) {
    await waitingForRenew;
  }
  // 先检查 token 是否过期，过期则 renew [从 localstorage 取]
  const { token, refreshToken, expireTime } = getLocalAuthData();

  let result = null;
  // console.log("base check", whiteList.includes(api.endpoint), api.endpoint);
  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 {
    result = await baseQuery(args, api, extraOptions);
  }
  if (result?.error) {
    console.error("api error", result.error, args, api.endpoint);
    if (errorWhiteList.includes(api.endpoint)) return result;
    switch (result.error.originalStatus || result.error.status) {
      case "FETCH_ERROR":
        {
          toast.error(`${api.endpoint}: Failed to fetch`);
        }
        break;
      case 400:
        toast.error("Bad Request");
        break;
      case 401:
        {
          if (whiteList401.includes(api.endpoint)) {
            return;
          }
          if (api.endpoint !== "login") {
            api.dispatch(resetAuthData());
            location.href = "/#/login";
            // toast.error("API Not Authenticated");
          }
        }
        break;
      case 403:
        {
          const whiteList403 = ["sendMsg", "login"];
          if (!whiteList403.includes(api.endpoint)) {
            toast.error("Request Not Allowed");
          }
        }

        break;
      case 404:
        {
          if (!whiteList404.includes(api.endpoint)) {
            toast.error("Request Not Found");
          }
        }
        break;
      case 413:
        {
          toast.error("File size too large");
        }
        break;
      case 415:
        {
          toast.error("Unsupported Media Type");
        }
        break;
      case 451:
        {
          // 证书错误
          if (api.endpoint !== "login") {
            // 退出登录
            api.dispatch(resetAuthData());
            location.href = "/#/login";
          }
          toast.error(result.error.data || "License Error");
        }
        break;
      case 500:
      case 503:
        {
          toast.error(result.error.data || "Server Error");
        }
        break;

      default:
        break;
    }
  }
  return result;
};

export default baseQueryWithTokenCheck;