import type { Tool } from "@modelcontextprotocol/sdk/types.js";
import type { Server } from "@modelcontextprotocol/sdk/server/index.js";
import {
  ateExtract,
  ateGenerate,
  ateRun,
  ateReport,
  ateHeal,
  ateImpact,
  runFullPipeline,
  analyzeFeedback,
  applyHeal,
  smartSelectRoutes,
  detectCoverageGaps,
  precommitCheck,
} from "@mandujs/ate";
import type { OracleLevel, AteMonitorEvent } from "@mandujs/ate";
import { eventBus } from "@mandujs/core/observability";
import {
  writePartialResults,
  createAteProgressTracker,
  type PartialRunResults,
} from "./ate-run.js";

export const ateToolDefinitions: Tool[] = [
  {
    name: "mandu.ate.extract",
    annotations: {
      readOnlyHint: false,
    },
    description:
      "ATE Step 1 — Extract: Statically analyze the Mandu project's AST to build an interaction graph of routes, slots, contracts, and data flow. " +
      "Identifies all testable interactions without running the server. " +
      "Output is stored in .mandu/ate/extract/ and used by mandu.ate.generate.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        tsconfigPath: { type: "string", description: "Path to tsconfig.json (default: tsconfig.json in repoRoot)" },
        routeGlobs: {
          type: "array",
          items: { type: "string" },
          description: "Glob patterns to limit which routes are analyzed (e.g. ['app/api/**', 'app/blog/**']). Omit for all routes.",
        },
        buildSalt: {
          type: "string",
          description: "Cache invalidation salt — change this to force re-extraction even if source hasn't changed",
        },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.ate.generate",
    annotations: {
      readOnlyHint: false,
    },
    description:
      "ATE Step 2 — Generate: Create Playwright test scenarios from the interaction graph produced by mandu.ate.extract. " +
      "Oracle level controls assertion depth: " +
      "L0 = no assertions (smoke test only), " +
      "L1 = basic HTTP status checks, " +
      "L2 = contract schema validation (response shape matches Zod contract), " +
      "L3 = full behavioral contract (side effects, state changes, error paths). " +
      "Output: .mandu/ate/tests/*.spec.ts ready to run with Playwright.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        oracleLevel: {
          type: "string",
          enum: ["L0", "L1", "L2", "L3"],
          description: "Assertion depth: L0=smoke, L1=HTTP status, L2=contract schema, L3=full behavioral",
        },
        onlyRoutes: {
          type: "array",
          items: { type: "string" },
          description: "Limit test generation to specific routeIds (e.g. ['api-users', 'blog-slug']). Omit for all routes.",
        },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.ate.run",
    annotations: {
      readOnlyHint: false,
    },
    description:
      "ATE Step 3 — Run: Execute the generated Playwright specs against a running Mandu dev server. " +
      "Collects test artifacts (screenshots, traces, results) in .mandu/ate/runs/{runId}/. " +
      "Requires the Mandu dev server to be running (use mandu_dev_start first). " +
      "Returns a runId for use with mandu.ate.report and mandu.ate.heal. " +
      "Streams notifications/progress per spec_done event (issue #238). " +
      "On timeout / kill, persists partial state under .mandu/reports/run-<runId>/results.json " +
      "so mandu.ate.heal remains reachable after the 10-min watchdog. " +
      "Issue #237 — scope filters (onlyFiles / onlyRoutes / grep) let callers " +
      "run a single spec or route and stay under the 10-min MCP watchdog; omit " +
      "them for the full suite.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        baseURL: {
          type: "string",
          description: "Dev server URL (default: http://localhost:3333). Must match the running mandu dev server.",
        },
        ci: { type: "boolean", description: "CI mode: stricter timeouts, no interactive prompts" },
        timeoutMs: {
          type: "number",
          description: "Hard timeout for the whole Playwright process in milliseconds (default: 600000).",
        },
        headless: { type: "boolean", description: "Run browsers headlessly (default: true)" },
        browsers: {
          type: "array",
          items: { type: "string", enum: ["chromium", "firefox", "webkit"] },
          description: "Browsers to test against (default: ['chromium'])",
        },
        onlyFiles: {
          type: "array",
          items: { type: "string" },
          description:
            "Issue #237 — explicit spec file paths (absolute or relative to repoRoot). " +
            "Forwarded to Playwright as positional <file> args.",
        },
        onlyRoutes: {
          type: "array",
          items: { type: "string" },
          description:
            "Issue #237 — route ids (e.g. 'api-signup'). Resolved to spec paths via " +
            "the spec-indexer. Unknown ids emit a warning; the remaining ids run.",
        },
        grep: {
          type: "string",
          description:
            "Issue #237 — pass-through to Playwright --grep. Applied on top of the " +
            "onlyFiles ∪ resolved(onlyRoutes) set.",
        },
        progressToken: {
          type: ["string", "number"],
          description:
            "Optional MCP progress token. When present, per-spec progress notifications are " +
            "sent with this token so the client can correlate them with the originating call.",
        },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.ate.report",
    annotations: {
      readOnlyHint: true,
    },
    description:
      "ATE Step 4 — Report: Generate a test report from run artifacts. " +
      "Produces pass/fail summary, coverage by route, and failure details. " +
      "Use the runId returned by mandu.ate.run. " +
      "If tests failed, follow up with mandu.ate.heal to get fix suggestions.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        runId: { type: "string", description: "Run ID returned by mandu.ate.run" },
        startedAt: { type: "string", description: "ISO timestamp when run started" },
        finishedAt: { type: "string", description: "ISO timestamp when run finished" },
        exitCode: { type: "number", description: "Playwright process exit code (0=pass, non-zero=fail)" },
        oracleLevel: {
          type: "string",
          enum: ["L0", "L1", "L2", "L3"],
          description: "Oracle level used during generation (for report context)",
        },
        format: {
          type: "string",
          enum: ["json", "html", "both"],
          description: "Report format: json (machine-readable), html (visual), both (default)",
        },
        impact: {
          type: "object",
          description: "Impact analysis context — set if tests were run on a subset of routes via mandu.ate.impact",
          properties: {
            mode: { type: "string", enum: ["full", "subset"] },
            changedFiles: { type: "array", items: { type: "string" } },
            selectedRoutes: { type: "array", items: { type: "string" } },
          },
          required: ["mode", "changedFiles", "selectedRoutes"],
        },
      },
      required: ["repoRoot", "runId", "startedAt", "finishedAt", "exitCode"],
    },
  },
  {
    name: "mandu.ate.heal",
    annotations: {
      readOnlyHint: true,
    },
    description:
      "ATE Step 5 — Heal: Analyze test failures from a run and generate safe diff suggestions for fixing the code. " +
      "Classifies failures by root cause (schema mismatch, missing handler, wrong status, selector stale, etc.) " +
      "and produces reviewable diffs — never auto-commits or overwrites files. " +
      "Requires a runId produced by mandu.ate.run (or the partial-results stub written " +
      "on timeout under .mandu/reports/run-<runId>/results.json). " +
      "See mandu.brain.status for LLM availability — template-based heals always run; " +
      "LLM-assisted analysis activates when active_tier is openai or anthropic. " +
      "Use mandu.ate.apply_heal to apply a specific suggestion after review. " +
      "Supports rollback via mandu_rollback if applied changes cause regressions.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        runId: { type: "string", description: "Run ID from mandu.ate.run with failures to analyze" },
      },
      required: ["repoRoot", "runId"],
    },
  },
  {
    name: "mandu.ate.impact",
    annotations: {
      readOnlyHint: true,
    },
    description:
      "ATE Impact Analysis (Phase B.3 v2). Calculates the minimal set of " +
      "routes / specs affected by changed files using git diff, classifies " +
      "contract changes (additive / breaking / renaming), and returns " +
      "`affected.specsToReRun`, `affected.specsLikelyBroken`, " +
      "`affected.missingCoverage`, plus a `suggestions` list keyed to " +
      "re_run / heal / regenerate / add_boundary_test. Stamped with " +
      "graphVersion for agent caching. Keeps v1 fields (changedFiles, " +
      "selectedRoutes, warnings) for backwards compatibility. " +
      "Pass `since: 'working'` for uncommitted changes, `since: 'staged'` " +
      "for staged changes, or a git rev (default: HEAD~1) for committed diffs.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        base: { type: "string", description: "Git base ref (legacy v1 — use `since` instead)" },
        head: { type: "string", description: "Git head ref (legacy v1 — defaults to HEAD)" },
        since: {
          type: "string",
          description:
            "v2 diff source: 'HEAD~1' | 'staged' | 'working' | any git rev. Default 'HEAD~1'.",
        },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.ate.auto_pipeline",
    annotations: {
      readOnlyHint: false,
    },
    description:
      "ATE Full Pipeline — Run the complete ATE cycle in one call: " +
      "Extract AST → Generate Playwright specs → Run tests → Create report → Suggest heals. " +
      "Recommended for: initial setup, scheduled CI runs, and full regression testing. " +
      "For incremental development, prefer individual steps (extract → generate → run → report → heal). " +
      "Set useImpactAnalysis=true to automatically limit tests to changed routes.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        baseURL: { type: "string", description: "Dev server URL (default: http://localhost:3333)" },
        oracleLevel: {
          type: "string",
          enum: ["L0", "L1", "L2", "L3"],
          description: "Assertion depth: L0=smoke, L1=HTTP status, L2=contract schema, L3=full behavioral",
        },
        ci: { type: "boolean", description: "CI mode: stricter timeouts" },
        timeoutMs: {
          type: "number",
          description: "Hard timeout for the whole Playwright process in milliseconds (default: 600000).",
        },
        useImpactAnalysis: {
          type: "boolean",
          description: "Run impact analysis first and only test changed routes (faster in CI)",
        },
        base: { type: "string", description: "Git base ref for impact analysis" },
        head: { type: "string", description: "Git head ref for impact analysis" },
        autoHeal: {
          type: "boolean",
          description: "Automatically run heal analysis after failures (produces diff suggestions, never auto-applies)",
        },
        tsconfigPath: { type: "string", description: "Path to tsconfig.json" },
        routeGlobs: { type: "array", items: { type: "string" }, description: "Limit extraction to specific route patterns" },
        buildSalt: { type: "string", description: "Cache invalidation salt for extraction" },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.ate.feedback",
    annotations: {
      readOnlyHint: true,
    },
    description:
      "ATE Feedback — Evaluate heal suggestions from a failed run and classify which fixes are safe to auto-apply. " +
      "Safe-to-auto-apply: selector-map updates (CSS selector changes). " +
      "Requires human review: contract schema changes, handler logic, route restructuring. " +
      "Returns priority ranking of suggestions to guide review order.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        runId: { type: "string", description: "Run ID with heal suggestions to evaluate" },
        autoApply: {
          type: "boolean",
          description: "If true, auto-apply only selector-map changes (CSS selectors) — other changes always require review",
        },
      },
      required: ["repoRoot", "runId"],
    },
  },
  {
    name: "mandu.ate.apply_heal",
    annotations: {
      destructiveHint: true,
      readOnlyHint: false,
    },
    description:
      "ATE Apply — Apply a specific heal suggestion diff to the codebase. " +
      "Always creates a backup snapshot first (use mandu_rollback to undo). " +
      "Run mandu.ate.feedback first to get the healIndex and confirm the fix is safe. " +
      "After applying, re-run mandu.ate.run to verify the fix resolved the failure.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        runId: { type: "string", description: "Run ID containing the heal suggestion to apply" },
        healIndex: {
          type: "number",
          description: "0-based index of the heal suggestion to apply (from mandu.ate.heal or mandu.ate.feedback results)",
        },
        createBackup: {
          type: "boolean",
          description: "Create a snapshot before applying (default: true, strongly recommended — enables mandu_rollback)",
        },
      },
      required: ["repoRoot", "runId", "healIndex"],
    },
  },
];

