From ffcc21975eedcff5b66f9905e0f305c4ef1457f3 Mon Sep 17 00:00:00 2001
From: dap <15891557205@163.com>
Date: Sat, 29 Mar 2025 15:52:11 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E6=96=87=E4=BB=B6/=E5=9B=BE?=
 =?UTF-8?q?=E7=89=87=E4=B8=8A=E4=BC=A0=E9=87=8D=E6=9E=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/web-antd/src/api/core/upload.ts          | 40 ++++----
 .../src/components/upload/src/file-upload.vue | 73 ++++-----------
 .../src/components/upload/src/hook.ts         | 92 ++++++++++---------
 .../components/upload/src/image-upload.vue    | 66 ++-----------
 .../src/components/upload/src/note.md         |  6 +-
 .../src/components/upload/src/props.d.ts      | 13 ++-
 .../views/演示使用自行删除/upload/index.vue   | 12 ++-
 7 files changed, 130 insertions(+), 172 deletions(-)

diff --git a/apps/web-antd/src/api/core/upload.ts b/apps/web-antd/src/api/core/upload.ts
index a74cb19f..d7270b0e 100644
--- a/apps/web-antd/src/api/core/upload.ts
+++ b/apps/web-antd/src/api/core/upload.ts
@@ -7,22 +7,6 @@ import { requestClient } from '#/api/request';
  */
 export type AxiosProgressEvent = AxiosRequestConfig['onUploadProgress'];
 
-/**
- * 通过单文件上传接口
- * @param file 上传的文件
- * @param onUploadProgress 上传进度事件 非必传
- * @returns 上传结果
- */
-export function uploadApi(
-  file: Blob | File,
-  onUploadProgress?: AxiosProgressEvent,
-) {
-  return requestClient.upload(
-    '/resource/oss/upload',
-    { file },
-    { onUploadProgress, timeout: 60_000 },
-  );
-}
 /**
  * 默认上传结果
  */
@@ -31,3 +15,27 @@ export interface UploadResult {
   fileName: string;
   ossId: string;
 }
+
+/**
+ * 通过单文件上传接口
+ * @param file 上传的文件
+ * @param otherData 其他请求参数 后端拓展可能会用到
+ * @param onUploadProgress 上传进度事件 非必传
+ * @returns 上传结果
+ */
+export function uploadApi(
+  file: Blob | File,
+  otherData?: Record<string, any>,
+  onUploadProgress?: AxiosProgressEvent,
+) {
+  return requestClient.upload<UploadResult>(
+    '/resource/oss/upload',
+    { file, ...otherData },
+    { onUploadProgress, timeout: 60_000 },
+  );
+}
+
+/**
+ * 上传api type
+ */
+export type UploadApi = typeof uploadApi;
diff --git a/apps/web-antd/src/components/upload/src/file-upload.vue b/apps/web-antd/src/components/upload/src/file-upload.vue
index 4e7f787f..04f946c5 100644
--- a/apps/web-antd/src/components/upload/src/file-upload.vue
+++ b/apps/web-antd/src/components/upload/src/file-upload.vue
@@ -5,6 +5,8 @@
 <script setup lang="ts">
 import type { UploadFile } from 'ant-design-vue';
 
+import type { BaseUploadProps } from './props';
+
 import { computed } from 'vue';
 
 import { $t, I18nT } from '@vben/locales';
@@ -12,68 +14,26 @@ import { $t, I18nT } from '@vben/locales';
 import { InboxOutlined, UploadOutlined } from '@ant-design/icons-vue';
 import { Upload } from 'ant-design-vue';
 
+import { uploadApi } from '#/api';
+
 import { defaultFileAcceptExts, defaultFilePreview } from './helper';
 import { useUpload } from './hook';
 
