253 lines
7.9 KiB
JavaScript
253 lines
7.9 KiB
JavaScript
import { spawn } from 'node:child_process'
|
|
import { createHash } from 'node:crypto'
|
|
import { chmodSync, mkdirSync, writeFileSync } from 'node:fs'
|
|
import net from 'node:net'
|
|
import { createRequire } from 'node:module'
|
|
import path from 'node:path'
|
|
import { fileURLToPath } from 'node:url'
|
|
|
|
// Why: Electron-based hosts (e.g. Claude Code, VS Code) set
|
|
// ELECTRON_RUN_AS_NODE=1 in their terminal environment. If this leaks into
|
|
// the electron-vite spawn, the Electron binary boots as plain Node and
|
|
// require('electron') returns the npm stub instead of the built-in API.
|
|
delete process.env.ELECTRON_RUN_AS_NODE
|
|
|
|
const require = createRequire(import.meta.url)
|
|
const repoRoot = path.resolve(path.dirname(fileURLToPath(import.meta.url)), '../..')
|
|
|
|
function getDevUserDataPath() {
|
|
if (process.env.ORCA_DEV_USER_DATA_PATH) {
|
|
return process.env.ORCA_DEV_USER_DATA_PATH
|
|
}
|
|
if (process.platform === 'darwin') {
|
|
return path.join(process.env.HOME ?? '', 'Library', 'Application Support', 'orca-dev')
|
|
}
|
|
if (process.platform === 'win32') {
|
|
return path.join(
|
|
process.env.APPDATA ?? path.join(process.env.USERPROFILE ?? '', 'AppData', 'Roaming'),
|
|
'orca-dev'
|
|
)
|
|
}
|
|
return path.join(
|
|
process.env.XDG_CONFIG_HOME ?? path.join(process.env.HOME ?? '', '.config'),
|
|
'orca-dev'
|
|
)
|
|
}
|
|
|
|
function prepareDevCliWrapper() {
|
|
const binDir = path.join(repoRoot, 'out', 'bin')
|
|
mkdirSync(binDir, { recursive: true })
|
|
const userDataPath = getDevUserDataPath()
|
|
const cliPath = path.join(repoRoot, 'out', 'cli', 'index.js')
|
|
const electronBin = getElectronExecutable()
|
|
|
|
if (process.platform === 'win32') {
|
|
writeFileSync(
|
|
path.join(binDir, 'orca-dev.cmd'),
|
|
`@echo off\r\nset "ORCA_USER_DATA_PATH=${userDataPath}"\r\nset "ORCA_APP_EXECUTABLE=${electronBin}"\r\nset "ORCA_APP_EXECUTABLE_NEEDS_APP_ROOT=1"\r\nnode "${cliPath}" %*\r\n`,
|
|
'utf8'
|
|
)
|
|
} else {
|
|
const wrapperPath = path.join(binDir, 'orca-dev')
|
|
writeFileSync(
|
|
wrapperPath,
|
|
`#!/usr/bin/env bash\nexport ORCA_USER_DATA_PATH=${JSON.stringify(userDataPath)}\nexport ORCA_APP_EXECUTABLE=${JSON.stringify(electronBin)}\nexport ORCA_APP_EXECUTABLE_NEEDS_APP_ROOT=1\nexec node ${JSON.stringify(cliPath)} "$@"\n`,
|
|
'utf8'
|
|
)
|
|
chmodSync(wrapperPath, 0o755)
|
|
}
|
|
|
|
process.env.PATH = `${binDir}${path.delimiter}${process.env.PATH ?? ''}`
|
|
console.log(`[orca-dev] Prepared wrapper in ${binDir}`)
|
|
}
|
|
|
|
function getElectronExecutable() {
|
|
if (process.platform === 'win32') {
|
|
return path.join(repoRoot, 'node_modules', 'electron', 'dist', 'electron.exe')
|
|
}
|
|
return path.join(repoRoot, 'node_modules', '.bin', 'electron')
|
|
}
|
|
|
|
if (process.env.ORCA_SKIP_DEV_CLI_PREPARE !== '1') {
|
|
prepareDevCliWrapper()
|
|
}
|
|
|
|
// Why: tests inject a tiny fake CLI here so they can verify Ctrl+C tears down
|
|
// the full child tree without depending on a real electron-vite install.
|
|
const electronViteCli =
|
|
process.env.ORCA_ELECTRON_VITE_CLI ||
|
|
path.join(path.dirname(require.resolve('electron-vite/package.json')), 'bin', 'electron-vite.js')
|
|
|
|
// Why: every `pn dev` should be attachable from agent-browser/playwright-cli
|
|
// without manual port juggling. Pick a best-effort deterministic port per
|
|
// worktree; falls back to a probe sweep if the deterministic pick or its
|
|
// neighbors are busy (multiple worktrees may share a machine).
|
|
const forwardedRaw = process.argv.slice(2)
|
|
function isPortFree(port) {
|
|
return new Promise((resolve) => {
|
|
const srv = net.createServer()
|
|
srv.once('error', () => {
|
|
// Why: error fires before listen binds; close() may throw — swallow it
|
|
// so the handle is released without leaking listeners across 64 probes.
|
|
try {
|
|
srv.close()
|
|
} catch {}
|
|
resolve(false)
|
|
})
|
|
srv.once('listening', () => srv.close(() => resolve(true)))
|
|
srv.listen(port, '127.0.0.1')
|
|
})
|
|
}
|
|
async function pickDebugPort() {
|
|
// Why: 32 bits of SHA1 (vs 16) reduces truncation bias; modulo 200 still
|
|
// collides routinely across many worktrees, hence the probe sweep below.
|
|
const seed = parseInt(createHash('sha1').update(repoRoot).digest('hex').slice(0, 8), 16)
|
|
const base = 9333 + (seed % 200) // deterministic base in 9333..9532; probe sweeps up to base+63
|
|
for (let i = 0; i < 64; i++) {
|
|
const p = base + i
|
|
if (await isPortFree(p)) {
|
|
return p
|
|
}
|
|
}
|
|
return null
|
|
}
|
|
function parseDebugPortEnv(raw) {
|
|
const n = Number.parseInt(raw, 10)
|
|
if (!Number.isInteger(n) || n < 1 || n > 65535 || String(n) !== raw.trim()) {
|
|
return null
|
|
}
|
|
return n
|
|
}
|
|
// Why: exact match (or `=` form) avoids false positives on hypothetical
|
|
// `--remote-debugging-port-*` flags; the bare flag also covers the
|
|
// space-separated form. `--remote-debugging-pipe` opts into pipe-based
|
|
// debugging — don't fight the user's choice by injecting a port.
|
|
const userPassedPort = forwardedRaw.some(
|
|
(a) =>
|
|
a === '--remote-debugging-port' ||
|
|
a.startsWith('--remote-debugging-port=') ||
|
|
a === '--remote-debugging-pipe'
|
|
)
|
|
// Why: --help/--version exit immediately; binding a probe socket and printing
|
|
// a debug-port line would be noise.
|
|
const isHelpOrVersion = forwardedRaw.some((a) => a === '--help' || a === '-h' || a === '--version')
|
|
let forwardedExtras = []
|
|
if (!userPassedPort && !isHelpOrVersion) {
|
|
const envPortRaw = process.env.REMOTE_DEBUGGING_PORT
|
|
let port = null
|
|
if (envPortRaw) {
|
|
port = parseDebugPortEnv(envPortRaw)
|
|
if (port === null) {
|
|
console.error(
|
|
`[orca-dev] Ignoring invalid REMOTE_DEBUGGING_PORT=${JSON.stringify(envPortRaw)}; falling back to probe.`
|
|
)
|
|
}
|
|
}
|
|
if (port === null) {
|
|
port = await pickDebugPort()
|
|
}
|
|
if (port !== null) {
|
|
forwardedExtras = [`--remote-debugging-port=${port}`]
|
|
// Why: stderr keeps stdout clean for downstream parsing; log uses
|
|
// 127.0.0.1 to match the interface we actually probed (localhost may
|
|
// resolve to ::1 on IPv6-first hosts).
|
|
console.error(`[orca-dev] Remote debugging on http://127.0.0.1:${port}`)
|
|
} else {
|
|
console.error(
|
|
'[orca-dev] No free debug port found in sweep; starting without --remote-debugging-port.'
|
|
)
|
|
}
|
|
}
|
|
const forwardedArgs = ['dev', ...forwardedRaw, ...forwardedExtras]
|
|
const child = spawn(process.execPath, [electronViteCli, ...forwardedArgs], {
|
|
stdio: 'inherit',
|
|
env: process.env,
|
|
// Why: electron-vite launches Electron as a descendant process. Giving the
|
|
// dev runner its own process group lets Ctrl+C kill the whole tree on macOS
|
|
// instead of leaving the Electron app alive after the terminal exits.
|
|
detached: process.platform !== 'win32'
|
|
})
|
|
|
|
let isShuttingDown = false
|
|
let forcedKillTimer = null
|
|
|
|
function signalExitCode(signal) {
|
|
if (signal === 'SIGINT') {
|
|
return 130
|
|
}
|
|
if (signal === 'SIGTERM') {
|
|
return 143
|
|
}
|
|
return 1
|
|
}
|
|
|
|
function terminateChild(signal) {
|
|
if (!child.pid) {
|
|
return
|
|
}
|
|
|
|
if (process.platform === 'win32') {
|
|
const taskkill = spawn('taskkill', ['/pid', String(child.pid), '/t', '/f'], {
|
|
stdio: 'ignore',
|
|
windowsHide: true
|
|
})
|
|
taskkill.unref()
|
|
return
|
|
}
|
|
|
|
try {
|
|
process.kill(-child.pid, signal)
|
|
} catch (error) {
|
|
const code = error && typeof error === 'object' && 'code' in error ? error.code : null
|
|
if (code !== 'ESRCH') {
|
|
throw error
|
|
}
|
|
}
|
|
}
|
|
|
|
function beginShutdown(signal) {
|
|
if (isShuttingDown) {
|
|
return
|
|
}
|
|
isShuttingDown = true
|
|
|
|
terminateChild(signal)
|
|
forcedKillTimer = setTimeout(() => {
|
|
terminateChild('SIGKILL')
|
|
}, 5000)
|
|
}
|
|
|
|
process.on('SIGINT', () => {
|
|
beginShutdown('SIGINT')
|
|
})
|
|
|
|
process.on('SIGTERM', () => {
|
|
beginShutdown('SIGTERM')
|
|
})
|
|
|
|
child.on('error', (error) => {
|
|
if (forcedKillTimer) {
|
|
clearTimeout(forcedKillTimer)
|
|
}
|
|
console.error(error)
|
|
process.exit(1)
|
|
})
|
|
|
|
child.on('exit', (code, signal) => {
|
|
if (forcedKillTimer) {
|
|
clearTimeout(forcedKillTimer)
|
|
}
|
|
|
|
if (isShuttingDown) {
|
|
process.exit(signalExitCode(signal ?? 'SIGINT'))
|
|
return
|
|
}
|
|
|
|
if (signal) {
|
|
process.exit(signalExitCode(signal))
|
|
return
|
|
}
|
|
|
|
process.exit(code ?? 1)
|
|
})
|