@tng-sh/js 0.0.9 → 0.1.1

package/bin/tng.js

@@ -7,11 +7,28 @@ const path = require('path');
 const { loadConfig } = require('../lib/config');
 const { saveTestFile } = require('../lib/saveFile');
 const { ping, getUserStats } = require('../index');
+const { applyFix } = require('../lib/fixApplier');
+
+// Handle EPIPE errors gracefully when child process (Go UI) exits
+process.stdout.on('error', (err) => {
+    if (err.code === 'EPIPE') {
+        // Ignore - this happens when Go UI exits and we try to write to closed pipe
+        process.exit(0);
+    }
+});
+
+process.on('uncaughtException', (err) => {
+    if (err.code === 'EPIPE') {
+        // Ignore broken pipe - child process exited
+        process.exit(0);
+    }
+    throw err; // Re-throw other errors
+});
 
 program
     .name('tng')
     .description('TNG - Automated Test Generation, and audit generation for JavaScript')
-    .version('0.0.9');
+    .version('0.1.1');
 
 /**
  * @command init
@@ -150,6 +167,39 @@ program
         }
     });
 
+/**
+ * @command fix
+ * Apply a specific fix to a file
+ */
+program
+    .command('fix')
+    .description('Apply a fix to a file')
+    .option('-f, --file <path>', 'File path to fix')
+    .option('-d, --data <json>', 'JSON data for the audit item containing fix info')
+    .action(async (options) => {
+        if (!options.file || !options.data) {
+            console.log(chalk.red('Error: Both --file and --data are required.'));
+            process.exit(1);
+        }
+
+        try {
+            const item = JSON.parse(options.data);
+            const result = await applyFix(options.file, item);
+            if (result.success) {
+                console.log(chalk.green(`✓ Fix applied to ${options.file}`));
+                if (result.backup_path) {
+                    console.log(chalk.dim(`  Backup created at ${result.backup_path}`));
+                }
+            } else {
+                console.log(chalk.red(`❌ Failed to apply fix: ${result.error}`));
+                process.exit(1);
+            }
+        } catch (e) {
+            console.log(chalk.red(`Error: ${e.message}`));
+            process.exit(1);
+        }
+    });
+
 
 
 /**
@@ -188,7 +238,7 @@ async function generateTest(filePath, methodName, testType, auditMode = false, j
         process.exit(1);
     }
 
-    const { submitAndPoll } = require('../index');
+    const { runAudit, generateTest: nativeGenerateTest } = require('../index');
 
     const action = auditMode ? 'Auditing' : 'Generating test for';
     const startMessage = `🔍 ${action} ${methodName} in ${filePath}...`;
@@ -201,31 +251,45 @@ async function generateTest(filePath, methodName, testType, auditMode = false, j
     } else {
         console.log(chalk.blue(startMessage));
         if (testType) {
-            console.log(chalk.cyan(`ℹ️  Type hint: ${testType}`));
+            console.log(chalk.cyan(`Type hint: ${testType}`));
         }
     }
 
     try {
-        const resultJson = submitAndPoll(
-            absolutePath,
-            methodName,
-            null, // class_name
-            testType || null,
-            auditMode, // audit_mode
-            JSON.stringify(config),
-            (msg, percent) => {
-                if (jsonSession) {
-                    // Emit progress events in JSON mode
-                    const reporter = new (require('../lib/jsonSession').JsonProgressReporter)();
-                    reporter.update(msg, percent);
-                } else {
-                    // Simple console progress for CLI mode
-                    if (percent % 10 === 0 || percent === 100) {
-                        console.log(chalk.cyan(`[${percent}%] ${msg}`));
-                    }
+        let resultJson;
+
+        const callback = (msg, percent) => {
+            if (jsonSession) {
+                // Emit progress events in JSON mode
+                const reporter = new (require('../lib/jsonSession').JsonProgressReporter)();
+                reporter.update(msg, percent);
+            } else {
+                // Simple console progress for CLI mode
+                if (percent % 10 === 0 || percent === 100) {
+                    console.log(chalk.cyan(`[${percent}%] ${msg}`));
                 }
             }
-        );
+        };
+
+        if (auditMode) {
+            resultJson = runAudit(
+                absolutePath,
+                methodName,
+                null, // class_name
+                testType || null,
+                JSON.stringify(config),
+                callback
+            );
+        } else {
+            resultJson = nativeGenerateTest(
+                absolutePath,
+                methodName,
+                null, // class_name
+                testType || null,
+                JSON.stringify(config),
+                callback
+            );
+        }
 
         if (auditMode) {
             // In audit mode, display the results

package/index.d.ts

@@ -3,10 +3,44 @@
 
 /* auto-generated by NAPI-RS */
 
+/**
+ * Analyzes a JavaScript/TypeScript file and returns its structural outline as a JSON string.
+ * Contains information about classes and methods found in the file.
+ */
 export declare function getFileOutline(filePath: string): string
