trace-sec 2.1.5 → 2.1.7

package/bin/postinstall.js

@@ -0,0 +1,121 @@
+#!/usr/bin/env node
+
+/**
+ * TRACE Post-Install Script
+ * Automatically provisions Python runtime and core dependencies during `npm install`.
+ */
+
+import { execSync } from 'child_process';
+import { existsSync, mkdirSync } from 'fs';
+import { resolve, join, dirname } from 'path';
+import { fileURLToPath } from 'url';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+const projectRoot = resolve(__dirname, '..');
+
+const isWin = process.platform === 'win32';
+const userHome = process.env.USERPROFILE || process.env.HOME || '';
+const dedicatedDir = join(userHome, '.trace');
+const dedicatedVenv = join(dedicatedDir, 'venv');
+const venvPython = join(dedicatedVenv, isWin ? 'Scripts' : 'bin', isWin ? 'python.exe' : 'python');
+const venvPip = join(dedicatedVenv, isWin ? 'Scripts' : 'bin', isWin ? 'pip.exe' : 'pip');
+const reqFile = join(projectRoot, 'requirements.txt');
+
+function findSystemPython() {
+  const candidates = isWin
+    ? [
+        'python',
+        'py -3',
+        'py',
+        'python3',
+        join(process.env.LOCALAPPDATA || '', 'Programs', 'Python', 'Python312', 'python.exe'),
+        join(process.env.LOCALAPPDATA || '', 'Programs', 'Python', 'Python311', 'python.exe'),
+        join(process.env.LOCALAPPDATA || '', 'Programs', 'Python', 'Python310', 'python.exe'),
+        join(process.env.PROGRAMFILES || '', 'Python312', 'python.exe'),
+        join(process.env.PROGRAMFILES || '', 'Python311', 'python.exe'),
+        join(process.env.PROGRAMFILES || '', 'Python310', 'python.exe'),
+      ]
+    : [
+        'python3',
+        'python',
+        '/usr/bin/python3',
+        '/usr/local/bin/python3',
+        '/opt/homebrew/bin/python3',
+      ];
+
+  for (const cmd of candidates) {
+    if (!cmd || !cmd.trim()) continue;
+    try {
+      const execCmd = cmd.includes(' ') && !cmd.startsWith('py ') ? `"${cmd}"` : cmd;
+      const ver = execSync(`${execCmd} -c "import sys; print(sys.version_info[0], sys.version_info[1])"`, {
+        stdio: ['ignore', 'pipe', 'ignore'],
+        encoding: 'utf-8',
+      }).trim();
+      const parts = ver.split(/\s+/).map(Number);
+      if (parts[0] === 3 && parts[1] >= 9) {
+        return execCmd;
+      }
+    } catch {
+      // try next candidate
+    }
+  }
+  return null;
+}
+
+try {
+  console.log('\n\x1b[32m\x1b[1m[TRACE]\x1b[0m Configuring environment for autonomous security verification...');
+
+  mkdirSync(dedicatedDir, { recursive: true });
+
+  const sysPython = findSystemPython();
+  if (!sysPython) {
+    console.log('  \x1b[38;2;255;158;59m•\x1b[0m Python 3.10+ not detected on PATH yet.');
+    console.log('  \x1b[38;2;255;158;59m•\x1b[0m Quick install:');
+    if (isWin) {
+      console.log('      \x1b[36mwinget install Python.Python.3.11\x1b[0m');
+    } else if (process.platform === 'darwin') {
+      console.log('      \x1b[36mbrew install python\x1b[0m');
+    } else {
+      console.log('      \x1b[36msudo apt install -y python3 python3-pip python3-venv\x1b[0m');
+    }
+    console.log('  Once installed, dependencies will configure automatically on first run.\n');
+    process.exit(0);
+  }
+
+  // Create virtualenv if not already created
+  if (!existsSync(venvPython)) {
+    console.log(`  \x1b[38;2;255;158;59m›\x1b[0m Initializing isolated Python environment in ${dedicatedVenv}...`);
+    execSync(`${sysPython} -m venv "${dedicatedVenv}"`, { stdio: 'inherit' });
+  }
+
+  // Check if dependencies are already satisfied
+  let depsOk = false;
+  try {
+    execSync(`"${venvPython}" -c "import typer, rich, pydantic, tree_sitter"`, {
+      stdio: 'ignore',
+    });
+    depsOk = true;
+  } catch {
+    depsOk = false;
+  }
+
+  if (!depsOk) {
+    console.log('  \x1b[38;2;255;158;59m›\x1b[0m Installing core Python dependencies into dedicated environment...');
+    if (existsSync(reqFile)) {
+      execSync(`"${venvPython}" -m pip install -r "${reqFile}"`, { stdio: 'inherit' });
+    } else {
+      execSync(`"${venvPython}" -m pip install typer rich pydantic pydantic-settings networkx httpx orjson tomli-w tree-sitter tree-sitter-language-pack onnxruntime transformers safetensors numpy`, { stdio: 'inherit' });
+    }
+    console.log('  \x1b[32m✓\x1b[0m Dependencies successfully attached and verified.');
+  } else {
+    console.log('  \x1b[32m✓\x1b[0m Dedicated runtime dependencies already up to date.');
+  }
+
+  console.log('\x1b[32m\x1b[1m✓ TRACE ready to run!\x1b[0m Type \x1b[36mtrace\x1b[0m or \x1b[36mnpx trace-sec\x1b[0m to launch.\n');
+} catch (err) {
+  // Never fail npm install completely on postinstall warning
+  console.log(`  \x1b[38;2;255;158;59m[TRACE]\x1b[0m Background setup notice: ${err.message}`);
+  console.log('  Dependencies will be verified when you run \x1b[36mtrace\x1b[0m.\n');
+}
+

