Function filesToTree

src/utils/promptGenerator.ts:127–182

(files: File[])

// 辅助函数：将文件列表转换为树形结构
export function filesToTree(files: File[]): FileNode[] {
  const root: FileNode[] = [];
  const pathMap: Record<string, FileNode> = {};
  
  // 首先创建所有目录节点
  for (const file of files) {
    const pathParts = file.webkitRelativePath.split('/');
    let currentPath = '';
    
    for (let i = 0; i < pathParts.length - 1; i++) {
      const part = pathParts[i];
      if (!part) continue;
      
      const parentPath = currentPath;
      currentPath = currentPath ? `${currentPath}/${part}` : part;
      
      if (!pathMap[currentPath]) {
        const dirNode: FileNode = {
          name: part,
          path: currentPath,
          isDirectory: true,
          children: []
        };
        
        pathMap[currentPath] = dirNode;
        
        if (parentPath) {
          pathMap[parentPath].children?.push(dirNode);
        } else {
          root.push(dirNode);
        }
      }
    }
  }
  
  // 然后添加文件节点
  for (const file of files) {
    const pathParts = file.webkitRelativePath.split('/');
    const fileName = pathParts[pathParts.length - 1];
    const parentPath = pathParts.slice(0, -1).join('/');
    
    const fileNode: FileNode = {
      name: fileName,
      path: file.webkitRelativePath,
      isDirectory: false,
    };
    
    if (parentPath && pathMap[parentPath]) {
      pathMap[parentPath].children?.push(fileNode);
    } else {
      root.push(fileNode);
    }
  }
  
  return root;
}

// 辅助函数：将单个文件转换为树节点