feat: refactor output and settings management, add file type statistics tracking

Author: 0-don

src/commands.ts

@@ -2,7 +2,6 @@ import * as fs from "fs";
 import * as path from "path";
 import * as vscode from "vscode";
 import { ContextCollector } from "./core";
-import { OutputManager } from "./output";
 import { parserRegistry } from "./parsers";
 import { FileContext } from "./types";
 import {
@@ -12,6 +11,8 @@ import {
   isTextFile,
   shouldIgnoreFile,
 } from "./utils";
+import { OutputManager } from "./lib/output";
+import { getFileTypeStats, showStatsNotification } from "./lib/stats";
 
 export class CommandHandler {
   private contextCollector = new ContextCollector();
@@ -189,11 +190,17 @@ export class CommandHandler {
       (sum, ctx) => sum + ctx.content.split("\n").length,
       0,
     );
+    const stats = getFileTypeStats(allContexts);
     await vscode.env.clipboard.writeText(output);
 
     const successMessage = `Copied ${allContexts.length} files (${totalLines} lines) - direct collection`;
     this.output.log(`✓ ${successMessage}`);
-    vscode.window.showInformationMessage(successMessage);
+    showStatsNotification(
+      allContexts.length,
+      totalLines,
+      stats,
+      "direct collection",
+    );
   }
 
   private async collectFilesFromDirectory(
@@ -300,6 +307,7 @@ export class CommandHandler {
       (sum, ctx) => sum + ctx.content.split("\n").length,
       0,
     );
+    const stats = getFileTypeStats(finalContexts);
     await vscode.env.clipboard.writeText(output);
 
     const programmingFiles = finalContexts.filter(
@@ -308,10 +316,10 @@ export class CommandHandler {
     const modeLabel = applyIgnorePatterns
       ? "smart filter"
       : `${programmingFiles} with imports, ${finalContexts.length - programmingFiles} text`;
-    const successMessage = `Copied ${finalContexts.length} files (${totalLines} lines) - ${modeLabel}`;
 
+    const successMessage = `Copied ${finalContexts.length} files (${totalLines} lines) - ${modeLabel}`;
     this.output.log(`✓ ${successMessage}`);
-    vscode.window.showInformationMessage(successMessage);
+    showStatsNotification(finalContexts.length, totalLines, stats, modeLabel);
   }
 
   async handleCollectAll(): Promise<void> {
@@ -351,12 +359,13 @@ export class CommandHandler {
             (sum, ctx) => sum + ctx.content.split("\n").length,
             0,
           );
+          const stats = getFileTypeStats(contexts);
           const output = formatContexts(contexts);
           await vscode.env.clipboard.writeText(output);
 
           const successMessage = `Copied ${contexts.length} files (${totalLines} lines)`;
           this.output.log(`✓ ${successMessage}`);
-          vscode.window.showInformationMessage(successMessage);
+          showStatsNotification(contexts.length, totalLines, stats);
         } catch (error) {
           const errorMessage = `Error: ${error}`;
           this.output.error(errorMessage, error);

src/core.ts

@@ -2,7 +2,7 @@ import * as fs from "fs";
 import * as micromatch from "micromatch";
 import * as path from "path";
 import { getIgnorePatterns } from "./config";
-import { OutputManager } from "./output";
+import { OutputManager } from "./lib/output";
 import { parserRegistry } from "./parsers";
 import { resolverRegistry } from "./resolvers";
 import { PythonResolver } from "./resolvers/python-resolver";

src/extension.ts

@@ -1,7 +1,7 @@
 import * as vscode from "vscode";
 import { CommandHandler } from "./commands";
-import { OutputManager } from "./output";
-import { SettingsInitializer } from "./settings-initializer";
+import { OutputManager } from "./lib/output";
+import { SettingsInitializer } from "./lib/settings-initializer";
 
 export async function activate(context: vscode.ExtensionContext) {
   const output = OutputManager.getInstance();

src/languages.ts src/lib/languages.tssrc/languages.ts renamed to src/lib/languages.ts

@@ -0,0 +1,81 @@
+import * as path from "path";
+import * as vscode from "vscode";
+import { FileContext } from "../types";
+
+interface FileTypeStats {
+  [ext: string]: { count: number; lines: number };
+}
+
+export function getFileTypeStats(contexts: FileContext[]): FileTypeStats {
+  const stats: FileTypeStats = {};
+
+  for (const ctx of contexts) {
+    const ext = path.extname(ctx.path) || "no extension";
+    const lines = ctx.content.split("\n").length;
+
+    if (!stats[ext]) {
+      stats[ext] = { count: 0, lines: 0 };
+    }
+
+    stats[ext].count++;
+    stats[ext].lines += lines;
+  }
+
+  return stats;
+}
+
+function showStatsQuickPick(
+  fileCount: number,
+  totalLines: number,
+  stats: FileTypeStats,
+  modeLabel?: string,
+): void {
+  const entries = Object.entries(stats).sort((a, b) => b[1].lines - a[1].lines);
+
+  const items = entries.map(([ext, { count, lines }]) => ({
+    label: ext,
+    description: `${count} files`,
+    detail: `${lines} lines`,
+  }));
+
+  const quickPick = vscode.window.createQuickPick();
+  quickPick.title = `✓ Copied ${fileCount} files (${totalLines} lines)${modeLabel ? ` - ${modeLabel}` : ""}`;
+  quickPick.items = items;
+  quickPick.canSelectMany = false;
+
+  quickPick.onDidHide(() => quickPick.dispose());
+  quickPick.show();
+}
+
+export function showStatsNotification(
+  fileCount: number,
+  totalLines: number,
+  stats: FileTypeStats,
+  modeLabel?: string,
+): void {
+  const statusBarItem = vscode.window.createStatusBarItem(
+    vscode.StatusBarAlignment.Left,
+    100,
+  );
+  statusBarItem.text = `$(list-unordered) File Stats (${fileCount} files, ${totalLines} lines)`;
+  statusBarItem.tooltip = "Click to view file type breakdown";
+  statusBarItem.command = "code-collector.showTempStats";
+
+  const disposable = vscode.commands.registerCommand(
+    "code-collector.showTempStats",
+    () => {
+      showStatsQuickPick(fileCount, totalLines, stats, modeLabel);
+    },
+  );
+
+  statusBarItem.show();
+
+  setTimeout(() => {
+    statusBarItem.dispose();
+    disposable.dispose();
+  }, 10000);
+
+  vscode.window.showInformationMessage(
+    `Copied ${fileCount} files (${totalLines} lines)${modeLabel ? ` - ${modeLabel}` : ""}`,
+  );
+}

src/output.ts src/lib/output.tssrc/output.ts renamed to src/lib/output.ts

@@ -1,9 +1,9 @@
 import { XMLParser } from "fast-xml-parser";
 import * as fs from "fs";
 import * as path from "path";
+import { jvmExtensions } from "../lib/languages";
 import { ResolverConfig } from "../types";
 import { BaseResolver } from "./base-resolver";
-import { jvmExtensions } from "../languages";
 
 interface MavenProject {
   project?: {

src/settings-initializer.ts src/lib/settings-initializer.tssrc/settings-initializer.ts renamed to src/lib/settings-initializer.ts

@@ -1,7 +1,7 @@
 import * as fs from "fs";
 import * as path from "path";
 import * as ts from "typescript";
-import { javascriptExtensions } from "../languages";
+import { javascriptExtensions } from "../lib/languages";
 import { ResolverConfig } from "../types";
 import { BaseResolver } from "./base-resolver";
 

src/lib/stats.ts

@@ -3,7 +3,7 @@ import * as fs from "fs";
 import * as os from "os";
 import * as path from "path";
 import * as util from "util";
-import { pythonExtensions } from "../languages";
+import { pythonExtensions } from "../lib/languages";
 import { ResolverConfig } from "../types";
 import { BaseResolver } from "./base-resolver";