patch_search_main.py

import re

with open('/workspace/search-main.js', 'r') as f:
    content = f.read()

# Replace getFileInfoFast to be async
old_getFileInfo = """// 快速获取文件信息 - 使用同步方法提高性能
function getFileInfoFast(filePath) {
  try {
    const stats = fsSync.statSync(filePath);
    const parsedPath = path.parse(filePath);

    return {
      name: parsedPath.base,
      path: filePath,
      dir: parsedPath.dir,
      ext: parsedPath.ext,
      size: stats.size,
      mtime: stats.mtime,
      birthtime: stats.birthtime,
      atime: stats.atime
    };
  } catch (error) {
    if (error.code !== 'EPERM' && error.code !== 'EACCES') {
      console.error('获取文件信息失败:', error);
    }
    return null;
  }
}"""

new_getFileInfo = """// 快速获取文件信息 - 使用异步方法不阻塞主线程
async function getFileInfoFast(filePath) {
  try {
    const stats = await fs.stat(filePath);
    const parsedPath = path.parse(filePath);

    return {
      name: parsedPath.base,
      path: filePath,
      dir: parsedPath.dir,
      ext: parsedPath.ext,
      size: stats.size,
      mtime: stats.mtime,
      birthtime: stats.birthtime,
      atime: stats.atime
    };
  } catch (error) {
    if (error.code !== 'EPERM' && error.code !== 'EACCES') {
      console.error('获取文件信息失败:', error);
    }
    return null;
  }
}"""

content = content.replace(old_getFileInfo, new_getFileInfo)

# Add export-files IPC handler
export_files_code = """
// 导出文件
ipcMain.handle('export-files', async (event, { files, destFolder }) => {
  let successCount = 0;
  let failCount = 0;
  
  // 限制并发数量
  const limit = 10;
  let active = 0;
  let index = 0;
  
  return new Promise((resolve, reject) => {
    const next = async () => {
      if (index >= files.length) {
        if (active === 0) {
          resolve({ successCount, failCount });
        }
        return;
      }
      
      const file = files[index++];
      active++;
      
      try {
        let targetPath = path.join(destFolder, file.name);
        
        // 处理文件名冲突
        let counter = 1;
        while (fsSync.existsSync(targetPath)) {
          const parsed = path.parse(file.name);
          targetPath = path.join(destFolder, `${parsed.name} (${counter})${parsed.ext}`);
          counter++;
        }
        
        // 使用原生的 copyFile，它在底层使用高效的系统调用
        await fs.copyFile(file.path, targetPath, fsSync.constants.COPYFILE_FICLONE);
        successCount++;
      } catch (err) {
        console.error('Copy failed:', err);
        failCount++;
      } finally {
        active--;
        
        // 发送进度
        if ((successCount + failCount) % 10 === 0 || (successCount + failCount) === files.length) {
          event.sender.send('export-progress', {
            current: successCount + failCount,
            total: files.length
          });
        }
        
        next();
      }
    };
    
    for (let i = 0; i < limit && i < files.length; i++) {
      next();
    }
  });
});
"""

# Insert export-files before // 打开文件夹
content = content.replace("// 打开文件夹", export_files_code + "\n// 打开文件夹")

with open('/workspace/search-main.js', 'w') as f:
    f.write(content)