douyin/app/src/utils/axiosHttp.ts

import axios, { AxiosInstance, AxiosRequestConfig, Method as _Method, AxiosResponse } from 'axios'; // 移除 AxiosProgressEvent
import qs from 'qs';
import Cookie from 'js-cookie';

declare interface _AxiosExtend {
  /**
   * 发起请求
   * @param url 请求地址
   * @param method 请求方法
   * @param params 请求参数
   * @param config 请求配置（新增上传进度回调）
   */
  request<T = any, R = AxiosResponse<T>>(
    url: string,
    method: Method,
    params?: Record<string | number, any> | FormData, // 支持 FormData 类型
    config?: AxiosRequestConfig & { onUploadProgress?: (progressEvent: ProgressEvent) => void } // 改用原生 ProgressEvent
  ): Promise<R>;
  /**
   * 设置token
   * @param value token值
   * @param expires 过期时间
   * - 类型为 number 时，表示 expires 毫秒后 token 过期
   * - 类型为 Date 时，表示在 expires 这个时间点 token 过期
   * @param name token 名称，默认为当前 http 实例的 xsrfCookieName 属性值
   */
  setAuthorization(value: string, expires: number | Date, name?: string): void;

  /**
   * 移出token
   * @param name token 名称， 默认为当前 http 实例的 xsrfCookieName 属性值
   */
  removeAuthorization(name?: string): void;
  /**
   * 校验 token 是否有效
   * @param name 需要校验的 token 名称，默认为当前 http 实例的 xsrfCookieName 属性值
   */
  checkAuthorization(name?: string): boolean;
}

export interface AxiosHttp extends Omit<AxiosInstance, 'request'>, _AxiosExtend { }

// 新增 post_form / POST_FORM 类型
export type Method = _Method | 'POST_JSON' | 'post_json' | 'PUT_JSON' | 'put_json' | 'POST_FORM' | 'post_form';

/**
 * 转表单格式
 * @param params
 * @returns
 */
export function toFormData(params?: Record<string | number, any>) {
  const formData = new FormData();
  if (!params) {
    return formData;
  }
  Object.entries(params).forEach(([key, value]) => {
    if (Array.isArray(value)) {
      value.forEach((val) => {
        formData.append(key, val);
      });
    } else {
      formData.set(key, value);
    }
  });
  return formData;
}

function toUrlencoded(params?: Record<string | number, any>) {
  const urlencoded = new URLSearchParams();
  for (const key in params) {
    if (params[key] !== undefined) {
      urlencoded.append(key, params[key]);
    }
  }
  return urlencoded;
}

/**
 * 创建 axios http
 * @param config
 * @returns
 */
function createAxiosHttp(config: AxiosRequestConfig): AxiosHttp {
  const _axios = axios.create(config);

  // 添加响应拦截器处理401状态
  _axios.interceptors.response.use(
    // 成功响应直接返回
    response => response,
    // 错误响应处理
    error => {
      // 检查是否是401未授权错误
      if (error.response && error.response.status === 401) {
        // 调用removeAuthorization方法清除token
        http.removeAuthorization();

        // 这里可以添加额外的处理，比如跳转到登录页
        // 示例: window.location.href = '/login';
      }
      return Promise.reject(error);
    }
  );

  const http: AxiosHttp = {
    ..._axios,
    request<T = any, R = AxiosResponse<T>>(
      url: string,
      method: Method,
      params?: Record<string | number, any> | FormData,
      config?: AxiosRequestConfig & { onUploadProgress?: (progressEvent: ProgressEvent) => void } // 改用原生 ProgressEvent
    ): Promise<R> {
      const _method = method.toUpperCase();
      // 处理上传进度配置
      const requestConfig: AxiosRequestConfig = {
        ...config,
        onUploadProgress: config?.onUploadProgress, // 透传上传进度回调
      };

      switch (_method) {
        case 'GET':
          return _axios.get(url, {
            params,
            paramsSerializer: (data) => {
              return qs.stringify(data, { indices: false, skipNulls: true });
            },
            ...requestConfig,
          });
        case 'POST':
          return _axios.post(url, toUrlencoded(params as Record<string | number, any>), requestConfig);
        case 'POST_JSON':
          return _axios.post(url, params, {
            ...requestConfig,
            headers: { 'Content-Type': 'application/json', ...requestConfig.headers },
          });
        // 新增：POST_FORM 类型（适配文件上传的 FormData）
        case 'POST_FORM':
          return _axios.post(url, params, {
            ...requestConfig,
            // FormData 不需要手动设置 Content-Type，axios 会自动处理为 multipart/form-data
            headers: { ...requestConfig.headers },
          });
        case 'PUT':
          return _axios.put(url, toFormData(params as Record<string | number, any>), requestConfig);
        case 'PUT_JSON':
          return _axios.put(url, params, {
            ...requestConfig,
            headers: { 'Content-Type': 'application/json', ...requestConfig.headers },
          });
        case 'DELETE':
          return _axios.delete(url, { data: toFormData(params as Record<string | number, any>), ...requestConfig });
        case 'HEAD':
          return _axios.head(url, { params, ...requestConfig });
        case 'OPTIONS':
          return _axios.options(url, { params, ...requestConfig });
        case 'PATCH':
          return _axios.patch(url, { params, ...requestConfig });
        case 'PURGE':
        case 'LINK':
        case 'UNLINK':
          const m = _method as _Method;
          return _axios.request({ url, method: m, params, ..._axios.defaults });
        default:
          return _axios.request({ url, method: 'GET', params, ..._axios.defaults });
      }
    },
    setAuthorization(token: string, expires: number | Date, name?: string): void {
      Cookie.set(name ?? _axios.defaults.xsrfCookieName!, token, { expires });
    },
    removeAuthorization(name?: string): void {
      Cookie.remove(name ?? _axios.defaults.xsrfCookieName!);
    },
    checkAuthorization(name?: string | undefined): boolean {
      return Boolean(Cookie.get(name ?? _axios.defaults.xsrfCookieName!));
    },
  };

  return http;
}

export default createAxiosHttp;