68 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
			
		
		
	
	
			68 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
import { request } from "@cqsjjb/jjb-common-lib/http";
 | 
						|
import { useState } from "react";
 | 
						|
 | 
						|
/**
 | 
						|
 * 上传文件 TODO
 | 
						|
 */
 | 
						|
function useUploadFile() {
 | 
						|
  // loading状态
 | 
						|
  const [loading, setLoading] = useState(false);
 | 
						|
 | 
						|
  // 上传文件
 | 
						|
  const uploadFile = (options) => {
 | 
						|
    setLoading(true);
 | 
						|
 | 
						|
    return new Promise((resolve, reject) => {
 | 
						|
      const { files = [], single = true, params = {} } = options;
 | 
						|
      const formData = new FormData();
 | 
						|
 | 
						|
      // 将文件添加到formData中
 | 
						|
      files.forEach((f) => {
 | 
						|
        f.originFileObj && formData.append("files", f.originFileObj);
 | 
						|
      });
 | 
						|
 | 
						|
      // 将额外携带的参数添加到formData中
 | 
						|
      Object.keys(params).forEach((key) => {
 | 
						|
        formData.append(key, params[key]);
 | 
						|
      });
 | 
						|
 | 
						|
      // 获取待上传的文件
 | 
						|
      const filesLength = formData.getAll("files").length;
 | 
						|
 | 
						|
      // 如果没有文件则返回老文件
 | 
						|
      if (filesLength === 0) {
 | 
						|
        setLoading(false);
 | 
						|
        resolve(
 | 
						|
          single
 | 
						|
            ? files?.[0]?.url || ""
 | 
						|
            : { files: files.map(f => f.url), id: "" },
 | 
						|
        );
 | 
						|
        return;
 | 
						|
      }
 | 
						|
 | 
						|
      // 发送请求
 | 
						|
      request(single ? "单文件上传的接口" : "多文件上传的接口", "post", formData, { "Content-Type": "multipart/form-data" })
 | 
						|
        .then((res) => {
 | 
						|
          resolve(
 | 
						|
            single
 | 
						|
              ? res.data.url
 | 
						|
              : {
 | 
						|
                  files: [...res.data.files.map(f => f.url), ...files.filter(f => !f.originFileObj).map(f => f.url)],
 | 
						|
                  id: res.data.id,
 | 
						|
                },
 | 
						|
          );
 | 
						|
        })
 | 
						|
        .catch((err) => {
 | 
						|
          reject(err);
 | 
						|
        })
 | 
						|
        .finally(() => {
 | 
						|
          setLoading(false);
 | 
						|
        });
 | 
						|
    });
 | 
						|
  };
 | 
						|
 | 
						|
  return [loading, uploadFile];
 | 
						|
}
 | 
						|
 | 
						|
export default useUploadFile;
 |