@toothfairyai/tfcode 2.0.1 → 2.0.2

package/bin/tfcode.js

@@ -1,19 +1,59 @@
 #!/usr/bin/env node
-const { spawn } = require('child_process')
+const { spawn, spawnSync } = require('child_process')
 const path = require('path')
 const fs = require('fs')
+const os = require('os')
 
-const binary = process.platform === 'win32' ? 'tfcode.exe' : 'tfcode'
-const binaryPath = path.join(__dirname, binary)
+const platform = os.platform() === 'darwin' ? 'darwin' : os.platform() === 'win32' ? 'windows' : 'linux'
+const arch = os.arch() === 'arm64' ? 'arm64' : 'x64'
+const binName = platform === 'windows' ? 'tfcode.exe' : 'tfcode'
 
-if (!fs.existsSync(binaryPath)) {
-  console.error('tfcode binary not found. Run: npm install @toothfairyai/tfcode')
-  process.exit(1)
+function needBaseline() {
+  if (arch !== 'x64' || platform === 'windows') return false
+  try {
+    if (platform === 'linux') {
+      return !fs.readFileSync('/proc/cpuinfo', 'utf8').toLowerCase().includes('avx2')
+    }
+    const r = spawnSync('sysctl', ['-n', 'hw.optional.avx2_0'], { encoding: 'utf8' })
+    return r.stdout.trim() !== '1'
+  } catch {
+    return false
+  }
+}
+
+function libcAbi() {
+  if (platform !== 'linux') return ''
+  try {
+    if (fs.existsSync('/etc/alpine-release')) return 'musl'
+    const r = spawnSync('ldd', ['--version'], { encoding: 'utf8' })
+    if ((r.stdout + r.stderr).toLowerCase().includes('musl')) return 'musl'
+  } catch {}
+  return ''
 }
 
-const child = spawn(binaryPath, process.argv.slice(2), {
-  stdio: 'inherit',
-  env: process.env
-})
+const baseline = needBaseline()
+const abi = libcAbi()
+let target = 'tfcode-' + platform + '-' + arch
+if (baseline) target += '-baseline'
+if (abi) target += '-' + abi
+
+let binaryPath
+try {
+  const dir = path.dirname(require.resolve('@toothfairyai/' + target + '/package.json'))
+  const candidate = path.join(dir, 'bin', binName)
+  if (fs.existsSync(candidate)) binaryPath = candidate
+} catch {}
+
+if (!binaryPath) {
+  const bundled = path.join(__dirname, binName)
+  if (fs.existsSync(bundled)) binaryPath = bundled
+}
+
+if (!binaryPath) {
+  console.error('tfcode binary not found for ' + target + '. Try: npm install -g --allow-scripts=@toothfairyai/tfcode')
+  process.exit(1)
+}
 
+const child = spawn(binaryPath, process.argv.slice(2), { stdio: 'inherit', env: process.env })
 child.on('exit', (code) => process.exit(code || 0))
+child.on('error', (err) => { console.error('Failed to start tfcode:', err.message); process.exit(1) })

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@toothfairyai/tfcode",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "bin": {
     "tfcode": "./bin/tfcode.js"
   },
@@ -9,18 +9,18 @@
   },
   "license": "MIT",
   "optionalDependencies": {
-    "@toothfairyai/tfcode-linux-arm64": "2.0.1",
-    "@toothfairyai/tfcode-windows-x64": "2.0.1",
-    "@toothfairyai/tfcode-linux-x64-baseline-musl": "2.0.1",
-    "@toothfairyai/tfcode-darwin-x64-baseline": "2.0.1",
-    "@toothfairyai/tfcode-linux-x64-musl": "2.0.1",
-    "@toothfairyai/tfcode-windows-x64-baseline": "2.0.1",
-    "@toothfairyai/tfcode-linux-arm64-musl": "2.0.1",
-    "@toothfairyai/tfcode-windows-arm64": "2.0.1",
-    "@toothfairyai/tfcode-linux-x64": "2.0.1",
-    "@toothfairyai/tfcode-darwin-x64": "2.0.1",
-    "@toothfairyai/tfcode-linux-x64-baseline": "2.0.1",
-    "@toothfairyai/tfcode-darwin-arm64": "2.0.1"
+    "@toothfairyai/tfcode-linux-arm64": "2.0.2",
+    "@toothfairyai/tfcode-windows-x64": "2.0.2",
+    "@toothfairyai/tfcode-linux-x64-baseline-musl": "2.0.2",
+    "@toothfairyai/tfcode-darwin-x64-baseline": "2.0.2",
+    "@toothfairyai/tfcode-linux-x64-musl": "2.0.2",
+    "@toothfairyai/tfcode-windows-x64-baseline": "2.0.2",
+    "@toothfairyai/tfcode-linux-arm64-musl": "2.0.2",
+    "@toothfairyai/tfcode-windows-arm64": "2.0.2",
+    "@toothfairyai/tfcode-linux-x64": "2.0.2",
+    "@toothfairyai/tfcode-darwin-x64": "2.0.2",
+    "@toothfairyai/tfcode-linux-x64-baseline": "2.0.2",
+    "@toothfairyai/tfcode-darwin-arm64": "2.0.2"
   },
   "engines": {
     "node": ">=18"