agent-orchestrator/packages/plugins/agent-opencode/src/index.ts

149 lines
4.7 KiB
TypeScript

import {
shellEscape,
type Agent,
type AgentSessionInfo,
type AgentLaunchConfig,
type ActivityState,
type PluginModule,
type RuntimeHandle,
type Session,
} from "@composio/ao-core";
import { execFile } from "node:child_process";
import { promisify } from "node:util";
const execFileAsync = promisify(execFile);
// =============================================================================
// Plugin Manifest
// =============================================================================
export const manifest = {
name: "opencode",
slot: "agent" as const,
description: "Agent plugin: OpenCode",
version: "0.1.0",
};
// =============================================================================
// Agent Implementation
// =============================================================================
function createOpenCodeAgent(): Agent {
return {
name: "opencode",
processName: "opencode",
getLaunchCommand(config: AgentLaunchConfig): string {
const parts: string[] = ["opencode"];
if (config.prompt) {
parts.push("run", shellEscape(config.prompt));
}
if (config.model) {
parts.push("--model", shellEscape(config.model));
}
return parts.join(" ");
},
getEnvironment(config: AgentLaunchConfig): Record<string, string> {
const env: Record<string, string> = {};
env["AO_SESSION_ID"] = config.sessionId;
// NOTE: AO_PROJECT_ID is the caller's responsibility (spawn.ts sets it)
if (config.issueId) {
env["AO_ISSUE_ID"] = config.issueId;
}
return env;
},
detectActivity(terminalOutput: string): ActivityState {
if (!terminalOutput.trim()) return "idle";
// OpenCode doesn't have rich terminal output patterns yet
return "active";
},
async getActivityState(session: Session, _readyThresholdMs?: number): Promise<ActivityState | null> {
// Check if process is running first
if (!session.runtimeHandle) return "exited";
const running = await this.isProcessRunning(session.runtimeHandle);
if (!running) return "exited";
// NOTE: OpenCode stores all session data in a single global SQLite database
// at ~/.local/share/opencode/opencode.db without per-workspace scoping. When
// multiple OpenCode sessions run in parallel, database modifications from any
// session will cause all sessions to appear active. Until OpenCode provides
// per-workspace session tracking, we return null (unknown) rather than guessing.
//
// TODO: Implement proper per-session activity detection when OpenCode supports it.
return null;
},
async isProcessRunning(handle: RuntimeHandle): Promise<boolean> {
try {
if (handle.runtimeName === "tmux" && handle.id) {
const { stdout: ttyOut } = await execFileAsync(
"tmux",
["list-panes", "-t", handle.id, "-F", "#{pane_tty}"],
{ timeout: 30_000 },
);
const ttys = ttyOut
.trim()
.split("\n")
.map((t) => t.trim())
.filter(Boolean);
if (ttys.length === 0) return false;
const { stdout: psOut } = await execFileAsync("ps", ["-eo", "pid,tty,args"], {
timeout: 30_000,
});
const ttySet = new Set(ttys.map((t) => t.replace(/^\/dev\//, "")));
const processRe = /(?:^|\/)opencode(?:\s|$)/;
for (const line of psOut.split("\n")) {
const cols = line.trimStart().split(/\s+/);
if (cols.length < 3 || !ttySet.has(cols[1] ?? "")) continue;
const args = cols.slice(2).join(" ");
if (processRe.test(args)) {
return true;
}
}
return false;
}
const rawPid = handle.data["pid"];
const pid = typeof rawPid === "number" ? rawPid : Number(rawPid);
if (Number.isFinite(pid) && pid > 0) {
try {
process.kill(pid, 0);
return true;
} catch (err: unknown) {
if (err instanceof Error && "code" in err && err.code === "EPERM") {
return true;
}
return false;
}
}
return false;
} catch {
return false;
}
},
async getSessionInfo(_session: Session): Promise<AgentSessionInfo | null> {
// OpenCode doesn't have JSONL session files for introspection yet
return null;
},
};
}
// =============================================================================
// Plugin Export
// =============================================================================
export function create(): Agent {
return createOpenCodeAgent();
}
export default { manifest, create } satisfies PluginModule<Agent>;