feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/agent-ide/chat-background-shared/src/hooks/use-upload-img.ts

@@ -0,0 +1,107 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+ 
+import { useEffect, useRef } from 'react';
+
+import { nanoid } from 'nanoid';
+import { withSlardarIdButton } from '@coze-studio/bot-utils';
+import { I18n } from '@coze-arch/i18n';
+import { Toast } from '@coze-arch/bot-semi';
+
+// import { type UploadState } from '../type';
+
+import { UploadController } from '../service/upload-controller';
+
+export const useUploadImage = ({
+  getUserId,
+  onUploadError,
+  onUploadAllSuccess,
+  onAuditError,
+}: {
+  getUserId: () => string;
+  onUploadAllSuccess: (param: { url: string; uri: string }[]) => void;
+  onUploadError: () => void;
+  onAuditError?: () => void;
+}) => {
+  const uploadControllerMap = useRef<Record<string, UploadController>>({});
+
+  const deleteUploadControllerById = (id: string) => {
+    delete uploadControllerMap.current[id];
+  };
+  const cancelUploadById = (id: string) => {
+    const controller = uploadControllerMap.current[id];
+    if (!controller) {
+      return;
+    }
+    controller.cancel();
+    deleteUploadControllerById(id);
+  };
+
+  const handleError = (_e: unknown, controllerId: string) => {
+    cancelUploadById(controllerId);
+    onUploadError();
+    Toast.error({
+      content: withSlardarIdButton(I18n.t('Upload_failed')),
+      showClose: false,
+    });
+  };
+
+  const onAuditFailed = () => {
+    if (onAuditError) {
+      onAuditError();
+    } else {
+      Toast.error({
+        content: I18n.t('inappropriate_contents'),
+        showClose: false,
+      });
+    }
+    onUploadError();
+  };
+
+  const uploadFileList = (fileList: File[]) => {
+    const controllerId = nanoid();
+
+    if (!fileList.length) {
+      return;
+    }
+
+    uploadControllerMap.current[controllerId] = new UploadController({
+      fileList,
+      controllerId,
+      userId: getUserId(),
+      onComplete: event => {
+        onUploadAllSuccess(event);
+      },
+      onUploadError: handleError,
+      onGetTokenError: handleError,
+      onGetUploadInstanceError: handleError,
+      onAuditFailed,
+    });
+  };
+
+  const clearAllSideEffect = () => {
+    Object.entries(uploadControllerMap.current).forEach(([, controller]) =>
+      controller.cancel(),
+    );
+    uploadControllerMap.current = {};
+  };
+
+  useEffect(() => clearAllSideEffect, []);
+
+  return {
+    uploadFileList,
+  };
+};