+/**
+ * Gathers project-wide metadata from package.json and related files.
+ * Returns a JSON string containing dependencies, frameworks, and project type.
+ */
 export declare function getProjectMetadata(projectRoot: string): string
+/**
+ * Searches the project for all locations where a specific method is called.
+ * Uses ripgrep under the hood for high performance.
+ */
 export declare function findCallSites(projectRoot: string, methodName: string): string
+/** Pings the TNG API to verify connectivity and API key validity. */
 export declare function ping(baseUrl: string, apiKey?: string | undefined | null): string
+/** Submits a test generation job to the API and returns the numeric job ID. */
 export declare function submitJob(baseUrl: string, apiKey: string, payloadJson: string): number
+/** Fetches usage statistics for the authenticated user from the API. */
 export declare function getUserStats(baseUrl: string, apiKey: string): string
-export declare function submitAndPoll(filePath: string, methodName: string, className: string | undefined | null, testType: string | undefined | null, auditMode: boolean, configJson: string, callback: (...args: any[]) => any): string
+/**
+ * Orchestrates the code audit process.
+ * Analyzes the source, builds context, and streams results via the provided callback.
+ */
+export declare function runAudit(filePath: string, methodName: string, className: string | undefined | null, testType: string | undefined | null, configJson: string, callback: (...args: any[]) => any): string
+/**
+ * Applies a single edit operation to a file.
+ * Uses backup and atomic write for safety.
+ */
+export declare function applyEdit(filePath: string, search: string, replace: string, lineHint?: number | undefined | null): string
+/**
+ * Applies multiple edit operations atomically.
+ * All succeed or all fail (with rollback).
+ */
+export declare function applyEditsAtomic(operationsJson: string): string
+/**
+ * Orchestrates the test generation process.
+ * 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

package/index.js

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

package/lib/auditWorker.js

@@ -0,0 +1,28 @@
+const { parentPort, workerData } = require('worker_threads');
+const { runAudit } = require('../index');
+const { loadConfig } = require('./config');
+
+const { filePath, methodName, className, testType } = workerData;
+const config = loadConfig();
+
+try {
+    const resultJson = runAudit(
+        filePath,
+        methodName,
+        className,
+        testType,
+        JSON.stringify(config),
+        (msg) => {
+            if (parentPort) {
+                parentPort.postMessage({ type: 'item', data: msg });
+            }
+        }
+    );
+    if (parentPort) {
+        parentPort.postMessage({ type: 'result', data: resultJson });
+    }
+} catch (e) {
+    if (parentPort) {
+        parentPort.postMessage({ type: 'error', data: e.message });
+    }
+}

package/lib/config.js

@@ -21,6 +21,7 @@ const loadConfig = () => {
 
         return Object.keys(filtered).length > 0 ? filtered : config;
     } catch (error) {
+        console.error(`Failed to load config from ${configFile}:`, error.message);
         return {};
     }
 };

package/lib/fixApplier.js

@@ -0,0 +1,59 @@
+const { applyEdit } = require('../index');
+
+/**
+ * Parses a SEARCH_AND_REPLACE block from a string
+ * @param {string} codeChange
+ * @returns {{search: string, replace: string} | null}
+ */
+function parseSearchReplace(codeChange) {
+    if (!codeChange) return null;
+
+    // Try to find SEARCH: and REPLACE: blocks
+    const searchIdx = codeChange.indexOf('SEARCH:');
+    const replaceIdx = codeChange.indexOf('REPLACE:');
+
+    if (searchIdx !== -1 && replaceIdx !== -1 && searchIdx < replaceIdx) {
+        const search = codeChange.substring(searchIdx + 7, replaceIdx).trim();
+        const replace = codeChange.substring(replaceIdx + 8).trim();
+        return { search, replace };
+    }
+
+    return null;
+}
+
+/**
+ * Appies a fix to a file
+ * @param {string} filePath
+ * @param {object} item
+ * @returns {object} result
+ */
+async function applyFix(filePath, item) {
+    try {
+        if (!item) {
+            return { success: false, error: 'Item is null or undefined' };
+        }
+        const codeChange = item.fix_code_change || (item.fix && item.fix.how && item.fix.how.code_change);
+        if (!codeChange) {
+            return { success: false, error: 'No code change found in fix data' };
+        }
+
+        const parsed = parseSearchReplace(codeChange);
+        if (!parsed) {
+            return { success: false, error: 'Failed to parse SEARCH_AND_REPLACE block' };
+        }
+
+        const resultJson = applyEdit(
+            filePath,
+            parsed.search,
+            parsed.replace,
+            typeof item.line_number === 'number' && item.line_number > 0 ? item.line_number : null
+        );
+
+        const result = JSON.parse(resultJson);
+        return result;
+    } catch (e) {
+        return { success: false, error: e.message };
+    }
+}
+
+module.exports = { applyFix, parseSearchReplace };