ai-agent-m/src/utils/request/http.ts

import { merge } from "lodash-es";
import { isFunction } from "@vue/shared";
import {
  HttpRequestOptions,
  RequestConfig,
  RequestEventStreamConfig,
  RequestOptions,
  UploadFileOption,
} from "./type";
import { RequestErrMsgEnum, RequestMethodsEnum } from "@/enums/requestEnums";
import requestCancel from "./cancel";
import { objectToQuery } from "../util";
import { isObject } from "mathjs";
import { Base64 } from "js-base64";
import { parse } from "jsonc-parser";

/**
 * @description 将流转换为utf-8
 */
function decodeUTF8(arr: Uint8Array) {
  let str = "";
  for (let i = 0; i < arr.length; i++) {
    str += String.fromCharCode(arr[i]);
  }
  return decodeURIComponent(escape(str));
}

function isStreamResponse(header?: any) {
  const contentType =
    header?.["Content-Type"] ||
    header?.["content-type"] ||
    header.get("content-type")!;
  if (typeof contentType !== "string") return false;
  return contentType.includes("text/event-stream");
}

export default class HttpRequest {
  private readonly options: HttpRequestOptions;
  constructor(options: HttpRequestOptions) {
    this.options = options;
  }
  /**
   * @description 重新请求
   */
  retryRequest(options: RequestOptions, config: RequestConfig) {
    const { retryCount, retryTimeout } = config;
    if (
      !retryCount ||
      options.method?.toUpperCase() == RequestMethodsEnum.POST
    ) {
      return Promise.reject();
    }
    uni.showLoading({ title: "加载中..." });
    config.hasRetryCount = config.hasRetryCount || 0;
    if (config.hasRetryCount >= retryCount) {
      return Promise.reject();
    }
    config.hasRetryCount++;
    config.requestHooks.requestInterceptorsHook = (options) => options;
    return new Promise((resolve) => setTimeout(resolve, retryTimeout))
      .then(() => this.request(options, config))
      .finally(() => uni.hideLoading());
  }
  /**
   * @description get请求
   */
  get<T = any>(
    options: RequestOptions,
    config?: Partial<RequestConfig>
  ): Promise<T> {
    return this.request({ ...options, method: RequestMethodsEnum.GET }, config);
  }

