import type { ChildProcess } from 'child_process';
import { EventEmitter } from 'events';
import fs from 'fs';
import os from 'os';
import path from 'path';
import { PassThrough } from 'stream';

import { afterEach, describe, expect, it, vi } from 'vitest';

import { OUTPUT_END_MARKER, OUTPUT_START_MARKER } from './agent-protocol.js';
import { runSpawnedAgentProcess } from './agent-runner-process.js';
import type { AgentInput, AgentOutput } from './agent-runner.js';
import type { RegisteredGroup } from './types.js';

function buildProcess(): ChildProcess {
  const proc = new EventEmitter() as ChildProcess & {
    stdout: PassThrough;
    stderr: PassThrough;
    kill: ReturnType<typeof vi.fn>;
    killed: boolean;
  };
  proc.stdout = new PassThrough();
  proc.stderr = new PassThrough();
  proc.killed = false;
  proc.kill = vi.fn(() => {
    proc.killed = true;
    return true;
  });
  return proc;
}

function timed<T>(promise: Promise<T>): Promise<T> {
  return Promise.race([
    promise,
    new Promise<never>((_, reject) =>
      setTimeout(() => reject(new Error('agent process did not resolve')), 100),
    ),
  ]);
}

const group: RegisteredGroup = {
  name: 'Agent Runner Test',
  folder: 'agent-runner-test',
  trigger: '@Agent',
  added_at: '2026-01-01T00:00:00.000Z',
  agentType: 'codex',
};

const input: AgentInput = {
  prompt: 'run',
  groupFolder: 'agent-runner-test',
  chatJid: 'dc:test',
  runId: 'run-1',
  isMain: false,
};

describe('runSpawnedAgentProcess', () => {
  let logsDir: string | null = null;

  afterEach(() => {
    if (logsDir) {
      fs.rmSync(logsDir, { recursive: true, force: true });
      logsDir = null;
    }
  });

  it('resolves even when streamed output delivery rejects', async () => {
    logsDir = fs.mkdtempSync(path.join(os.tmpdir(), 'ejclaw-agent-runner-'));
    const proc = buildProcess();
    const onOutput = vi.fn<(_: AgentOutput) => Promise<void>>(async () => {
      throw new Error('delivery failed');
    });

    const resultPromise = runSpawnedAgentProcess({
      proc,
      group,
      input,
      processName: 'test-agent',
      logsDir,
      startTime: Date.now(),
      onOutput,
    });

    (proc.stdout as PassThrough).write(
      [
        OUTPUT_START_MARKER,
        JSON.stringify({ status: 'success', result: 'hello' }),
        OUTPUT_END_MARKER,
      ].join('\n'),
    );
    proc.emit('close', 0, null);

    await expect(timed(resultPromise)).resolves.toMatchObject({
      status: 'success',
      result: null,
    });
    expect(onOutput).toHaveBeenCalledOnce();
  });
});