@hzab/data-model 1.8.9-alpha.1 → 2.0.0-alpha.0

package/src/axios.ts

@@ -0,0 +1,210 @@
+import axiosDef, { AxiosResponse, InternalAxiosRequestConfig } from "axios";
+import Cookies from "js-cookie";
+import _ from "lodash";
+
+export const TOKEN = "token";
+
+const axios = axiosDef.create();
+
+/**
+ * setToken 选项接口
+ */
+export interface SetTokenOptions {
+  /** 是否同时设置 cookie */
+  hasCookie?: boolean;
+  /** cookie 属性配置 */
+  cookieAttrs?: Cookies.CookieAttributes;
+}
+
+/**
+ * getToken 选项接口
+ */
+export interface GetTokenOptions {
+  /** 是否从 cookie 获取 */
+  hasCookie?: boolean;
+}
+
+/**
+ * 处理错误回调函数的选项接口
+ */
+export interface HandleErrCbOptions {
+  /** 直接替换回调函数 */
+  replaceErrCb?: boolean;
+  /** 日志前缀 */
+  logPrefix?: string;
+}
+
+/**
+ * 设置 token
+ * @param token token 值
+ * @param opt 选项
+ */
+export function setToken(token = "", opt?: SetTokenOptions) {
+  const { hasCookie = false } = opt || {};
+  const _t = token || "";
+  if (hasCookie) {
+    const { cookieAttrs } = opt || {};
+    const cookieOpt = _.merge(
+      {
+        expires: 365,
+      },
+      cookieAttrs,
+    );
+    Cookies.set(TOKEN, _t, cookieOpt);
+  }
+  localStorage.setItem(TOKEN, _t);
+  setAxAuthorization(_t);
+}
+
+/**
+ * 获取 token
+ * @param opt 选项
+ * @returns token 值
+ */
+export function getToken(opt?: GetTokenOptions): string {
+  const { hasCookie = false } = opt || {};
+  let token = "";
+  if (hasCookie) {
+    token = Cookies.get(TOKEN) || "";
+    if (token) {
+      return token;
+    }
+  }
+  token = localStorage.getItem(TOKEN);
+  return token;
+}
+
+export const temp = { axiosList: [axios, axiosDef] };
+
+/**
+ * 设置 axios 列表，用于批量设置 axios 相关配置
+ * @param axiosList axios 实例列表
+ * @returns 更新后的 axios 实例列表
+ */
+export function setAxios(axiosList: (typeof axios | typeof axiosDef)[]) {
+  temp.axiosList = [...temp.axiosList, ...axiosList];
+  return temp.axiosList;
+}
+
+/**
+ * axios 请求拦截器
+ * @param cb 成功回调
+ * @param errCb 错误回调
+ * @param opt 选项
+ */
+export function setAxRequest(
+  cb: (config: InternalAxiosRequestConfig) => InternalAxiosRequestConfig | Promise<InternalAxiosRequestConfig>,
+  errCb?: (error: any) => Promise<any>,
+  opt?: HandleErrCbOptions,
+) {
+  temp.axiosList?.forEach((_ax) => {
+    _ax.interceptors.request.use(cb, handleErrCb(errCb, { logPrefix: "Error axios response: ", ...opt }));
+  });
+}
+
+/**
+ * 响应拦截器
+ * @param cb 成功回调
+ * @param errCb 错误回调
+ * @param opt 选项
+ */
+export function setAxResponse(
+  cb: (response: AxiosResponse) => AxiosResponse | Promise<AxiosResponse>,
+  errCb?: (error: any) => Promise<any>,
+  opt?: HandleErrCbOptions,
+) {
+  temp.axiosList?.forEach((_ax) => {
+    _ax.interceptors.response.use(cb, handleErrCb(errCb, { logPrefix: "Error axios response: ", ...opt }));
+  });
+}
+
+/**
+ * 处理错误回调函数
+ * @param errCb 错误回调
+ * @param opt 选项
+ * @returns 处理后的错误回调
+ */
+export function handleErrCb(errCb?: (error: any) => Promise<any>, opt?: HandleErrCbOptions) {
+  const { replaceErrCb, logPrefix } = opt || {};
+  const isFn = typeof errCb === "function";
+
+  if (replaceErrCb && isFn) {
+    return errCb;
+  }
+
+  return (error: any) => {
+    if (axiosDef.isCancel(error)) {
+      console.info("axios 请求已取消:", error);
+      return Promise.reject(error);
+    }
+    console.error(logPrefix, error);
+    return isFn ? errCb(error) : Promise.reject(error);
+  };
+}
+
+/**
+ * 设置 axios.defaults
+ * @param key 配置键
+ * @param val 配置值
+ */
+export function setAxDefaults(key: string, val: any) {
+  temp.axiosList?.forEach((_ax) => {
+    (_ax.defaults as any)[key] = val;
+  });
+}
+
+/**
+ * 设置 headers
+ * @param key header 键
+ * @param val header 值
+ */
+export function setAxHeaders(key: string, val: any) {
+  temp.axiosList?.forEach((_ax) => {
+    (_ax.defaults.headers as any)[key] = val;
+  });
+}
+
+/**
+ * 设置 timeout
+ * @param timeout 超时时间
+ */
+export function setAxTimeout(timeout = 0) {
+  setAxDefaults("timeout", timeout);
+}
+
+/**
+ * 设置 Authorization token
+ * @param token token 值
+ */
+export function setAxAuthorization(token: string) {
+  setAxHeaders("Authorization", token);
+}
+
+/**
+ * 设置 baseUrl
+ * @param url 基础 URL
+ */
+export function setAxBaseUrl(url: string) {
+  setAxDefaults("baseURL", url);
+}
+
+/**
+ * 获取 axios.CancelToken.source，用于取消请求
+ * @returns CancelToken source
+ */
+export function getCancelTokenSource() {
+  return axiosDef.CancelToken.source();
+}
+
+/**
+ * 检查是否是 cancel 的结果
+ * @param res
+ * @returns
+ */
+export function isCancel(res) {
+  return axiosDef.isCancel(res);
+}
+
+export { axios, axiosDef };
+
+export default axios;