From 8faf1a233eaaf6ac1236f372f53c05f8a79b6673 Mon Sep 17 00:00:00 2001
From: lyonbot <lyonbot@gmail.com>
Date: Sun, 19 Jan 2025 14:46:33 +0800
Subject: [PATCH] feat: drag'n'drop with directory

---
 web/src/components/ContentInput.tsx | 17 ++++++----
 web/src/utils/fileEntry.ts          | 50 +++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+), 6 deletions(-)
 create mode 100644 web/src/utils/fileEntry.ts

diff --git a/web/src/components/ContentInput.tsx b/web/src/components/ContentInput.tsx
index a053e0b..3fe256c 100644
--- a/web/src/components/ContentInput.tsx
+++ b/web/src/components/ContentInput.tsx
@@ -2,6 +2,7 @@ import { memo, useEffect, useMemo, useRef, useState } from 'react';
 import { useConsistCallback } from '../utils/useConsistCallback';
 import { isEqual } from '../utils/isEqual';
 import { createThumbnail } from '../utils/createThumbnail';
+import { getFilesFromDataTransferItem } from '../utils/fileEntry';
 
 interface Props {
   text?: string;
@@ -105,10 +106,13 @@ export const ContentInput = memo<Props>((props) => {
       e.preventDefault();
       e.stopPropagation();
 
-      const droppedFiles = Array.from(e.dataTransfer!.files);
-      if (droppedFiles.length) {
-        handleFilesChange((prev) => [...prev, ...droppedFiles]);
-      }
+      const promises = Array.from(e.dataTransfer?.items || [], getFilesFromDataTransferItem)
+      Promise.all(promises).then((files) => {
+        const droppedFiles = ([] as File[]).concat(...files);
+        if (droppedFiles.length) {
+          handleFilesChange((prev) => [...prev, ...droppedFiles]);
+        }
+      });
     };
 
     window.addEventListener('drop', handleDrop, true);
@@ -184,7 +188,7 @@ export const ContentInput = memo<Props>((props) => {
           Add file
         </button>
 
-        {files.map((file, index) => <UploadedFileItem file={file} index={index} removeFile={removeFile} key={index} />)}
+        {files.map((file, index) => <AttachedFileItem file={file} index={index} removeFile={removeFile} key={index} />)}
 
         <button onClick={doClear} className='btn-gray' key='clearBtn'>
           Clear
@@ -209,7 +213,7 @@ declare global {
   }
 }
 
-function UploadedFileItem({ index, file, removeFile }: { index: number; file: File; removeFile: (index: number) => void; }) {
+function AttachedFileItem({ index, file, removeFile }: { index: number; file: File; removeFile: (index: number) => void; }) {
   const [url, setUrl] = useState<string>();
   useEffect(() => {
     const url = URL.createObjectURL(file);
@@ -250,3 +254,4 @@ export async function addThumbnail(file: File) {
 
   return true;
 }
+
diff --git a/web/src/utils/fileEntry.ts b/web/src/utils/fileEntry.ts
new file mode 100644
index 0000000..2383d95
--- /dev/null
+++ b/web/src/utils/fileEntry.ts
@@ -0,0 +1,50 @@
+export async function getFilesFromDataTransferItem(item: DataTransferItem) {
+  const entry = item.webkitGetAsEntry?.();
+  const allFiles: File[] = [];
+  if (entry?.isDirectory) {
+    const dir = entry as FileSystemDirectoryEntry;
+    await readDirectory(dir, allFiles, `${dir.name}/`);
+    return allFiles;
+  }
+
+  if (entry?.isFile) {
+    const file = await getFile(entry as FileSystemFileEntry);
+    allFiles.push(file);
+    return allFiles;
+  }
+
+  return [];
+}
+
+function readDirectory(directory: FileSystemDirectoryEntry, allFiles: File[], pathPrefix: string): Promise<void> {
+  return new Promise((resolve) => {
+    const reader = directory.createReader();
+    reader.readEntries(async (entries) => {
+      for (const entry of entries) {
+        if (entry.isDirectory) {
+          await readDirectory(entry as FileSystemDirectoryEntry, allFiles, `${pathPrefix}${entry.name}/`);
+        } else if (entry.isFile) {
+          const originalFile = await getFile(entry as FileSystemFileEntry);
+          const file = new File(
+            [originalFile],
+            `${pathPrefix}${entry.name}`,
+            {
+              type: originalFile.type,
+              lastModified: originalFile.lastModified,
+            }
+          );
+          allFiles.push(file);
+        }
+      }
+      resolve();
+    });
+  });
+}
+
+function getFile(fileEntry: FileSystemFileEntry): Promise<File> {
+  return new Promise((resolve, reject) => {
+    fileEntry.file((file) => {
+      resolve(file);
+    }, reject);
+  });
+}
\ No newline at end of file