  /**
   * @description post请求
   */
  post<T = any>(
    options: RequestOptions,
    config?: Partial<RequestConfig>
  ): Promise<T> {
    return this.request(
      { ...options, method: RequestMethodsEnum.POST },
      config
    );
  }
  /**
   * @description 事件流
   */
  eventStream(options: RequestOptions, config?: RequestEventStreamConfig) {
    let mergeOptions: RequestOptions = merge(
      {},
      this.options.requestOptions,
      options
    );
    const mergeConfig: RequestEventStreamConfig = merge(
      {},
      this.options,
      config
    );
    const { requestInterceptorsHook, responseInterceptorsHook } =
      mergeConfig.requestHooks || {};
    if (requestInterceptorsHook && isFunction(requestInterceptorsHook)) {
      mergeOptions = requestInterceptorsHook(
        mergeOptions,
        mergeConfig as RequestConfig
      );
    }

    const { onmessage, onclose, onstart } = mergeConfig;
    return new Promise((resolve, reject) => {
      try {
        // #ifdef H5
        const decoder = new TextDecoder();
        const abortController = new AbortController();
        const push = async (controller: any, reader: any) => {
          try {
            const { value, done } = await reader.read();
            if (done) {
              controller.close();
              onclose?.();
            } else {
              const tempval = Base64.decode(decoder.decode(value));
              onmessage?.(tempval);
              controller.enqueue(tempval);
              push(controller, reader);
            }
          } catch (error) {
            onclose?.();
          }
        };

        if (mergeOptions.method!.toUpperCase() == RequestMethodsEnum.GET) {
          if (isObject(mergeOptions.data)) {
            mergeOptions.data = objectToQuery(
              mergeOptions.data as Record<string, any>
            );
          }
          mergeOptions.url = `${mergeOptions.url}?${mergeOptions.data}`;
        }
        let body = undefined;
        if (mergeOptions.method!.toUpperCase() == RequestMethodsEnum.POST) {
          body = isObject(mergeOptions.data)
            ? JSON.stringify(mergeOptions.data)
            : (mergeOptions.data as string);
        }
        fetch(mergeOptions.url, {
          ...mergeOptions,
          signal: abortController.signal,
          body,
          headers: {
            "content-type": "application/json; charset=utf-8",
            ...mergeOptions.header,
            Accept: "text/event-stream",
          },
        })
          .then(async (response) => {
            if (response.status == 200) {
              if (isStreamResponse(response.headers)) {
                const reader = response.body!.getReader();
                onstart?.(abortController);
                new ReadableStream({
                  start(controller) {
                    push(controller, reader);
                  },
                });
              } else {
                //@ts-ignore
                response.data = await response.json();
                return response;
              }
            } else {
              reject(response.statusText);
            }
          })
          .then(async (response: any) => {
            if (!response) {
              resolve(response);
              return;
            }
            if (
              responseInterceptorsHook &&
              isFunction(responseInterceptorsHook)
            ) {
              try {
                response = await responseInterceptorsHook(
                  response,
                  mergeConfig as RequestConfig,
                  mergeOptions
                );
                resolve(response);
              } catch (error) {
                reject(error);
              }
              return;
            }
            resolve(response);
          })
          .catch((error) => {
            reject(error);
          });
        // #endif

        // #ifdef MP-WEIXIN
        let header: Record<string, any> = {};
        const requestTask = uni.request({
          ...mergeOptions,
          enableChunked: true,
          responseType: "arraybuffer",
          async success(response) {
            if (response.statusCode !== 200) {
              reject(response);
              return;
            }
            resolve(response);
          },
          fail(error) {
            reject(error);
          },
          complete() {
            onclose?.();
          },
        });
        onstart?.(requestTask);
        //@ts-ignore
        requestTask.onHeadersReceived((response) => {
          header = response.header;
        });
        //@ts-ignore
        requestTask.onChunkReceived(async (response) => {
          const arrayBuffer = response.data;
          const uint8Array = new Uint8Array(arrayBuffer);
          const base64str = decodeUTF8(uint8Array);
          const str = Base64.decode(base64str);
          if (isStreamResponse(header)) {
            console.log(str);
            onmessage?.(str);
          } else {
            const data = parse(str);
            if (
              responseInterceptorsHook &&
              isFunction(responseInterceptorsHook)
            ) {
              try {
                const response = await responseInterceptorsHook(
                  { data, statusCode: 200 },
                  mergeConfig as RequestConfig,
                  mergeOptions
                );
                resolve(response);
              } catch (error) {
                reject(error);
              }
              return;
            }
          }
        });
        // #endif
      } catch (e) {
        reject();
      }
    });
  }
  /**
   * @description 上传图片
   */
  uploadFile(options: UploadFileOption, config?: Partial<RequestConfig>) {
    let mergeOptions: RequestOptions = merge(
      {},
      this.options.requestOptions,
      options
    );
    const mergeConfig: RequestConfig = merge({}, this.options, config);
    const {
      requestInterceptorsHook,
      responseInterceptorsHook,
      responseInterceptorsCatchHook,
    } = mergeConfig.requestHooks || {};
    if (requestInterceptorsHook && isFunction(requestInterceptorsHook)) {
      mergeOptions = requestInterceptorsHook(mergeOptions, mergeConfig);
    }
    return new Promise((resolve, reject) => {
      const task = uni.uploadFile({
        ...mergeOptions,
        header: {
          ...mergeOptions.header,
        },
        async success(response) {
          if (response.statusCode == 200) {
            response.data = parse(response.data);
            if (
              responseInterceptorsHook &&
              isFunction(responseInterceptorsHook)
            ) {
              try {
                response = await responseInterceptorsHook(
                  response,
                  mergeConfig,
                  mergeOptions
                );
                resolve(response);
              } catch (error) {
                reject(error);
              }
              return;
            }

            resolve(response);
          } else {
            reject(response.errMsg);
          }
        },
        fail(err) {
          if (err.errMsg == RequestErrMsgEnum.ABORT) {
            return;
          }
          if (
            responseInterceptorsCatchHook &&
            isFunction(responseInterceptorsCatchHook)
          ) {
            reject(responseInterceptorsCatchHook(mergeOptions, err));
            return;
          }
          reject(err);
        },
      });
      task.onProgressUpdate(({ progress }) => {
        mergeConfig.onProgress?.(progress);
      });
    });
  }
  /**
   * @description 请求函数
   */
  async request(
    options: RequestOptions,
    config?: Partial<RequestConfig>
  ): Promise<any> {
    let mergeOptions: RequestOptions = merge(
      {},
      this.options.requestOptions,
      options
    );
    const mergeConfig: RequestConfig = merge({}, this.options, config);
    const {
      requestInterceptorsHook,
      responseInterceptorsHook,
      responseInterceptorsCatchHook,
    } = mergeConfig.requestHooks || {};
    if (requestInterceptorsHook && isFunction(requestInterceptorsHook)) {
      mergeOptions = requestInterceptorsHook(mergeOptions, mergeConfig);
    }
    return new Promise((resolve, reject) => {
      const requestTask = uni.request({
        ...mergeOptions,
        async success(response) {
          if (response.statusCode !== 200) {
            return reject();
          }
          if (
            responseInterceptorsHook &&
            isFunction(responseInterceptorsHook)
          ) {
            try {
              response = await responseInterceptorsHook(
                response,
                mergeConfig,
                mergeOptions
              );
              resolve(response);
            } catch (error) {
              reject(error);
            }
            return;
          }
          resolve(response);
        },
        fail: async (err) => {
          if (err.errMsg == RequestErrMsgEnum.TIMEOUT) {
            this.retryRequest(mergeOptions, mergeConfig)
              .then((res) => resolve(res))
              .catch((err) => reject(err));
            return;
          }
          if (err.errMsg == RequestErrMsgEnum.ABORT) {
            return;
          }
          if (
            responseInterceptorsCatchHook &&
            isFunction(responseInterceptorsCatchHook)
          ) {
            reject(await responseInterceptorsCatchHook(mergeOptions, err));
            return;
          }
          reject(err);
        },
        complete(err) {
          if (err.errMsg !== RequestErrMsgEnum.ABORT) {
            requestCancel.remove(options.url);
          }
        },
      });
      const { ignoreCancel } = mergeConfig;
      !ignoreCancel && requestCancel.add(options.url, requestTask);
    });
  }
}