import { runSSH, b64 } from './ssh';

import type { DelegateParams, DelegateResult } from './types';

const HOST = process.env.ZPM_CODEX_HOST || 'REDACTED-IP';

const USER = process.env.ZPM_CODEX_USER || 'zion';

export async function codexAdapter(p: DelegateParams): Promise<DelegateResult> {

  const start = Date.now();

  const cwd = p.cwd || '/tmp';

  const prompt = p.context ? `${p.context}\n\n---\n\n${p.task}` : p.task;

  const cmd =

    `export PATH="$HOME/.local/bin:/usr/local/bin:$PATH"; cd ${JSON.stringify(cwd)} 2>/dev/null; ` +

    `f=$(mktemp); printf %s ${b64(prompt)} | base64 -d > "$f"; ` +

    `codex exec --skip-git-repo-check "$(cat "$f")" 2>&1; rc=$?; rm -f "$f"; exit $rc`;

  try {

    const r = await runSSH(HOST, USER, cmd, p.timeoutMs ?? 240_000);

    return { ok: r.code === 0, output: extractCodexReply(r.stdout) || r.stderr.trim(), artifacts: [], error: r.code === 0 ? undefined : `exit ${r.code}`, durationMs: Date.now() - start };

  } catch (e) {

    return { ok: false, output: '', artifacts: [], error: (e as Error).message, durationMs: Date.now() - start };

  }

}

function extractCodexReply(raw: string): string {

  const s = raw.trim();

  const i = s.lastIndexOf('\ncodex\n');

  let body = i >= 0 ? s.slice(i + 7) : s;

  body = body.replace(/\ntokens used[\s\S]*$/i, '').trim();

  return body || s;

}