@probebrowser/trace 1.5.0 → 1.5.1

package/dist/index.js

@@ -8,11 +8,8 @@ import { program } from 'commander';
 import chalk from 'chalk';
 import ora from 'ora';
 import readline from 'readline';
-import { Trace } from '@probebrowser/sdk';
-import { exec } from 'child_process';
-import { promisify } from 'util';
-const execAsync = promisify(exec);
-const VERSION = '1.5.0';
+import { Trace } from '../../sdk/dist/index.js';
+const VERSION = '1.0.4';
 // ============================================
 // LOGGING HELPERS
 // ============================================
@@ -136,7 +133,6 @@ program
     .option('--api-key <key>', 'API key')
     .option('--headless', 'Run in headless mode', true)
     .option('--no-headless', 'Run with visible browser')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
     .option('--json', 'Output as JSON')
     .option('-v, --verbose', 'Verbose output')
     .action(async (url, options) => {
@@ -149,7 +145,6 @@ program
             apiKey: getApiKey(options),
             headless: options.headless,
             verbose: options.verbose,
-            connectPort: options.connect ? parseInt(options.connect) : undefined,
         });
         spinner.text = `Connecting to ${url}`;
         await trace.connect(url);
@@ -199,7 +194,6 @@ program
     .description('Full debug scan of a URL')
     .option('--api-url <url>', 'API endpoint', 'https://dev-intelli-api.azurewebsites.net')
     .option('--api-key <key>', 'API key')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
     .option('--json', 'Output as JSON')
     .action(async (url, options) => {
     const spinner = ora('Connecting...').start();
@@ -209,7 +203,6 @@ program
         trace = new Trace({
             apiUrl: options.apiUrl,
             apiKey: getApiKey(options),
-            connectPort: options.connect ? parseInt(options.connect) : undefined,
         });
         spinner.text = `Connecting to ${url}`;
         await trace.connect(url);
@@ -256,715 +249,36 @@ program
 program
     .command('errors <url>')
     .alias('e')
-    .description('AI-powered error analysis with code context and fix suggestions')
-    .option('--api-url <url>', 'API endpoint', 'https://dev-intelli-api.azurewebsites.net')
-    .option('--api-key <key>', 'API key')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--raw', 'Show raw errors without AI analysis')
-    .option('--json', 'Output as JSON')
-    .action(async (url, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            apiUrl: options.apiUrl,
-            apiKey: getApiKey(options),
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
-        spinner.text = `Connecting to ${url}`;
-        await trace.connect(url);
-        // Raw mode - old behavior without AI
-        if (options.raw) {
-            spinner.text = 'Checking for errors...';
-            const errors = await trace.getConsoleErrors();
-            const groups = await trace.getErrorGroups();
-            await trace.disconnect();
-            spinner.stop();
-            if (options.json) {
-                console.log(JSON.stringify({ errors, groups }, null, 2));
-                return;
-            }
-            console.log();
-            console.log(chalk.bold(`🔴 Console Errors (${errors.length})`));
-            console.log(chalk.gray('─'.repeat(60)));
-            if (errors.length === 0) {
-                log.success('No console errors found!');
-            }
-            else {
-                for (const err of errors.slice(0, 20)) {
-                    console.log();
-                    console.log(`  ${chalk.red('●')} ${chalk.bold(err.errorType || 'Error')}`);
-                    console.log(`    ${err.message.substring(0, 200)}`);
-                    if (err.location) {
-                        console.log(`    ${chalk.dim(err.location)}`);
-                    }
-                }
-            }
-            console.log();
-            return;
-        }
-        // AI-Powered Analysis (new behavior)
-        spinner.text = chalk.cyan('🔍 Analyzing errors with AI...');
-        const analysis = await trace.analyzeErrors();
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify(analysis, null, 2));
-            return;
-        }
-        console.log();
-        console.log(chalk.bold.cyan('🧠 AI Error Analysis'));
-        console.log(chalk.gray('─'.repeat(60)));
-        console.log();
-        if (analysis.totalErrors === 0) {
-            console.log(chalk.green('✓ No errors detected on this page!'));
-            console.log();
-            return;
-        }
-        // Summary
-        console.log(chalk.yellow(`Found ${analysis.totalErrors} error(s)`));
-        console.log(chalk.dim(analysis.summary));
-        console.log();
-        // Each analyzed error
-        for (const err of analysis.errors) {
-            console.log(chalk.red('━'.repeat(60)));
-            console.log();
-            console.log(`  ${chalk.red('❌')} ${chalk.bold(err.summary)}`);
-            console.log();
-            console.log(`  ${chalk.cyan('Why:')} ${err.explanation}`);
-            if (err.location) {
-                console.log(`  ${chalk.cyan('Location:')} ${chalk.dim(err.location)}`);
-            }
-            if (err.codeSnippet) {
-                console.log();
-                console.log(`  ${chalk.cyan('Code:')}`);
-                // Indent code snippet
-                const lines = err.codeSnippet.split('\n');
-                for (const line of lines.slice(0, 10)) {
-                    console.log(`    ${chalk.dim(line)}`);
-                }
-            }
-            if (err.suggestedFix) {
-                console.log();
-                console.log(`  ${chalk.green('Fix:')} ${err.suggestedFix}`);
-            }
-            if (err.codeChange) {
-                console.log();
-                console.log(`  ${chalk.green('Suggested Code Change:')}`);
-                const changes = err.codeChange.split('\n');
-                for (const line of changes.slice(0, 5)) {
-                    console.log(`    ${chalk.green(line)}`);
-                }
-            }
-            console.log();
-        }
-        // Recommendations
-        if (analysis.recommendations.length > 0) {
-            console.log(chalk.cyan('📋 Recommendations:'));
-            for (const rec of analysis.recommendations) {
-                console.log(`  ${chalk.cyan('→')} ${rec}`);
-            }
-            console.log();
-        }
-        console.log(chalk.dim('Powered by Trace Intelligence Engine'));
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        if (isAuthError(error)) {
-            log.authError();
-            process.exit(1);
-        }
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// EVAL COMMAND - Execute JavaScript
-// ============================================
-program
-    .command('eval <expression>')
-    .description('Execute JavaScript in the browser context')
-    .requiredOption('-u, --url <url>', 'URL to connect to')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--json', 'Output as JSON')
-    .action(async (expression, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
-        spinner.text = `Connecting to ${options.url}`;
-        await trace.connect(options.url);
-        spinner.text = 'Evaluating...';
-        const result = await trace.evaluate(expression);
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify({ expression, result }, null, 2));
-            return;
-        }
-        console.log();
-        console.log(chalk.cyan('Expression:'), expression);
-        console.log(chalk.green('Result:'));
-        console.log(JSON.stringify(result, null, 2));
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// EXPLORE COMMAND - Tree Object Exploration
-// ============================================
-program
-    .command('explore <expression>')
-    .alias('x')
-    .description('Explore an object with tree-like property display')
-    .requiredOption('-u, --url <url>', 'URL to connect to')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--depth <n>', 'Max depth to expand', '3')
-    .option('--json', 'Output as JSON')
-    .action(async (expression, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
-        spinner.text = `Connecting to ${options.url}`;
-        await trace.connect(options.url);
-        spinner.text = 'Inspecting...';
-        // Evaluate and get detailed object info
-        const maxDepth = parseInt(options.depth) || 3;
-        const inspectCode = `
-                (function inspect(obj, depth = 0, maxDepth = ${maxDepth}, path = '') {
-                    if (depth > maxDepth) return { _truncated: true, _type: typeof obj };
-                    if (obj === null) return { _value: null, _type: 'null' };
-                    if (obj === undefined) return { _value: undefined, _type: 'undefined' };
-                    
-                    const type = typeof obj;
-                    
-                    if (type === 'function') {
-                        return { _type: 'function', _name: obj.name || 'anonymous', _length: obj.length };
-                    }
-                    if (type !== 'object') {
-                        return { _value: obj, _type: type };
-                    }
-                    if (Array.isArray(obj)) {
-                        return {
-                            _type: 'array',
-                            _length: obj.length,
-                            _items: obj.slice(0, 10).map((item, i) => ({ key: i, ...inspect(item, depth + 1, maxDepth, path + '[' + i + ']') })),
-                            _truncated: obj.length > 10
-                        };
-                    }
-                    if (obj instanceof Date) {
-                        return { _type: 'Date', _value: obj.toISOString() };
-                    }
-                    if (obj instanceof RegExp) {
-                        return { _type: 'RegExp', _value: obj.toString() };
-                    }
-                    if (obj instanceof Error) {
-                        return { _type: 'Error', _name: obj.name, _message: obj.message, _stack: obj.stack?.split('\\n').slice(0, 5) };
-                    }
-                    
-                    // Regular object
-                    const keys = Object.keys(obj).slice(0, 20);
-                    const result = {
-                        _type: obj.constructor?.name || 'Object',
-                        _keys: keys.length,
-                        _properties: {}
-                    };
-                    for (const key of keys) {
-                        try {
-                            result._properties[key] = inspect(obj[key], depth + 1, maxDepth, path + '.' + key);
-                        } catch (e) {
-                            result._properties[key] = { _error: 'Cannot access' };
-                        }
-                    }
-                    if (Object.keys(obj).length > 20) {
-                        result._truncated = true;
-                    }
-                    return result;
-                })(${expression})
-            `;
-        const result = await trace.evaluate(inspectCode);
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify({ expression, result }, null, 2));
-            return;
-        }
-        console.log();
-        console.log(chalk.bold.cyan('🔍 Object Inspector'));
-        console.log(chalk.gray('─'.repeat(60)));
-        console.log(chalk.dim(`Expression: ${expression}`));
-        console.log();
-        // Pretty print the tree
-        function printTree(obj, indent = '', isLast = true, keyName = '') {
-            const prefix = indent + (isLast ? '└─ ' : '├─ ');
-            const childIndent = indent + (isLast ? '   ' : '│  ');
-            if (!obj || typeof obj !== 'object') {
-                console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${chalk.yellow(String(obj))}`);
-                return;
-            }
-            if (obj._error) {
-                console.log(`${prefix}${chalk.cyan(keyName)}: ${chalk.red(obj._error)}`);
-                return;
-            }
-            if (obj._truncated && !obj._properties && !obj._items) {
-                console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${chalk.dim(`[${obj._type}...]`)}`);
-                return;
-            }
-            if (obj._type && obj._value !== undefined) {
-                const typeColor = obj._type === 'string' ? chalk.green :
-                    obj._type === 'number' ? chalk.yellow :
-                        obj._type === 'boolean' ? chalk.magenta : chalk.white;
-                const displayValue = obj._type === 'string' ? `"${obj._value}"` : String(obj._value);
-                console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${typeColor(displayValue)} ${chalk.dim(`(${obj._type})`)}`);
-                return;
-            }
-            if (obj._type === 'function') {
-                console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${chalk.magenta(`ƒ ${obj._name}(${obj._length} args)`)}`);
-                return;
-            }
-            if (obj._type === 'array') {
-                console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${chalk.dim(`Array(${obj._length})`)}${obj._truncated ? chalk.dim(' [truncated]') : ''}`);
-                const items = obj._items || [];
-                items.forEach((item, i) => {
-                    printTree(item, childIndent, i === items.length - 1, String(item.key));
-                });
-                return;
-            }
-            if (obj._type === 'Error') {
-                console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${chalk.red(`${obj._name}: ${obj._message}`)}`);
-                return;
-            }
-            // Object with properties
-            const typeName = obj._type || 'Object';
-            console.log(`${prefix}${keyName ? chalk.cyan(keyName) + ': ' : ''}${chalk.dim(typeName)}${obj._truncated ? chalk.dim(' [truncated]') : ''}`);
-            if (obj._properties) {
-                const keys = Object.keys(obj._properties);
-                keys.forEach((key, i) => {
-                    printTree(obj._properties[key], childIndent, i === keys.length - 1, key);
-                });
-            }
-        }
-        printTree(result);
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// LOGS COMMAND - All Console Logs
-// ============================================
-program
-    .command('logs <url>')
-    .alias('l')
-    .description('Get all console logs from a URL')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--filter <type>', 'Filter by type: log, info, warn, error, debug')
-    .option('--limit <n>', 'Limit number of logs shown', '50')
-    .option('--json', 'Output as JSON')
-    .action(async (url, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
-        spinner.text = `Connecting to ${url}`;
-        await trace.connect(url);
-        spinner.text = 'Collecting logs...';
-        const summary = await trace.getConsoleSummary();
-        await trace.disconnect();
-        spinner.stop();
-        let logs = summary?.logs || [];
-        // Apply filter
-        if (options.filter) {
-            logs = logs.filter((l) => l.type === options.filter);
-        }
-        // Apply limit
-        const limit = parseInt(options.limit) || 50;
-        logs = logs.slice(-limit);
-        if (options.json) {
-            console.log(JSON.stringify({ total: logs.length, logs }, null, 2));
-            return;
-        }
-        console.log();
-        console.log(chalk.bold(`📋 Console Logs (${logs.length})`));
-        console.log(chalk.gray('─'.repeat(60)));
-        if (logs.length === 0) {
-            console.log(chalk.dim('No logs captured.'));
-        }
-        else {
-            for (const log of logs) {
-                const typeColors = {
-                    error: chalk.red,
-                    warn: chalk.yellow,
-                    log: chalk.white,
-                    info: chalk.blue,
-                    debug: chalk.gray
-                };
-                const color = typeColors[log.type] || chalk.white;
-                const icon = log.type === 'error' ? '●' : log.type === 'warn' ? '▲' : '○';
-                console.log();
-                console.log(`  ${color(icon)} ${color.bold(log.type.toUpperCase())}`);
-                console.log(`    ${log.message.substring(0, 200)}`);
-                if (log.location) {
-                    console.log(`    ${chalk.dim(log.location)}`);
-                }
-            }
-        }
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// STREAM COMMAND - Real-time Console Streaming
-// ============================================
-program
-    .command('stream <url>')
-    .alias('s')
-    .description('Stream console logs in real-time (like tail -f)')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--filter <type>', 'Filter by type: log, info, warn, error, debug')
-    .action(async (url, options) => {
-    let trace = null;
-    try {
-        console.log();
-        console.log(chalk.bold.cyan('🔴 Live Console Stream'));
-        console.log(chalk.gray('─'.repeat(60)));
-        console.log(chalk.dim(`Connecting to ${url}...`));
-        console.log(chalk.dim('Press Ctrl+C to stop'));
-        console.log();
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined,
-            verbose: false
-        });
-        await trace.connect(url);
-        console.log(chalk.green('✓ Connected. Streaming logs...'));
-        console.log();
-        // Get initial logs
-        const summary = await trace.getConsoleSummary();
-        let seenIds = new Set((summary?.logs || []).map((l) => l.id));
-        // Poll for new logs
-        const pollInterval = setInterval(async () => {
-            try {
-                const current = await trace.getConsoleSummary();
-                const logs = current?.logs || [];
-                for (const log of logs) {
-                    if (!seenIds.has(log.id)) {
-                        seenIds.add(log.id);
-                        // Apply filter
-                        if (options.filter && log.type !== options.filter) {
-                            continue;
-                        }
-                        const time = new Date(log.timestamp).toLocaleTimeString();
-                        const typeColors = {
-                            error: chalk.red,
-                            warn: chalk.yellow,
-                            log: chalk.white,
-                            info: chalk.blue,
-                            debug: chalk.gray
-                        };
-                        const color = typeColors[log.type] || chalk.white;
-                        console.log(`${chalk.dim(`[${time}]`)} ${color(log.type.toUpperCase().padEnd(5))} ${log.message.substring(0, 150)}`);
-                    }
-                }
-            }
-            catch (e) {
-                // Ignore polling errors
-            }
-        }, 500);
-        // Handle Ctrl+C
-        process.on('SIGINT', async () => {
-            clearInterval(pollInterval);
-            console.log();
-            console.log(chalk.dim('Disconnecting...'));
-            if (trace)
-                await trace.disconnect().catch(() => { });
-            process.exit(0);
-        });
-        // Keep alive
-        await new Promise(() => { });
-    }
-    catch (error) {
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// NETWORK COMMAND
-// ============================================
-program
-    .command('network <url>')
-    .alias('n')
-    .description('Analyze network requests from a URL')
-    .option('--failed', 'Show only failed requests')
-    .option('--json', 'Output as JSON')
-    .action(async (url, options) => {
-    const spinner = ora('Connecting...').start();
-    try {
-        const trace = new Trace({ headless: true });
-        spinner.text = `Connecting to ${url}`;
-        await trace.connect(url);
-        spinner.text = 'Capturing network...';
-        // Wait a bit for network activity
-        await new Promise(r => setTimeout(r, 2000));
-        const summary = await trace.getNetworkSummary();
-        const failed = await trace.getNetworkFailed();
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify({ summary, failed }, null, 2));
-            return;
-        }
-        console.log();
-        console.log(chalk.bold('🌐 Network Summary'));
-        console.log(chalk.gray('─'.repeat(60)));
-        console.log();
-        log.table({
-            'Total Requests': summary.total || 0,
-            'Failed': summary.failed || 0,
-            'Cached': summary.cached || 0,
-            'Total Size': `${Math.round((summary.totalSize || 0) / 1024)} KB`,
-            'Avg Duration': `${Math.round(summary.avgDuration || 0)} ms`,
-        });
-        console.log();
-        if (failed.length > 0 || options.failed) {
-            console.log(chalk.bold(`❌ Failed Requests (${failed.length})`));
-            console.log();
-            for (const req of failed.slice(0, 10)) {
-                console.log(`  ${chalk.red('●')} ${chalk.bold(req.status || 'ERR')} ${req.method || 'GET'}`);
-                console.log(`    ${req.url?.substring(0, 80)}`);
-                if (req.errorText) {
-                    console.log(`    ${chalk.dim(req.errorText)}`);
-                }
-                console.log();
-            }
-        }
-    }
-    catch (error) {
-        spinner.stop();
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// COOKIES COMMAND
-// ============================================
-program
-    .command('cookies <url>')
-    .description('View and manage browser cookies')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--clear', 'Clear all cookies')
-    .option('--delete <name>', 'Delete a specific cookie by name')
-    .option('--json', 'Output as JSON')
-    .action(async (url, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
-        spinner.text = `Connecting to ${url}`;
-        await trace.connect(url);
-        if (options.clear) {
-            spinner.text = 'Clearing cookies...';
-            await trace.executeTool('clear_all_cookies', {});
-            await trace.disconnect();
-            spinner.stop();
-            console.log(chalk.green('✓ All cookies cleared'));
-            return;
-        }
-        if (options.delete) {
-            spinner.text = `Deleting cookie: ${options.delete}`;
-            await trace.executeTool('delete_cookies', { name: options.delete });
-            await trace.disconnect();
-            spinner.stop();
-            console.log(chalk.green(`✓ Deleted cookie: ${options.delete}`));
-            return;
-        }
-        spinner.text = 'Fetching cookies...';
-        const result = await trace.executeTool('get_cookies', {});
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify(result.data, null, 2));
-            return;
-        }
-        const cookies = result.data?.cookies || [];
-        console.log();
-        console.log(chalk.bold(`🍪 Cookies (${cookies.length})`));
-        console.log(chalk.gray('─'.repeat(60)));
-        if (cookies.length === 0) {
-            console.log(chalk.dim('No cookies found.'));
-        }
-        else {
-            for (const cookie of cookies) {
-                console.log();
-                console.log(`  ${chalk.cyan(cookie.name)}`);
-                console.log(`    Value: ${chalk.dim(cookie.value)}`);
-                console.log(`    Domain: ${cookie.domain}${cookie.path}`);
-                console.log(`    Expires: ${cookie.expires}`);
-                console.log(`    ${cookie.httpOnly ? chalk.yellow('HttpOnly') : ''} ${cookie.secure ? chalk.green('Secure') : ''} ${chalk.dim(cookie.sameSite)}`);
-            }
-        }
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// STORAGE COMMAND
-// ============================================
-program
-    .command('storage <url>')
-    .description('View localStorage and sessionStorage')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--session', 'View sessionStorage instead of localStorage')
-    .option('--clear', 'Clear storage')
-    .option('--json', 'Output as JSON')
-    .action(async (url, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
-        spinner.text = `Connecting to ${url}`;
-        await trace.connect(url);
-        const storageType = options.session ? 'Session' : 'Local';
-        const toolName = options.session ? 'get_session_storage' : 'get_local_storage';
-        if (options.clear) {
-            spinner.text = `Clearing ${storageType}Storage...`;
-            await trace.executeTool('clear_storage', { session: !!options.session });
-            await trace.disconnect();
-            spinner.stop();
-            console.log(chalk.green(`✓ ${storageType}Storage cleared`));
-            return;
-        }
-        spinner.text = `Fetching ${storageType}Storage...`;
-        const result = await trace.executeTool(toolName, {});
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify(result.data, null, 2));
-            return;
-        }
-        const items = result.data?.items || [];
-        console.log();
-        console.log(chalk.bold(`💾 ${storageType}Storage (${items.length} items)`));
-        console.log(chalk.gray('─'.repeat(60)));
-        if (items.length === 0) {
-            console.log(chalk.dim('No items found.'));
-        }
-        else {
-            for (const item of items) {
-                console.log();
-                console.log(`  ${chalk.cyan(item.key)} ${chalk.dim(`(${item.size} bytes)`)}`);
-                console.log(`    ${chalk.dim(item.value)}`);
-            }
-        }
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// SERVICE WORKERS COMMAND
-// ============================================
-program
-    .command('sw <url>')
-    .description('View and manage service workers')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--unregister', 'Unregister all service workers')
+    .description('Get console errors from a URL')
     .option('--json', 'Output as JSON')
     .action(async (url, options) => {
     const spinner = ora('Connecting...').start();
-    let trace = null;
     try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
+        const trace = new Trace({ headless: true });
         spinner.text = `Connecting to ${url}`;
         await trace.connect(url);
-        if (options.unregister) {
-            spinner.text = 'Unregistering service workers...';
-            await trace.executeTool('unregister_service_worker', {});
-            await trace.disconnect();
-            spinner.stop();
-            console.log(chalk.green('✓ All service workers unregistered'));
-            return;
-        }
-        spinner.text = 'Fetching service workers...';
-        const result = await trace.executeTool('get_service_workers', {});
+        spinner.text = 'Checking for errors...';
+        const errors = await trace.getConsoleErrors();
+        const groups = await trace.getErrorGroups();
         await trace.disconnect();
         spinner.stop();
         if (options.json) {
-            console.log(JSON.stringify(result.data, null, 2));
+            console.log(JSON.stringify({ errors, groups }, null, 2));
             return;
         }
-        const workers = result.data?.workers || [];
         console.log();
-        console.log(chalk.bold(`⚙️ Service Workers (${workers.length})`));
+        console.log(chalk.bold(`🔴 Console Errors (${errors.length})`));
         console.log(chalk.gray('─'.repeat(60)));
-        if (workers.length === 0) {
-            console.log(chalk.dim('No service workers registered.'));
+        if (errors.length === 0) {
+            log.success('No console errors found!');
         }
         else {
-            for (const sw of workers) {
+            for (const err of errors.slice(0, 20)) {
                 console.log();
-                console.log(`  ${chalk.cyan('Scope:')} ${sw.scope}`);
-                if (sw.active) {
-                    console.log(`    ${chalk.green('Active:')} ${sw.active.scriptURL} (${sw.active.state})`);
-                }
-                if (sw.waiting) {
-                    console.log(`    ${chalk.yellow('Waiting:')} ${sw.waiting.scriptURL}`);
-                }
-                if (sw.installing) {
-                    console.log(`    ${chalk.blue('Installing:')} ${sw.installing.scriptURL}`);
+                console.log(`  ${chalk.red('●')} ${chalk.bold(err.errorType || 'Error')}`);
+                console.log(`    ${err.message.substring(0, 200)}`);
+                if (err.location) {
+                    console.log(`    ${chalk.dim(err.location)}`);
                 }
             }
         }
@@ -972,123 +286,63 @@ program
     }
     catch (error) {
         spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
         log.error(`Failed: ${error}`);
         process.exit(1);
     }
 });
 // ============================================
-// SECURITY COMMAND
+// NETWORK COMMAND
 // ============================================
 program
-    .command('security <url>')
-    .description('Check page security status')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
+    .command('network <url>')
+    .alias('n')
+    .description('Analyze network requests from a URL')
+    .option('--failed', 'Show only failed requests')
     .option('--json', 'Output as JSON')
     .action(async (url, options) => {
     const spinner = ora('Connecting...').start();
-    let trace = null;
     try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
-        });
+        const trace = new Trace({ headless: true });
         spinner.text = `Connecting to ${url}`;
         await trace.connect(url);
-        spinner.text = 'Checking security...';
-        const securityInfo = await trace.executeTool('get_security_info', {});
-        const mixedContent = await trace.executeTool('check_mixed_content', {});
+        spinner.text = 'Capturing network...';
+        // Wait a bit for network activity
+        await new Promise(r => setTimeout(r, 2000));
+        const summary = await trace.getNetworkSummary();
+        const failed = await trace.getNetworkFailed();
         await trace.disconnect();
         spinner.stop();
         if (options.json) {
-            console.log(JSON.stringify({ security: securityInfo.data, mixedContent: mixedContent.data }, null, 2));
+            console.log(JSON.stringify({ summary, failed }, null, 2));
             return;
         }
-        const info = securityInfo.data || {};
-        const mixed = mixedContent.data || {};
         console.log();
-        console.log(chalk.bold('🔒 Security Status'));
+        console.log(chalk.bold('🌐 Network Summary'));
         console.log(chalk.gray('─'.repeat(60)));
         console.log();
-        const secureIcon = info.secure ? chalk.green('✓') : chalk.red('✗');
-        console.log(`  ${secureIcon} ${info.secure ? 'Connection is secure' : 'Connection is NOT secure'}`);
-        console.log(`    Protocol: ${info.protocol}`);
-        console.log(`    URL: ${info.url}`);
-        console.log();
-        if (mixed.hasMixedContent) {
-            console.log(chalk.yellow(`  ⚠️ Mixed Content Issues (${mixed.count})`));
-            for (const issue of (mixed.issues || []).slice(0, 5)) {
-                console.log(`    - ${issue.tag}: ${issue.src}`);
-            }
-        }
-        else {
-            console.log(chalk.green('  ✓ No mixed content issues'));
-        }
-        console.log();
-    }
-    catch (error) {
-        spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
-        log.error(`Failed: ${error}`);
-        process.exit(1);
-    }
-});
-// ============================================
-// PROFILE COMMAND - CPU Profiler
-// ============================================
-program
-    .command('profile <url>')
-    .description('CPU profile the page (like Performance tab)')
-    .option('--connect <port>', 'Connect to existing Chrome instance on port (e.g. 9222)')
-    .option('--duration <ms>', 'Profile duration in milliseconds', '3000')
-    .option('--json', 'Output as JSON')
-    .action(async (url, options) => {
-    const spinner = ora('Connecting...').start();
-    let trace = null;
-    try {
-        trace = new Trace({
-            headless: true,
-            connectPort: options.connect ? parseInt(options.connect) : undefined
+        log.table({
+            'Total Requests': summary.total || 0,
+            'Failed': summary.failed || 0,
+            'Cached': summary.cached || 0,
+            'Total Size': `${Math.round((summary.totalSize || 0) / 1024)} KB`,
+            'Avg Duration': `${Math.round(summary.avgDuration || 0)} ms`,
         });
-        spinner.text = `Connecting to ${url}`;
-        await trace.connect(url);
-        const duration = parseInt(options.duration) || 3000;
-        spinner.text = `Profiling for ${duration}ms...`;
-        const result = await trace.executeTool('get_profile_snapshot', { duration_ms: duration });
-        await trace.disconnect();
-        spinner.stop();
-        if (options.json) {
-            console.log(JSON.stringify(result.data, null, 2));
-            return;
-        }
-        const profile = result.data || {};
-        console.log();
-        console.log(chalk.bold(`🔥 CPU Profile (${profile.totalTime})`));
-        console.log(chalk.gray('─'.repeat(60)));
-        console.log(`Samples: ${profile.sampleCount}`);
         console.log();
-        const hotPaths = profile.hotPaths || [];
-        if (hotPaths.length === 0) {
-            console.log(chalk.dim('No significant CPU activity captured.'));
-        }
-        else {
-            console.log(chalk.bold('Hot Paths:'));
+        if (failed.length > 0 || options.failed) {
+            console.log(chalk.bold(`❌ Failed Requests (${failed.length})`));
             console.log();
-            for (const path of hotPaths.slice(0, 15)) {
-                const barLength = Math.round(parseFloat(path.percentage) / 3);
-                const bar = '█'.repeat(barLength) + '░'.repeat(Math.max(0, 20 - barLength));
-                console.log(`  ${chalk.yellow(bar)} ${path.percentage.padStart(5)}%  ${chalk.cyan(path.function)}`);
-                console.log(`  ${' '.repeat(20)}        ${chalk.dim(path.file)}:${path.line || '?'}`);
+            for (const req of failed.slice(0, 10)) {
+                console.log(`  ${chalk.red('●')} ${chalk.bold(req.status || 'ERR')} ${req.method || 'GET'}`);
+                console.log(`    ${req.url?.substring(0, 80)}`);
+                if (req.errorText) {
+                    console.log(`    ${chalk.dim(req.errorText)}`);
+                }
+                console.log();
             }
         }
-        console.log();
     }
     catch (error) {
         spinner.stop();
-        if (trace)
-            await trace.disconnect().catch(() => { });
         log.error(`Failed: ${error}`);
         process.exit(1);
     }
@@ -1653,525 +907,181 @@ program
     }
 });
 // ============================================
-// FILES COMMAND - Direct code access
-// ============================================
-program
-    .command('files')
-    .alias('f')
-    .description('Show project file tree (no browser needed)')
-    .option('-d, --depth <n>', 'Directory depth', '3')
-    .option('-p, --path <dir>', 'Project directory', process.cwd())
-    .option('--json', 'Output as JSON')
-    .action(async (options) => {
-    const projectPath = path.resolve(options.path);
-    const depth = parseInt(options.depth, 10);
-    console.log();
-    console.log(chalk.bold('📁 Project Files'));
-    console.log(chalk.gray('─'.repeat(60)));
-    console.log(chalk.dim(`Path: ${projectPath}`));
-    console.log();
-    try {
-        const tree = getFileTree(projectPath, depth);
-        if (options.json) {
-            console.log(JSON.stringify(tree, null, 2));
-            return;
-        }
-        printTree(tree, '');
-        console.log();
-    }
-    catch (error) {
-        log.error(`Failed to read directory: ${error}`);
-        process.exit(1);
-    }
-});
-program
-    .command('read <file>')
-    .description('Read a file from the project')
-    .option('-p, --path <dir>', 'Project directory', process.cwd())
-    .option('-l, --lines <range>', 'Line range (e.g., 1-50)')
-    .action(async (file, options) => {
-    const projectPath = path.resolve(options.path);
-    const filePath = path.resolve(projectPath, file);
-    console.log();
-    console.log(chalk.bold(`📄 ${file}`));
-    console.log(chalk.gray('─'.repeat(60)));
-    try {
-        if (!fs.existsSync(filePath)) {
-            log.error(`File not found: ${file}`);
-            process.exit(1);
-        }
-        const content = fs.readFileSync(filePath, 'utf-8');
-        const lines = content.split('\n');
-        let startLine = 1;
-        let endLine = lines.length;
-        if (options.lines) {
-            const [s, e] = options.lines.split('-').map(Number);
-            startLine = s || 1;
-            endLine = e || lines.length;
-        }
-        for (let i = startLine - 1; i < Math.min(endLine, lines.length); i++) {
-            const lineNum = chalk.dim(`${String(i + 1).padStart(4)} │`);
-            console.log(`${lineNum} ${lines[i]}`);
-        }
-        console.log();
-    }
-    catch (error) {
-        log.error(`Failed to read file: ${error}`);
-        process.exit(1);
-    }
-});
-program
-    .command('search <query>')
-    .description('Search for text in the codebase')
-    .option('-p, --path <dir>', 'Project directory', process.cwd())
-    .option('-n, --max <n>', 'Max results', '20')
-    .action(async (query, options) => {
-    const projectPath = path.resolve(options.path);
-    const maxResults = parseInt(options.max, 10);
-    console.log();
-    console.log(chalk.bold(`🔍 Searching: "${query}"`));
-    console.log(chalk.gray('─'.repeat(60)));
-    try {
-        const results = searchInFiles(projectPath, query, maxResults);
-        if (results.length === 0) {
-            console.log(chalk.dim('  No matches found'));
-        }
-        else {
-            for (const match of results) {
-                console.log();
-                console.log(`  ${chalk.cyan(match.file)}:${chalk.yellow(match.line)}`);
-                console.log(`    ${match.content.trim()}`);
-            }
-        }
-        console.log();
-        console.log(chalk.dim(`Found ${results.length} matches`));
-        console.log();
-    }
-    catch (error) {
-        log.error(`Search failed: ${error}`);
-        process.exit(1);
-    }
-});
-function printTree(tree, prefix) {
-    const entries = Object.entries(tree);
-    for (let i = 0; i < entries.length; i++) {
-        const [name, value] = entries[i];
-        const isLast = i === entries.length - 1;
-        const connector = isLast ? '└── ' : '├── ';
-        const isDir = typeof value === 'object' && value !== null;
-        const icon = isDir ? '📁' : '📄';
-        console.log(`${prefix}${connector}${icon} ${name}`);
-        if (isDir) {
-            const newPrefix = prefix + (isLast ? '    ' : '│   ');
-            printTree(value, newPrefix);
-        }
-    }
-}
-// ============================================
-// CONNECT COMMAND - IDE Bridge for Extension
+// IDE BRIDGE - WebSocket Server
 // ============================================
-import { WebSocketServer } from 'ws';
-import * as fs from 'fs';
-import * as path from 'path';
 program
     .command('connect')
-    .alias('c')
-    .description('Start IDE bridge for Chrome extension (enables source code access)')
-    .option('-p, --port <port>', 'WebSocket port', '8765')
+    .alias('ide')
+    .description('Start IDE Bridge WebSocket server for extension <-> VS Code connection')
+    .option('-p, --port <number>', 'WebSocket server port', '8765')
     .action(async (options) => {
+    const { WebSocketServer } = await import('ws');
+    const fs = await import('fs/promises');
+    const path = await import('path');
+    const { glob } = await import('glob');
     const port = parseInt(options.port);
-    const projectPath = process.cwd();
-    console.log();
-    console.log(chalk.bold.cyan('🔗 Trace IDE Bridge'));
-    console.log(chalk.gray('─'.repeat(55)));
-    console.log();
-    console.log(`📁 Project: ${chalk.green(projectPath)}`);
-    console.log(`🌐 Port: ${chalk.cyan(port)}`);
-    console.log();
-    // Read package.json for project info
-    let projectInfo = { projectPath };
-    try {
-        const pkgPath = path.join(projectPath, 'package.json');
-        if (fs.existsSync(pkgPath)) {
-            const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
-            projectInfo = {
-                ...projectInfo,
-                name: pkg.name,
-                version: pkg.version,
-                description: pkg.description
-            };
-            console.log(`📦 Package: ${chalk.yellow(pkg.name)} v${pkg.version}`);
-        }
-    }
-    catch (e) {
-        // No package.json, that's fine
-    }
-    // Start WebSocket server
     const wss = new WebSocketServer({ port });
-    let clientCount = 0;
     console.log();
-    console.log(chalk.green('✓') + ' WebSocket server started');
-    console.log(chalk.dim('Waiting for extension to connect...'));
+    console.log(chalk.bold.cyan('🔌 IDE Bridge Server'));
+    console.log(chalk.gray('─'.repeat(60)));
+    console.log(`  ${chalk.green('✓')} WebSocket server running on ${chalk.cyan(`ws://localhost:${port}`)}`);
+    console.log(`  ${chalk.dim('→')} Project path: ${chalk.white(process.cwd())}`);
     console.log();
-    console.log(chalk.gray('─'.repeat(55)));
+    console.log(chalk.dim('Waiting for extension to connect...'));
     console.log(chalk.dim('Press Ctrl+C to stop'));
     console.log();
     wss.on('connection', (ws) => {
-        clientCount++;
-        console.log(chalk.green('●') + ` Extension connected (${clientCount} client${clientCount > 1 ? 's' : ''})`);
+        console.log(chalk.green('✓') + ' Extension connected');
         ws.on('message', async (data) => {
             try {
                 const message = JSON.parse(data.toString());
                 const { id, type } = message;
-                let response = { id };
+                let response = {};
                 switch (type) {
                     case 'GET_PROJECT_INFO':
-                        response.data = projectInfo;
-                        break;
-                    case 'READ_FILE':
-                        try {
-                            const filePath = path.resolve(projectPath, message.filePath);
-                            // Security: ensure file is within project
-                            if (!filePath.startsWith(projectPath)) {
-                                response.error = 'Access denied';
-                            }
-                            else if (fs.existsSync(filePath)) {
-                                response.data = {
-                                    content: fs.readFileSync(filePath, 'utf-8'),
-                                    exists: true,
-                                    path: filePath
-                                };
-                            }
-                            else {
-                                response.data = { exists: false };
-                            }
-                        }
-                        catch (e) {
-                            response.error = e.message;
-                        }
+                        response = {
+                            projectPath: process.cwd(),
+                            projectName: path.basename(process.cwd()),
+                        };
                         break;
                     case 'GET_SOURCE':
                         try {
-                            const filePath = path.resolve(projectPath, message.filePath);
-                            if (!filePath.startsWith(projectPath)) {
-                                response.error = 'Access denied';
-                            }
-                            else if (fs.existsSync(filePath)) {
-                                const content = fs.readFileSync(filePath, 'utf-8');
-                                const lines = content.split('\n');
-                                const start = Math.max(0, (message.lineStart || 1) - 1);
-                                const end = message.lineEnd ? Math.min(lines.length, message.lineEnd) : lines.length;
-                                response.data = {
-                                    lines: lines.slice(start, end),
-                                    startLine: start + 1,
-                                    endLine: end,
-                                    totalLines: lines.length
-                                };
-                            }
-                            else {
-                                response.error = 'File not found';
-                            }
+                            const { filePath, lineStart, lineEnd } = message;
+                            const fullPath = path.resolve(process.cwd(), filePath);
+                            const content = await fs.readFile(fullPath, 'utf-8');
+                            const lines = content.split('\n');
+                            const start = Math.max(0, (lineStart || 1) - 1);
+                            const end = lineEnd ? Math.min(lines.length, lineEnd) : lines.length;
+                            response = {
+                                filePath,
+                                content: lines.slice(start, end).join('\n'),
+                                lineStart: start + 1,
+                                lineEnd: end,
+                            };
                         }
-                        catch (e) {
-                            response.error = e.message;
+                        catch (error) {
+                            response = { error: error.message };
                         }
                         break;
-                    case 'GET_ERROR_CONTEXT':
+                    case 'READ_FILE':
                         try {
-                            const filePath = path.resolve(projectPath, message.filePath);
-                            if (!filePath.startsWith(projectPath)) {
-                                response.error = 'Access denied';
-                            }
-                            else if (fs.existsSync(filePath)) {
-                                const content = fs.readFileSync(filePath, 'utf-8');
-                                const lines = content.split('\n');
-                                const targetLine = message.line || 1;
-                                const contextLines = message.contextLines || 5;
-                                const start = Math.max(0, targetLine - contextLines - 1);
-                                const end = Math.min(lines.length, targetLine + contextLines);
-                                response.data = {
-                                    lines: lines.slice(start, end).map((line, i) => ({
-                                        number: start + i + 1,
-                                        content: line,
-                                        isError: start + i + 1 === targetLine
-                                    })),
-                                    errorLine: targetLine,
-                                    filePath: message.filePath
-                                };
-                            }
-                            else {
-                                response.error = 'File not found';
-                            }
+                            const { filePath } = message;
+                            const fullPath = path.resolve(process.cwd(), filePath);
+                            const content = await fs.readFile(fullPath, 'utf-8');
+                            response = content; // Return content directly
                         }
-                        catch (e) {
-                            response.error = e.message;
+                        catch (error) {
+                            response = { error: error.message };
                         }
                         break;
-                    case 'GET_FILE_TREE':
+                    case 'FIND_FILES':
                         try {
-                            const depth = message.depth || 3;
-                            const tree = getFileTree(projectPath, depth);
-                            response.data = { tree };
+                            const { pattern } = message;
+                            // Use glob to find files matching the pattern
+                            const files = await glob(pattern, {
+                                cwd: process.cwd(),
+                                ignore: ['**/node_modules/**', '**/dist/**', '**/build/**', '**/.git/**', '**/__pycache__/**', '**/.next/**'],
+                                nodir: true, // Only files, not directories
+                            });
+                            response = files; // Return array of matching file paths
                         }
-                        catch (e) {
-                            response.error = e.message;
+                        catch (error) {
+                            response = { error: error.message };
                         }
                         break;
                     case 'SEARCH_CODE':
                         try {
-                            const query = message.query;
-                            const matches = searchInFiles(projectPath, query, 20);
-                            response.data = { matches };
-                        }
-                        catch (e) {
-                            response.error = e.message;
-                        }
-                        break;
-                    // ========== DEEP DEBUGGING TOOLS ==========
-                    case 'GIT_BLAME':
-                        try {
-                            const filePath = message.filePath;
-                            const line = message.line;
-                            // git blame -L n,n --porcelain file
-                            const { stdout } = await execAsync(`git blame -L ${line},${line} --porcelain "${filePath}"`, { cwd: projectPath });
-                            // Parse porcelain output
-                            const lines = stdout.split('\n');
-                            const commitHash = lines[0].split(' ')[0];
-                            const author = lines.find(l => l.startsWith('author '))?.substring(7);
-                            const email = lines.find(l => l.startsWith('author-mail '))?.substring(12);
-                            const date = lines.find(l => l.startsWith('author-time '))?.substring(12);
-                            const summary = lines.find(l => l.startsWith('summary '))?.substring(8);
-                            response.data = {
-                                commit: commitHash,
-                                author,
-                                email,
-                                date: new Date(parseInt(date) * 1000).toISOString().split('T')[0],
-                                message: summary
-                            };
-                        }
-                        catch (e) {
-                            response.error = `Git blame failed: ${e.message}`;
-                        }
-                        break;
-                    case 'GIT_RECENT_CHANGES':
-                        try {
-                            const filePath = message.filePath;
-                            const days = message.days || 7;
-                            // git log -n 10 --pretty=format:"%h|%an|%ad|%s" --date=short file
-                            const { stdout } = await execAsync(`git log -n 10 --since="${days} days ago" --pretty=format:"%h|%an|%ad|%s" --date=short "${filePath}"`, { cwd: projectPath });
-                            response.data = {
-                                history: stdout.split('\n').filter(Boolean).map(line => {
-                                    const [hash, author, date, message] = line.split('|');
-                                    return { hash, author, date, message };
-                                })
-                            };
-                        }
-                        catch (e) {
-                            response.error = `Git log failed: ${e.message}`;
-                        }
-                        break;
-                    case 'GET_IMPORTS':
-                        try {
-                            const filePath = path.resolve(projectPath, message.filePath);
-                            if (fs.existsSync(filePath)) {
-                                const content = fs.readFileSync(filePath, 'utf-8');
-                                // Basic match for imports
-                                const importRegex = /import\s+(?:[\w*\s{},]*)\s+from\s+['"]([^'"]+)['"]/g;
-                                const imports = [];
-                                let match;
-                                while ((match = importRegex.exec(content)) !== null) {
-                                    imports.push(match[1]);
+                            const { query, options = {} } = message;
+                            const pattern = options.filePattern || '**/*.{js,ts,jsx,tsx,vue,css,html,py,go,java,rb,rs}';
+                            const files = await glob(pattern, {
+                                cwd: process.cwd(),
+                                ignore: ['**/node_modules/**', '**/dist/**', '**/build/**', '**/.git/**'],
+                            });
+                            const results = [];
+                            for (const file of files.slice(0, 100)) { // Limit to 100 files
+                                try {
+                                    const fullPath = path.resolve(process.cwd(), file);
+                                    const content = await fs.readFile(fullPath, 'utf-8');
+                                    if (content.toLowerCase().includes(query.toLowerCase())) {
+                                        const lines = content.split('\n');
+                                        const matches = lines
+                                            .map((line, idx) => ({ line, lineNumber: idx + 1 }))
+                                            .filter(({ line }) => line.toLowerCase().includes(query.toLowerCase()))
+                                            .slice(0, 5); // Max 5 matches per file
+                                        results.push({
+                                            file,
+                                            matches: matches.map((m) => ({
+                                                line: m.lineNumber,
+                                                text: m.line.trim(),
+                                            })),
+                                        });
+                                    }
                                 }
-                                response.data = { imports };
-                            }
-                            else {
-                                response.error = 'File not found';
+                                catch { }
                             }
+                            response = { results };
                         }
-                        catch (e) {
-                            response.error = e.message;
+                        catch (error) {
+                            response = { error: error.message };
                         }
                         break;
-                    case 'FIND_USAGES':
+                    case 'GET_FILE_TREE':
                         try {
-                            const query = message.query;
-                            // git grep -n "query"
-                            const { stdout } = await execAsync(`git grep -n "${query}"`, { cwd: projectPath });
-                            response.data = {
-                                usages: stdout.split('\n').filter(Boolean).slice(0, 20).map(line => {
-                                    const parts = line.split(':');
-                                    return {
-                                        file: parts[0],
-                                        line: parseInt(parts[1]),
-                                        content: parts.slice(2).join(':').trim()
-                                    };
-                                })
-                            };
+                            const { depth = 3 } = message;
+                            const files = await glob('**/*', {
+                                cwd: process.cwd(),
+                                ignore: ['**/node_modules/**', '**/dist/**', '**/build/**', '**/.git/**'],
+                                maxDepth: depth,
+                            });
+                            response = { tree: files };
                         }
-                        catch (e) {
-                            // grep returns 1 provided no matches found
-                            if (e.code === 1)
-                                response.data = { usages: [] };
-                            else
-                                response.error = `Grep failed: ${e.message}`;
+                        catch (error) {
+                            response = { error: error.message };
                         }
                         break;
-                    case 'GET_RELATED_FILES':
+                    case 'GET_ERROR_CONTEXT':
                         try {
-                            const filePath = message.filePath;
-                            // Find files often committed together
-                            // git log --name-only --pretty=format:"" file | sort | uniq -c | sort -nr | head -n 10
-                            const cmd = `git log --name-only --pretty=format:"" "${filePath}" | sort | uniq -c | sort -nr | head -n 10`;
-                            const { stdout } = await execAsync(cmd, { cwd: projectPath });
-                            response.data = {
-                                related: stdout.split('\n').filter(Boolean).map(line => {
-                                    const parts = line.trim().split(' ');
-                                    return {
-                                        count: parseInt(parts[0]),
-                                        file: parts[1]
-                                    };
-                                }).filter(r => r.file && r.file !== filePath)
+                            const { filePath: file, line, contextLines = 10 } = message;
+                            const fullPath = path.resolve(process.cwd(), file);
+                            const content = await fs.readFile(fullPath, 'utf-8');
+                            const lines = content.split('\n');
+                            const start = Math.max(0, line - contextLines);
+                            const end = Math.min(lines.length, line + contextLines);
+                            response = {
+                                file,
+                                line,
+                                context: lines.slice(start, end).join('\n'),
+                                lineStart: start + 1,
                             };
                         }
-                        catch (e) {
-                            response.error = `Analysis failed: ${e.message}`;
-                        }
-                        break;
-                    case 'GET_ENV_VARS':
-                        try {
-                            const filePath = path.resolve(projectPath, message.filePath);
-                            if (fs.existsSync(filePath)) {
-                                const content = fs.readFileSync(filePath, 'utf-8');
-                                // Match process.env.VAR or import.meta.env.VAR
-                                const envRegex = /(?:process\.env\.|import\.meta\.env\.)([A-Z_][A-Z0-9_]*)/g;
-                                const vars = new Set();
-                                let match;
-                                while ((match = envRegex.exec(content)) !== null) {
-                                    vars.add(match[1]);
-                                }
-                                response.data = { envVars: Array.from(vars) };
-                            }
-                            else {
-                                response.error = 'File not found';
-                            }
-                        }
-                        catch (e) {
-                            response.error = e.message;
+                        catch (error) {
+                            response = { error: error.message };
                         }
                         break;
                     default:
-                        response.error = `Unknown message type: ${type}`;
+                        response = { error: `Unknown command: ${type}` };
                 }
-                ws.send(JSON.stringify(response));
+                ws.send(JSON.stringify({ id, type, data: response }));
             }
-            catch (e) {
-                console.error(chalk.red('Parse error:'), e.message);
+            catch (error) {
+                console.error(chalk.red('Error:'), error.message);
+                ws.send(JSON.stringify({
+                    id: JSON.parse(data.toString()).id,
+                    error: error.message
+                }));
             }
         });
         ws.on('close', () => {
-            clientCount--;
-            console.log(chalk.yellow('●') + ` Extension disconnected (${clientCount} client${clientCount > 1 ? 's' : ''})`);
-        });
-        ws.on('error', (error) => {
-            console.error(chalk.red('WebSocket error:'), error.message);
+            console.log(chalk.yellow('⚠') + ' Extension disconnected');
         });
     });
-    wss.on('error', (error) => {
-        if (error.code === 'EADDRINUSE') {
-            console.log(chalk.red(`✗ Port ${port} is already in use`));
-            console.log(chalk.dim('Try: trace connect --port 8766'));
-        }
-        else {
-            console.error(chalk.red('Server error:'), error.message);
-        }
-        process.exit(1);
-    });
-    // Keep alive
     process.on('SIGINT', () => {
         console.log();
-        console.log(chalk.dim('Stopping...'));
+        console.log(chalk.dim('Shutting down...'));
         wss.close();
         process.exit(0);
     });
 });
-// Helper: Get file tree
-function getFileTree(dir, depth, currentDepth = 0) {
-    if (currentDepth >= depth)
-        return null;
-    const result = {};
-    const ignorePatterns = ['node_modules', '.git', 'dist', 'build', '.next', 'coverage', '.cache'];
-    try {
-        const items = fs.readdirSync(dir);
-        for (const item of items.slice(0, 50)) { // Limit items
-            if (ignorePatterns.includes(item) || item.startsWith('.'))
-                continue;
-            const fullPath = path.join(dir, item);
-            try {
-                const stat = fs.statSync(fullPath);
-                if (stat.isDirectory()) {
-                    result[item] = getFileTree(fullPath, depth, currentDepth + 1);
-                }
-                else {
-                    result[item] = 'file';
-                }
-            }
-            catch (e) {
-                // Skip inaccessible files
-            }
-        }
-    }
-    catch (e) {
-        // Skip inaccessible directories
-    }
-    return result;
-}
-// Helper: Simple search in files
-function searchInFiles(dir, query, maxResults) {
-    const results = [];
-    const extensions = ['.js', '.ts', '.jsx', '.tsx', '.vue', '.svelte', '.css', '.html', '.json'];
-    const ignorePatterns = ['node_modules', '.git', 'dist', 'build', '.next'];
-    function searchDir(currentDir) {
-        if (results.length >= maxResults)
-            return;
-        try {
-            const items = fs.readdirSync(currentDir);
-            for (const item of items) {
-                if (results.length >= maxResults)
-                    return;
-                if (ignorePatterns.includes(item) || item.startsWith('.'))
-                    continue;
-                const fullPath = path.join(currentDir, item);
-                try {
-                    const stat = fs.statSync(fullPath);
-                    if (stat.isDirectory()) {
-                        searchDir(fullPath);
-                    }
-                    else if (extensions.some(ext => item.endsWith(ext))) {
-                        const content = fs.readFileSync(fullPath, 'utf-8');
-                        const lines = content.split('\n');
-                        for (let i = 0; i < lines.length && results.length < maxResults; i++) {
-                            if (lines[i].includes(query)) {
-                                results.push({
-                                    file: path.relative(dir, fullPath),
-                                    line: i + 1,
-                                    content: lines[i].trim().substring(0, 200)
-                                });
-                            }
-                        }
-                    }
-                }
-                catch (e) {
-                    // Skip inaccessible files
-                }
-            }
-        }
-        catch (e) {
-            // Skip inaccessible directories
-        }
-    }
-    searchDir(dir);
-    return results;
-}
 // ============================================
 // RUN CLI
 // ============================================