export function ateTools(projectRoot: string, server?: Server) {
  /**
   * Shared subscription helper for `mandu.ate.run`. Wraps ateRun (which
   * drives Playwright) with eventBus listeners so per-spec progress
   * notifications flow through the MCP transport and a partial
   * results.json is persisted on timeout / kill. Downstream consumers
   * can then hand the runId to `mandu.ate.heal` even when the 10-min
   * watchdog fired mid-run.
   */
  const runWithObservability = async (
    input: Parameters<typeof ateRun>[0],
    opts: { progressToken?: string | number } = {},
  ) => {
    const started = new Date().toISOString();

    const tracker = createAteProgressTracker({
      progressToken: opts.progressToken,
      sendProgress: async (progress, total, message) => {
        if (!server) return;
        const snap = tracker.snapshot();
        const token = opts.progressToken ?? snap.runId;
        if (!token) return;
        try {
          await server.notification({
            method: "notifications/progress",
            params: { progressToken: token, progress, total, message },
          });
        } catch {
          /* transport offline — never fail the run */
        }
      },
    });

    const unsubscribe = eventBus.on("ate", (event) => {
      try {
        const data = event.data as unknown as AteMonitorEvent | undefined;
        if (!data || typeof data.kind !== "string") return;
        tracker.handle(data);
      } catch {
        /* swallow — never break the run */
      }
    });

    try {
      return await ateRun(input);
    } catch (err) {
      const message = err instanceof Error ? err.message : String(err);
      const isTimeout = /timeout|timed out|타임아웃/i.test(message);
      const snap = tracker.snapshot();
      const partial: PartialRunResults = {
        runId: snap.runId ?? `unknown-${Date.now()}`,
        status: isTimeout ? "timed_out" : "error",
        graphVersion: snap.graphVersion,
        completedSpecs: snap.completedSpecs,
        inProgressSpec: snap.inProgressSpec,
        failures: snap.failures,
        startedAt: started,
        killedAt: new Date().toISOString(),
        error: message,
      };
      const resultsPath = writePartialResults(input.repoRoot, partial);
      return {
        ok: false,
        error: `ateRun failed: ${message}`,
        partial,
        resultsPath,
        runId: partial.runId,
      };
    } finally {
      try {
        unsubscribe();
      } catch {
        /* no-op */
      }
    }
  };
  // Reserved for future use (progress capability detection). Not used
  // during registration today but documented on the closure so the
  // next caller understands the parameter shape.
  void projectRoot;

  return {
    "mandu.ate.extract": async (args: Record<string, unknown>) => {
      const { repoRoot, tsconfigPath, routeGlobs, buildSalt } = args as {
        repoRoot: string;
        tsconfigPath?: string;
        routeGlobs?: string[];
        buildSalt?: string;
      };
      return await ateExtract({ repoRoot, tsconfigPath, routeGlobs, buildSalt });
    },
    "mandu.ate.generate": async (args: Record<string, unknown>) => {
      const { repoRoot, oracleLevel, onlyRoutes } = args as {
        repoRoot: string;
        oracleLevel?: OracleLevel;
        onlyRoutes?: string[];
      };
      return ateGenerate({ repoRoot, oracleLevel, onlyRoutes });
    },
    "mandu.ate.run": async (args: Record<string, unknown>) => {
      const {
        repoRoot,
        baseURL,
        ci,
        headless,
        browsers,
        onlyFiles,
        onlyRoutes,
        grep,
        timeoutMs,
        progressToken,
      } = args as {
        repoRoot: string;
        baseURL?: string;
        ci?: boolean;
        headless?: boolean;
        browsers?: ("chromium" | "firefox" | "webkit")[];
        onlyFiles?: string[];
        onlyRoutes?: string[];
        grep?: string;
        timeoutMs?: number;
        progressToken?: string | number;
      };
      return await runWithObservability(
        {
          repoRoot,
          baseURL,
          ci,
          headless,
          browsers,
          onlyFiles,
          onlyRoutes,
          grep,
          timeoutMs,
        },
        { progressToken },
      );
    },
    "mandu.ate.report": async (args: Record<string, unknown>) => {
      const { repoRoot, runId, startedAt, finishedAt, exitCode, oracleLevel, format, impact } = args as {
        repoRoot: string;
        runId: string;
        startedAt: string;
        finishedAt: string;
        exitCode: number;
        oracleLevel?: OracleLevel;
        format?: "json" | "html" | "both";
        impact?: { changedFiles: string[]; selectedRoutes: string[]; mode: "full" | "subset" };
      };
      return await ateReport({
        repoRoot,
        runId,
        startedAt,
        finishedAt,
        exitCode,
        oracleLevel: oracleLevel ?? "L1",
        format: format ?? "both",
        impact,
      });
    },
    "mandu.ate.heal": async (args: Record<string, unknown>) => {
      const { repoRoot, runId } = args as { repoRoot: string; runId: string };
      return ateHeal({ repoRoot, runId });
    },
    "mandu.ate.impact": async (args: Record<string, unknown>) => {
      const { repoRoot, base, head, since } = args as {
        repoRoot: string;
        base?: string;
        head?: string;
        since?: "HEAD~1" | "staged" | "working" | string;
      };

      // Phase B.3 — try the v2 impact pipeline first so callers get
      // `affected`, `suggestions`, `contractDiffs`, `graphVersion` in
      // addition to the v1 fields. Fall back to v1 on failure so the
      // tool contract stays backwards compatible.
      try {
        const { computeImpactV2 } = await import("@mandujs/ate");
        const v2 = await computeImpactV2({
          repoRoot,
          since: since ?? base,
        });
        return { ok: true, ...v2 };
      } catch {
        // Fall through to v1.
      }
      return ateImpact({ repoRoot, base, head });
    },
    "mandu.ate.auto_pipeline": async (args: Record<string, unknown>) => {
      const {
        repoRoot, baseURL, oracleLevel, ci, useImpactAnalysis,
        base, head, autoHeal, tsconfigPath, routeGlobs, buildSalt,
        timeoutMs,
      } = args as {
        repoRoot: string;
        baseURL?: string;
        oracleLevel?: OracleLevel;
        ci?: boolean;
        useImpactAnalysis?: boolean;
        base?: string;
        head?: string;
        autoHeal?: boolean;
        tsconfigPath?: string;
        routeGlobs?: string[];
        buildSalt?: string;
        timeoutMs?: number;
      };
      return await runFullPipeline({
        repoRoot, baseURL, oracleLevel, ci, useImpactAnalysis,
        base, head, autoHeal, tsconfigPath, routeGlobs, buildSalt, timeoutMs,
      });
    },
    "mandu.ate.feedback": async (args: Record<string, unknown>) => {
      const { repoRoot, runId, autoApply } = args as {
        repoRoot: string;
        runId: string;
        autoApply?: boolean;
      };
      const result = analyzeFeedback({ repoRoot, runId, autoApply });
      return {
        ok: true,
        category: result.category,
        autoApplicable: result.autoApplicable,
        priority: result.priority,
        reasoning: result.reasoning,
        suggestions: result.suggestions,
      };
    },
    "mandu.ate.apply_heal": async (args: Record<string, unknown>) => {
      const { repoRoot, runId, healIndex, createBackup } = args as {
        repoRoot: string;
        runId: string;
        healIndex: number;
        createBackup?: boolean;
      };
      const result = applyHeal({ repoRoot, runId, healIndex, createBackup });
      return {
        ok: result.success,
        ...result,
      };
    },
  };
}

