feat: update version to 0.0.63 and enhance collectAll command to accept URI and selected files

Author: 0-don

package.json

@@ -2,7 +2,7 @@
   "name": "code-collector",
   "displayName": "Code Collector",
   "description": "Gather code context for AI based on imports",
-  "version": "0.0.62",
+  "version": "0.0.63",
   "publisher": "0-don",
   "icon": "assets/icon.png",
   "repository": {

src/commands.ts

@@ -307,7 +307,10 @@ export class CommandHandler {
     showStatsNotification(finalContexts.length, totalLines, stats, modeLabel);
   }
 
-  async handleCollectAll(): Promise<void> {
+  async handleCollectAll(
+    uri: vscode.Uri,
+    selectedFiles?: vscode.Uri[],
+  ): Promise<void> {
     return vscode.window.withProgress(
       {
         location: vscode.ProgressLocation.Notification,
@@ -317,40 +320,82 @@ export class CommandHandler {
       async (progress, token) => {
         try {
           this.output.clear();
-          this.output.log("Starting full workspace collection...");
+          this.output.log("Starting full collection...");
 
-          const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
-          if (!workspaceFolder) {
+          const workspaceRoot = getWorkspaceRoot();
+          if (!workspaceRoot) {
             const message = "No workspace folder open";
             this.output.error(message);
             vscode.window.showErrorMessage(message);
             return;
           }
 
-          if (token.isCancellationRequested) {
-            return;
+          const targetFolders: string[] = [];
+
+          if (selectedFiles?.length) {
+            for (const selectedUri of selectedFiles) {
+              const fsPath = selectedUri.fsPath;
+              if (fs.existsSync(fsPath)) {
+                const stat = fs.statSync(fsPath);
+                if (stat.isDirectory()) {
+                  targetFolders.push(fsPath);
+                } else if (stat.isFile()) {
+                  targetFolders.push(path.dirname(fsPath));
+                }
+              }
+            }
+          } else if (uri) {
+            const fsPath = uri.fsPath;
+            if (fs.existsSync(fsPath)) {
+              const stat = fs.statSync(fsPath);
+              if (stat.isDirectory()) {
+                targetFolders.push(fsPath);
+              } else if (stat.isFile()) {
+                targetFolders.push(path.dirname(fsPath));
+              }
+            }
           }
 
-          const contexts = await this.contextCollector.collectAllFiles(
-            workspaceFolder.uri.fsPath,
-            () => !token.isCancellationRequested,
-          );
+          if (targetFolders.length === 0) {
+            targetFolders.push(workspaceRoot);
+          }
+
+          const uniqueFolders = [...new Set(targetFolders)];
+          this.output.log(`Collecting from ${uniqueFolders.length} folder(s)`);
+
+          const allContexts: FileContext[] = [];
+
+          for (const folder of uniqueFolders) {
+            if (token.isCancellationRequested) {
+              return;
+            }
+
+            const contexts = await this.contextCollector.collectAllFiles(
+              folder,
+              () => !token.isCancellationRequested,
+            );
+            allContexts.push(...contexts);
+          }
 
           if (token.isCancellationRequested) {
             return;
           }
 
-          const totalLines = contexts.reduce(
+          const uniqueContexts = Array.from(
+            new Map(allContexts.map((ctx) => [ctx.path, ctx])).values(),
+          );
+
+          const totalLines = uniqueContexts.reduce(
             (sum, ctx) => sum + ctx.content.split("\n").length,
             0,
           );
-          const stats = getFileTypeStats(contexts);
-          const output = formatContexts(contexts);
+          const stats = getFileTypeStats(uniqueContexts);
+          const output = formatContexts(uniqueContexts);
           await vscode.env.clipboard.writeText(output);
 
-          const successMessage = `Copied ${contexts.length} files (${totalLines} lines)`;
+          const successMessage = `Copied ${uniqueContexts.length} files (${totalLines} lines)`;
           this.output.log(`✓ ${successMessage}`);
-          showStatsNotification(contexts.length, totalLines, stats);
+          showStatsNotification(uniqueContexts.length, totalLines, stats);
         } catch (error) {
           const errorMessage = `Error: ${error}`;
           this.output.error(errorMessage, error);

src/extension.ts

@@ -31,7 +31,8 @@ export async function activate(context: vscode.ExtensionContext) {
 
   const collectAllDisposable = vscode.commands.registerCommand(
     "code-collector.collectAll",
-    () => commandHandler.handleCollectAll(),
+    (uri: vscode.Uri, selectedFiles?: vscode.Uri[]) =>
+      commandHandler.handleCollectAll(uri, selectedFiles),
   );
 
   const showOutputDisposable = vscode.commands.registerCommand(