opencode/packages/cli/src/services/server.ts

import { NodeFileSystem } from "@effect/platform-node"
import { Service } from "@opencode-ai/client/effect"
import { ClientError, isUnauthorizedError, OpenCode } from "@opencode-ai/client/promise"
import { InstallationVersion } from "@opencode-ai/core/installation/version"
import { Effect, Redacted } from "effect"
import { Env } from "../env"
import { ServiceConfig } from "./service-config"
import { Standalone } from "./standalone"

export type Args = {
  readonly server?: string
  readonly standalone?: boolean
  readonly mismatch?: "replace" | "ignore" | "error"
  readonly onStart?: Service.StartOptions["onStart"]
}

export type Resolved = {
  readonly endpoint: Service.Endpoint
  readonly reconnect?: (attempt: number) => Promise<Service.Endpoint>
  readonly reload?: () => Promise<void>
}

export const resolve = Effect.fn("cli.server.resolve")(function* (args: Args) {
  if (args.server !== undefined && args.standalone)
    return yield* Effect.fail(new Error("--server and --standalone cannot be combined"))
  if (args.server !== undefined) {
    const password = yield* Env.password
    const endpoint = {
      url: args.server,
      auth: password
        ? { type: "basic" as const, username: "opencode", password: Redacted.value(password) }
        : undefined,
    } satisfies Service.Endpoint
    const client = OpenCode.make({ baseUrl: endpoint.url, headers: Service.headers(endpoint) })
    const health = yield* Effect.tryPromise({
      try: () => client.health.get({ signal: AbortSignal.timeout(5_000) }),
      catch: (cause) => connectError(endpoint, cause),
    })
    if (health.version !== InstallationVersion)
      process.stderr.write(
        `Warning: Server at ${endpoint.url} has version ${health.version}; this client is ${InstallationVersion}. Continuing anyway.\n`,
      )
    return { endpoint } satisfies Resolved
  }
  if (args.standalone) {
    return { endpoint: yield* Standalone.start() } satisfies Resolved
  }

  const options = yield* ServiceConfig.options()
  const endpoint = yield* resolveManaged({ ...options, onStart: args.onStart }, args.mismatch ?? "replace")
  const reconnectOptions = { ...options, version: undefined }
  return {
    endpoint,
    reconnect: (attempt) =>
      Effect.runPromise(
        Effect.gen(function* () {
          if (attempt > 3) return yield* Service.start(reconnectOptions)
          const endpoint = yield* Service.discover(reconnectOptions)
          if (endpoint !== undefined) return endpoint
          return yield* Effect.fail(new Error("Background server is unavailable"))
        }).pipe(Effect.provide(NodeFileSystem.layer)),
      ),
    reload: () =>
      Effect.runPromise(
        Effect.gen(function* () {
          yield* Service.stop(options)
          yield* Service.start(options)
        }).pipe(Effect.provide(NodeFileSystem.layer)),
      ),
  } satisfies Resolved
})

const resolveManaged = Effect.fnUntraced(function* (
  options: Service.StartOptions,
  mismatch: NonNullable<Args["mismatch"]>,
) {
  if (mismatch === "replace") return yield* Service.start(options)
  if (mismatch === "ignore") return yield* Service.start({ ...options, version: undefined })

  const compatible = yield* Service.discover(options)
  if (compatible !== undefined) return compatible
  const existing = yield* Service.discover({ ...options, version: undefined })
  if (existing !== undefined) return yield* Effect.fail(new Error("Background server version does not match this client"))
  return yield* Service.start(options)
})

function connectError(endpoint: Service.Endpoint, cause: unknown) {
  if (isUnauthorizedError(cause)) {
    return new Error(
      endpoint.auth === undefined
        ? `Server at ${endpoint.url} requires a password; set OPENCODE_PASSWORD`
        : `Server at ${endpoint.url} rejected the password`,
      { cause },
    )
  }
  if (cause instanceof ClientError && cause.reason === "Transport")
    return new Error(`Could not reach server at ${endpoint.url}`, { cause })
  return new Error(`Server at ${endpoint.url} did not provide a compatible V2 health response`, { cause })
}

export * as Server from "./server"