openAgent

1import { NodeFileSystem } from "@effect/platform-node"
2import { HttpRecorder } from "@opencode-ai/http-recorder"
3import { HttpRecorderInternal } from "@opencode-ai/http-recorder/internal"
4import { Layer } from "effect"
5import { FetchHttpClient } from "effect/unstable/http"
6import * as path from "node:path"
7import { fileURLToPath } from "node:url"
8import { LLMClient, RequestExecutor } from "../src/route"
9import type { Service as LLMClientService } from "../src/route/client"
10import type { Service as RequestExecutorService } from "../src/route/executor"
11import type { Service as WebSocketExecutorService } from "../src/route/transport/websocket"
12import {
13  recordedEffectGroup,
14  type RecordedCaseOptions as RunnerCaseOptions,
15  type RecordedGroupOptions,
16} from "./recorded-runner"
17import { webSocketCassetteLayer } from "./recorded-websocket"
18 
19const __dirname = path.dirname(fileURLToPath(import.meta.url))
20const FIXTURES_DIR = path.resolve(__dirname, "fixtures", "recordings")
21 
22type RecordedEnv = RequestExecutorService | WebSocketExecutorService | LLMClientService
23 
24type RecordedTestsOptions = RecordedGroupOptions & {
25  readonly options?: HttpRecorder.RecorderOptions
26}
27 
28type RecordedCaseOptions = RunnerCaseOptions & {
29  readonly options?: HttpRecorder.RecorderOptions
30}
31 
32const mergeOptions = (
33  base: HttpRecorder.RecorderOptions | undefined,
34  override: HttpRecorder.RecorderOptions | undefined,
35) => {
36  if (!base) return override
37  if (!override) return base
38  return {
39    ...base,
40    ...override,
41    metadata: base.metadata || override.metadata ? { ...base.metadata, ...override.metadata } : undefined,
42    redact:
43      base.redact || override.redact
44        ? {
45            ...base.redact,
46            ...override.redact,
47            headers: [...(base.redact?.headers ?? []), ...(override.redact?.headers ?? [])],
48            allowRequestHeaders: [
49              ...(base.redact?.allowRequestHeaders ?? []),
50              ...(override.redact?.allowRequestHeaders ?? []),
51            ],
52            allowResponseHeaders: [
53              ...(base.redact?.allowResponseHeaders ?? []),
54              ...(override.redact?.allowResponseHeaders ?? []),
55            ],
56            queryParameters: [...(base.redact?.queryParameters ?? []), ...(override.redact?.queryParameters ?? [])],
57            jsonFields: [...(base.redact?.jsonFields ?? []), ...(override.redact?.jsonFields ?? [])],
58          }
59        : undefined,
60  }
61}
62 
63export const recordedTests = (options: RecordedTestsOptions) =>
64  recordedEffectGroup<RecordedEnv, never, RecordedTestsOptions, RecordedCaseOptions>({
65    duplicateLabel: "recorded cassette",
66    options,
67    cassetteExists: (cassette) => HttpRecorderInternal.hasCassetteSync(cassette, { directory: FIXTURES_DIR }),
68    layer: ({ cassette, metadata, options, caseOptions, recording }) => {
69      const recorderOptions = mergeOptions(options.options, caseOptions.options)
70      const recorderMetadata = {
71        ...recorderOptions?.metadata,
72        ...metadata,
73      }
74      const mode = recording ? "record" : "replay"
75      const cassetteService = HttpRecorderInternal.Cassette.fileSystem({ directory: FIXTURES_DIR }).pipe(
76        Layer.provide(NodeFileSystem.layer),
77      )
78      const requestExecutor = RequestExecutor.layer.pipe(
79        Layer.provide(
80          HttpRecorderInternal.recordingLayer(cassette, {
81            mode,
82            metadata: recorderMetadata,
83            redactor: HttpRecorderInternal.Redactor.make(recorderOptions?.redact),
84            match: recorderOptions?.match,
85          }).pipe(Layer.provide(FetchHttpClient.layer)),
86        ),
87      )
88      const deps = Layer.mergeAll(
89        requestExecutor,
90        webSocketCassetteLayer(cassette, { metadata: recorderMetadata, mode }),
91      )
92      return Layer.mergeAll(deps, LLMClient.layer.pipe(Layer.provide(deps))).pipe(Layer.provide(cassetteService))
93    },
94  })
95