feat: support drop folder

buxuku · buxuku · commit dd2d29b571ae · 2025-05-07T17:12:31.000+08:00
diff --git a/main/helpers/ipcHandlers.ts b/main/helpers/ipcHandlers.ts
@@ -1,7 +1,70 @@
 import { ipcMain, BrowserWindow, dialog, shell } from 'electron';
 import * as fs from 'fs';
+import * as path from 'path';
 import { createMessageSender } from './messageHandler';
 
+// 定义支持的文件扩展名常量
+export const MEDIA_EXTENSIONS = [
+  // 视频格式
+  '.mp4', '.avi', '.mov', '.mkv', '.flv', '.wmv', '.webm',
+  // 音频格式
+  '.mp3', '.wav', '.ogg', '.aac', '.wma', '.flac', '.m4a',
+  '.aiff', '.ape', '.opus', '.ac3', '.amr', '.au', '.mid',
+  // 其他常见视频格式
+  '.3gp', '.asf', '.rm', '.rmvb', '.vob', '.ts', '.mts', '.m2ts',
+];
+
+export const SUBTITLE_EXTENSIONS = [
+  // 字幕格式
+  '.srt', '.vtt', '.ass', '.ssa',
+];
+
+// 判断文件是否为媒体文件
+export function isMediaFile(filePath: string): boolean {
+  const ext = path.extname(filePath).toLowerCase();
+  return MEDIA_EXTENSIONS.includes(ext);
+}
+
+// 判断文件是否为字幕文件
+export function isSubtitleFile(filePath: string): boolean {
+  const ext = path.extname(filePath).toLowerCase();
+  return SUBTITLE_EXTENSIONS.includes(ext);
+}
+
+// 递归获取文件夹中的符合任务类型的文件
+async function getMediaFilesFromDirectory(directoryPath: string, taskType: string): Promise<string[]> {
+  // 根据任务类型选择扩展名
+  const supportedExtensions = taskType === 'translate' 
+    ? SUBTITLE_EXTENSIONS 
+    : MEDIA_EXTENSIONS;
+  
+  const files: string[] = [];
+  
+  try {
+    const entries = await fs.promises.readdir(directoryPath, { withFileTypes: true });
+    
+    for (const entry of entries) {
+      const fullPath = path.join(directoryPath, entry.name);
+      
+      if (entry.isDirectory()) {
+        // 递归处理子目录
+        const subDirFiles = await getMediaFilesFromDirectory(fullPath, taskType);
+        files.push(...subDirFiles);
+      } else if (entry.isFile()) {
+        // 检查文件扩展名是否受支持
+        const ext = path.extname(entry.name).toLowerCase();
+        if (supportedExtensions.includes(ext)) {
+          files.push(fullPath);
+        }
+      }
+    }
+  } catch (error) {
+    console.error(`读取目录 ${directoryPath} 时出错:`, error);
+  }
+  
+  return files;
+}
+
 export function setupIpcHandlers(mainWindow: BrowserWindow) {
   ipcMain.on('message', async (event, arg) => {
     event.reply('message', `${arg} World!`);
@@ -11,40 +74,12 @@ export function setupIpcHandlers(mainWindow: BrowserWindow) {
     const { fileType } = data;
     console.log(fileType, 'fileType');
     const name = fileType === 'srt' ? 'Subtitle Files' : 'Media Files';
-    const extensions =
-      fileType === 'srt'
-        ? ['srt']
-        : [
-            'mp4',
-            'avi',
-            'mov',
-            'mkv',
-            'flv',
-            'wmv',
-            'webm',
-            'mp3',
-            'wav',
-            'ogg',
-            'aac',
-            'wma',
-            'flac',
-            'm4a',
-            'aiff',
-            'ape',
-            'opus',
-            'ac3',
-            'amr',
-            'au',
-            'mid',
-            '3gp',
-            'asf',
-            'rm',
-            'rmvb',
-            'vob',
-            'ts',
-            'mts',
-            'm2ts',
-          ];
+    
+    // 使用已定义的常量获取扩展名
+    const extensions = fileType === 'srt'
+      ? SUBTITLE_EXTENSIONS.map(ext => ext.substring(1)) // 移除前面的点
+      : MEDIA_EXTENSIONS.map(ext => ext.substring(1));
+    
     const result = await dialog.showOpenDialog({
       properties: ['openFile', 'multiSelections'],
       filters: [
@@ -69,20 +104,33 @@ export function setupIpcHandlers(mainWindow: BrowserWindow) {
     shell.openExternal(url);
   });
 
-  ipcMain.handle('getDroppedFiles', async (event, files) => {
-    // 验证文件是否存在和可访问
-    const validPaths = await Promise.all(
-      files.map(async (filePath) => {
-        try {
-          await fs.promises.access(filePath);
-          return filePath;
-        } catch {
-          return null;
+  ipcMain.handle('getDroppedFiles', async (event, { files, taskType }) => {
+    // 处理文件和文件夹
+    const allValidPaths: string[] = [];
+    
+    for (const filePath of files) {
+      try {
+        const stats = await fs.promises.stat(filePath);
+        
+        if (stats.isDirectory()) {
+          // 如果是文件夹，递归获取所有符合任务类型的文件
+          const filteredFiles = await getMediaFilesFromDirectory(filePath, taskType);
+          allValidPaths.push(...filteredFiles);
+        } else if (stats.isFile()) {
+          // 如果是文件，根据任务类型过滤
+          // 根据任务类型决定添加哪种文件
+          if ((taskType === 'translate' && isSubtitleFile(filePath)) || 
+              (taskType !== 'translate' && isMediaFile(filePath))) {
+            allValidPaths.push(filePath);
+          }
         }
-      }),
-    );
-
-    return validPaths.filter(Boolean);
+      } catch {
+        // 如果访问失败，跳过此路径
+        continue;
+      }
+    }
+    
+    return allValidPaths;
   });
 
   ipcMain.handle('selectDirectory', async () => {
diff --git a/renderer/pages/[locale]/home.tsx b/renderer/pages/[locale]/home.tsx
@@ -47,11 +47,48 @@ export default function Component() {
     e.preventDefault();
     setIsDragging(false);
 
-    const droppedFiles = filterSupportedFiles(Array.from(e.dataTransfer.files));
+    // 获取所有拖放的项目（文件和文件夹）
+    const items = e.dataTransfer.items;
+    const paths: string[] = [];
 
-    if (droppedFiles.length > 0) {
-      const fileList = droppedFiles.map((file) => file.path);
-      window?.ipc?.invoke('getDroppedFiles', fileList).then((filePaths) => {
+    if (items) {
+      // 使用DataTransferItemList接口获取所有文件和文件夹
+      for (let i = 0; i < items.length; i++) {
+        const item = items[i];
+        // 如果是文件系统条目
+        if (item.kind === 'file') {
+          const fileSystemEntry = item.webkitGetAsEntry();
+          // 将文件或文件夹的路径添加到列表中
+          if (fileSystemEntry) {
+            // @ts-ignore - 获取文件路径，这是Electron特有的属性
+            const path = item.getAsFile()?.path;
+            if (path) {
+              paths.push(path);
+            }
+          }
+        }
+      }
+    } else {
+      // 回退到标准File API（不支持WebkitGetAsEntry的情况）
+      const files = e.dataTransfer.files;
+      for (let i = 0; i < files.length; i++) {
+        // @ts-ignore - 获取文件路径，这是Electron特有的属性
+        const path = files[i].path;
+        if (path) {
+          paths.push(path);
+        }
+      }
+    }
+
+    if (paths.length > 0) {
+      // 根据translateOnly任务类型决定是否只处理字幕文件
+      const isTranslateOnly = formData.taskType === 'translateOnly';
+      const taskType = isTranslateOnly ? 'translate' : 'media';
+      
+      window?.ipc?.invoke('getDroppedFiles', { 
+        files: paths, 
+        taskType 
+      }).then((filePaths) => {
         const newFiles = filePaths.map((filePath) => ({
           uuid: Math.random().toString(36).substring(2),
           filePath,
diff --git a/renderer/public/locales/en/home.json b/renderer/public/locales/en/home.json
@@ -70,8 +70,8 @@
   "noLimit": "Maximize context semantic(-1)",
   "noContext": "Minimize subtitle repetition(0)",
   "importSubtitles": "Import Subtitles",
-  "dragSubtitleHere": "Drag subtitle files here, or click to upload",
-  "dragMediaHere": "Drag media files here, or click to upload",
+  "dragSubtitleHere": "Drag subtitle files or folders here, or click to upload",
+  "dragMediaHere": "Drag media files or folders here, or click to upload",
   "supportedSubtitleFormats": "Supported subtitle formats: SRT",
   "supportedMediaFormats": "Supported media formats: MP4, AVI, MKV, MOV, MP3, WAV"
 }
diff --git a/renderer/public/locales/zh/home.json b/renderer/public/locales/zh/home.json
@@ -76,8 +76,8 @@
   "noLimit": "尽可能关联上下文语义(-1)",
   "noContext": "尽可能减少字幕重复问题(0)",
   "importSubtitles": "导入字幕",
-  "dragSubtitleHere": "拖放字幕文件到这里，或者点击上传",
-  "dragMediaHere": "拖放媒体文件到这里，或者点击上传",
+  "dragSubtitleHere": "拖放字幕文件或目录到这里，或者点击上传",
+  "dragMediaHere": "拖放媒体文件或目录到这里，或者点击上传",
   "supportedSubtitleFormats": "支持的字幕格式：SRT",
   "supportedMediaFormats": "支持的媒体格式：MP4, AVI, MKV, MOV, MP3, WAV"
 }
