import { merge } from '@formily/shared';
import { useSetState } from 'ahooks';
import { default as useReq } from 'ahooks/lib/useRequest';
import { Options } from 'ahooks/lib/useRequest/src/types';
import { AxiosRequestConfig } from 'axios';
import cloneDeep from 'lodash/cloneDeep';
import { useContext } from 'react';
import { APIClientContext } from '../context';
import { assign } from './assign';

type FunctionService = (...args: any[]) => Promise<any>;

type ResourceActionOptions<P = any> = {
  resource?: string;
  resourceOf?: any;
  action?: string;
  params?: P;
};

export function useRequest<P>(
  service: AxiosRequestConfig<P> | ResourceActionOptions<P> | FunctionService,
  options: Options<any, any> & { uid?: string } = {},
) {
  // 缓存用途
  const [state, setState] = useSetState({});
  const api = useContext(APIClientContext);
  if (typeof service === 'function') {
    const result = useReq(service, {
      ...options,
      onSuccess(...args) {
        options.onSuccess?.(...args);
        if (options.uid) {
          api.services[options.uid] = result;
        }
      },
    });
    return { ...result, state, setState };
  }
  const result = useReq(
    async (params = {}) => {
      const { resource } = service as ResourceActionOptions;
      let args = cloneDeep(service);
      if (resource) {
        args.params = args.params || {};
        assign(args.params, params);
      } else {
        args = merge(args, params);
      }
      const response = await api.request(args);
      return response?.data;
    },
    {
      ...options,
      onSuccess(...args) {
        options.onSuccess?.(...args);
        if (options.uid) {
          api.services[options.uid] = result;
        }
      },
    },
  );
  return { ...result, state, setState };
}