pantheon-opencode 1.0.0

package/scripts/install/venv.mjs

@@ -0,0 +1,116 @@
+#!/usr/bin/env node
+
+/**
+ * venv.mjs — Virtual environment setup + dependency install
+ *
+ * Usage:
+ *   node scripts/install/venv.mjs --target ~/.config/opencode
+ *   node scripts/install/venv.mjs --target ~/.config/opencode --dry-run
+ *   node scripts/install/venv.mjs --target ~/.config/opencode --skip-install
+ */
+
+import { spawnSync } from 'node:child_process'
+import { existsSync } from 'node:fs'
+import { dirname, join } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __dirname = dirname(fileURLToPath(import.meta.url))
+const ROOT = join(__dirname, '..', '..')
+
+/**
+ * Set up a Python virtual environment and install MCP dependencies.
+ *
+ * @param {string} target - Target installation directory (e.g. ~/.config/opencode)
+ * @param {{ dryRun?: boolean, skipInstall?: boolean }} [options]
+ * @returns {{ venvPath: string, python: string }}
+ */
+export function setupVenv(target, { dryRun = false, skipInstall = false } = {}) {
+  const venvPath = join(target, '.venv')
+  const pythonBin = join(venvPath, 'bin', 'python3')
+
+  // Step 1: Create venv if not exists
+  if (!existsSync(pythonBin)) {
+    if (!dryRun) {
+      console.log('  Creating .venv...')
+      const result = spawnSync('python3', ['-m', 'venv', venvPath], {
+        stdio: 'inherit',
+        timeout: 30_000,
+      })
+      if (result.status !== 0) {
+        throw new Error('Failed to create virtual environment')
+      }
+    }
+    console.log('  ✅ .venv created')
+  } else {
+    console.log('  ⏭️  .venv already exists')
+  }
+
+  // Step 2: Install MCP dependencies
+  if (!skipInstall) {
+    const pip = join(venvPath, 'bin', 'pip')
+    const reqFile = join(ROOT, 'src', 'mcp', 'requirements-mcp.txt')
+
+    if (!existsSync(reqFile)) {
+      throw new Error(`Requirements file not found: ${reqFile}`)
+    }
+
+    console.log('  Installing MCP dependencies...')
+
+    if (!dryRun) {
+      // Upgrade pip first (avoids resolver warnings)
+      spawnSync(pip, ['install', '--upgrade', 'pip'], {
+        stdio: 'ignore',
+        timeout: 30_000,
+      })
+
+      // Install requirements
+      const r = spawnSync(pip, ['install', '-r', reqFile], {
+        stdio: 'inherit',
+        timeout: 120_000,
+      })
+
+      if (r.status !== 0) {
+        // If failed, try with --break-system-packages (PEP 668 workaround)
+        console.log('  ⚠️  First attempt failed, retrying with --break-system-packages...')
+        const r2 = spawnSync(pip, ['install', '-r', reqFile, '--break-system-packages'], {
+          stdio: 'inherit',
+          timeout: 120_000,
+        })
+        if (r2.status !== 0) {
+          throw new Error(
+            'Failed to install MCP dependencies. ' +
+            'Try: pip install -r src/mcp/requirements-mcp.txt --break-system-packages'
+          )
+        }
+      }
+    }
+    console.log('  ✅ MCP dependencies installed')
+  } else {
+    console.log('  ⏭️  Dependency install skipped (--skip-install)')
+  }
+
+  return { venvPath, python: pythonBin }
+}
+
+// ---------------------------------------------------------------------------
+// CLI entry point
+// ---------------------------------------------------------------------------
+function main() {
+  const args = process.argv.slice(2)
+  const targetIdx = args.indexOf('--target')
+  const target = targetIdx !== -1 ? args[targetIdx + 1] : process.cwd()
+  const dryRun = args.includes('--dry-run')
+  const skipInstall = args.includes('--skip-install')
+
+  try {
+    const result = setupVenv(target, { dryRun, skipInstall })
+    console.log(`  📍 Python: ${result.python}`)
+  } catch (err) {
+    console.error(`\n  ❌ ${err.message}`)
+    process.exit(1)
+  }
+}
+
+if (process.argv[1] && import.meta.url === `file://${process.argv[1]}`) {
+  main()
+}