import { buildURL } from '../helpers/buildURL'; import { combineURL } from '../helpers/combineURL'; import { CancelToken } from '../request/cancel'; import { AxiosTransformer } from '../request/transformData'; import { AxiosAdapter, AxiosAdapterRequestMethod, AxiosAdapterPlatformTask, AxiosAdapterRequestConfig, AxiosAdapterResponseData, } from '../adpater/createAdapter'; import InterceptorManager, { Interceptor } from './InterceptorManager'; import { mergeConfig } from './mergeConfig'; import AxiosDomain, { AxiosDomainRequestHandler } from './AxiosDomain'; /** * 请求方法 */ export type AxiosRequestMethod = | AxiosAdapterRequestMethod | 'options' | 'get' | 'head' | 'post' | 'put' | 'patch' | 'delete' | 'trace' | 'connect'; /** * 请求头 */ export interface AxiosRequestHeaders extends AnyObject { /** * 通用请求头 */ common?: AnyObject; /** * options 请求头 */ options?: AnyObject; /** * get 请求头 */ get?: AnyObject; /** * head 请求头 */ head?: AnyObject; /** * post 请求头 */ post?: AnyObject; /** * put 请求头 */ put?: AnyObject; /** * delete 请求头 */ delete?: AnyObject; /** * trace 请求头 */ trace?: AnyObject; /** * connect 请求头 */ connect?: AnyObject; } /** * 表单数据(上传会用到) */ export interface AxiosRequestFormData extends AnyObject { /** * 文件名 */ name: string; /** * 文件路径 */ filePath: string; } /** * 请求数据 */ export type AxiosRequestData = | string | AnyObject | ArrayBuffer | AxiosRequestFormData; /** * 响应数据 */ export type AxiosResponseData = number | AxiosAdapterResponseData; /** * 进度对象 */ export interface AxiosProgressEvent extends AnyObject { /** * 上传进度百分比 */ progress: number; } /** * 下载进度对象 */ export interface AxiosDownloadProgressEvent extends AxiosProgressEvent { /** * 已经下载的数据长度,单位 Bytes */ totalBytesWritten: number; /** * 预预期需要下载的数据总长度,单位 Bytes */ totalBytesExpectedToWrite: number; } /** * 监听下载进度 */ export interface AxiosDownloadProgressCallback { (event: AxiosDownloadProgressEvent): void; } /** * 上传进度对象 */ export interface AxiosUploadProgressEvent extends AxiosProgressEvent { /** * 已经上传的数据长度,单位 Bytes */ totalBytesSent: number; /** * 预期需要上传的数据总长度,单位 Bytes */ totalBytesExpectedToSend: number; } /** * 监听上传进度 */ export interface AxiosUploadProgressCallback { (event: AxiosUploadProgressEvent): void; } /** * 请求配置 */ export interface AxiosRequestConfig extends Partial< Omit > { /** * 请求适配器 */ adapter?: AxiosAdapter; /** * 基础路径 */ baseURL?: string; /** * 请求的 URL */ url?: string; /** * 请求参数 */ params?: AnyObject; /** * 请求数据 */ data?: AxiosRequestData; /** * 请求头 */ headers?: AxiosRequestHeaders; /** * 请求方法 */ method?: AxiosRequestMethod; /** * 取消令牌 */ cancelToken?: CancelToken; /** * 下载文件 */ download?: boolean; /** * 上传文件 */ upload?: boolean; /** * 请求参数系列化函数 */ paramsSerializer?: (params?: AnyObject) => string; /** * 校验状态码 */ validateStatus?: (status: number) => boolean; /** * 转换请求数据 */ transformRequest?: AxiosTransformer; /** * 转换响应数据 */ transformResponse?: AxiosTransformer; /** * 错误处理 */ errorHandler?: (error: unknown) => Promise; /** * 监听下载进度 */ onDownloadProgress?: AxiosUploadProgressCallback; /** * 监听上传进度 */ onUploadProgress?: AxiosUploadProgressCallback; } /** * 响应体 */ export interface AxiosResponse< TData extends AxiosResponseData = AxiosResponseData, > extends AnyObject { /** * 状态码 */ status: number; /** * 状态字符 */ statusText: string; /** * 响应头 */ headers: AnyObject; /** * 响应数据 */ data: TData; /** * 请求配置 */ config: AxiosRequestConfig; /** * 请求任务 */ request?: AxiosAdapterPlatformTask; } /** * 错误体 */ export interface AxiosResponseError extends AnyObject { /** * 状态码 */ status: number; /** * 状态字符 */ statusText: string; /** * 响应头 */ headers: AnyObject; /** * 错误数据 */ data: AnyObject; /** * 失败的请求,指没能够成功响应的请求 */ isFail: true; /** * 请求配置 */ config: AxiosRequestConfig; /** * 请求任务 */ request?: AxiosAdapterPlatformTask; } /** * Axios 构造函数 */ export interface AxiosConstructor { new (config: AxiosRequestConfig): Axios; } export default class Axios extends AxiosDomain { /** * 拦截器 */ interceptors = { /** * 请求拦截器 */ request: new InterceptorManager(), /** * 响应拦截器 */ response: new InterceptorManager(), }; constructor(defaults: AxiosRequestConfig = {}) { super(defaults, (...args) => this.#processRequest(...args)); } getUri(config: AxiosRequestConfig): string { const { url, params, paramsSerializer } = mergeConfig( this.defaults, config, ); return buildURL(url, params, paramsSerializer).replace(/^\?/, ''); } /** * 派生领域 */ fork = (config: AxiosRequestConfig = {}) => { config.baseURL = combineURL(this.defaults.baseURL, config.baseURL); const domain = new AxiosDomain( mergeConfig(this.defaults, config), (...args) => this.#processRequest(...args), ); domain.flush = this.middleware.wrap(domain.flush); return domain; }; #processRequest( config: AxiosRequestConfig, requestHandlerFn: AxiosDomainRequestHandler, ) { const requestHandler = { resolved: requestHandlerFn, }; const errorHandler = { rejected: config.errorHandler, }; const chain: ( | Partial> | Partial> )[] = []; this.interceptors.request.forEach((requestInterceptor) => { chain.unshift(requestInterceptor); }); chain.push(requestHandler); this.interceptors.response.forEach((responseInterceptor) => { chain.push(responseInterceptor); }); chain.push(errorHandler); return chain.reduce( (next, { resolved, rejected }) => next.then( // @ts-ignore resolved, rejected, ), Promise.resolve(config), ) as Promise; } }