import { expect } from "bun:test" import type { SessionConfigOption, SessionConfigSelectOption } from "@agentclientprotocol/sdk" import { Duration, Effect } from "effect" import type { AcpHandle } from "../../lib/cli-process" type JsonRpcRequest = { readonly jsonrpc: "2.0" readonly id: number readonly method: string readonly params?: unknown } type JsonRpcResponse = { readonly jsonrpc: "2.0" readonly id: number readonly result?: T readonly error?: unknown } type JsonRpcNotification = { readonly jsonrpc: "2.0" readonly method: string readonly params?: T } export type AcpClient = { readonly request: (method: string, params?: unknown) => Effect.Effect, unknown> readonly receive: Effect.Effect readonly waitForNotification: ( method: string, predicate: (params: T) => boolean, timeoutMs?: number, ) => Effect.Effect, unknown> } export function createAcpClient(acp: AcpHandle): AcpClient { const state = { nextId: 1 } const request = (method: string, params?: unknown) => Effect.gen(function* () { const id = state.nextId++ const message: JsonRpcRequest = params === undefined ? { jsonrpc: "2.0", id, method } : { jsonrpc: "2.0", id, method, params } yield* acp.send(message) while (true) { const received = yield* acp.receive.pipe(Effect.timeout(Duration.seconds(15))) if (isJsonRpcResponse(received) && received.id === id) return received } }) const waitForNotification = (method: string, predicate: (params: T) => boolean, timeoutMs = 15_000) => Effect.gen(function* () { while (true) { const received = yield* acp.receive.pipe(Effect.timeout(Duration.millis(timeoutMs))) if (!isJsonRpcNotification(received)) continue if (received.method === method && predicate(received.params as T)) return received } }) return { request, receive: acp.receive, waitForNotification, } } export function expectOk(response: JsonRpcResponse) { expect(response.error).toBeUndefined() expect(response.result).toBeDefined() return response.result as T } export function selectConfigOption(options: SessionConfigOption[] | null | undefined, id: string) { return options?.find( (option): option is Extract => option.id === id && option.type === "select", ) } export function firstAlternateValue(option: Extract) { return flattenSelectOptions(option).find((item) => item.value !== option.currentValue)?.value } export function flattenSelectOptions(option: Extract) { return option.options.flatMap((item): SessionConfigSelectOption[] => ("value" in item ? [item] : item.options)) } function isJsonRpcResponse(input: unknown): input is JsonRpcResponse { if (!input || typeof input !== "object") return false return "id" in input && "jsonrpc" in input } function isJsonRpcNotification(input: unknown): input is JsonRpcNotification { if (!input || typeof input !== "object") return false return "method" in input && !("id" in input) && "jsonrpc" in input }