opencode-v2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56// The OpenCode 2 provider plugin: one agent, on the `opencode serve` this
// plugin starts or on a server the user runs (see agent.ts and NOTES.md).
import * as Effect from "effect/Effect";
import * as Scope from "effect/Scope";
import * as Stream from "effect/Stream";
import * as z from "zod";
import { Agents, Env, Settings, runPlugin } from "convergence/effect";
import { OpenCodeV2Agent } from "./agent.ts";
import type { Config } from "./agent.ts";

/// The login environment the agent reads: the server's credentials.
export const ENV_VARS = [
  "OPENCODE_SERVER_USERNAME",
  "OPENCODE_SERVER_PASSWORD",
  "CONVERGENCE_OPENCODE_SERVER_PASSWORD",
];

const loginEnv = Effect.fn("OpenCodeV2.loginEnv")(function* () {
  const env = yield* Env;
  const values: Record<string, string> = {};
  for (const name of ENV_VARS) {
    const value = yield* env.get(name);
    if (value) values[name] = value;
  }
  return values;
});

const setting = z.string().nullish();

/// The binary and server address chosen in Settings; empty means the
/// `opencode` on the login PATH and a server this plugin starts.
const readConfig = Effect.fn("OpenCodeV2.readConfig")(function* () {
  const settings = yield* Settings;
  const binaryPath = yield* settings.get("binaryPath", setting);
  const serverUrl = yield* settings.get("serverUrl", setting);
  return { binaryPath: binaryPath || null, serverUrl: serverUrl || null } satisfies Config;
});

/// Registers the agent and keeps it in step with its settings. `agents`
/// receives the agent, for lifecycle tests.
export const registerAgent = Effect.fn("OpenCodeV2.registerAgent")(function* (agents: OpenCodeV2Agent[] = []) {
  const scope = yield* Scope.Scope;
  const settings = yield* Settings;
  const agent = new OpenCodeV2Agent(scope, yield* readConfig(), yield* loginEnv());
  const registration = yield* (yield* Agents).register(agent.definition());
  agent.emit = registration.emit;
  agents.push(agent);
  yield* settings.changes.pipe(
    Stream.filter((change) => change.key === "binaryPath" || change.key === "serverUrl"),
    Stream.runForEach(() => readConfig().pipe(Effect.flatMap((config) => agent.reconfigure(config)))),
    Effect.forkIn(scope),
  );
  yield* Effect.addFinalizer(() => agent.shutdown());
});

export const activate = runPlugin(registerAgent());