云存储文件夹内容批量复制教程：基于文件列表与迭代操作

const admin = require('firebase-admin');

// 确保已初始化Firebase Admin SDK
// admin.initializeApp({
//   credential: admin.credential.applicationDefault(),
//   storageBucket: 'your-project-id.appspot.com' // 替换为你的存储桶名称
// });

/**
 * 列出指定前缀下的所有文件。
 * @param {string} prefix 要列出的文件前缀（即“文件夹”路径）。
 * @returns {Promise<Array<Object>>} 包含文件对象的数组。
 */
async function listFilesInPrefix(prefix) {
  const bucket = admin.storage().bucket();
  let files = [];
  try {
    // getFiles 方法返回一个数组，第一个元素是文件对象数组，第二个是下一个查询令牌（用于分页）
    [files] = await bucket.getFiles({ prefix: prefix });
    console.log(`在 '${prefix}' 下找到 ${files.length} 个文件。`);
    return files;
  } catch (error) {
    console.error(`列出文件失败，前缀：${prefix}，错误：`, error);
    throw error;
  }
}

// 示例用法
// listFilesInPrefix('documents/1/')
//   .then(files => {
//     files.forEach(file => console.log(file.name));
//   })
//   .catch(err => console.error('获取文件列表失败:', err));

/**
 * 批量复制云存储“文件夹”内容到新位置。
 * 由于云存储没有传统意义上的文件夹，此操作通过列出前缀匹配的文件并逐一复制实现。
 * @param {string} sourcePrefix 源“文件夹”路径前缀，例如 'documents/docId/' 或 'documents/docId'
 * @param {string} destinationPrefix 目标“文件夹”路径前缀，例如 'users/userId/documents/docId/' 或 'users/userId/documents/docId'
 * @returns {Promise<void>}
 */
async function copyFolderContents(sourcePrefix, destinationPrefix) {
  const bucket = admin.storage().bucket();

  // 确保前缀以斜杠结尾，以便后续路径处理的一致性
  const effectiveSourcePrefix = sourcePrefix.endsWith('/') ? sourcePrefix : sourcePrefix + '/';
  const effectiveDestinationPrefix = destinationPrefix.endsWith('/') ? destinationPrefix : destinationPrefix + '/';

  let filesToCopy;

  try {
    // 1. 列出源前缀下的所有文件
    [filesToCopy] = await bucket.getFiles({ prefix: effectiveSourcePrefix });

    if (filesToCopy.length === 0) {
      console.log(`源路径 '${effectiveSourcePrefix}' 下没有找到文件。`);
      return;
    }

    console.log(`找到 ${filesToCopy.length} 个文件，开始复制...`);

    // 2. 为每个文件创建复制操作的Promise
    const copyPromises = filesToCopy.map(async (file) => {
      // 从原始文件路径中提取相对文件名
      const relativeFileName = file.name.substring(effectiveSourcePrefix.length);
      // 构建新的目标文件路径
      const destinationPath = effectiveDestinationPrefix + relativeFileName;

      try {
        await file.copy(bucket.file(destinationPath));
        console.log(`成功复制：${file.name} -> ${destinationPath}`);
        return { status: 'fulfilled', value: { source: file.name, destination: destinationPath } };
      } catch (error) {
        console.error(`复制失败：${file.name} -> ${destinationPath}，错误：`, error);
        return { status: 'rejected', reason: new Error(`复制文件 ${file.name} 失败: ${error.message}`) };
      }
    });

    // 3. 使用 Promise.allSettled 等待所有复制操作完成
    // Promise.allSettled 会等待所有 Promise 完成，无论成功或失败
    const results = await Promise.allSettled(copyPromises);

    const failures = results.filter(result => result.status === 'rejected');
    if (failures.length > 0) {
      console.warn(`批量复制操作完成，但有 ${failures.length} 个文件复制失败。`);
      failures.forEach(fail => console.error('失败详情:', fail.reason));
      throw new Error(`批量复制操作未能完全成功，有 ${failures.length} 个文件复制失败。`);
    } else {
      console.log(`所有 ${filesToCopy.length} 个文件成功复制。`);
    }

  } catch (err) {
    console.error(`执行批量复制时发生错误:`, err);
    throw err; // 重新抛出以便调用者处理
  }
}

// 示例用法
// 假设要将 'documents/1/' 下的所有内容复制到 'users/testID/documents/1/'
// copyFolderContents('documents/1/', 'users/testID/documents/1/')
//   .then(() => console.log('批量复制操作完成。'))
//   .catch(err => console.error('批量复制操作失败:', err));