openpencil/server/utils/opencode-client.ts

42 lines
1.2 KiB
TypeScript
Raw Normal View History

/**
* Shared OpenCode client manager.
* Reuses an existing server on port 4096; starts one on a random port as fallback.
* Tracks spawned servers so they can be cleaned up on process exit.
*/
const activeServers = new Set<{ close(): void }>()
// Clean up spawned OpenCode servers on process exit
function cleanup() {
for (const server of activeServers) {
try { server.close() } catch { /* ignore */ }
}
activeServers.clear()
}
process.on('beforeExit', cleanup)
process.on('SIGTERM', cleanup)
process.on('SIGINT', cleanup)
export async function getOpencodeClient() {
const { createOpencodeClient, createOpencode } = await import('@opencode-ai/sdk/v2')
// Try connecting to an existing server first
try {
const client = createOpencodeClient()
await client.config.providers() // probe
return { client, server: undefined }
} catch {
// No running server — start a temporary one on a random port
const oc = await createOpencode({ port: 0 })
activeServers.add(oc.server)
return { client: oc.client, server: oc.server }
}
}
export function releaseOpencodeServer(server: { close(): void } | undefined) {
if (!server) return
try { server.close() } catch { /* ignore */ }
activeServers.delete(server)
}