import fs from "node:fs/promises";

import path from "node:path";

import {

abortAndDrainAgentHarnessRun,

nativeHookRelayTesting,

queueAgentHarnessMessage,

resetAgentEventsForTest,

const activeAppServerAttemptsForTest = new Set<{

abortController?: AbortController;

promise: Promise<unknown>;

sessionId: string;

sessionKey?: string;

}>();

type RunCodexAppServerAttemptOptions = NonNullable<

const entry = {

abortController,

promise: undefined as unknown as Promise<unknown>,

sessionId: params.sessionId,

sessionKey: params.sessionKey,

};

const promise = runCodexAppServerAttemptImpl(

trackedParams,

}

async function drainActiveAppServerAttemptsForTest(): Promise<void> {

vi.useRealTimers();

const attempts = [...activeAppServerAttemptsForTest];

if (attempts.length === 0) {

return;

}

for (const attempt of attempts) {

attempt.abortController?.abort("test_cleanup");

}

await Promise.race([

Promise.allSettled(attempts.map((attempt) => attempt.promise)),

new Promise<void>((resolve) => {

setTimeout(resolve, 5_000);

const drainedSessions = new Set<string>();

const sessionDrains = attempts.flatMap((attempt) => {

if (!attempt.sessionId || drainedSessions.has(attempt.sessionId)) {

return [];

}

drainedSessions.add(attempt.sessionId);

return [

abortAndDrainAgentHarnessRun({

sessionId: attempt.sessionId,

sessionKey: attempt.sessionKey,

settleMs: 1_000,

forceClear: true,

reason: "test_cleanup",

}).catch(() => undefined),

];

});

const drainResult = await Promise.race([

Promise.allSettled([...attempts.map((attempt) => attempt.promise), ...sessionDrains]).then(

() => "settled" as const,

),

new Promise<"timeout">((resolve) => {

setTimeout(() => resolve("timeout"), 5_000);

}),

]);

if (drainResult === "settled") {

activeAppServerAttemptsForTest.clear();

}

}

export function createParams(sessionFile: string, workspaceDir: string): EmbeddedRunAttemptParams {

resetGlobalHookRunner();

clearInternalHooks();

defaultCodexAppInventoryCache.clear();

vi.useRealTimers();

vi.restoreAllMocks();

vi.useRealTimers();

vi.unstubAllEnvs();

await closeCodexSandboxExecServersForTests();

await fs.rm(tempDir, { recursive: true, force: true });