-interface Props {
-  /**
-   * 文件上传失败 是否从展示列表中删除
-   * @default true
-   */
-  removeOnError?: boolean;
-  /**
-   * 上传成功 是否展示提示信息
-   * @default true
-   */
-  showSuccessMsg?: boolean;
-  /**
-   * 删除文件前是否需要确认
-   * @default false
-   */
-  removeConfirm?: boolean;
-  /**
-   * 同antdv参数
-   */
-  accept?: string;
-  /**
-   * 附带的请求参数
-   */
-  data?: any;
-  /**
-   * 最大上传图片数量
-   * maxCount为1时 会被绑定为string而非string[]
-   * @default 1
-   */
-  maxCount?: number;
-  /**
-   * 文件最大 单位M
-   * @default 5
-   */
-  maxSize?: number;
-  /**
-   * 是否禁用
-   * @default false
-   */
-  disabled?: boolean;
-  /**
-   * 是否显示文案 请上传不超过...
-   * @default true
-   */
-  helpMessage?: boolean;
-  /**
-   * 是否支持多选文件，ie10+ 支持。开启后按住 ctrl 可选择多个文件。
-   * @default false
-   */
-  multiple?: boolean;
+interface FileUploadProps extends BaseUploadProps {
   /**
    * 自定义文件预览逻辑 比如: 你可以改为下载
    * @param file file
    */
   preview?: (file: UploadFile) => Promise<void> | void;
+  /**
+   * 是否支持拖拽上传
+   * @default false
+   */
   enableDragUpload?: boolean;
 }
 
-const props = withDefaults(defineProps<Props>(), {
+const props = withDefaults(defineProps<FileUploadProps>(), {
+  api: () => uploadApi,
   removeOnError: true,
   showSuccessMsg: true,
   removeConfirm: false,
@@ -84,8 +44,11 @@ const props = withDefaults(defineProps<Props>(), {
   disabled: false,
   helpMessage: true,
   preview: defaultFilePreview,
+  enableDragUpload: false,
+  directory: false,
 });
 
+/** 返回不同的上传组件 */
 const CurrentUploadComponent = computed(() => {
   if (props.enableDragUpload) {
     return Upload.Dragger;
@@ -99,8 +62,7 @@ const ossIdList = defineModel<string | string[]>('value', {
 });
 
 const {
-  uploadUrl,
-  headers,
+  customRequest,
   acceptFormat,
   handleChange,
   handleRemove,
@@ -113,15 +75,14 @@ const {
   <div>
     <CurrentUploadComponent
       v-model:file-list="innerFileList"
-      :action="uploadUrl"
-      :headers="headers"
-      :data="data"
       :accept="accept"
       :disabled="disabled"
+      :directory="directory"
       :max-count="maxCount"
       :progress="{ showInfo: true }"
       :multiple="multiple"
       :before-upload="beforeUpload"
+      :custom-request="customRequest"
       @preview="preview"
       @change="handleChange"
       @remove="handleRemove"
diff --git a/apps/web-antd/src/components/upload/src/hook.ts b/apps/web-antd/src/components/upload/src/hook.ts
index 8d9500f9..6ac9f840 100644
--- a/apps/web-antd/src/components/upload/src/hook.ts
+++ b/apps/web-antd/src/components/upload/src/hook.ts
@@ -1,21 +1,21 @@
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
 import type { UploadChangeParam, UploadFile } from 'ant-design-vue';
 import type { FileType } from 'ant-design-vue/es/upload/interface';
+import type { UploadRequestOption } from 'ant-design-vue/es/vc-upload/interface';
 
 import type { ModelRef } from 'vue';
 
-import type { HttpResponse } from '@vben/request';
+import type { BaseUploadProps } from './props';
 
-import type { UploadResult } from '#/api';
+import type { AxiosProgressEvent, UploadResult } from '#/api';
 import type { OssFile } from '#/api/system/oss/model';
 
 import { computed, ref, watch } from 'vue';
 
-import { useAppConfig } from '@vben/hooks';
 import { $t } from '@vben/locales';
-import { useAccessStore } from '@vben/stores';
 
 import { message, Modal } from 'ant-design-vue';
+import { isFunction } from 'lodash-es';
 
 import { ossInfo } from '#/api/system/oss';
 
@@ -85,29 +85,9 @@ export function useUpload(
   props: Readonly<BaseUploadProps>,
   bindValue: ModelRef<string | string[]>,
 ) {
-  const { apiURL, clientId } = useAppConfig(
-    import.meta.env,
-    import.meta.env.PROD,
-  );
-
-  // 内部维护fileList
+  // 组件内部维护fileList
   const innerFileList = ref<UploadFile[]>([]);
 
-  /** oss上传地址 */
-  const uploadUrl = `${apiURL}/resource/oss/upload`;
-
-  const accessStore = useAccessStore();
-
-  /**
-   * header参数 需要带上token和clientId
-   */
-  const headers = computed(() => {
-    return {
-      Authorization: `Bearer ${accessStore.accessToken}`,
-      clientId,
-    };
-  });
-
   const acceptFormat = computed(() => {
     return props.accept
       ?.split(',')
@@ -121,6 +101,11 @@ export function useUpload(
   });
 
   function handleChange(info: UploadChangeParam) {
+    /**
+     * 移除当前文件
+     * @param currentFile 当前文件
+     * @param currentFileList 当前所有文件list
+     */
     function removeCurrentFile(
       currentFile: UploadChangeParam['file'],
       currentFileList: UploadChangeParam['fileList'],
@@ -140,19 +125,9 @@ export function useUpload(
         if (!currentFile.response) {
           return;
         }
-        // 获取返回结果
-        const response = currentFile.response as HttpResponse<UploadResult>;
-        // 上传异常
-        if (response.code !== 200) {
-          message.error(response.msg);
-          removeCurrentFile(currentFile, fileList);
-          return;
-        }
-        // 上传成功 做转换
-        if (props.showSuccessMsg) {
-          message.success($t('component.upload.uploadSuccess'));
-        }
-        const { ossId, fileName, url } = response.data;
+        // 获取返回结果 为customRequest的reslove参数
+        // 只有success才会走到这里
+        const { ossId, fileName, url } = currentFile.response as UploadResult;
         currentFile.url = url;
         currentFile.fileName = fileName;
         currentFile.uid = ossId;
@@ -166,7 +141,6 @@ export function useUpload(
       }
       // 上传失败 网络原因导致httpStatus 不等于200
       case 'error': {
-        message.error($t('component.upload.uploadError'));
         removeCurrentFile(currentFile, fileList);
       }
     }
@@ -207,6 +181,12 @@ export function useUpload(
     });
   }
 
+  /**
+   * 上传前检测文件大小
+   * 拖拽时候前置会有浏览器自身的accept校验 校验失败不会执行此方法
+   * @param file file
+   * @returns file | false
+   */
   function beforeUpload(file: FileType) {
     const isLtMax = file.size / 1024 / 1024 < props.maxSize!;
     if (!isLtMax) {
@@ -217,6 +197,37 @@ export function useUpload(
     return file;
   }
 
+  /**
+   * 自定义上传实现
+   * @param info
+   */
+  async function customRequest(info: UploadRequestOption<any>) {
+    const { api } = props;
+    if (!isFunction(api)) {
+      console.warn('upload api must exist and be a function');
+      return;
+    }
+    try {
+      // 进度条事件
+      const progressEvent: AxiosProgressEvent = (e) => {
+        const percent = Math.trunc((e.loaded / e.total!) * 100);
+        info.onProgress!({ percent });
+      };
+      const res = await api(
+        info.file as File,
+        props?.data ?? {},
+        progressEvent,
+      );
+      info.onSuccess!(res);
+      if (props.showSuccessMsg) {
+        message.success($t('component.upload.uploadSuccess'));
+      }
+    } catch (error: any) {
+      console.error(error);
+      info.onError!(error);
+    }
+  }
+
   /**
    * 这里只监听list地址变化 即重新赋值才会触发watch
    * immediate用于初始化触发
@@ -244,11 +255,10 @@ export function useUpload(
   );
 
   return {
-    uploadUrl,
-    headers,
     handleChange,
     handleRemove,
     beforeUpload,
+    customRequest,
     innerFileList,
     acceptFormat,
   };
diff --git a/apps/web-antd/src/components/upload/src/image-upload.vue b/apps/web-antd/src/components/upload/src/image-upload.vue
index 058f4ec9..b81f9a9a 100644
--- a/apps/web-antd/src/components/upload/src/image-upload.vue
+++ b/apps/web-antd/src/components/upload/src/image-upload.vue
@@ -5,72 +5,28 @@
 <script setup lang="ts">
 import type { UploadListType } from 'ant-design-vue/es/upload/interface';
 
+import type { BaseUploadProps } from './props';
+
 import { $t, I18nT } from '@vben/locales';
 
 import { PlusOutlined } from '@ant-design/icons-vue';
 import { Image, ImagePreviewGroup, Upload } from 'ant-design-vue';
 
+import { uploadApi } from '#/api';
+
 import { defaultImageAcceptExts } from './helper';
 import { useImagePreview, useUpload } from './hook';
 
-interface Props {
-  /**
-   * 文件上传失败 是否从展示列表中删除
-   * @default true
-   */
-  removeOnError?: boolean;
-  /**
-   * 上传成功 是否展示提示信息
-   * @default true
-   */
-  showSuccessMsg?: boolean;
-  /**
-   * 删除文件前是否需要确认
-   * @default false
-   */
-  removeConfirm?: boolean;
-  /**
-   * 同antdv参数
-   */
-  accept?: string;
-  /**
-   * 附带的请求参数
-   */
-  data?: any;
-  /**
-   * 最大上传图片数量
-   * maxCount为1时 会被绑定为string而非string[]
-   * @default 1
-   */
-  maxCount?: number;
-  /**
-   * 文件最大 单位M
-   * @default 5
-   */
-  maxSize?: number;
-  /**
-   * 是否禁用
-   * @default false
-   */
-  disabled?: boolean;
+interface ImageUploadProps extends BaseUploadProps {
   /**
    * 同antdv的listType
    * @default picture-card
    */
   listType?: UploadListType;
-  /**
-   * 是否显示文案 请上传不超过...
-   * @default true
-   */
-  helpMessage?: boolean;
-  /**
-   * 是否支持多选文件，ie10+ 支持。开启后按住 ctrl 可选择多个文件。
-   * @default false
-   */
-  multiple?: boolean;
 }
 
-const props = withDefaults(defineProps<Props>(), {
+const props = withDefaults(defineProps<ImageUploadProps>(), {
+  api: () => uploadApi,
   removeOnError: true,
   showSuccessMsg: true,
   removeConfirm: false,
@@ -89,13 +45,12 @@ const ossIdList = defineModel<string | string[]>('value', {
 });
 
 const {
-  uploadUrl,
-  headers,
   acceptFormat,
   handleChange,
   handleRemove,
   beforeUpload,
   innerFileList,
+  customRequest,
 } = useUpload(props, ossIdList);
 
 const { previewVisible, previewImage, handleCancel, handlePreview } =
@@ -107,15 +62,14 @@ const { previewVisible, previewImage, handleCancel, handlePreview } =
     <Upload
       v-model:file-list="innerFileList"
       :list-type="listType"
-      :action="uploadUrl"
-      :headers="headers"
-      :data="data"
       :accept="accept"
       :disabled="disabled"
+      :directory="directory"
       :max-count="maxCount"
       :progress="{ showInfo: true }"
       :multiple="multiple"
       :before-upload="beforeUpload"
+      :custom-request="customRequest"
       @preview="handlePreview"
       @change="handleChange"
       @remove="handleRemove"
diff --git a/apps/web-antd/src/components/upload/src/note.md b/apps/web-antd/src/components/upload/src/note.md
index 23dd610e..261cc450 100644
--- a/apps/web-antd/src/components/upload/src/note.md
+++ b/apps/web-antd/src/components/upload/src/note.md
@@ -19,4 +19,8 @@ return new Promise<FileType>((resolve) =>
 );
 ```
 
-根本原因在于`file-typ`库可能不支持Safari 去掉可以正常上传
+根本原因在于`file-typ`库的`fileTypeFromBlob`方法不支持Safari 去掉可以正常上传
+
+safari不支持`ReadableStreamBYOBReader`api
+
+详见: https://github.com/sindresorhus/file-type/issues/690
diff --git a/apps/web-antd/src/components/upload/src/props.d.ts b/apps/web-antd/src/components/upload/src/props.d.ts
index eba5bcd6..5ae0e509 100644
--- a/apps/web-antd/src/components/upload/src/props.d.ts
+++ b/apps/web-antd/src/components/upload/src/props.d.ts
@@ -1,4 +1,10 @@
-interface BaseUploadProps {
+import type { UploadApi } from '#/api';
+
+export interface BaseUploadProps {
+  /**
+   * 上传接口
+   */
+  api?: UploadApi;
   /**
    * 文件上传失败 是否从展示列表中删除
    * @default true
@@ -48,4 +54,9 @@ interface BaseUploadProps {
    * @default false
    */
   multiple?: boolean;
+  /**
+   * 是否支持上传文件夹
+   * @default false
+   */
+  directory?: boolean;
 }
diff --git a/apps/web-antd/src/views/演示使用自行删除/upload/index.vue b/apps/web-antd/src/views/演示使用自行删除/upload/index.vue
index a4745113..d7f5f006 100644
--- a/apps/web-antd/src/views/演示使用自行删除/upload/index.vue
+++ b/apps/web-antd/src/views/演示使用自行删除/upload/index.vue
@@ -9,7 +9,7 @@ import { Alert, Card, Modal } from 'ant-design-vue';
 
 import { FileUpload, ImageUpload } from '#/components/upload';
 
-const singleImageId = ref('1905537674682916865');
+const singleImageId = ref('');
 const singleFileId = ref('1905191167882518529');
 const multipleImageId = ref<string[]>(['1905537674682916865']);
 const multipleFileId = ref<string[]>(['1905191167882518529']);
@@ -74,6 +74,16 @@ function handlePreview(file: UploadFile) {
       <Card title="文件禁用上传" size="small">
         <FileUpload :disabled="true" :max-count="3" :help-message="false" />
       </Card>
+
+      <Card title="文件夹上传" size="small">
+        <FileUpload
+          v-model:value="multipleFileId"
+          :max-count="3"
+          :directory="true"
+          accept="*"
+        />
+        当前绑定值: {{ multipleFileId }}
+      </Card>
     </div>
   </Page>
 </template>