'use strict';

const { execFile, execFileSync } = require('child_process');
const fs = require('fs');
const os = require('os');
const path = require('path');

// AI agent 工具介接：依 antigravity → codex → claude 順序偵測可用工具，
// 以非互動模式（stdin 餵提示）執行 sub agent 並取回最終回覆。

const TOOLS = [
  {
    name: 'antigravity',
    // 需人工確認：antigravity 的非互動執行參數尚未驗證（開發機無此工具），此處先比照 codex exec 的形式。
    buildArgs: ({ model }) => ['exec', ...(model ? ['-m', model] : []), '-'],
    resultFrom: 'stdout',
  },
  {
    name: 'codex',
    buildArgs: ({ model, lastMessageFile }) => [
      'exec',
      '--skip-git-repo-check',
      '--sandbox', 'read-only',
      '--output-last-message', lastMessageFile,
      ...(model ? ['-m', model] : []),
      '-',
    ],
    resultFrom: 'lastMessageFile',
  },
  {
    name: 'claude',
    buildArgs: ({ model }) => ['-p', '--output-format', 'text', ...(model ? ['--model', model] : [])],
    resultFrom: 'stdout',
  },
];

/**
 * 依固定優先序（antigravity → codex → claude）偵測本機可用的 AI CLI 工具。
 *
 * 逐一以同步方式執行 `<tool> --version`（逾時 30 秒），第一個成功者即中選，
 * 並取其 stdout 第一行作為版本字串；偵測失敗（未安裝、不可執行、逾時）
 * 則靜默換下一個工具。本函式不會拋出例外。
 *
 * 注意：antigravity 的非互動執行參數尚未驗證（需人工確認），本函式僅確認
 * `--version` 可執行，不保證後續 runAgent 的參數組合正確。
 *
 * @returns {{ name: string, buildArgs: Function, resultFrom: string, version: string } | null}
 *   中選工具的描述物件（TOOLS 項目加上 version 欄位）；所有工具皆不可用時回傳 null。
 * @remarks
 * 使用情境：action 主流程（步驟 3）啟動審查前呼叫一次，取得工具描述後交給
 * runAgent 執行；若回傳 null，主流程會記 ERR 並以失敗收場（無工具即無法審查）。
 */
function detectTool() {
  for (const tool of TOOLS) {
    try {
      const version = execFileSync(tool.name, ['--version'], { encoding: 'utf8', timeout: 30_000 })
        .trim()
        .split('\n')[0];
      return { ...tool, version };
    } catch {
      // 不可用（未安裝或無法執行）→ 換下一個。
    }
  }
  return null;
}

/**
 * 以非互動模式執行一次 sub agent：把提示從 stdin 餵給偵測到的 AI CLI 工具，
 * 等子行程結束後回傳最終文字回覆。
 *
 * 依 tool.resultFrom 決定結果來源：stdout（antigravity、claude），或
 * codex 專用的 --output-last-message 暫存檔（codex exec 的 stdout 夾雜過程
 * 訊息，改讀工具寫出的最終回覆檔，讀取後即刪除）。
 *
 * 本函式永不 reject：任何失敗（非零退出碼、逾時、maxBuffer 超限）都以
 * { ok: false, error } resolve，由呼叫端決定降級行為；並對 child.stdin
 * 掛空 error handler，避免工具提早結束時 EPIPE 造成整個 action 噴例外。
 *
 * 注意：antigravity 的非互動參數尚未驗證（需人工確認），以該工具執行時
 * 可能因參數不符而以 ok: false 收場。
 *
 * @param {{ name: string, buildArgs: Function, resultFrom: string }} tool
 *   工具描述物件（通常來自 detectTool() 的回傳值）。
 * @param {Object} options 執行選項（解構參數）。
 * @param {string} [options.model] 指定模型名稱；未給時不帶模型參數，使用工具預設模型。
 * @param {string} options.prompt 要餵給 agent 的完整提示文字，經 stdin 寫入。
 * @param {string} [options.cwd] 子行程工作目錄；影響工具讀取檔案的相對路徑基準。
 * @param {number} [options.timeoutMs=600000] 子行程逾時毫秒數（預設 10 分鐘），逾時即終止並回報 ok: false。
 * @returns {Promise<{ ok: boolean, output: string, stderr: string, error: Error | null }>}
 *   ok 表示子行程是否成功結束；output 為最終回覆文字（codex 取自
 *   --output-last-message 檔，其餘取 stdout）；stderr 供除錯；error 為失敗原因（成功時為 null）。
 * @remarks
 * 使用情境：審查流程對每個角色組好提示後呼叫本函式，
 * 例如 `const r = await runAgent(tool, { model, prompt, cwd: workspace });`
 * 再以 `r.ok ? extractJson(r.output) : null` 取回結構化 findings，
 * 失敗時記 log 並跳過該角色，不中斷整個 action。
 */
