1import * as NodeCrypto from "@effect/platform-node/NodeCrypto";
2import * as NodeServices from "@effect/platform-node/NodeServices";
3import { GrokSettings } from "@t3tools/provider-grok/settings";
4import * as Effect from "effect/Effect";
5import * as FileSystem from "effect/FileSystem";
6import * as Layer from "effect/Layer";
7import * as Path from "effect/Path";
8import * as Schema from "effect/Schema";
9import * as ChildProcessSpawner from "effect/process/ChildProcessSpawner";
10import * as HostProcess from "@t3tools/shared/HostProcess";
11import { resolveSelfInvocation } from "@t3tools/shared/nodeRuntime";
12
13import * as TestProviderHost from "@t3tools/provider-testing/TestProviderHost";
14import { GROK_ACP_CANCEL_META, GROK_ACP_INITIALIZE_META } from "@t3tools/provider-grok/testing";
15import { makeXAiPromptCompletionRuntime } from "@t3tools/provider-grok/testing";
16import * as IdAllocator from "@t3tools/provider-core/server/IdAllocator";
17import * as ProviderContinuationRequests from "@t3tools/provider-core/server/ProviderContinuationRequests";
18import * as ProviderAdapterRegistry from "../ProviderAdapterRegistry.ts";
19import type { ProviderReplayGate } from "@t3tools/provider-testing/replayGate";
20import type { OrchestratorV2ProviderReplayHarness } from "../testkit/ProviderReplayHarness.ts";
21import {
22 type AcpReplayTranscript,
23 AcpReplayTranscriptDecodeError,
24 decodeAcpReplayTranscript,
25 makeAcpReplayCompletenessAssertion,
26 makeAcpReplayRuntime,
27} from "./AcpAdapterV2.testkit.ts";
28import {
29 GROK_DEFAULT_INSTANCE_ID,
30 GROK_PROVIDER,
31 makeGrokAdapterV2,
32} from "@t3tools/provider-grok/testing";
33
34const DEFAULT_GROK_SETTINGS = Schema.decodeSync(GrokSettings)({});
35
36function layerGrokProviderAdapterRegistryReplay(
37 transcript: AcpReplayTranscript,
38 options: { readonly replayGate?: ProviderReplayGate } = {},
39) {
40 const layerHost = TestProviderHost.layer().pipe(
41 Layer.provide(Layer.mergeAll(NodeServices.layer, NodeCrypto.layer)),
42 );
43
44 return ProviderAdapterRegistry.layerFromAdaptersEffect(
45 Effect.gen(function* () {
46 const fileSystem = yield* FileSystem.FileSystem;
47 const path = yield* Path.Path;
48 const childProcessSpawner = yield* ChildProcessSpawner.ChildProcessSpawner;
49 const hostPlatform = yield* HostProcess.Platform;
50 const continuationRequests = yield* ProviderContinuationRequests.ProviderContinuationRequests;
51 const replayGate = options.replayGate;
52 const replayDir = yield* fileSystem
53 .makeTempDirectoryScoped({
54 prefix: `t3-orchestration-v2-grok-replay-${transcript.scenario}-`,
55 })
56 .pipe(Effect.orDie);
57 const statusPath = path.join(replayDir, "status.json");
58 const scriptPath = yield* path
59 .fromFileUrl(new URL("../../../scripts/acp-replay-agent.ts", import.meta.url))
60 .pipe(Effect.orDie);
61 const adapter = yield* makeGrokAdapterV2({
62 instanceId: GROK_DEFAULT_INSTANCE_ID,
63 settings: DEFAULT_GROK_SETTINGS,
64 environment: {},
65 hostPlatform,
66 selfInvocation: yield* resolveSelfInvocation(),
67 makeRuntime: (runtimeInput) =>
68 makeAcpReplayRuntime({
69 transcript,
70 statusPath,
71 scriptPath,
72 childProcessSpawner,
73 fileSystem,
74 ...(replayGate === undefined ? {} : { replayGate }),
75 cancelMeta: GROK_ACP_CANCEL_META,
76 initializeMeta: GROK_ACP_INITIALIZE_META,
77 })(runtimeInput).pipe(Effect.flatMap(makeXAiPromptCompletionRuntime)),
78 continuationRequests,
79 assertComplete: makeAcpReplayCompletenessAssertion(fileSystem, statusPath, transcript),
80 ...(replayGate === undefined
81 ? {}
82 : {
83 testHooks: {
84 onDeferredFinalizeScheduled: (debounce) =>
85 Effect.sync(() => replayGate.recordFinishArmed(debounce)),
86 },
87 }),
88 });
89 return [adapter];
90 }),
91 ).pipe(
92 Layer.provide(
93 Layer.mergeAll(layerHost, NodeServices.layer, NodeCrypto.layer, IdAllocator.layer),
94 ),
95 // Held inbound lines must not outlive the scenario and wedge teardown.
96 Layer.merge(
97 Layer.effectDiscard(
98 Effect.addFinalizer(() => Effect.sync(() => options.replayGate?.releaseAll())),
99 ),
100 ),
101 );
102}
103
104export const GrokOrchestratorReplayHarness: OrchestratorV2ProviderReplayHarness<
105 AcpReplayTranscript,
106 AcpReplayTranscriptDecodeError
107> = {
108 driver: GROK_PROVIDER,
109 decodeTranscript: (transcript) => decodeAcpReplayTranscript(transcript, GROK_PROVIDER),
110 makeProviderAdapterRegistryLayer: layerGrokProviderAdapterRegistryReplay,
111};