@tng-sh/js 0.1.2 → 0.1.3

package/bin/tng.js

@@ -28,7 +28,7 @@ process.on('uncaughtException', (err) => {
 program
     .name('tng')
     .description('TNG - Automated Test Generation, and audit generation for JavaScript')
-    .version('0.1.2');
+    .version('0.1.3');
 
 /**
  * @command init
@@ -210,10 +210,51 @@ program
     .option('-f, --file <path>', 'JavaScript file path')
     .option('-t, --type <type>', 'Component type (react_component, express_handler, etc) [required]')
     .option('-a, --audit', 'Run audit mode instead of test generation')
+    .option('--trace', 'Run symbolic trace visualization')
     .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);
+
+                    // Launch Go UI
+                    const session = new GoUISession();
+                    const binaryPath = session._binaryPath;
+
+                    const { spawnSync } = require('child_process');
+                    spawnSync(binaryPath, ['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;
+            }
+
             if (!options.type && !options.audit) {
                 console.log(chalk.red('Error: --type <type> is required.'));
                 process.exit(1);

package/index.d.ts

@@ -44,3 +44,5 @@ export declare function applyEditsAtomic(operationsJson: string): string
  * Analyzes code, submits a job, and polls for the final generated test.
  */
 export declare function generateTest(filePath: string, methodName: string, className: string | undefined | null, testType: string | undefined | null, configJson: string, callback: (...args: any[]) => any): string
+/** Analyzes a method and generates a symbolic execution trace. */
+export declare function getSymbolicTrace(filePath: string, methodName: string, className?: string | undefined | null): string

package/index.js

@@ -310,7 +310,7 @@ if (!nativeBinding) {
   throw new Error(`Failed to load native binding`)
 }
 
-const { getFileOutline, getProjectMetadata, findCallSites, ping, submitJob, getUserStats, runAudit, applyEdit, applyEditsAtomic, generateTest } = nativeBinding
+const { getFileOutline, getProjectMetadata, findCallSites, ping, submitJob, getUserStats, runAudit, applyEdit, applyEditsAtomic, generateTest, getSymbolicTrace } = nativeBinding
 
 module.exports.getFileOutline = getFileOutline
 module.exports.getProjectMetadata = getProjectMetadata
@@ -322,3 +322,4 @@ module.exports.runAudit = runAudit
 module.exports.applyEdit = applyEdit
 module.exports.applyEditsAtomic = applyEditsAtomic
 module.exports.generateTest = generateTest
+module.exports.getSymbolicTrace = getSymbolicTrace

package/lib/generateTestsUi.js

@@ -37,6 +37,9 @@ class GenerateTestsUI {
             } else if (choice === 'xray') {
                 const result = await this._showFileSelection(false, true);
                 if (result === 'exit') return 'exit';
+            } else if (choice === 'trace') {
+                const result = await this._showFileSelection(false, false, true);
+                if (result === 'exit') return 'exit';
             } else if (choice === 'stats') {
                 await this._showStats();
             } else if (choice === 'about') {
@@ -98,7 +101,7 @@ class GenerateTestsUI {
         });
     }
 
-    async _showFileSelection(isAudit = false, isXray = false) {
+    async _showFileSelection(isAudit = false, isXray = false, isTrace = false) {
         const files = await this._getUserFiles();
 
         if (files.length === 0) {
@@ -112,19 +115,23 @@ class GenerateTestsUI {
             path: path.dirname(file)
         }));
 
-        const title = isXray ? 'Select File for X-Ray' : (isAudit ? 'Select JavaScript File to Audit' : 'Select JavaScript File');
+        let title = 'Select JavaScript File';
+        if (isAudit) title = 'Select JavaScript File to Audit';
+        else if (isXray) title = 'Select File for X-Ray';
+        else if (isTrace) title = 'Select File for Symbolic Trace';
+
         const selectedName = this.goUiSession.showListView(title, items);
 
         if (selectedName === 'back') return 'back';
         if (!selectedName || selectedName === 'exit') return 'exit';
 
         const selectedFile = path.resolve(cwd, selectedName);
-        const result = await this._showMethodsForFile(selectedFile, isAudit, isXray);
+        const result = await this._showMethodsForFile(selectedFile, isAudit, isXray, isTrace);
         if (result === 'main_menu') return 'main_menu';
         return result;
     }
 
-    async _showMethodsForFile(filePath, isAudit = false, isXray = false) {
+    async _showMethodsForFile(filePath, isAudit = false, isXray = false, isTrace = false) {
         let outline;
         try {
             const result = getFileOutline(filePath);
@@ -147,14 +154,23 @@ class GenerateTestsUI {
             methodData: m
         }));
 
-        const title = isXray ? `Select Method to X-Ray for ${fileName}` : (isAudit ? `Select Method to Audit for ${fileName}` : `Select Method for ${fileName}`);
+        let title = 'Select Method';
+        if (isAudit) title = `Select Method to Audit for ${fileName}`;
+        else if (isXray) title = `Select Method to X-Ray for ${fileName}`;
+        else if (isTrace) title = `Select Method to Trace for ${fileName}`;
+
         const selectedDisplay = this.goUiSession.showListView(title, items);
 
-        if (selectedDisplay === 'back' || !selectedDisplay) return this._showFileSelection(isAudit, isXray);
+        if (selectedDisplay === 'back' || !selectedDisplay) return this._showFileSelection(isAudit, isXray, isTrace);
 
         const selectedMethod = items.find(i => i.name === selectedDisplay)?.methodData;
 
         if (selectedMethod) {
+            if (isTrace) {
+                await this._launchTrace(filePath, selectedMethod.name);
+                return this._showFileSelection(isAudit, isXray, isTrace);
+            }
+
             if (isXray) {
                 const choice = await this._generateTestsForMethod(filePath, selectedMethod, 'visualize', false, true);
                 if (choice === 'main_menu') return 'main_menu';
@@ -657,6 +673,41 @@ class GenerateTestsUI {
             return [];
         }
     }
+
+    async _launchTrace(filePath, methodName) {
+        const { getSymbolicTrace } = require('../index');
+        const fs = require('fs');
+        const path = require('path');
+        const chalk = require('chalk');
+
+        try {
+            // 1. Generate Trace (with Spinner)
+            const result = this.goUiSession.showSpinner(`Tracing ${methodName}...`, () => {
+                try {
+                    const traceJson = getSymbolicTrace(filePath, methodName, null);
+                    const tmpDir = require('os').tmpdir();
+                    const f = path.join(tmpDir, `trace-${Date.now()}.json`);
+                    fs.writeFileSync(f, traceJson);
+                    return { success: true, file: f };
+                } catch (e) {
+                    return { success: false, message: e.message };
+                }
+            });
+
+            if (result && result.success && result.file) {
+                // 2. Show Trace UI
+                const { spawnSync } = require('child_process');
+                spawnSync(this.goUiSession._binaryPath, ['trace-results', '--file', result.file], {
+                    stdio: 'inherit'
+                });
+
+                try { fs.unlinkSync(result.file); } catch (e) { }
+            }
+
+        } catch (e) {
+            console.error(chalk.red(`Trace error: ${e.message}`));
+        }
+    }
 }
 
 module.exports = GenerateTestsUI;

package/package.json

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