mirror of
https://github.com/anomalyco/opencode.git
synced 2026-04-25 15:24:58 +00:00
102 lines
2.9 KiB
TypeScript
102 lines
2.9 KiB
TypeScript
import { Installation } from "@/installation"
|
|
import { Server } from "@/server/server"
|
|
import { Log } from "@/util/log"
|
|
import { Instance } from "@/project/instance"
|
|
import { InstanceBootstrap } from "@/project/bootstrap"
|
|
import { Rpc } from "@/util/rpc"
|
|
import { upgrade } from "@/cli/upgrade"
|
|
import { Config } from "@/config/config"
|
|
import { GlobalBus } from "@/bus/global"
|
|
import { Flag } from "@/flag/flag"
|
|
import { writeHeapSnapshot } from "node:v8"
|
|
import { Heap } from "@/cli/heap"
|
|
import { AppRuntime } from "@/effect/app-runtime"
|
|
|
|
await Log.init({
|
|
print: process.argv.includes("--print-logs"),
|
|
dev: Installation.isLocal(),
|
|
level: (() => {
|
|
if (Installation.isLocal()) return "DEBUG"
|
|
return "INFO"
|
|
})(),
|
|
})
|
|
|
|
Heap.start()
|
|
|
|
process.on("unhandledRejection", (e) => {
|
|
Log.Default.error("rejection", {
|
|
e: e instanceof Error ? e.message : e,
|
|
})
|
|
})
|
|
|
|
process.on("uncaughtException", (e) => {
|
|
Log.Default.error("exception", {
|
|
e: e instanceof Error ? e.message : e,
|
|
})
|
|
})
|
|
|
|
// Subscribe to global events and forward them via RPC
|
|
GlobalBus.on("event", (event) => {
|
|
Rpc.emit("global.event", event)
|
|
})
|
|
|
|
let server: Awaited<ReturnType<typeof Server.listen>> | undefined
|
|
|
|
export const rpc = {
|
|
async fetch(input: { url: string; method: string; headers: Record<string, string>; body?: string }) {
|
|
const headers = { ...input.headers }
|
|
const auth = getAuthorizationHeader()
|
|
if (auth && !headers["authorization"] && !headers["Authorization"]) {
|
|
headers["Authorization"] = auth
|
|
}
|
|
const request = new Request(input.url, {
|
|
method: input.method,
|
|
headers,
|
|
body: input.body,
|
|
})
|
|
const response = await Server.Default().app.fetch(request)
|
|
const body = await response.text()
|
|
return {
|
|
status: response.status,
|
|
headers: Object.fromEntries(response.headers.entries()),
|
|
body,
|
|
}
|
|
},
|
|
snapshot() {
|
|
const result = writeHeapSnapshot("server.heapsnapshot")
|
|
return result
|
|
},
|
|
async server(input: { port: number; hostname: string; mdns?: boolean; cors?: string[] }) {
|
|
if (server) await server.stop(true)
|
|
server = await Server.listen(input)
|
|
return { url: server.url.toString() }
|
|
},
|
|
async checkUpgrade(input: { directory: string }) {
|
|
await Instance.provide({
|
|
directory: input.directory,
|
|
init: () => AppRuntime.runPromise(InstanceBootstrap),
|
|
fn: async () => {
|
|
await upgrade().catch(() => {})
|
|
},
|
|
})
|
|
},
|
|
async reload() {
|
|
await AppRuntime.runPromise(Config.Service.use((cfg) => cfg.invalidate(true)))
|
|
},
|
|
async shutdown() {
|
|
Log.Default.info("worker shutting down")
|
|
|
|
await Instance.disposeAll()
|
|
if (server) await server.stop(true)
|
|
},
|
|
}
|
|
|
|
Rpc.listen(rpc)
|
|
|
|
function getAuthorizationHeader(): string | undefined {
|
|
const password = Flag.OPENCODE_SERVER_PASSWORD
|
|
if (!password) return undefined
|
|
const username = Flag.OPENCODE_SERVER_USERNAME ?? "opencode"
|
|
return `Basic ${btoa(`${username}:${password}`)}`
|
|
}
|