opencode/packages/server/test/usage.test.ts

import fs from "node:fs/promises"
import path from "node:path"
import { expect } from "bun:test"
import { UsageLimits } from "@opencode/core/usage-limits"
import { Money } from "@opencode/schema/money"
import { makeLocationNode } from "@opencode/util/effect/app-node"
import { Effect, Layer } from "effect"
import { tmpdir } from "../../core/test/fixture/tmpdir"
import { it } from "../../core/test/lib/effect"
import { ServerFetch } from "../src/fetch"

/** Requests a location's usage limits from a server whose UsageLimits answers with `get`. */
const request = (get: UsageLimits.Interface["get"]) =>
  Effect.gen(function* () {
    const tmp = yield* Effect.acquireDisposable(Effect.promise(() => tmpdir("opencode-usage-endpoint-")))
    const global = path.join(tmp.path, "global")
    yield* Effect.promise(() => fs.mkdir(global))
    const handler = yield* ServerFetch.make(
      {
        database: { path: ":memory:" },
        config: { directory: global },
        fs: { filewatcher: false },
        models: { fetch: false },
      },
      {
        overrides: [
          UsageLimits.node.replace(
            makeLocationNode({
              service: UsageLimits.Service,
              layer: Layer.succeed(UsageLimits.Service, UsageLimits.Service.of({ get })),
              deps: [],
            }),
          ),
        ],
      },
    )
    const url = new URL("/api/experimental/usage/limits", "http://opencode.local")
    url.searchParams.set("location[directory]", tmp.path)
    const response = yield* Effect.promise(() => handler(new Request(url)))
    return { status: response.status, body: yield* Effect.promise(() => response.json()) }
  })

it.live("returns the usage limits of the OpenCode Console workspace a location is signed in to", () =>
  Effect.gen(function* () {
    const limits: UsageLimits.Info = {
      workspace: { id: "wrk_anomaly", usageURL: "https://opencode.ai/console/wrk_anomaly/activity/overview" },
      budgets: [
        {
          id: "rule_monthly",
          limit: Money.USD.make(50),
          spent: Money.USD.make(12.5),
          scope: "each",
          per: "set",
          coverage: "all",
          models: [],
          vendors: [],
          picked: false,
          shared: false,
          resetsAt: 1793491200000,
        },
      ],
    }
    const result = yield* request(() => Effect.succeed(limits))

    expect(result.status).toBe(200)
    expect(result.body).toMatchObject({ data: limits })
  }),
)

it.live("returns null for a location that isn't signed in to OpenCode Console", () =>
  Effect.gen(function* () {
    const result = yield* request(() => Effect.succeed(undefined))

    expect(result.status).toBe(200)
    expect(result.body).toMatchObject({ data: null })
  }),
)

it.live("reports usage limits Console can't provide as unavailable", () =>
  Effect.gen(function* () {
    const result = yield* request(() =>
      Effect.fail(new UsageLimits.UnavailableError({ message: "OpenCode Console budgets are unavailable" })),
    )

    expect(result.status).toBe(503)
    expect(result.body).toMatchObject({ message: "OpenCode Console budgets are unavailable", service: "opencode" })
  }),
)