package/bin/trace.js

@@ -5,7 +5,7 @@
  * Node.js & NPX launcher for TRACE Python engine.
  */
 
-import { spawn } from 'child_process';
+import { spawn, execSync } from 'child_process';
 import { existsSync, mkdirSync, cpSync, writeFileSync } from 'fs';
 import { resolve, join, dirname } from 'path';
 import { fileURLToPath } from 'url';
@@ -84,17 +84,60 @@ if (args[0] === 'install-skill' || args[0] === 'setup-agent' || args[0] === 'ins
 // Find virtual environment python or system python
 const isWin = process.platform === 'win32';
 const userHome = process.env.USERPROFILE || process.env.HOME || '';
+const dedicatedVenv = join(userHome, '.trace', 'venv');
+const dedicatedPython = join(dedicatedVenv, isWin ? 'Scripts' : 'bin', isWin ? 'python.exe' : 'python');
+const dedicatedPip = join(dedicatedVenv, isWin ? 'Scripts' : 'bin', isWin ? 'pip.exe' : 'pip');
+const reqFile = join(projectRoot, 'requirements.txt');
+
+function findSystemPython() {
+  const candidates = isWin
+    ? [
+        'python',
+        'py -3',
+        'py',
+        'python3',
+        join(process.env.LOCALAPPDATA || '', 'Programs', 'Python', 'Python312', 'python.exe'),
+        join(process.env.LOCALAPPDATA || '', 'Programs', 'Python', 'Python311', 'python.exe'),
+        join(process.env.LOCALAPPDATA || '', 'Programs', 'Python', 'Python310', 'python.exe'),
+        join(process.env.PROGRAMFILES || '', 'Python312', 'python.exe'),
+        join(process.env.PROGRAMFILES || '', 'Python311', 'python.exe'),
+        join(process.env.PROGRAMFILES || '', 'Python310', 'python.exe'),
+      ]
+    : [
+        'python3',
+        'python',
+        '/usr/bin/python3',
+        '/usr/local/bin/python3',
+        '/opt/homebrew/bin/python3',
+      ];
+
+  for (const cmd of candidates) {
+    if (!cmd || !cmd.trim()) continue;
+    try {
+      const execCmd = cmd.includes(' ') && !cmd.startsWith('py ') ? `"${cmd}"` : cmd;
+      const ver = execSync(`${execCmd} -c "import sys; print(sys.version_info[0], sys.version_info[1])"`, {
+        stdio: ['ignore', 'pipe', 'ignore'],
+        encoding: 'utf-8',
+      }).trim();
+      const parts = ver.split(/\s+/).map(Number);
+      if (parts[0] === 3 && parts[1] >= 9) {
+        return execCmd;
+      }
+    } catch {
+      // try next
+    }
+  }
+  return null;
+}
+
 const candidatePythons = [
   process.env.TRACE_PYTHON,
+  dedicatedPython,
   join(process.cwd(), '.venv', isWin ? 'Scripts' : 'bin', isWin ? 'python.exe' : 'python'),
   join(projectRoot, '.venv', isWin ? 'Scripts' : 'bin', isWin ? 'python.exe' : 'python'),
-  isWin ? 'D:\\Startup\\TRACE\\.venv\\Scripts\\python.exe' : null,
-  join(userHome, '.trace', 'venv', isWin ? 'Scripts' : 'bin', isWin ? 'python.exe' : 'python'),
-  join(userHome, '.trace', 'models', 'venv', isWin ? 'Scripts' : 'bin', isWin ? 'python.exe' : 'python'),
-  isWin ? 'python' : 'python3',
 ];
 
-let pythonBin = isWin ? 'python' : 'python3';
+let pythonBin = null;
 for (const cand of candidatePythons) {
   if (cand && existsSync(cand)) {
     pythonBin = cand;
@@ -102,6 +145,70 @@ for (const cand of candidatePythons) {
   }
 }
 
+if (!pythonBin) {
+  pythonBin = findSystemPython();
+}
+
+if (!pythonBin) {
+  console.error('\n\x1b[31m\x1b[1m[TRACE ERROR]\x1b[0m Python 3.10+ is required to execute TRACE.');
+  console.error('Please install Python to proceed:');
+  if (isWin) {
+    console.error('  \x1b[36mwinget install Python.Python.3.11\x1b[0m  (or from https://www.python.org/)\n');
+  } else if (process.platform === 'darwin') {
+    console.error('  \x1b[36mbrew install python\x1b[0m\n');
+  } else {
+    console.error('  \x1b[36msudo apt install -y python3 python3-pip python3-venv\x1b[0m\n');
+  }
+  process.exit(1);
+}
+
+// Verify that the resolved Python environment has required TRACE dependencies
+let depsOk = false;
+try {
+  execSync(`"${pythonBin}" -c "import typer, rich, pydantic, tree_sitter"`, { stdio: 'ignore' });
+  depsOk = true;
+} catch (e) {
+  depsOk = false;
+}
+
+if (!depsOk) {
+  if (existsSync(dedicatedPython)) {
+    try {
+      execSync(`"${dedicatedPython}" -c "import typer, rich, pydantic, tree_sitter"`, { stdio: 'ignore' });
+      pythonBin = dedicatedPython;
+      depsOk = true;
+    } catch (e) {
+      depsOk = false;
+    }
+  }
+
+  if (!depsOk) {
+    console.log('\n\x1b[32m\x1b[1m[TRACE]\x1b[0m Initializing TRACE runtime dependencies in dedicated environment (~/.trace/venv)...');
+    try {
+      mkdirSync(join(userHome, '.trace'), { recursive: true });
+      if (!existsSync(dedicatedPython)) {
+        console.log('  \x1b[38;2;255;158;59m›\x1b[0m Creating Python virtual environment in ~/.trace/venv...');
+        const sysPy = findSystemPython();
+        if (!sysPy) {
+          throw new Error('System Python 3.10+ not found. Please install Python from https://www.python.org/');
+        }
+        execSync(`${sysPy} -m venv "${dedicatedVenv}"`, { stdio: 'inherit' });
+      }
+      console.log('  \x1b[38;2;255;158;59m›\x1b[0m Installing core dependencies via pip...');
+      if (existsSync(reqFile)) {
+        execSync(`"${dedicatedPython}" -m pip install -r "${reqFile}"`, { stdio: 'inherit' });
+      } else {
+        execSync(`"${dedicatedPython}" -m pip install typer rich pydantic pydantic-settings networkx httpx orjson tomli-w tree-sitter tree-sitter-language-pack onnxruntime transformers safetensors numpy`, { stdio: 'inherit' });
+      }
+      pythonBin = dedicatedPython;
+      console.log('\x1b[32m✓ Runtime dependencies provisioned successfully.\x1b[0m\n');
+    } catch (e) {
+      console.warn(`[TRACE] Automatic environment setup warning: ${e.message}`);
+      console.warn(`If command fails, install directly via: pip install git+https://github.com/VK-Amogh/TRACE.git\n`);
+    }
+  }
+}
+
 const cliArgs = ['-m', 'trace_engine.cli', ...args];
 
 const proc = spawn(pythonBin, cliArgs, {

package/package.json

@@ -1,19 +1,26 @@
 {
   "name": "trace-sec",
-  "version": "2.1.5",
+  "version": "2.1.7",
   "description": "Threat Reconnaissance & Attack-path Correlation Engine — Security Verification Layer for Coding Agents with Fine-Tuned SecureBERT 2.0 & Laya AI",
   "type": "module",
   "main": "bin/trace.js",
+  "scripts": {
+    "postinstall": "node bin/postinstall.js"
+  },
   "bin": {
-    "trace": "./bin/trace.js",
-    "trace-sec": "./bin/trace.js",
-    "trace-security": "./bin/trace.js"
+    "trace": "bin/trace.js",
+    "trace-sec": "bin/trace.js",
+    "trace-security": "bin/trace.js"
   },
   "files": [
     "bin",
     "src",
     "README.md",
-    "pyproject.toml"
+    "pyproject.toml",
+    "requirements.txt",
+    "!**/__pycache__",
+    "!**/*.pyc",
+    "!**/*.pyo"
   ],
   "repository": {
     "type": "git",

package/requirements.txt

@@ -0,0 +1,15 @@
+typer>=0.12.0
+rich>=13.7.0
+pydantic>=2.7.0
+pydantic-settings>=2.2.0
+networkx>=3.2.0
+httpx>=0.27.0
+orjson>=3.9.0
+tomli-w>=1.0.0
+tree-sitter>=0.21.0
+tree-sitter-language-pack>=0.2.0
+transformers>=4.40.0
+safetensors>=0.4.0
+onnxruntime>=1.17.0
+numpy>=1.24.0
+torch>=2.0.0