ragflow/web/src/utils/registerServer.ts

import { RequestMethod } from 'umi-request';

type Service<T extends string> = Record<T, (params: any) => any>;

const registerServer = <T extends string>(
  opt: Record<T, { url: string; method: string }>,
  request: RequestMethod,
) => {
  const server: Service<T> = {} as Service<T>;
  for (let key in opt) {
    server[key] = (params: any, urlAppendix?: string) => {
      let url = opt[key].url;
      if (urlAppendix) {
        url = url + '/' + urlAppendix;
      }
      if (opt[key].method === 'post' || opt[key].method === 'POST') {
        return request(url, {
          method: opt[key].method,
          data: params,
        });
      }

      if (opt[key].method === 'get' || opt[key].method === 'GET') {
        return request.get(url, {
          params,
        });
      }
    };
  }
  return server;
};

export default registerServer;