// ─── Phase 5: AI Agent Integration Tools ───

export const atePhase5ToolDefinitions: Tool[] = [
  {
    name: "mandu.test.smart",
    annotations: { readOnlyHint: true },
    description:
      "Intelligently select which routes to test based on git diff analysis. " +
      "Prioritizes contract changes (HIGH), guard violations (HIGH), API routes (MEDIUM), and shared code (LOW). " +
      "Returns a sorted list of routes with reasoning for each selection.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
        changedFiles: {
          type: "array",
          items: { type: "string" },
          description: "List of changed file paths. If not provided, uses git diff HEAD.",
        },
        maxRoutes: { type: "number", description: "Maximum routes to select (default: 10)" },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.test.coverage",
    annotations: { readOnlyHint: true },
    description:
      "Detect missing test coverage in the interaction graph. " +
      "Finds route transitions, API calls, form actions, and island interactions that have no corresponding test. " +
      "Returns gaps with suggested test scenarios and overall coverage percentage.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
      },
      required: ["repoRoot"],
    },
  },
  {
    name: "mandu.test.precommit",
    annotations: { readOnlyHint: true },
    description:
      "Pre-commit check: analyze staged files to determine if tests should be run before committing. " +
      "Returns whether testing is recommended and which routes are affected.",
    inputSchema: {
      type: "object",
      properties: {
        repoRoot: { type: "string", description: "Absolute path to the Mandu project root" },
      },
      required: ["repoRoot"],
    },
  },
];

export function createAtePhase5Handlers() {
  return {
    "mandu.test.smart": async (args: Record<string, unknown>) => {
      const { repoRoot, changedFiles, maxRoutes } = args as {
        repoRoot: string;
        changedFiles?: string[];
        maxRoutes?: number;
      };
      return smartSelectRoutes({ repoRoot, changedFiles, maxRoutes });
    },
    "mandu.test.coverage": async (args: Record<string, unknown>) => {
      const { repoRoot } = args as { repoRoot: string };
      return detectCoverageGaps(repoRoot);
    },
    "mandu.test.precommit": async (args: Record<string, unknown>) => {
      const { repoRoot } = args as { repoRoot: string };
      return precommitCheck(repoRoot);
    },
  };
}