@tng-sh/js 0.2.1 → 0.2.4

package/bin/tng.js

@@ -28,7 +28,7 @@ process.on('uncaughtException', (err) => {
 program
     .name('tng')
     .description('TNG - Advanced Code Audit, Test Generation, Visualization, Clone Detection, and Dead Code Analysis for JavaScript/TypeScript')
-    .version('0.2.1');
+    .version('0.2.4');
 
 /**
  * Copy text to system clipboard
@@ -256,139 +256,172 @@ program
     .option('--all', 'Run dead code detection across the entire repo (respects .gitignore)')
     .option('--xray', 'Generate X-Ray visualization (Mermaid flowchart)')
     .option('--impact', 'Run impact analysis (blast radius check) on a method')
+    .option('--callsites', 'Find in-repo call sites for a method')
+    .option('--class <name>', 'Class/module name to disambiguate methods when multiple classes define the same method in a file')
     .option('--json', 'Output results as JSON events (machine-readable)')
 
     .action(async (options) => {
-        if (options.method && options.file) {
-            if (options.trace) {
-                const { getSymbolicTrace } = require('../index');
-                const GoUISession = require('../lib/goUiSession');
-
-                try {
-                    const traceJson = getSymbolicTrace(
-                        path.resolve(options.file),
-                        options.method,
-                        null
-                    );
-
-                    if (options.json) {
-                        console.log(traceJson);
-                        return;
-                    }
-
-                    // Create temp file for the trace
-                    const tmpDir = require('os').tmpdir();
-                    const tmpFile = path.join(tmpDir, `trace-${Date.now()}.json`);
-                    fs.writeFileSync(tmpFile, traceJson);
+        if (options.callsites) {
+            if (!options.method) {
+                console.log(chalk.red('Error: --method is required for call site analysis.'));
+                process.exit(1);
+            }
 
-                    // Launch Go UI
-                    const session = new GoUISession();
-                    const binaryPath = session._binaryPath;
+            try {
+                const { findCallSites } = require('../index');
+                const projectRoot = process.cwd();
+                const resultJson = findCallSites(projectRoot, options.method);
 
-                    const { spawnSync } = require('child_process');
-                    spawnSync(binaryPath, ['js-trace-results', '--file', tmpFile], {
-                        stdio: 'inherit'
-                    });
+                if (options.json) {
+                    console.log(resultJson);
+                    return;
+                }
 
-                    // Cleanup
-                    try { fs.unlinkSync(tmpFile); } catch (e) { }
+                const parsed = JSON.parse(resultJson);
+                const sites = Array.isArray(parsed) ? parsed : (parsed.call_sites || parsed.sites || parsed);
+                if (!sites || sites.length === 0) {
+                    console.log(chalk.green('\nNo call sites found in this project. You might be safe!'));
+                    return;
+                }
 
-                } catch (e) {
-                    console.log(chalk.red(`Trace failed: ${e.message}`));
-                    process.exit(1);
+                console.log(chalk.blue(`\nFound ${sites.length} call sites:\n`));
+                sites.slice(0, 200).forEach((site) => {
+                    const file = site.file || 'unknown';
+                    const line = site.line || 0;
+                    const content = site.content || '';
+                    console.log(`${chalk.yellow(file)}:${chalk.cyan(line)} ${content}`);
+                });
+                if (sites.length > 200) {
+                    console.log(chalk.dim(`\n... ${sites.length - 200} more`));
                 }
                 return;
+            } catch (e) {
+                console.log(chalk.red(`Error: ${e.message}`));
+                process.exit(1);
             }
+        }
+        // 1. Symbolic Trace
+        if (options.method && options.file && options.trace) {
+            const { getSymbolicTrace } = require('../index');
+            const GoUISession = require('../lib/goUiSession');
 
-            if (options.xray) {
-                const config = loadConfig();
-                const { generateTest: nativeGenerateTest } = require('../index');
+            try {
+            const traceJson = getSymbolicTrace(
+                path.resolve(options.file),
+                options.method,
+                options.class || null
+            );
 
-                try {
-                    const resultJson = nativeGenerateTest(
-                        path.resolve(options.file),
-                        options.method,
-                        null,
-                        'visualize',
-                        JSON.stringify(config),
-                        (msg, percent) => {
-                            if (options.json) return;
-                            if (percent % 20 === 0) console.log(chalk.dim(`[${percent}%] ${msg}`));
-                        }
-                    );
-
-                    if (options.json) {
-                        console.log(resultJson);
-                        return;
-                    }
+                if (options.json) {
+                    console.log(traceJson);
+                    return;
+                }
+
+                // Create temp file for the trace
+                const tmpDir = require('os').tmpdir();
+                const tmpFile = path.join(tmpDir, `trace-${Date.now()}.json`);
+                fs.writeFileSync(tmpFile, traceJson);
+
+                // Launch Go UI
+                const session = new GoUISession();
+                const binaryPath = session._binaryPath;
+
+                const { spawnSync } = require('child_process');
+                spawnSync(binaryPath, ['js-trace-results', '--file', tmpFile], {
+                    stdio: 'inherit'
+                });
+
+                // Cleanup
+                try { fs.unlinkSync(tmpFile); } catch (e) { }
+
+            } catch (e) {
+                console.log(chalk.red(`Trace failed: ${e.message}`));
+                process.exit(1);
+            }
+            return;
+        }
 
-                    // For global --xray, we just launch the Premium UI directly if not JSON
-                    const GoUISession = require('../lib/goUiSession');
-                    const session = new GoUISession();
-                    let mermaidCode = "";
-                    let explanation = "";
-                    try {
-                        const parsed = JSON.parse(resultJson);
-                        mermaidCode = parsed.mermaid_code || resultJson;
-                        explanation = parsed.explanation || "";
-                    } catch (e) {
-                        mermaidCode = resultJson;
+        // 2. X-Ray
+        if (options.method && options.file && options.xray) {
+            const config = loadConfig();
+            const { generateTest: nativeGenerateTest } = require('../index');
+
+            try {
+                const resultJson = nativeGenerateTest(
+                    path.resolve(options.file),
+                    options.method,
+                    options.class || null,
+                    'visualize',
+                    JSON.stringify(config),
+                    (msg, percent) => {
+                        if (options.json) return;
+                        if (percent % 20 === 0) console.log(chalk.dim(`[${percent}%] ${msg}`));
                     }
+                );
 
-                    // Copy to clipboard
-                    copyToClipboard(mermaidCode);
+                if (options.json) {
+                    console.log(resultJson);
+                    return;
+                }
 
-                    await session.showXrayResults(options.method, '', mermaidCode, explanation);
+                const GoUISession = require('../lib/goUiSession');
+                const session = new GoUISession();
+                let mermaidCode = "";
+                let explanation = "";
+                try {
+                    const parsed = JSON.parse(resultJson);
+                    mermaidCode = parsed.mermaid_code || resultJson;
+                    explanation = parsed.explanation || "";
                 } catch (e) {
-                    console.log(chalk.red(`X-Ray failed: ${e.message}`));
-                    process.exit(1);
+                    mermaidCode = resultJson;
                 }
-                return;
+
+                copyToClipboard(mermaidCode);
+                await session.showXrayResults(options.method, '', mermaidCode, explanation);
+            } catch (e) {
+                console.log(chalk.red(`X-Ray failed: ${e.message}`));
+                process.exit(1);
             }
+            return;
+        }
 
+        // 3. Impact Analysis
+        if (options.method && options.file && options.impact) {
+            await runImpact(options.file, options.method, options.json, options.class || null);
+            return;
+        }
 
-            if (options.impact) {
-                await runImpact(options.file, options.method, options.json);
-                return;
-            }
+        // 4. Clone Detection
+        if (options.file && options.clones) {
+            await runClones(options.file, options.level || 'all', options.json);
+            return;
+        }
 
-            if (!options.type && !options.audit) {
-                console.log(chalk.red('Error: --type <type> is required.'));
-                process.exit(1);
-            }
-            generateTest(options.file, options.method, options.type, options.audit, options.json);
-        } else if (options.file && !options.method) {
-            if (options.clones) {
-                await runClones(options.file, options.level || 'all', options.json);
-            } else if (options.deadcode) {
-                if (options.all) {
-                    await runDeadCodeRepo(options.json);
-                } else {
-                    await runDeadCode(options.file, options.json);
-                }
+        // 5. Dead Code Analysis
+        if (options.deadcode) {
+            if (options.all) {
+                await runDeadCodeRepo(options.json);
+            } else if (options.file) {
+                await runDeadCode(options.file, options.json);
             } else {
-                console.log(chalk.yellow('Specify a method with -m, use --outline to see methods, or run "tng i" for full selection.'));
-            }
-            if (options.xray) {
-                // ... (existing xray logic) ...
-                return;
-            }
-
-            if (options.impact) {
-                await runImpact(options.file, options.method, options.json);
-                return;
+                console.log(chalk.red('Error: --file <path> or --all is required for deadcode analysis.'));
+                process.exit(1);
             }
+            return;
+        }
 
+        // 6. Test Generation / Audit (Method + File required)
+        if (options.method && options.file) {
             if (!options.type && !options.audit) {
                 console.log(chalk.red('Error: --type <type> is required.'));
                 process.exit(1);
             }
-            generateTest(options.file, options.method, options.type, options.audit, options.json);
-        } else if (options.deadcode && options.all && !options.file && !options.method) {
-            await runDeadCodeRepo(options.json);
-        } else if (options.file && !options.method) {
-            launchInteractive();
+            generateTest(options.file, options.method, options.type, options.audit, options.json, options.class || null);
+            return;
         }
+
+        // 7. Interactive Fallback
+        launchInteractive();
     });
 
 /**
@@ -674,7 +707,7 @@ program
 /**
  * Logic to generate test or run audit (delegates to native binary)
  */
-async function generateTest(filePath, methodName, testType, auditMode = false, jsonMode = false) {
+async function generateTest(filePath, methodName, testType, auditMode = false, jsonMode = false, className = null) {
     const config = loadConfig();
 
     // Initialize JSON session if requested
@@ -744,7 +777,7 @@ async function generateTest(filePath, methodName, testType, auditMode = false, j
             resultJson = runAudit(
                 absolutePath,
                 methodName,
-                null, // class_name
+                className,
                 testType || null,
                 JSON.stringify(config),
                 callback
@@ -753,7 +786,7 @@ async function generateTest(filePath, methodName, testType, auditMode = false, j
             resultJson = nativeGenerateTest(
                 absolutePath,
                 methodName,
-                null, // class_name
+                className,
                 testType || null,
                 JSON.stringify(config),
                 callback
@@ -824,7 +857,7 @@ program.on('--help', () => {
 /**
  * Logic to run impact analysis
  */
-async function runImpact(filePath, methodName, jsonMode = false) {
+async function runImpact(filePath, methodName, jsonMode = false, className = null) {
     const { analyzeImpact } = require('../index');
     const absolutePath = path.resolve(filePath);
     const projectRoot = process.cwd();
@@ -836,7 +869,7 @@ async function runImpact(filePath, methodName, jsonMode = false) {
 
     if (jsonMode) {
         try {
-            const resultJson = analyzeImpact(projectRoot, absolutePath, methodName);
+            const resultJson = analyzeImpact(projectRoot, absolutePath, methodName, className);
             console.log(resultJson);
         } catch (e) {
             console.error(JSON.stringify({ error: e.message }));
@@ -848,7 +881,7 @@ async function runImpact(filePath, methodName, jsonMode = false) {
     console.log(chalk.blue(`💥 Analyzing impact for ${methodName} in ${filePath}...`));
 
     try {
-        const resultJson = analyzeImpact(projectRoot, absolutePath, methodName);
+        const resultJson = analyzeImpact(projectRoot, absolutePath, methodName, className);
         const result = JSON.parse(resultJson);
 
         if (result.status === 'error') {

package/index.d.ts

@@ -69,4 +69,4 @@ export declare function listDeadcodeFiles(projectRoot: string): string
  * Analyzes the impact of changes in a specific method compared to git HEAD.
  * Returns a JSON string containing breaking changes and impacted files.
  */
-export declare function analyzeImpact(projectRoot: string, filePath: string, methodName: string): string
+export declare function analyzeImpact(projectRoot: string, filePath: string, methodName: string, className?: string | undefined | null): string

package/lib/generateTestsUi.js

@@ -6,7 +6,6 @@ const GoUISession = require('./goUiSession');
 const { getFileOutline, generateTest, ping, getUserStats } = require('../index');
 const { loadConfig } = require('./config');
 const { saveTestFile } = require('./saveFile');
-const { applyFix } = require('./fixApplier');
 
 class GenerateTestsUI {
     constructor(cliMode = false) {
@@ -127,7 +126,7 @@ class GenerateTestsUI {
 
         let title = 'Select JavaScript File';
         if (isAudit) title = 'Select JavaScript File to Audit';
-        else if (isImpact) title = 'Select JavaScript File to Impact Audit';
+        else if (isImpact) title = 'Select JavaScript File for Regression Check';
         else if (isXray) title = 'Select File for X-Ray';
         else if (isTrace) title = 'Select File for Symbolic Trace';
 
@@ -160,14 +159,14 @@ class GenerateTestsUI {
 
         const fileName = path.basename(filePath);
         const items = methods.map(m => ({
-            name: m.class_name ? `${m.class_name}.${m.name}` : m.name,
+            name: m.class_name ? `${m.class_name}#${m.name}` : m.name,
             path: `Function in ${fileName}`,
             methodData: m
         }));
 
         let title = 'Select Method';
         if (isAudit) title = `Select Method to Audit for ${fileName}`;
-        else if (isImpact) title = `Select Method to Impact Audit for ${fileName}`;
+        else if (isImpact) title = `Select Method for Regression Check for ${fileName}`;
         else if (isXray) title = `Select Method to X-Ray for ${fileName}`;
         else if (isTrace) title = `Select Method to Trace for ${fileName}`;
 
@@ -179,7 +178,7 @@ class GenerateTestsUI {
 
         if (selectedMethod) {
             if (isTrace) {
-                await this._launchTrace(filePath, selectedMethod.name);
+                await this._launchTrace(filePath, selectedMethod.name, selectedMethod.class_name || null);
                 return this._showFileSelection(isAudit, isXray, isTrace, isImpact);
             }
 
@@ -259,7 +258,7 @@ class GenerateTestsUI {
                 // Artificial delay for UX
                 await new Promise(r => setTimeout(r, 300));
 
-                const json = analyzeImpact(projectRoot, filePath, method.name);
+                const json = analyzeImpact(projectRoot, filePath, method.name, method.class_name || null);
                 progress.update('Comparing logic...', { percent: 80 });
 
                 return json;
@@ -272,7 +271,7 @@ class GenerateTestsUI {
         }
 
         await this.goUiSession.showImpactResults(impactResult);
-        return { message: 'Impact audit complete' };
+        return { message: 'Regression check complete' };
     }
 
     async _handleAuditFlow(filePath, method, testType) {
@@ -600,7 +599,7 @@ class GenerateTestsUI {
         }
     }
 
-    async _launchTrace(filePath, methodName) {
+    async _launchTrace(filePath, methodName, className) {
         const { getSymbolicTrace } = require('../index');
         const fs = require('fs');
         const path = require('path');
@@ -610,7 +609,7 @@ class GenerateTestsUI {
             // 1. Generate Trace (with Spinner)
             const result = this.goUiSession.showSpinner(`Tracing ${methodName}...`, () => {
                 try {
-                    const traceJson = getSymbolicTrace(filePath, methodName, null);
+                    const traceJson = getSymbolicTrace(filePath, methodName, className || null);
                     const tmpDir = require('os').tmpdir();
                     const f = path.join(tmpDir, `trace-${Date.now()}.json`);
                     fs.writeFileSync(f, traceJson);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@tng-sh/js",
-    "version": "0.2.1",
+    "version": "0.2.4",
     "description": "TNG JavaScript CLI",
     "repository": {
         "type": "git",