feat: 增加右键选中文件打包功能

Author: yefansky

package.json

@@ -24,7 +24,8 @@
     "onCommand:codeReDesign.analyzeSingleFile",
     "onCommand:codeReDesign.redesignSingleFile",
     "onCommand:codeReDesign.startChat",
-    "onCommand:codeReDesign.continueChat"
+    "onCommand:codeReDesign.continueChat",
+    "onCommand:codeReDesign.packupToCvb"
   ],
   "repository": {
     "type": "git",
@@ -121,6 +122,13 @@
         "when": "false",
         "category": "CodeReDesign",
         "icon": "${bug}"
+      },
+      {
+        "command": "codeReDesign.packupToCvb",
+        "title": "CodeReDesign: Packup to CVB",
+        "when": "false",
+        "category": "CodeReDesign",
+        "icon": "${add}"
       }
     ],
     "viewsContainers": {
@@ -194,6 +202,10 @@
           "command": "codeReDesign.redesignSingleFile",
           "group": "cvb@1",
           "when": "resourceExtname in codeReDesign.supportedSourceFileTypeExt"
+        },
+        {
+          "command": "codeReDesign.packupToCvb",
+          "group": "cvb@1"
         }
       ]
     },

src/extension.ts

@@ -11,6 +11,7 @@ import { activateGuide } from './guide';
 import {ChatPanel} from './chatPanel';
 import { isUnderTokenLimit, initTokenizer } from './deepseekTokenizer';
 import * as ragService from './ragService';
+import {collectSupportedFiles} from './languageMapping';
 
 let currentOperationController: AbortController | null = null;
 
@@ -211,6 +212,36 @@ export function activate(context: vscode.ExtensionContext) {
         vscode.window.showInformationMessage(`CVB file generated at: ${cvbFilePath}`);
     });
 
+    // New command for the context menu
+    let packupToCvbCommand = vscode.commands.registerCommand('codeReDesign.packupToCvb', async (uri: vscode.Uri, selectedUris: vscode.Uri[]) => {
+        // Collect URIs (prioritize selectedUris for multi-selection)
+        const uris: vscode.Uri[] = selectedUris && selectedUris.length > 0 ? selectedUris : uri ? [uri] : [];
+
+        if (uris.length === 0) {
+            vscode.window.showErrorMessage('No files or folders selected.');
+            return;
+        }
+
+        // Collect all supported files (recursively for folders)
+        const filePaths = await collectSupportedFiles(uris);
+
+        const userRequest = await vscode.window.showInputBox({
+            prompt: 'Enter your refactoring request',
+            placeHolder: 'e.g., Move all mouse event handling code to a single file',
+        });
+
+        if (!userRequest) {
+            return;
+        }
+
+        try {
+            const cvbFilePath = generateCvb(filePaths, userRequest);
+            vscode.window.showInformationMessage(`CVB file generated at: ${cvbFilePath}`);
+        } catch (error) {
+            vscode.window.showErrorMessage(`Failed to generate CVB file: ${(error as Error).message}`);
+        }
+    });
+
     // 注册命令:上传 CVB 并调用 API
     let redesignCvbCommand = vscode.commands.registerCommand('codeReDesign.redesignCvb', async () => {
         const workspaceFolders = vscode.workspace.workspaceFolders;
@@ -367,7 +398,7 @@ export function activate(context: vscode.ExtensionContext) {
         }
     });
 
-    context.subscriptions.push(generateCvbCommand, redesignCvbCommand, applyCvbCommand, stopOperation, analyzeCodeCommand, outputChannel, startChatCommand);
+    context.subscriptions.push(generateCvbCommand, redesignCvbCommand, applyCvbCommand, stopOperation, analyzeCodeCommand, startChatCommand, packupToCvbCommand);
 
     setupCvbAsMarkdown(context);
 

src/languageMapping.ts

@@ -1,3 +1,7 @@
+import * as vscode from 'vscode';
+import * as fs from 'fs';
+import * as path from 'path';
+
 // 语言映射表
 export const g_objLanguageMapping: { [key: string]: string } = {
   'cpp': 'c++',
@@ -36,4 +40,47 @@ export function getLanguageFromPath(filePath: string): string {
 }
 
 export const SOURCE_FILE_EXTENSIONS_WITH_DOT = Object.keys(g_objLanguageMapping)
-  .map(ext => `.${ext}`);
+  .map(ext => `.${ext}`);
+
+  /**
+ * Recursively collect all supported files from the given URIs (files or folders).
+ * @param uris Array of URIs (files or folders) to process.
+ * @returns Array of file paths with supported extensions.
+ */
+export async function collectSupportedFiles(uris: vscode.Uri[]): Promise<string[]> {
+  const filePaths: string[] = [];
+
+  async function traverseFolder(folderPath: string): Promise<void> {
+    const entries = await fs.promises.readdir(folderPath, { withFileTypes: true });
+    for (const entry of entries) {
+      const fullPath = path.join(folderPath, entry.name);
+      if (entry.isDirectory()) {
+        // Recursively traverse subfolders
+        await traverseFolder(fullPath);
+      } else if (entry.isFile()) {
+        // Check if the file has a supported extension
+        const ext = path.extname(entry.name).toLowerCase();
+        if (SOURCE_FILE_EXTENSIONS_WITH_DOT.includes(ext)) {
+          filePaths.push(fullPath);
+        }
+      }
+    }
+  }
+
+  // Process each URI
+  for (const uri of uris) {
+    const stat = await fs.promises.stat(uri.fsPath);
+    if (stat.isDirectory()) {
+      // If it's a folder, traverse it recursively
+      await traverseFolder(uri.fsPath);
+    } else if (stat.isFile()) {
+      // If it's a file, check its extension
+      const ext = path.extname(uri.fsPath).toLowerCase();
+      if (SOURCE_FILE_EXTENSIONS_WITH_DOT.includes(ext)) {
+        filePaths.push(uri.fsPath);
+      }
+    }
+  }
+
+  return filePaths;
+}