function runAgent(tool, { model, prompt, cwd, timeoutMs = 600_000 }) {
  return new Promise((resolve) => {
    const lastMessageFile = path.join(
      os.tmpdir(),
      `ai-review-${process.pid}-${Math.random().toString(36).slice(2)}.txt`,
    );
    const args = tool.buildArgs({ model, lastMessageFile });
    const child = execFile(
      tool.name,
      args,
      { cwd, encoding: 'utf8', timeout: timeoutMs, maxBuffer: 64 * 1024 * 1024 },
      (error, stdout, stderr) => {
        let output = stdout || '';
        // codex exec 的 stdout 夾雜過程訊息，改讀 --output-last-message 寫出的最終回覆。
        if (tool.resultFrom === 'lastMessageFile' && fs.existsSync(lastMessageFile)) {
          const last = fs.readFileSync(lastMessageFile, 'utf8').trim();
          if (last) output = last;
          fs.rmSync(lastMessageFile, { force: true });
        }
        resolve({ ok: !error, output, stderr: stderr || '', error });
      },
    );
    child.stdin.on('error', () => {
      // 工具提早結束時避免 EPIPE 讓整個 action 噴例外。
    });
    child.stdin.write(prompt);
    child.stdin.end();
  });
}

/**
 * 從 agent 的自由文字回覆中萃取 JSON，容忍 Markdown code fence 與前後雜訊。
 *
 * 處理順序：先取第一個 ``` 或 ```json fence 的內文；再依序以
 * 「第一個 [ 到最後一個 ]」、「第一個 { 到最後一個 }」的最大範圍切片
 * 嘗試 JSON.parse（陣列優先）；最後退而直接 parse 整段文字。
 * 本函式不會拋出例外，所有 parse 失敗一律回傳 null。
 *
 * @param {string} text agent 回覆的原始文字；可為空或 null/undefined。
 * @returns {any | null} 解析成功的 JSON 值（通常為 findings 陣列或物件）；無法解析時為 null。
 * @remarks
 * 使用情境：搭配 runAgent 使用——LLM 即使被要求輸出純 JSON，實務上仍常
 * 包在 ```json fence 內或前後夾說明文字，例如
 * `const findings = extractJson(result.output) ?? [];`
 * 可穩定取回 review findings；回傳 null 時呼叫端應視為該次回覆無效並降級處理。
 */
function extractJson(text) {
  if (!text) return null;
  let t = text.trim();
  const fence = /```(?:json)?\s*([\s\S]*?)```/i.exec(t);
  if (fence) t = fence[1].trim();
  for (const [open, close] of [['[', ']'], ['{', '}']]) {
    const start = t.indexOf(open);
    const end = t.lastIndexOf(close);
    if (start !== -1 && end > start) {
      try {
        return JSON.parse(t.slice(start, end + 1));
      } catch {
        // 換下一種括號組合再試。
      }
    }
  }
  try {
    return JSON.parse(t);
  } catch {
    return null;
  }
}

module.exports = { TOOLS, detectTool, runAgent, extractJson };