372 lines
11 KiB
TypeScript
372 lines
11 KiB
TypeScript
import { randomUUID } from "node:crypto"
|
|
import { EventEmitter } from "node:events"
|
|
import { existsSync, mkdirSync, rmSync } from "node:fs"
|
|
import * as http from "node:http"
|
|
import { createServer } from "node:net"
|
|
import { homedir, tmpdir } from "node:os"
|
|
import { join } from "node:path"
|
|
import { getCACertificates, setDefaultCACertificates } from "node:tls"
|
|
import type { Event } from "electron"
|
|
import { app, BrowserWindow } from "electron"
|
|
|
|
import contextMenu from "electron-context-menu"
|
|
|
|
import type { InitStep, ServerReadyData, SqliteMigrationProgress, WslConfig } from "../preload/types"
|
|
import { checkAppExists, resolveAppPath, wslPath } from "./apps"
|
|
import { CHANNEL, UPDATER_ENABLED } from "./constants"
|
|
import { registerIpcHandlers, sendDeepLinks, sendMenuCommand, sendSqliteMigrationProgress } from "./ipc"
|
|
import { initLogging } from "./logging"
|
|
import { parseMarkdown } from "./markdown"
|
|
import { createMenu } from "./menu"
|
|
import {
|
|
getDefaultServerUrl,
|
|
getWslConfig,
|
|
preferAppEnv,
|
|
setDefaultServerUrl,
|
|
setWslConfig,
|
|
spawnLocalServer,
|
|
type SidecarListener,
|
|
} from "./server"
|
|
import {
|
|
createLoadingWindow,
|
|
createMainWindow,
|
|
registerRendererProtocol,
|
|
setBackgroundColor,
|
|
setDockIcon,
|
|
} from "./windows"
|
|
import { migrate } from "./migrate"
|
|
import { checkUpdate, checkForUpdates, installUpdate, setupAutoUpdater } from "./updater"
|
|
import { Deferred, Effect, Fiber } from "effect"
|
|
|
|
const APP_NAMES: Record<string, string> = {
|
|
dev: "OpenCode Dev",
|
|
beta: "OpenCode Beta",
|
|
prod: "OpenCode",
|
|
}
|
|
const APP_IDS: Record<string, string> = {
|
|
dev: "ai.opencode.desktop.dev",
|
|
beta: "ai.opencode.desktop.beta",
|
|
prod: "ai.opencode.desktop",
|
|
}
|
|
const TEST_ONBOARDING = process.env.OPENCODE_TEST_ONBOARDING === "1"
|
|
|
|
let logger: ReturnType<typeof initLogging>
|
|
let mainWindow: BrowserWindow | null = null
|
|
let server: SidecarListener | null = null
|
|
|
|
const initEmitter = new EventEmitter()
|
|
let initStep: InitStep = { phase: "server_waiting" }
|
|
|
|
const pendingDeepLinks: string[] = []
|
|
|
|
function useEnvProxy() {
|
|
try {
|
|
// Electron 41.2 runs Node 24.14.1; latest @types/node@24 is 24.12.2.
|
|
;(http as any).setGlobalProxyFromEnv()
|
|
} catch (error) {
|
|
logger.warn("failed to load proxy environment", error)
|
|
}
|
|
}
|
|
|
|
function emitDeepLinks(urls: string[]) {
|
|
if (urls.length === 0) return
|
|
pendingDeepLinks.push(...urls)
|
|
if (mainWindow) sendDeepLinks(mainWindow, urls)
|
|
}
|
|
|
|
function setInitStep(step: InitStep) {
|
|
initStep = step
|
|
logger.log("init step", { step })
|
|
initEmitter.emit("step", step)
|
|
}
|
|
|
|
async function killSidecar() {
|
|
if (!server) return
|
|
const current = server
|
|
server = null
|
|
await current.stop()
|
|
}
|
|
|
|
function ensureLoopbackNoProxy() {
|
|
const loopback = ["127.0.0.1", "localhost", "::1"]
|
|
const upsert = (key: string) => {
|
|
const items = (process.env[key] ?? "")
|
|
.split(",")
|
|
.map((value: string) => value.trim())
|
|
.filter((value: string) => Boolean(value))
|
|
|
|
for (const host of loopback) {
|
|
if (items.some((value: string) => value.toLowerCase() === host)) continue
|
|
items.push(host)
|
|
}
|
|
|
|
process.env[key] = items.join(",")
|
|
}
|
|
|
|
upsert("NO_PROXY")
|
|
upsert("no_proxy")
|
|
}
|
|
|
|
const main = Effect.gen(function* () {
|
|
contextMenu({ showSaveImageAs: true, showLookUpSelection: false, showSearchWithGoogle: false })
|
|
|
|
// on macOS apps run in `/` which can cause issues with ripgrep
|
|
try {
|
|
process.chdir(homedir())
|
|
} catch {}
|
|
|
|
process.env.OPENCODE_DISABLE_EMBEDDED_WEB_UI = "true"
|
|
|
|
const appId = app.isPackaged ? APP_IDS[CHANNEL] : "ai.opencode.desktop.dev"
|
|
const onboardingTestRoot = ((): string | undefined => {
|
|
if (!TEST_ONBOARDING) return
|
|
|
|
const root = join(tmpdir(), `opencode-onboarding-${randomUUID()}`)
|
|
rmSync(root, { recursive: true, force: true })
|
|
;["data", "config", "cache", "state", "desktop", "session"].forEach((dir) =>
|
|
mkdirSync(join(root, dir), { recursive: true }),
|
|
)
|
|
process.env.OPENCODE_DB = ":memory:"
|
|
process.env.XDG_DATA_HOME = join(root, "data")
|
|
process.env.XDG_CONFIG_HOME = join(root, "config")
|
|
process.env.XDG_CACHE_HOME = join(root, "cache")
|
|
process.env.XDG_STATE_HOME = join(root, "state")
|
|
return root
|
|
})()
|
|
app.setName(app.isPackaged ? APP_NAMES[CHANNEL] : "OpenCode Dev")
|
|
app.setAppUserModelId(appId)
|
|
app.setPath(
|
|
"userData",
|
|
onboardingTestRoot ? join(onboardingTestRoot, "desktop") : join(app.getPath("appData"), appId),
|
|
)
|
|
if (onboardingTestRoot) app.setPath("sessionData", join(onboardingTestRoot, "session"))
|
|
logger = initLogging()
|
|
|
|
try {
|
|
setDefaultCACertificates([...new Set([...getCACertificates("default"), ...getCACertificates("system")])])
|
|
} catch (error) {
|
|
logger.warn("failed to load system certificates", error)
|
|
}
|
|
|
|
logger.log("app starting", {
|
|
version: app.getVersion(),
|
|
packaged: app.isPackaged,
|
|
onboardingTest: Boolean(onboardingTestRoot),
|
|
})
|
|
|
|
ensureLoopbackNoProxy()
|
|
useEnvProxy()
|
|
app.commandLine.appendSwitch("proxy-bypass-list", "<-loopback>")
|
|
if (!app.isPackaged) app.commandLine.appendSwitch("remote-debugging-port", "9222")
|
|
|
|
if (!app.requestSingleInstanceLock()) {
|
|
app.quit()
|
|
return
|
|
}
|
|
|
|
preferAppEnv(app.getPath("userData"))
|
|
|
|
app.on("second-instance", (_event: Event, argv: string[]) => {
|
|
const urls = argv.filter((arg: string) => arg.startsWith("opencode://"))
|
|
if (urls.length) {
|
|
logger.log("deep link received via second-instance", { urls })
|
|
emitDeepLinks(urls)
|
|
}
|
|
if (mainWindow) {
|
|
mainWindow.show()
|
|
mainWindow.focus()
|
|
}
|
|
})
|
|
|
|
app.on("open-url", (event: Event, url: string) => {
|
|
event.preventDefault()
|
|
logger.log("deep link received via open-url", { url })
|
|
emitDeepLinks([url])
|
|
})
|
|
|
|
app.on("before-quit", () => {
|
|
void killSidecar()
|
|
})
|
|
|
|
app.on("will-quit", () => {
|
|
void killSidecar()
|
|
})
|
|
|
|
for (const signal of ["SIGINT", "SIGTERM"] as const) {
|
|
process.on(signal, () => {
|
|
void killSidecar().finally(() => app.exit(0))
|
|
})
|
|
}
|
|
|
|
const serverReady = Deferred.makeUnsafe<ServerReadyData>()
|
|
const loadingComplete = Deferred.makeUnsafe<void>()
|
|
|
|
registerIpcHandlers({
|
|
killSidecar: () => killSidecar(),
|
|
awaitInitialization: Effect.fnUntraced(
|
|
function* (sendStep) {
|
|
sendStep(initStep)
|
|
const listener = (step: InitStep) => sendStep(step)
|
|
initEmitter.on("step", listener)
|
|
try {
|
|
logger.log("awaiting server ready")
|
|
const res = yield* Deferred.await(serverReady)
|
|
logger.log("server ready", { url: res.url })
|
|
return res
|
|
} finally {
|
|
initEmitter.off("step", listener)
|
|
}
|
|
},
|
|
(e) => Effect.runPromise(e),
|
|
),
|
|
getWindowConfig: () => ({ updaterEnabled: UPDATER_ENABLED }),
|
|
consumeInitialDeepLinks: () => pendingDeepLinks.splice(0),
|
|
getDefaultServerUrl: () => getDefaultServerUrl(),
|
|
setDefaultServerUrl: (url) => setDefaultServerUrl(url),
|
|
getWslConfig: () => Promise.resolve(getWslConfig()),
|
|
setWslConfig: (config: WslConfig) => setWslConfig(config),
|
|
getDisplayBackend: async () => null,
|
|
setDisplayBackend: async () => undefined,
|
|
parseMarkdown: async (markdown) => parseMarkdown(markdown),
|
|
checkAppExists: (appName) => checkAppExists(appName),
|
|
wslPath: async (path, mode) => wslPath(path, mode),
|
|
resolveAppPath: async (appName) => resolveAppPath(appName),
|
|
loadingWindowComplete: () => Deferred.doneUnsafe(loadingComplete, Effect.void),
|
|
runUpdater: async (alertOnFail) => checkForUpdates(alertOnFail, killSidecar),
|
|
checkUpdate: async () => checkUpdate(),
|
|
installUpdate: async () => installUpdate(killSidecar),
|
|
setBackgroundColor: (color) => setBackgroundColor(color),
|
|
})
|
|
|
|
yield* Effect.promise(() => app.whenReady())
|
|
|
|
if (!TEST_ONBOARDING) migrate()
|
|
app.setAsDefaultProtocolClient("opencode")
|
|
registerRendererProtocol()
|
|
setDockIcon()
|
|
setupAutoUpdater()
|
|
|
|
const needsMigration = ((): boolean => {
|
|
if (process.env.OPENCODE_DB === ":memory:") return false
|
|
|
|
const xdg = process.env.XDG_DATA_HOME
|
|
const base = xdg && xdg.length > 0 ? xdg : join(homedir(), ".local", "share")
|
|
return !existsSync(join(base, "opencode", "opencode.db"))
|
|
})()
|
|
let overlay: BrowserWindow | null = null
|
|
|
|
const port = yield* Effect.gen(function* () {
|
|
const fromEnv = process.env.OPENCODE_PORT
|
|
if (fromEnv) {
|
|
const parsed = Number.parseInt(fromEnv, 10)
|
|
if (!Number.isNaN(parsed)) return parsed
|
|
}
|
|
|
|
const res = yield* Deferred.make<number, unknown>()
|
|
const server = createServer()
|
|
server.on("error", (e) => Deferred.failSync(res, () => e))
|
|
server.listen(0, "127.0.0.1", () => {
|
|
const address = server.address()
|
|
if (typeof address !== "object" || !address) {
|
|
server.close()
|
|
Deferred.failSync(res, () => new Error("Failed to get port"))
|
|
return
|
|
}
|
|
const port = address.port
|
|
server.close(() => Effect.runSync(Deferred.succeed(res, port)))
|
|
})
|
|
|
|
return yield* Deferred.await(res)
|
|
})
|
|
const hostname = "127.0.0.1"
|
|
const url = `http://${hostname}:${port}`
|
|
const password = randomUUID()
|
|
|
|
const loadingTask = yield* Effect.gen(function* () {
|
|
logger.log("sidecar connection started", { url })
|
|
|
|
initEmitter.on("sqlite", (progress: SqliteMigrationProgress) => {
|
|
setInitStep({ phase: "sqlite_waiting" })
|
|
if (overlay) sendSqliteMigrationProgress(overlay, progress)
|
|
if (mainWindow) sendSqliteMigrationProgress(mainWindow, progress)
|
|
})
|
|
|
|
logger.log("spawning sidecar", { url })
|
|
const { listener, health } = yield* Effect.promise(() =>
|
|
spawnLocalServer(
|
|
hostname,
|
|
port,
|
|
password,
|
|
() => {
|
|
ensureLoopbackNoProxy()
|
|
useEnvProxy()
|
|
},
|
|
{
|
|
needsMigration,
|
|
userDataPath: app.getPath("userData"),
|
|
onSqliteProgress: (progress) => initEmitter.emit("sqlite", progress),
|
|
onStdout: (message) => logger.log("sidecar stdout", { message }),
|
|
onStderr: (message) => logger.warn("sidecar stderr", { message }),
|
|
onExit: (code) => logger.warn("sidecar exited", { code }),
|
|
},
|
|
),
|
|
)
|
|
server = listener
|
|
yield* Deferred.succeed(serverReady, {
|
|
url,
|
|
username: "opencode",
|
|
password,
|
|
})
|
|
|
|
yield* Effect.promise(() => health.wait).pipe(
|
|
Effect.timeout("30 seconds"),
|
|
Effect.catch((e) =>
|
|
Effect.sync(() => {
|
|
logger.error("sidecar health check failed", e.toString())
|
|
}),
|
|
),
|
|
)
|
|
|
|
logger.log("loading task finished")
|
|
}).pipe(Effect.forkChild)
|
|
|
|
if (needsMigration) {
|
|
const show = yield* loadingTask.pipe(
|
|
Fiber.await,
|
|
Effect.timeout("1 second"),
|
|
Effect.as(false),
|
|
Effect.catch(() => Effect.succeed(true)),
|
|
)
|
|
if (show) {
|
|
overlay = createLoadingWindow()
|
|
yield* Effect.sleep("1 second")
|
|
}
|
|
}
|
|
|
|
yield* Fiber.await(loadingTask)
|
|
setInitStep({ phase: "done" })
|
|
|
|
if (overlay) yield* Deferred.await(loadingComplete)
|
|
|
|
mainWindow = createMainWindow()
|
|
if (mainWindow) {
|
|
createMenu({
|
|
trigger: (id) => mainWindow && sendMenuCommand(mainWindow, id),
|
|
checkForUpdates: () => {
|
|
void checkForUpdates(true, killSidecar)
|
|
},
|
|
reload: () => mainWindow?.reload(),
|
|
relaunch: () => {
|
|
void killSidecar().finally(() => {
|
|
app.relaunch()
|
|
app.exit(0)
|
|
})
|
|
},
|
|
})
|
|
}
|
|
|
|
overlay?.close()
|
|
})
|
|
|
|
Effect.runFork(main)
|