Refactor: 퍼센테이지 복구

frontend/src/components/ImagePreSignedForm/index.tsx

@@ -5,7 +5,6 @@ import useAuthStore from '@/stores/useAuthStore';
 import { CircleCheckBig, CircleDashed, CircleX, X } from 'lucide-react';
 import { FixedSizeList } from 'react-window';
 import useUploadFiles from '@/hooks/useUploadFiles';
-import useUploadImagePresignedQuery from '@/queries/images/useUploadImagePresignedQuery';
 import { unzipFilesWithPath, extractFilesRecursivelyWithPath } from '@/utils/fileUtils';
 
 interface ImagePreSignedFormProps {
@@ -34,20 +33,16 @@ export default function ImagePreSignedForm({
   const [isUploading, setIsUploading] = useState<boolean>(false);
   const [isUploaded, setIsUploaded] = useState<boolean>(false);
   const [isFailed, setIsFailed] = useState<boolean>(false);
-  const [progress, setProgress] = useState<number>(0);
+  const [uploadStatus, setUploadStatus] = useState<('uploading' | 'success' | 'failed' | null)[]>([]);
-  const [uploadStatus, setUploadStatus] = useState<(boolean | null)[]>([]);
 
-  // Ensure to destructure the uploadFiles function properly from the hook
   const { uploadFiles } = useUploadFiles();
-  const uploadImageFile = useUploadImagePresignedQuery();
+
   const handleClose = () => {
     onClose();
     setFiles([]);
     setInputKey((prevKey) => prevKey + 1);
-    setIsUploading(false);
-    setIsUploaded(false);
     setIsFailed(false);
-    setProgress(0);
+    setIsUploaded(false);
     setUploadStatus([]);
   };
 
@@ -92,9 +87,10 @@ export default function ImagePreSignedForm({
     event.stopPropagation();
     setIsDragging(false);
 
+    let processedFiles: { path: string; file: File }[] = [];
+
     if (uploadType === 'folder') {
       const droppedItems = event.dataTransfer.items;
-      let processedFiles: { path: string; file: File }[] = [];
 
       for (let i = 0; i < droppedItems.length; i++) {
         const item = droppedItems[i];
@@ -106,20 +102,17 @@ export default function ImagePreSignedForm({
           }
         }
       }
-
-      setFiles((prevFiles) => [...prevFiles, ...processedFiles]);
-      setUploadStatus((prevStatus) => [...prevStatus, ...processedFiles.map(() => null)]);
     } else {
       const droppedFiles = event.dataTransfer.files;
       if (droppedFiles) {
-        const processedFiles: { path: string; file: File }[] = [];
         for (const file of Array.from(droppedFiles)) {
           processedFiles.push({ path: file.name, file });
         }
-        setFiles((prevFiles) => [...prevFiles, ...processedFiles]);
-        setUploadStatus((prevStatus) => [...prevStatus, ...processedFiles.map(() => null)]);
       }
     }
+
+    setFiles((prevFiles) => [...prevFiles, ...processedFiles]);
+    setUploadStatus((prevStatus) => [...prevStatus, ...processedFiles.map(() => null)]);
   };
 
   const handleRemoveFile = (index: number) => {
@@ -130,72 +123,52 @@ export default function ImagePreSignedForm({
   const handleUpload = async () => {
     if (files.length > 0) {
       setIsUploading(true);
-      setIsUploaded(false);
       setIsFailed(false);
+      setIsUploaded(false);
+
+      setUploadStatus(files.map(() => 'uploading'));
 
       let finalFiles: { path: string; file: File }[] = [];
 
       for (const file of files) {
         if (file.file.type === 'application/zip' || file.file.type === 'application/x-zip-compressed') {
-          console.log('업로드 전에 ZIP 파일 해제:', file.file.name);
           const unzippedFiles = await unzipFilesWithPath(file.file);
-          console.log('해제된 파일:', unzippedFiles);
           finalFiles = [...finalFiles, ...unzippedFiles];
         } else {
           finalFiles.push(file);
         }
       }
 
-      if (uploadType === 'file') {
+      try {
-        uploadImageFile.mutate(
+        await uploadFiles({
-          {
+          files: finalFiles,
-            memberId,
+          projectId,
-            projectId,
+          folderId,
-            folderId,
+          memberId,
-            files: finalFiles.map(({ file }) => file), // Extract only the file
+          onProgress: (progress) => {
-            progressCallback: (index: number) => {
+            setUploadStatus((prevStatus) => {
-              setUploadStatus((prevStatus) => {
+              const completedFiles = Math.round((progress / 100) * files.length);
-                const newStatus = [...prevStatus];
+              const newStatus = prevStatus.map((status, index) => (index < completedFiles ? 'success' : status));
-                newStatus[index] = true; // Mark as uploaded
+              return newStatus;
-                return newStatus;
+            });
-              });
-            },
           },
-          {
+          useSingleUpload: uploadType === 'file',
-            onSuccess: () => {
+        });
-              handleRefetch();
-              setIsUploaded(true);
-            },
-            onError: () => {
-              setIsFailed(true);
-              setUploadStatus((prevStatus) => prevStatus.map((status) => (status === null ? false : status)));
-            },
-          }
-        );
-      } else {
-        try {
-          await uploadFiles({
-            files: finalFiles,
-            projectId,
-            folderId,
-            memberId,
-            onProgress: (progressValue: number) => {
-              setProgress(progressValue);
-            },
-          });
 
-          setUploadStatus(finalFiles.map(() => true));
+        setUploadStatus((prevStatus) => prevStatus.map(() => 'success'));
-          setIsUploaded(true);
+        setIsUploaded(true);
-          handleRefetch();
+        handleRefetch();
-        } catch (error) {
+      } catch (error) {
-          setIsFailed(true);
+        setUploadStatus((prevStatus) => prevStatus.map((status) => (status === 'uploading' ? 'failed' : status)));
-          setUploadStatus(finalFiles.map(() => false));
+        setIsFailed(true);
-          console.error('업로드 실패:', error);
+        console.error('업로드 실패:', error);
-        }
       }
     }
   };
 
+  // 전체 진행 상황 계산
+  const totalProgress = Math.round((uploadStatus.filter((status) => status !== null).length / files.length) * 100);
+
   useEffect(() => {
     onFileCount(files.length);
   }, [files, onFileCount]);
@@ -237,46 +210,44 @@ export default function ImagePreSignedForm({
         </div>
       )}
       {files.length > 0 && (
-        <ul className="m-0 max-h-[260px] list-none overflow-y-auto p-0">
+        <FixedSizeList
-          <FixedSizeList
+          height={260}
-            height={260}
+          itemCount={files.length}
-            itemCount={files.length}
+          itemSize={40}
-            itemSize={40}
+          width="100%"
-            width="100%"
+        >
-          >
+          {({ index, style }) => (
-            {({ index, style }) => (
+            <div
-              <li
+              key={index}
-                key={index}
+              className="flex items-center justify-between border-b border-gray-200 p-2"
-                className="flex items-center justify-between p-1"
+              style={style}
-                style={style}
+            >
-              >
+              <span className="truncate">{files[index].path}</span>
-                <span className="truncate">{files[index].path}</span>
+              <div className="flex items-center">
-                {isUploading ? (
+                {uploadStatus[index] === 'success' ? (
-                  <div className="p-2">
+                  <CircleCheckBig
-                    {uploadStatus[index] === true ? (
+                    className="stroke-green-500"
-                      <CircleCheckBig
+                    size={16}
-                        className="stroke-green-500"
+                    strokeWidth="2"
-                        size={16}
+                  />
-                        strokeWidth="2"
+                ) : uploadStatus[index] === 'failed' ? (
-                      />
+                  <CircleX
-                    ) : uploadStatus[index] === false ? (
+                    className="stroke-red-500"
-                      <CircleX
+                    size={16}
-                        className="stroke-red-500"
+                    strokeWidth="2"
-                        size={16}
+                  />
-                        strokeWidth="2"
+                ) : uploadStatus[index] === 'uploading' ? (
-                      />
+                  <CircleDashed
-                    ) : (
+                    className="animate-spin stroke-gray-500"
-                      <CircleDashed
+                    size={16}
-                        className="stroke-gray-500"
+                    strokeWidth="2"
-                        size={16}
+                  />
-                        strokeWidth="2"
+                ) : null}
-                      />
+                {!isUploading && (
-                    )}
-                  </div>
-                ) : (
                   <button
-                    className="cursor-pointer p-2"
+                    className="ml-2 cursor-pointer p-1"
                     onClick={() => handleRemoveFile(index)}
+                    disabled={uploadStatus[index] === 'success'}
                   >
                     <X
                       color="red"
@@ -285,18 +256,23 @@ export default function ImagePreSignedForm({
                     />
                   </button>
                 )}
-              </li>
+              </div>
-            )}
+            </div>
-          </FixedSizeList>
+          )}
-        </ul>
+        </FixedSizeList>
       )}
       {isUploading ? (
         <Button
           onClick={handleClose}
           variant={isFailed ? 'red' : 'blue'}
-          disabled={!isUploaded && !isFailed}
         >
-          {isFailed ? '업로드 실패 (닫기)' : isUploaded ? '업로드 완료 (닫기)' : `업로드 중... ${progress}%`}
+          {isFailed
+            ? '업로드 실패 (닫기)'
+            : isUploaded
+              ? '업로드 완료 (닫기)'
+              : totalProgress === 0
+                ? '업로드 준비 중...'
+                : `업로드 중... ${totalProgress}%`}
         </Button>
       ) : (
         <Button

frontend/src/hooks/useUploadFiles.ts

@@ -11,55 +11,74 @@ export default function useUploadFiles() {
     folderId,
     memberId,
     onProgress,
+    useSingleUpload = false,
   }: {
     files: { path: string; file: File }[];
     projectId: number;
     folderId: number;
     memberId: number;
     onProgress: (progress: number) => void;
+    useSingleUpload?: boolean;
   }) => {
     const folderIdMap: { [path: string]: number } = { '': folderId };
 
-    const foldersToCreate = Array.from(new Set(files.map(({ path }) => path.split('/').slice(0, -1).join('/'))));
+    if (!useSingleUpload) {
-    foldersToCreate.sort();
+      const foldersToCreate = Array.from(new Set(files.map(({ path }) => path.split('/').slice(0, -1).join('/'))));
+      foldersToCreate.sort();
 
-    for (const folderPath of foldersToCreate) {
+      for (const folderPath of foldersToCreate) {
-      if (folderPath) {
+        if (folderPath) {
-        const pathSegments = folderPath.split('/');
+          const pathSegments = folderPath.split('/');
-        const parentPath = pathSegments.slice(0, -1).join('/');
+          const parentPath = pathSegments.slice(0, -1).join('/');
-        const folderName = pathSegments[pathSegments.length - 1];
+          const folderName = pathSegments[pathSegments.length - 1];
 
-        const parentId = folderIdMap[parentPath] || folderId;
+          const parentId = folderIdMap[parentPath] || folderId;
 
-        const newFolder = await createFolderMutation.mutateAsync({
+          const newFolder = await createFolderMutation.mutateAsync({
-          projectId,
+            projectId,
-          folderData: {
+            folderData: {
-            title: folderName,
+              title: folderName,
-            parentId: parentId,
+              parentId: parentId,
-          },
+            },
-        });
+          });
 
-        folderIdMap[folderPath] = newFolder.id;
+          folderIdMap[folderPath] = newFolder.id;
+        }
       }
     }
 
-    let progress = 0;
+    let completedFiles = 0;
     const totalFiles = files.length;
 
-    for (const { path, file } of files) {
+    if (useSingleUpload) {
-      const folderPath = path.split('/').slice(0, -1).join('/');
-      const targetFolderId = folderIdMap[folderPath] || folderId;
-
       await uploadImageMutation.mutateAsync({
         memberId,
         projectId,
-        folderId: targetFolderId,
+        folderId,
-        files: [file],
+        files: files.map(({ file }) => file),
-        progressCallback: (value) => {
+        progressCallback: (progressValue: number) => {
-          progress += value / totalFiles;
+          const progress = (progressValue / totalFiles) * 100;
           onProgress(Math.round(progress));
         },
       });
+    } else {
+      for (const { path, file } of files) {
+        const folderPath = path.split('/').slice(0, -1).join('/');
+        const targetFolderId = folderIdMap[folderPath] || folderId;
+
+        await uploadImageMutation.mutateAsync({
+          memberId,
+          projectId,
+          folderId: targetFolderId,
+          files: [file],
+          progressCallback: (progressValue: number) => {
+            const progress = ((completedFiles + progressValue / 100) / totalFiles) * 100;
+            onProgress(Math.round(progress));
+          },
+        });
+
+        completedFiles += 1;
+      }
     }
   };