zy-react-library/hooks/useGetFile/index.js

import { request } from "@cqsjjb/jjb-common-lib/http";
import { useState } from "react";
import { UPLOAD_FILE_TYPE_ENUM } from "../../enum/uploadFile/gwj";
import { addingPrefixToFile } from "../../utils";

/**
 * 获取文件
 */
function useGetFile(returnType = "object") {
  // loading状态
  const [loading, setLoading] = useState(false);

  // 获取文件
  const getFile = (options) => {
    if (!options)
      throw new Error("请传入 options");

    setLoading(true);

    return new Promise((resolve, reject) => {
      const { eqType, eqForeignKey } = options;

      if (!eqType)
        throw new Error("请传入 options.eqType");

      // 检查eqType是否在UPLOAD_FILE_TYPE_ENUM中
      if (!Object.values(UPLOAD_FILE_TYPE_ENUM).includes(eqType))
        throw new Error("传入的 eqType 不在 UPLOAD_FILE_TYPE_ENUM 中");

      if (eqForeignKey === undefined || eqForeignKey === null)
        throw new Error("请传入 options.eqForeignKey");

      // 发送请求
      request(
        "/basic-info/imgFiles/listAll",
        "get",
        { eqType, eqForeignKey },
      )
        .then((res) => {
          resolve(addingPrefixToFile(res.data).map(item => ({ ...item, type: item.type.toString() })));
        })
        .catch((err) => {
          reject(err);
        })
        .finally(() => {
          setLoading(false);
        });
    });
  };

  if (returnType === "array")
    return [loading, getFile];
  return { loading, getFile };
}

export default useGetFile;