diff options
| author | Mark Jaquith <mark.jaquith@gusto.com> | 2026-09-11 16:36:18 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2026-09-11 22:36:18 +0200 |
| commit | fb5525fc2234a246dfdfc9e0d7ed73c824cbd8a6 (patch) | |
| tree | f989f8175e8afdd89824fd94477b1b1645af05ff /src/integration/assets/opencode | |
| parent | 38d002de59331b27aee2f5838639ea27a04878f8 (diff) | |
fix: opencode v2 lifecycle reporting (#3757)
* fix: support opencode v2 lifecycle reporting
* fix: repair Japanese docs
* fix: ignore payload-less OpenCode events
* fix: harden opencode v2 integration install and reporting
Follow-ups on top of the V2 lifecycle support:
- create `cli.json` when OpenCode has no V1 TUI preferences (`tui.json` or
`kv.json`) to migrate, instead of only registering into an existing file
- settle stalled socket attempts with a plain connect timer, and resend the
latest lifecycle state after a failed delivery so the pane cannot get stuck
- resolve the OpenCode state directory from `XDG_STATE_HOME`
- remove the managed `herdr-opencode` directory on uninstall
- document failed executions reporting `blocked`, keep the integration test
environment independent of an inherited `XDG_STATE_HOME`, and simplify
`reconcileBlockers` to reassign its map
---------
Co-authored-by: Jonathan Liebig <jonathan.liebig@gmail.com>
Diffstat (limited to 'src/integration/assets/opencode')
| -rw-r--r-- | src/integration/assets/opencode/herdr-agent-state.js | 11 | ||||
| -rw-r--r-- | src/integration/assets/opencode/herdr-agent-state.test.ts | 10 | ||||
| -rw-r--r-- | src/integration/assets/opencode/herdr-tui-session.js | 243 | ||||
| -rw-r--r-- | src/integration/assets/opencode/herdr-tui-session.test.ts | 224 | ||||
| -rw-r--r-- | src/integration/assets/opencode/tui.js | 5 |
5 files changed, 475 insertions, 18 deletions
diff --git a/src/integration/assets/opencode/herdr-agent-state.js b/src/integration/assets/opencode/herdr-agent-state.js index 45032335..73a70534 100644 --- a/src/integration/assets/opencode/herdr-agent-state.js +++ b/src/integration/assets/opencode/herdr-agent-state.js @@ -2,7 +2,7 @@ // managed by herdr; reinstalling or updating the integration overwrites this file. // add custom hooks/plugins beside this file instead of editing it. // HERDR_INTEGRATION_ID=opencode -// HERDR_INTEGRATION_VERSION=11 +// HERDR_INTEGRATION_VERSION=12 import net from "node:net"; @@ -199,3 +199,12 @@ export const HerdrAgentStatePlugin = async () => { }, }; }; + +// V1 (1.18.29+) calls server(). V2 calls setup() instead. Its shared server +// cannot attribute sessions using its process environment: the pane-local TUI +// owns both selection and lifecycle reporting there, including remote servers. +export default { + id: "herdr.opencode", + server: HerdrAgentStatePlugin, + setup() {}, +}; diff --git a/src/integration/assets/opencode/herdr-agent-state.test.ts b/src/integration/assets/opencode/herdr-agent-state.test.ts index a3eea848..38c1f05c 100644 --- a/src/integration/assets/opencode/herdr-agent-state.test.ts +++ b/src/integration/assets/opencode/herdr-agent-state.test.ts @@ -227,6 +227,16 @@ function requestMethod(request: unknown): unknown { return isRecord(request) ? request.method : undefined; } +test("dual server entrypoint keeps V1 hooks and never reports from the V2 shared server", async () => { + const module = await import(`./herdr-agent-state.js?test=${++importCounter}`); + expect(module.default.server).toBe(module.HerdrAgentStatePlugin); + expect(await module.default.setup({})).toBeUndefined(); + expect(requests).toHaveLength(0); + const hooks = await module.default.server(); + await hooks["chat.message"]({ sessionID: "v1-root" }); + expect(requests.map(requestState)).toEqual(["working"]); +}); + function requestState(request: unknown): unknown { return requestParam(request, "state"); } diff --git a/src/integration/assets/opencode/herdr-tui-session.js b/src/integration/assets/opencode/herdr-tui-session.js index f4f08d57..4e5631ac 100644 --- a/src/integration/assets/opencode/herdr-tui-session.js +++ b/src/integration/assets/opencode/herdr-tui-session.js @@ -1,7 +1,7 @@ // installed by herdr // managed by herdr; reinstalling or updating the integration overwrites this file. // HERDR_INTEGRATION_ID=opencode-tui -// HERDR_INTEGRATION_VERSION=11 +// HERDR_INTEGRATION_VERSION=12 import net from "node:net"; @@ -10,11 +10,11 @@ const AGENT = "opencode"; const ROUTE_POLL_INTERVAL_MS = 100; const SELECTION_RETRY_DELAYS_MS = [100, 400, 1_000]; -function requestOnce(sessionID) { +function requestOnce(sessionID, state, seq, isCurrent = () => true) { const paneId = process.env.HERDR_PANE_ID; const socketPath = process.env.HERDR_SOCKET_PATH; if (!paneId || !socketPath) { - return Promise.resolve(); + return Promise.resolve(true); } const socketEndpoint = @@ -23,35 +23,50 @@ function requestOnce(sessionID) { id: `${SOURCE}:tui:${Date.now()}:${Math.floor(Math.random() * 1_000_000) .toString() .padStart(6, "0")}`, - method: "pane.report_agent_session", + method: state === undefined ? "pane.report_agent_session" : "pane.report_agent", params: { pane_id: paneId, source: SOURCE, agent: AGENT, agent_session_id: sessionID, - session_start_source: "select", + ...(state === undefined ? { session_start_source: "select" } : { state, seq }), }, }; return new Promise((resolve) => { + let settled = false; + let timer; + const settle = (delivered) => { + if (settled) return; + settled = true; + clearTimeout(timer); + client.destroy(); + resolve(delivered); + }; const client = net.createConnection(socketEndpoint, () => { + if (!isCurrent()) { + settle(false); + return; + } client.write(`${JSON.stringify(request)}\n`); }); - const finish = () => { - client.destroy(); - resolve(); - }; - client.setTimeout(500, finish); - client.on("data", finish); - client.on("error", finish); - client.on("end", finish); - client.on("close", resolve); + // A plain timer, not socket.setTimeout, so a connection that never finishes + // connecting still settles and cannot block later reports behind the queue. + timer = setTimeout(() => settle(false), 500); + timer.unref?.(); + client.on("data", () => settle(true)); + client.on("error", () => settle(false)); + client.on("end", () => settle(false)); + client.on("close", () => settle(false)); }); } export default { id: "herdr.opencode.session-selection", + // Keep this plain object dependency-free: V1 and V2 expose different SDK + // packages, but both loaders accept their own lifecycle entry on this object. + setup, tui: async (api) => { if ( process.env.HERDR_ENV !== "1" || @@ -111,3 +126,203 @@ export default { api.lifecycle.onDispose(() => clearInterval(routePoll)); }, }; + +function setup(api) { + if (process.env.HERDR_ENV !== "1" || !process.env.HERDR_SOCKET_PATH || !process.env.HERDR_PANE_ID) return; + + let disposed = false; + let selected; + let generation = 0; + let sequence = Date.now() * 1000; + let chain = Promise.resolve(); + let retryIndex = 0; + let nextSelectionAt = 0; + let state = "idle"; + let retryTimer; + const sessions = new Map(); + let blockers = new Map(); + // Event callbacks may precede cache updates. Retain each delta until the + // cache reflects it, so late hydration cannot undo a reply or lose an ask. + const blockerChanges = new Map(); + + function root(id) { + const seen = new Set(); + while (typeof id === "string" && !seen.has(id)) { + seen.add(id); + const session = api.data.session.get(id) ?? sessions.get(id); + if (!session) return; + if (!session.parentID) return id; + id = session.parentID; + } + } + + function current() { + const route = api.ui.router.current(); + return route.type === "session" ? root(route.sessionID) : undefined; + } + + // Selection and lifecycle use one queue. Recheck attribution at dispatch, + // not just when receiving the event, and reject A -> B -> A stale work too. + function enqueue(value) { + const sessionID = selected; + const revision = generation; + const isCurrent = () => !disposed && revision === generation && !!sessionID && current() === sessionID; + chain = chain.then(async () => { + if (!isCurrent()) return; + const delivered = await requestOnce(sessionID, value, value === undefined ? undefined : ++sequence, isCurrent); + if (!delivered) scheduleStateRetry(); + }).catch(() => {}); + } + + // A dropped report must not strand the pane on a stale state once the + // selection retry schedule has run out: resend the latest state until the + // socket accepts it or the selection is no longer current. + function scheduleStateRetry() { + if (disposed || retryTimer) return; + retryTimer = setTimeout(() => { + retryTimer = undefined; + publish(); + }, 500); + retryTimer.unref?.(); + } + + function publish() { + enqueue(blockers.size ? "blocked" : state); + } + + function changeBlocker(id, kind, requestID, present) { + if (typeof requestID !== "string") return; + const key = `${kind}:${requestID}`; + blockerChanges.set(key, { id, kind, present }); + if (present) blockers.set(key, id); + else blockers.delete(key); + } + + function reconcileBlockers() { + const next = new Map(); + const hydrated = new Set(); + const members = new Set([selected, ...api.data.session.family(selected), ...blockers.values()]); + for (const member of members) { + if (root(member) !== selected) continue; + for (const kind of ["permission", "form"]) { + const items = api.data.session[kind].list(member); + if (items === undefined) { + for (const [key, owner] of blockers) { + if (owner === member && key.startsWith(`${kind}:`)) next.set(key, owner); + } + continue; + } + hydrated.add(`${kind}:${member}`); + for (const item of items) next.set(`${kind}:${item.id}`, member); + } + } + for (const [key, change] of blockerChanges) { + if (hydrated.has(`${change.kind}:${change.id}`) && next.has(key) === change.present) { + blockerChanges.delete(key); + } else if (change.present) { + next.set(key, change.id); + } else { + next.delete(key); + } + } + const changed = (blockers.size > 0) !== (next.size > 0); + blockers = next; + return changed; + } + + function syncSelection() { + if (disposed) return; + const id = current(); + if (id !== selected) { + selected = id; + generation += 1; + retryIndex = 0; + nextSelectionAt = 0; + blockers.clear(); + blockerChanges.clear(); + if (id) { + state = api.data.session.status(id) === "running" ? "working" : "idle"; + } + } + if (!id) return; + const blockersChanged = reconcileBlockers(); + if (Date.now() < nextSelectionAt) { + if (blockersChanged) publish(); + return; + } + enqueue(undefined); + publish(); + const delay = SELECTION_RETRY_DELAYS_MS[retryIndex++]; + nextSelectionAt = delay === undefined ? Number.POSITIVE_INFINITY : Date.now() + delay; + } + + function receive({ details: event }) { + if (disposed) return; + const data = event.data; + if (data == null) return; + if (event.type === "session.created") { + sessions.set(data.sessionID, { id: data.sessionID, parentID: data.parentID }); + } + if (event.type === "session.deleted") { + const affected = data.sessionID === selected || [...blockers.values()].includes(data.sessionID); + sessions.delete(data.sessionID); + // Deletion is delivered after the cache can remove the session. Use + // stored ownership rather than looking up the deleted child's ancestry. + for (const [key, owner] of blockers) if (owner === data.sessionID) blockers.delete(key); + for (const [key, change] of blockerChanges) { + if (change.id === data.sessionID) blockerChanges.delete(key); + } + syncSelection(); + if (selected && affected) publish(); + return; + } + syncSelection(); + const id = event.type === "form.created" ? data.form.sessionID : data.sessionID; + if (!selected || root(id) !== selected) return; + switch (event.type) { + case "permission.asked": + changeBlocker(id, "permission", data.id, true); + break; + case "permission.replied": + changeBlocker(id, "permission", data.requestID, false); + break; + case "form.created": + changeBlocker(id, "form", data.form.id, true); + break; + case "form.replied": + case "form.cancelled": + changeBlocker(id, "form", data.id, false); + break; + case "session.execution.started": + if (id !== selected) return; + state = "working"; + break; + case "session.execution.succeeded": + case "session.execution.interrupted": + if (id !== selected) return; + state = "idle"; + break; + case "session.execution.failed": + if (id !== selected) return; + state = "blocked"; + break; + default: + return; + } + publish(); + } + + const unsubscribe = api.data.listen(receive); + syncSelection(); + const poll = setInterval(syncSelection, ROUTE_POLL_INTERVAL_MS); + return () => { + disposed = true; + generation += 1; + clearTimeout(retryTimer); + clearInterval(poll); + unsubscribe(); + sessions.clear(); + blockers.clear(); + blockerChanges.clear(); + }; +} diff --git a/src/integration/assets/opencode/herdr-tui-session.test.ts b/src/integration/assets/opencode/herdr-tui-session.test.ts index 1bcb1f81..dd031527 100644 --- a/src/integration/assets/opencode/herdr-tui-session.test.ts +++ b/src/integration/assets/opencode/herdr-tui-session.test.ts @@ -3,15 +3,25 @@ import { afterEach, beforeEach, expect, mock, test } from "bun:test"; const requests: unknown[] = []; const activeDisposers: Array<() => void> = []; const requestWaiters: Array<() => void> = []; +const stateWaiters: Array<() => void> = []; let importCounter = 0; +let holdConnections = false; +let failConnections = false; +const connections: Array<() => void> = []; mock.module("node:net", () => ({ default: { createConnection(_path: string, onConnect: () => void) { const handlers = new Map<string, () => void>(); const client = { + destroyed: false, write(input: string) { - requests.push(JSON.parse(input.trim())); + if (client.destroyed) return; + const request = JSON.parse(input.trim()); + requests.push(request); + if (isRecord(request) && isRecord(request.params) && request.params.state !== undefined) { + stateWaiters.shift()?.(); + } requestWaiters.shift()?.(); queueMicrotask(() => client.emit("data")); }, @@ -19,12 +29,16 @@ mock.module("node:net", () => ({ on(event: string, handler: () => void) { handlers.set(event, handler); }, - destroy() {}, + destroy() { + client.destroyed = true; + }, emit(event: string) { handlers.get(event)?.(); }, }; - queueMicrotask(onConnect); + if (holdConnections) connections.push(onConnect); + else if (failConnections) queueMicrotask(() => client.emit("error")); + else queueMicrotask(onConnect); return client; }, }, @@ -33,6 +47,10 @@ mock.module("node:net", () => ({ beforeEach(() => { requests.length = 0; requestWaiters.length = 0; + stateWaiters.length = 0; + holdConnections = false; + failConnections = false; + connections.length = 0; process.env.HERDR_ENV = "1"; process.env.HERDR_SOCKET_PATH = "test.sock"; process.env.HERDR_PANE_ID = "test:p1"; @@ -93,6 +111,10 @@ function waitForNextRequest(): Promise<void> { return new Promise((resolve) => requestWaiters.push(resolve)); } +function waitForStateReport(): Promise<void> { + return new Promise((resolve) => stateWaiters.push(resolve)); +} + test("reports a root session when only the local route changes", async () => { const plugin = await loadPlugin(); const tui = fakeApi(); @@ -179,3 +201,199 @@ function requestParam(request: unknown, name: string): unknown { function isRecord(value: unknown): value is Record<string, unknown> { return typeof value === "object" && value !== null; } + +function v2Api() { + const sessions = new Map([ + ["a", { id: "a" }], + ["b", { id: "b" }], + ["child", { id: "child", parentID: "a" }], + ]); + let route = { type: "session", sessionID: "a" }; + const listeners = new Set<(event: unknown) => void>(); + const permissions = new Map<string, Array<{ id: string }> | undefined>(); + const forms = new Map<string, Array<{ id: string }> | undefined>(); + return { + api: { + ui: { router: { current: () => route } }, + data: { + session: { + get: (id: string) => sessions.get(id), + family: () => [...sessions.keys()], + status: () => "idle", + permission: { list: (id: string) => permissions.get(id) }, + form: { list: (id: string) => forms.get(id) }, + }, + listen: (handler: (event: unknown) => void) => { + listeners.add(handler); + return () => listeners.delete(handler); + }, + }, + }, + select(sessionID: string) { route = { type: "session", sessionID }; }, + home() { route = { type: "home", sessionID: "" }; }, + emit(type: string, data?: object) { + for (const listener of listeners) listener({ details: { type, data } }); + }, + listeners, + sessions, + permissions, + forms, + }; +} + +const flushReports = () => new Promise((resolve) => setTimeout(resolve, 10)); +const states = () => requests.filter((r) => requestParam(r, "state") !== undefined) + .map((r) => requestParam(r, "state")); + +test("V2 ignores events without data", async () => { + const plugin = await loadPlugin(); + const tui = v2Api(); + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + requests.length = 0; + expect(() => tui.emit("legacy.event")).not.toThrow(); + tui.emit("session.execution.started", { sessionID: "a" }); + await flushReports(); + expect(states()).toEqual(["working"]); +}); + +test("V2 completes and interrupts without legacy idle events", async () => { + for (const terminal of ["succeeded", "interrupted", "failed"]) { + const plugin = await loadPlugin(); + const tui = v2Api(); + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + requests.length = 0; + tui.emit("session.execution.started", { sessionID: "a" }); + tui.emit(`session.execution.${terminal}`, { sessionID: "a" }); + await flushReports(); + expect(states()).toEqual(["working", terminal === "failed" ? "blocked" : "idle"]); + dispose(); + } +}); + +test("V2 aggregates root and child blockers and ignores other roots and child completion", async () => { + const plugin = await loadPlugin(); + const tui = v2Api(); + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + requests.length = 0; + tui.emit("session.execution.started", { sessionID: "a" }); + tui.emit("permission.asked", { sessionID: "a", id: "permission-a" }); + tui.emit("form.created", { form: { sessionID: "child", id: "form-child" } }); + tui.emit("permission.replied", { sessionID: "a", requestID: "permission-a" }); + tui.emit("session.execution.succeeded", { sessionID: "child" }); + tui.emit("session.execution.started", { sessionID: "b" }); + tui.emit("permission.asked", { sessionID: "b", id: "other" }); + await flushReports(); + expect(states().at(-1)).toBe("blocked"); + expect(requests.every((r) => requestParam(r, "agent_session_id") === "a")).toBe(true); + tui.emit("form.cancelled", { sessionID: "child", id: "form-child" }); + tui.emit("session.execution.succeeded", { sessionID: "a" }); + await flushReports(); + expect(states().slice(-2)).toEqual(["working", "idle"]); +}); + +test("V2 discards queued reports after selection changes and stops on disposal", async () => { + const plugin = await loadPlugin(); + const tui = v2Api(); + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + requests.length = 0; + tui.emit("session.execution.started", { sessionID: "a" }); + tui.select("b"); + tui.emit("session.execution.started", { sessionID: "b" }); + await flushReports(); + expect(requests.every((r) => requestParam(r, "agent_session_id") === "b")).toBe(true); + requests.length = 0; + tui.emit("session.execution.succeeded", { sessionID: "b" }); + tui.home(); + await flushReports(); + expect(requests).toHaveLength(0); + dispose(); + expect(tui.listeners.size).toBe(0); + tui.select("a"); + await new Promise((resolve) => setTimeout(resolve, 250)); + expect(requests).toHaveLength(0); +}); + +test("V2 reconciles late blocker hydration without reviving an already-replied request", async () => { + const plugin = await loadPlugin(); + const tui = v2Api(); + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + tui.permissions.set("child", [{ id: "late" }]); + await waitForStateReport(); + expect(states().at(-1)).toBe("blocked"); + tui.emit("permission.replied", { sessionID: "child", requestID: "late" }); + await waitForStateReport(); + expect(states().at(-1)).toBe("idle"); + tui.permissions.set("child", []); + tui.forms.set("child", [{ id: "second" }]); + await waitForStateReport(); + expect(states().at(-1)).toBe("blocked"); + tui.sessions.delete("child"); + tui.emit("session.deleted", { sessionID: "child" }); + await flushReports(); + expect(states().at(-1)).toBe("idle"); +}); + +test("V2 never writes a delayed connection after disposal or a session switch", async () => { + for (const action of ["dispose", "switch"]) { + const plugin = await loadPlugin(); + const tui = v2Api(); + holdConnections = true; + requests.length = 0; + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + expect(connections.length).toBeGreaterThan(0); + if (action === "dispose") dispose(); + else tui.select("b"); + holdConnections = false; + for (const connect of connections.splice(0)) connect(); + await flushReports(); + expect(requests).toHaveLength(0); + dispose(); + } +}); + +test("V2 settles a connection that never completes", async () => { + const plugin = await loadPlugin(); + const tui = v2Api(); + holdConnections = true; + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + const started = Date.now(); + while (connections.length <= 1 && Date.now() - started < 2_000) { + await new Promise((resolve) => setTimeout(resolve, 25)); + } + expect(connections.length).toBeGreaterThan(1); + dispose(); +}); + +test("V2 resends the latest state after a failed delivery", async () => { + const plugin = await loadPlugin(); + const tui = v2Api(); + const dispose = await plugin.setup(tui.api); + activeDisposers.push(dispose); + await flushReports(); + // Exhaust the selection retry schedule so only the event report remains. + await new Promise((resolve) => setTimeout(resolve, 1_600)); + requests.length = 0; + tui.emit("session.execution.started", { sessionID: "a" }); + await flushReports(); + failConnections = true; + tui.emit("session.execution.succeeded", { sessionID: "a" }); + const resend = waitForStateReport(); + await new Promise((resolve) => setTimeout(resolve, 700)); + failConnections = false; + await resend; + expect(states().at(-1)).toBe("idle"); + dispose(); +}); diff --git a/src/integration/assets/opencode/tui.js b/src/integration/assets/opencode/tui.js new file mode 100644 index 00000000..cc1ffe6a --- /dev/null +++ b/src/integration/assets/opencode/tui.js @@ -0,0 +1,5 @@ +// installed by herdr +// HERDR_INTEGRATION_ID=opencode-tui-v2 +// HERDR_INTEGRATION_VERSION=12 +// V2 resolves the directory's tui entrypoint; V1 uses the original file. +export { default } from "../herdr-tui-session.js"; |
