From a0e6f03db0e27fcfa9b63e07ecde48f81874f80e Mon Sep 17 00:00:00 2001
From: yuanyxh <15766118362@139.com>
Date: Tue, 21 May 2024 16:30:52 +0800
Subject: [PATCH] feat: add export file

---
 src/assets/svgs/bxs--file-export.svg      |  1 +
 src/filehandle/components/FileContent.tsx | 26 +++++++++++++++-
 src/filehandle/utils/fileManager.ts       | 38 ++++++++++++++++++-----
 types/filesystem.d.ts                     |  2 +-
 4 files changed, 58 insertions(+), 9 deletions(-)
 create mode 100644 src/assets/svgs/bxs--file-export.svg

diff --git a/src/assets/svgs/bxs--file-export.svg b/src/assets/svgs/bxs--file-export.svg
new file mode 100644
index 0000000..72a4578
--- /dev/null
+++ b/src/assets/svgs/bxs--file-export.svg
@@ -0,0 +1 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="1em" height="1em" viewBox="0 0 24 24"><path fill="currentColor" d="M18 22a2 2 0 0 0 2-2v-5l-5 4v-3H8v-2h7v-3l5 4V8l-6-6H6a2 2 0 0 0-2 2v16a2 2 0 0 0 2 2zM13 4l5 5h-5z"/></svg>
\ No newline at end of file
diff --git a/src/filehandle/components/FileContent.tsx b/src/filehandle/components/FileContent.tsx
index 8f991f2..18ced19 100644
--- a/src/filehandle/components/FileContent.tsx
+++ b/src/filehandle/components/FileContent.tsx
@@ -13,7 +13,7 @@ import AddFileModal from './AddFileModal';
 import { FileSystemContext } from './FilePanel';
 import styles from './styles/FileContent.module.less';
 import type { FileInfo } from '../utils/fileManager';
-import { FileType } from '../utils/fileManager';
+import { exportDirectory, exportFile, FileType } from '../utils/fileManager';
 import { fileOperationWarning } from '../utils/operationWarning';
 
 function MountWebdavModal(props: { open: boolean; close(): void }) {
@@ -270,6 +270,22 @@ const FileContent: React.FC<IFileContentProps> = (props) => {
     });
   };
 
+  const handleExportFile = async () => {
+    const curr = selection[0];
+
+    if (!curr) return void 0;
+
+    try {
+      await (curr.handle.kind === 'directory'
+        ? exportDirectory(curr.handle)
+        : exportFile(curr.handle));
+
+      success(`已为你导出文件 ${curr.name}。`);
+    } catch (err) {
+      error((err as Error).message);
+    }
+  };
+
   const handleOk = async (name: string, type: FileType) => {
     try {
       await create(name, type);
@@ -334,6 +350,14 @@ const FileContent: React.FC<IFileContentProps> = (props) => {
               icon: <Icon icon="ri--import-line" color="var(--color-primary)" />,
               onClick: importDirectory
             },
+            {
+              name: '导出文件',
+              icon: <Icon icon="bxs--file-export" color="var(--color-primary)" />,
+              style: {
+                display: selection.length ? void 0 : 'none'
+              },
+              onClick: handleExportFile
+            },
             {
               name: '删除',
               icon: <Icon icon="material-symbols--delete" color="var(--color-primary)" />,
diff --git a/src/filehandle/utils/fileManager.ts b/src/filehandle/utils/fileManager.ts
index 15bedbf..a472071 100644
--- a/src/filehandle/utils/fileManager.ts
+++ b/src/filehandle/utils/fileManager.ts
@@ -69,8 +69,6 @@ export async function getHandleType(handle: DH | FH) {
 }
 
 export async function importFile(directory: DH, options?: FilePickerOptions) {
-  if (!window.showOpenFilePicker) return false;
-
   const files = await window.showOpenFilePicker(options);
 
   await Promise.all(
@@ -81,8 +79,6 @@ export async function importFile(directory: DH, options?: FilePickerOptions) {
 }
 
 export async function importDirectory(directory: DH, options?: DirectoryPickerOptions) {
-  if (!window.showDirectoryPicker) return false;
-
   const folder = await window.showDirectoryPicker(options);
 
   await move(folder, await createDirectory(directory, folder.name));
@@ -90,6 +86,25 @@ export async function importDirectory(directory: DH, options?: DirectoryPickerOp
   return true;
 }
 
+export async function exportFile(file: FH) {
+  const handle = await window.showSaveFilePicker({
+    suggestedName: file.name,
+    types: [{ description: 'export file', accept: {} }]
+  });
+
+  await writeFile(handle, await file.getFile());
+
+  return true;
+}
+
+export async function exportDirectory(directory: DH) {
+  const handle = await window.showDirectoryPicker({ mode: 'readwrite' });
+
+  await moveDirectoryWithSelf(directory, handle);
+
+  return true;
+}
+
 export async function writeFile(file: FH, data: FileDataType) {
   const writable = await file.createWritable();
   await writable.write(data);
@@ -123,15 +138,24 @@ export async function moveFile(origin: DH, target: DH, name: string, copy = true
   return handle;
 }
 
+export async function moveDirectoryWithSelf(origin: DH, target: DH) {
+  const _target = await target.getDirectoryHandle(origin.name, { create: true });
+
+  await move(origin, _target);
+
+  return true;
+}
+
 export async function moveDirectory(origin: DH, target: DH, name: string, copy = true) {
   const _origin = await origin.getDirectoryHandle(name);
   const _target = await createDirectory(target, name);
 
-  for await (const [key, handle] of _origin.entries()) {
+  const entries = _origin.entries();
+  for await (const [key, handle] of entries) {
     if ((await getHandleType(handle)) === FileType.FILE) {
-      return await moveFile(_origin, _target, key, copy);
+      await moveFile(_origin, _target, key, copy);
     } else {
-      return await moveDirectory(_origin, _target, key);
+      await moveDirectory(_origin, _target, key);
     }
   }
 
diff --git a/types/filesystem.d.ts b/types/filesystem.d.ts
index 987c520..8f8f0b2 100644
--- a/types/filesystem.d.ts
+++ b/types/filesystem.d.ts
@@ -99,7 +99,7 @@ declare interface SaveFilePickerOptions {
 }
 
 declare interface Window {
-  showDirectoryPicker: (options?: DirectoryPickerOptions) => Promise<FileSystemDirectoryHandle>;
+  showDirectoryPicker(options?: DirectoryPickerOptions): Promise<FileSystemDirectoryHandle>;
   showOpenFilePicker(options?: FilePickerOptions): Promise<FileSystemFileHandle[]>;
   showSaveFilePicker(options?: SaveFilePickerOptions): Promise<FileSystemFileHandle>;
 }