fix(process): prevent orphaned opencode subprocesses on shutdown (#15924)

pull/15954/head^2
Dax 2026-03-03 22:14:28 -05:00 committed by GitHub
parent 2a0be8316b
commit 3ebebe0a96
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 100 additions and 74 deletions

View File

@ -20,6 +20,17 @@
Prefer single word names for variables and functions. Only use multiple words if necessary. Prefer single word names for variables and functions. Only use multiple words if necessary.
### Naming Enforcement (Read This)
THIS RULE IS MANDATORY FOR AGENT WRITTEN CODE.
- Use single word names by default for new locals, params, and helper functions.
- Multi-word names are allowed only when a single word would be unclear or ambiguous.
- Do not introduce new camelCase compounds when a short single-word alternative is clear.
- Before finishing edits, review touched lines and shorten newly introduced identifiers where possible.
- Good short names to prefer: `pid`, `cfg`, `err`, `opts`, `dir`, `root`, `child`, `state`, `timeout`.
- Examples to avoid unless truly required: `inputPID`, `existingClient`, `connectTimeout`, `workerPath`.
```ts ```ts
// Good // Good
const foo = 1 const foo = 1

View File

@ -5,8 +5,8 @@ import { type rpc } from "./worker"
import path from "path" import path from "path"
import { fileURLToPath } from "url" import { fileURLToPath } from "url"
import { UI } from "@/cli/ui" import { UI } from "@/cli/ui"
import { iife } from "@/util/iife"
import { Log } from "@/util/log" import { Log } from "@/util/log"
import { withTimeout } from "@/util/timeout"
import { withNetworkOptions, resolveNetworkOptions } from "@/cli/network" import { withNetworkOptions, resolveNetworkOptions } from "@/cli/network"
import { Filesystem } from "@/util/filesystem" import { Filesystem } from "@/util/filesystem"
import type { Event } from "@opencode-ai/sdk/v2" import type { Event } from "@opencode-ai/sdk/v2"
@ -45,6 +45,20 @@ function createEventSource(client: RpcClient): EventSource {
} }
} }
async function target() {
if (typeof OPENCODE_WORKER_PATH !== "undefined") return OPENCODE_WORKER_PATH
const dist = new URL("./cli/cmd/tui/worker.js", import.meta.url)
if (await Filesystem.exists(fileURLToPath(dist))) return dist
return new URL("./worker.ts", import.meta.url)
}
async function input(value?: string) {
const piped = process.stdin.isTTY ? undefined : await Bun.stdin.text()
if (!value) return piped
if (!piped) return value
return piped + "\n" + value
}
export const TuiThreadCommand = cmd({ export const TuiThreadCommand = cmd({
command: "$0 [project]", command: "$0 [project]",
describe: "start opencode tui", describe: "start opencode tui",
@ -97,23 +111,17 @@ export const TuiThreadCommand = cmd({
} }
// Resolve relative paths against PWD to preserve behavior when using --cwd flag // Resolve relative paths against PWD to preserve behavior when using --cwd flag
const baseCwd = process.env.PWD ?? process.cwd() const root = process.env.PWD ?? process.cwd()
const cwd = args.project ? path.resolve(baseCwd, args.project) : process.cwd() const cwd = args.project ? path.resolve(root, args.project) : process.cwd()
const localWorker = new URL("./worker.ts", import.meta.url) const file = await target()
const distWorker = new URL("./cli/cmd/tui/worker.js", import.meta.url)
const workerPath = await iife(async () => {
if (typeof OPENCODE_WORKER_PATH !== "undefined") return OPENCODE_WORKER_PATH
if (await Filesystem.exists(fileURLToPath(distWorker))) return distWorker
return localWorker
})
try { try {
process.chdir(cwd) process.chdir(cwd)
} catch (e) { } catch {
UI.error("Failed to change directory to " + cwd) UI.error("Failed to change directory to " + cwd)
return return
} }
const worker = new Worker(workerPath, { const worker = new Worker(file, {
env: Object.fromEntries( env: Object.fromEntries(
Object.entries(process.env).filter((entry): entry is [string, string] => entry[1] !== undefined), Object.entries(process.env).filter((entry): entry is [string, string] => entry[1] !== undefined),
), ),
@ -121,76 +129,88 @@ export const TuiThreadCommand = cmd({
worker.onerror = (e) => { worker.onerror = (e) => {
Log.Default.error(e) Log.Default.error(e)
} }
const client = Rpc.client<typeof rpc>(worker)
process.on("uncaughtException", (e) => {
Log.Default.error(e)
})
process.on("unhandledRejection", (e) => {
Log.Default.error(e)
})
process.on("SIGUSR2", async () => {
await client.call("reload", undefined)
})
const prompt = await iife(async () => { const client = Rpc.client<typeof rpc>(worker)
const piped = !process.stdin.isTTY ? await Bun.stdin.text() : undefined const error = (e: unknown) => {
if (!args.prompt) return piped Log.Default.error(e)
return piped ? piped + "\n" + args.prompt : args.prompt }
}) const reload = () => {
client.call("reload", undefined).catch((err) => {
Log.Default.warn("worker reload failed", {
error: err instanceof Error ? err.message : String(err),
})
})
}
process.on("uncaughtException", error)
process.on("unhandledRejection", error)
process.on("SIGUSR2", reload)
let stopped = false
const stop = async () => {
if (stopped) return
stopped = true
process.off("uncaughtException", error)
process.off("unhandledRejection", error)
process.off("SIGUSR2", reload)
await withTimeout(client.call("shutdown", undefined), 5000).catch((error) => {
Log.Default.warn("worker shutdown failed", {
error: error instanceof Error ? error.message : String(error),
})
})
worker.terminate()
}
const prompt = await input(args.prompt)
const config = await Instance.provide({ const config = await Instance.provide({
directory: cwd, directory: cwd,
fn: () => TuiConfig.get(), fn: () => TuiConfig.get(),
}) })
// Check if server should be started (port or hostname explicitly set in CLI or config) const network = await resolveNetworkOptions(args)
const networkOpts = await resolveNetworkOptions(args) const external =
const shouldStartServer =
process.argv.includes("--port") || process.argv.includes("--port") ||
process.argv.includes("--hostname") || process.argv.includes("--hostname") ||
process.argv.includes("--mdns") || process.argv.includes("--mdns") ||
networkOpts.mdns || network.mdns ||
networkOpts.port !== 0 || network.port !== 0 ||
networkOpts.hostname !== "127.0.0.1" network.hostname !== "127.0.0.1"
let url: string const transport = external
let customFetch: typeof fetch | undefined ? {
let events: EventSource | undefined url: (await client.call("server", network)).url,
fetch: undefined,
if (shouldStartServer) { events: undefined,
// Start HTTP server for external access }
const server = await client.call("server", networkOpts) : {
url = server.url url: "http://opencode.internal",
} else { fetch: createWorkerFetch(client),
// Use direct RPC communication (no HTTP) events: createEventSource(client),
url = "http://opencode.internal" }
customFetch = createWorkerFetch(client)
events = createEventSource(client)
}
const tuiPromise = tui({
url,
config,
directory: cwd,
fetch: customFetch,
events,
args: {
continue: args.continue,
sessionID: args.session,
agent: args.agent,
model: args.model,
prompt,
fork: args.fork,
},
onExit: async () => {
await client.call("shutdown", undefined)
},
})
setTimeout(() => { setTimeout(() => {
client.call("checkUpgrade", { directory: cwd }).catch(() => {}) client.call("checkUpgrade", { directory: cwd }).catch(() => {})
}, 1000) }, 1000).unref?.()
await tuiPromise try {
await tui({
url: transport.url,
config,
directory: cwd,
fetch: transport.fetch,
events: transport.events,
args: {
continue: args.continue,
sessionID: args.session,
agent: args.agent,
model: args.model,
prompt,
fork: args.fork,
},
onExit: stop,
})
} finally {
await stop()
}
} finally { } finally {
unguard?.() unguard?.()
} }

View File

@ -137,12 +137,7 @@ export const rpc = {
async shutdown() { async shutdown() {
Log.Default.info("worker shutting down") Log.Default.info("worker shutting down")
if (eventStream.abort) eventStream.abort.abort() if (eventStream.abort) eventStream.abort.abort()
await Promise.race([ await Instance.disposeAll()
Instance.disposeAll(),
new Promise((resolve) => {
setTimeout(resolve, 5000)
}),
])
if (server) server.stop(true) if (server) server.stop(true)
}, },
} }