27 lines
1.0 KiB
TypeScript
27 lines
1.0 KiB
TypeScript
import { HttpRecorderInternal } from "@opencode-ai/http-recorder/internal"
|
|
import { Effect, Layer } from "effect"
|
|
import { WebSocketExecutor } from "../src/route"
|
|
import type { Service as WebSocketExecutorService } from "../src/route/transport/websocket"
|
|
|
|
const liveWebSocket = WebSocketExecutor.open
|
|
|
|
export const webSocketCassetteLayer = (
|
|
cassette: string,
|
|
input: { readonly metadata?: Record<string, unknown>; readonly mode: HttpRecorderInternal.RecordReplayMode },
|
|
): Layer.Layer<WebSocketExecutorService, never, HttpRecorderInternal.Cassette.Service> =>
|
|
Layer.effect(
|
|
WebSocketExecutor.Service,
|
|
Effect.gen(function* () {
|
|
const cassetteService = yield* HttpRecorderInternal.Cassette.Service
|
|
const executor = yield* HttpRecorderInternal.makeWebSocketExecutor({
|
|
name: cassette,
|
|
mode: input.mode,
|
|
metadata: input.metadata,
|
|
cassette: cassetteService,
|
|
live: { open: liveWebSocket },
|
|
compareClientMessagesAsJson: true,
|
|
})
|
|
return WebSocketExecutor.Service.of(executor)
|
|
}),
|
|
)
|