/**
 * ContractVisualToolsNudge — mid-run reminder when the user asked for contract
 * or visual-regression tests but no matching exec (or run_tests/verify_fix)
 * has run yet.
 *
 * Soft: max 1 fire. Pairs with evaluateInventedContractVisualCompletionGate.
 */

export const CONTRACT_VISUAL_TOOLS_NUDGE_MAX_ATTEMPTS = 1;

const CONTRACT_VISUAL_USER_RE =
  /(?:\bcontract tests?\b|\bvisual(?: regression)? tests?\b|\bpact\b|\bschemathesis\b|\bchromatic\b|\bpercy\b|\bloki\b|契约测试|视觉回归)/iu;

const EXEC_TOOLS = new Set(['exec', 'exec_background']);

export interface ContractVisualToolsNudgeRequest {
  userText: string;
  toolCallsByName: Record<string, number>;
  messages?: Array<{ role?: string; content?: unknown }>;
  totalToolCalls: number;
  attempts: number;
}

export type ContractVisualToolsNudgeResult =
  | { fire: false }
  | { fire: true; correction: string };

function sawContractVisualEvidence(
  byName: Record<string, number>,
  messages: Array<{ role?: string; content?: unknown }> | undefined,
): boolean {
  if ((byName.run_tests ?? 0) > 0 || (byName.verify_fix ?? 0) > 0) return true;
  if (!messages?.length) return false;
  for (const m of messages) {
    if (!m || m.role !== 'assistant' || !Array.isArray(m.content)) continue;
    for (const block of m.content) {
      const b = block as { type?: string; name?: string; input?: unknown };
      if (b?.type !== 'tool_use' || !b.name || !EXEC_TOOLS.has(b.name)) continue;
      const input = b.input;
      if (!input || typeof input !== 'object') continue;
      const o = input as Record<string, unknown>;
      let cmd = '';
      for (const key of ['command', 'cmd', 'input'] as const) {
        if (typeof o[key] === 'string' && String(o[key]).trim()) {
          cmd = String(o[key]);
          break;
        }
      }
      if (
        /\bpact\b/i.test(cmd) ||
        /\bschemathesis\b/i.test(cmd) ||
        /\bdredd\b/i.test(cmd) ||
        /\bchromatic\b/i.test(cmd) ||
        /\bpercy\b/i.test(cmd) ||
        /\bloki\b/i.test(cmd) ||
        /\bbackstop\b/i.test(cmd) ||
        /\bnpm run (?:test:)?(?:contract|visual|chromatic)\b|\bpnpm (?:run )?(?:test:)?(?:contract|visual|chromatic)\b|\byarn (?:test:)?(?:contract|visual|chromatic)\b/i.test(
          cmd,
        )
      ) {
        return true;
      }
    }
  }
  return false;
}

export function evaluateContractVisualToolsNudge(
  request: ContractVisualToolsNudgeRequest,
): ContractVisualToolsNudgeResult {
  if (request.attempts >= CONTRACT_VISUAL_TOOLS_NUDGE_MAX_ATTEMPTS) return { fire: false };
  if (request.totalToolCalls < 1) return { fire: false };
  if (sawContractVisualEvidence(request.toolCallsByName, request.messages)) return { fire: false };

  const user = (request.userText || '').trim();
  if (!user || !CONTRACT_VISUAL_USER_RE.test(user)) return { fire: false };

  if (
    /(?:what is|how does|文档|原理|介绍)/iu.test(user) &&
    !/(?:run|please|now|帮我|请|现在|跑)/iu.test(user)
  ) {
    return { fire: false };
  }

  return {
    fire: true,
    correction:
      '[System] The user asked for contract or visual-regression tests, and tools have already run without a matching command ' +
      '(`pact`, `schemathesis`, `chromatic`, `percy`, `npm run test:contract`, etc.) or `run_tests`/`verify_fix`. ' +
      'Run the real suite and report its output, or clearly say it was not run. ' +
      'Do not invent contract/visual results.',
  };
}