/**
 * Schematic field-placement & layout-check tools.
 *
 * Move Reference/Value field labels, audit a schematic for layout problems, and
 * auto-position fields so they don't overlap bodies, wires, or net labels.
 */
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";

export function registerSchematicLayoutTools(server: McpServer, callKicadScript: Function) {
  // Move a single Reference/Value field
  server.tool(
    "set_schematic_property_position",
    "Move a component's Reference or Value field label to an absolute (x, y) coordinate (mm), optionally rotating or hiding it. Only 'Reference' and 'Value' are supported. Use autoplace_schematic_fields to place all of them automatically.",
    {
      schematicPath: z.string().describe("Path to the .kicad_sch file"),
      reference: z.string().describe("Component reference designator (e.g., R1, U2)"),
      property: z.enum(["Reference", "Value"]).describe("Which field to move"),
      x: z.number().describe("New X position in mm (absolute schematic coordinate)"),
      y: z.number().describe("New Y position in mm (absolute schematic coordinate)"),
      angle: z.number().optional().default(0).describe("Text angle in degrees (default 0)"),
      visible: z
        .boolean()
        .optional()
        .default(true)
        .describe("Whether the field is visible (default true)"),
    },
    async (args: any) => {
      const result = await callKicadScript("set_schematic_property_position", args);
      return {
        content: [
          {
            type: "text",
            text: result.success ? result.message : `Failed: ${result.message || "Unknown error"}`,
          },
        ],
      };
    },
  );

  // Move many fields in one file read/write
  server.tool(
    "batch_set_schematic_property_positions",
    "Move many Reference/Value field labels in a single file read/write — far faster than repeated set_schematic_property_position calls. Pass an 'updates' array; each item is {reference, property:'Reference'|'Value', x, y, angle?, visible?}. Returns per-item applied/failed lists.",
    {
      schematicPath: z.string().describe("Path to the .kicad_sch file"),
      updates: z
        .array(
          z.object({
            reference: z.string(),
            property: z.enum(["Reference", "Value"]),
            x: z.number(),
            y: z.number(),
            angle: z.number().optional().default(0),
            visible: z.boolean().optional().default(true),
          }),
        )
        .describe("List of field moves to apply"),
    },
    async (args: any) => {
      const result = await callKicadScript("batch_set_schematic_property_positions", args);
      if (result.success === false && result.message) {
        return { content: [{ type: "text", text: `Failed: ${result.message}` }] };
      }
      const lines = [
        `Applied ${result.applied_count} field move(s), ${result.failed_count} failed.`,
      ];
      for (const f of result.failed || []) {
        lines.push(`  ✗ ${f.reference}.${f.property}: ${f.reason}`);
      }
      return { content: [{ type: "text", text: lines.join("\n") }] };
    },
  );

  // Auto-position all Ref/Value fields
  server.tool(
    "autoplace_schematic_fields",
    "Automatically reposition every component's Reference and Value field so they sit outside the component body AND outside any net labels attached to its pins, avoiding collisions with other components and already-placed fields. Like KiCAD's built-in field auto-placement but net-label aware. Optionally limit to specific references.",
    {
      schematicPath: z.string().describe("Path to the .kicad_sch file"),
      references: z
        .array(z.string())
        .optional()
        .describe("Only reposition these references (default: all components)"),
      clearance: z
        .number()
        .optional()
        .describe(
          "Gap in mm between the body/label extent and field text (default one 1.27mm grid unit)",
        ),
    },
    async (args: any) => {
      const result = await callKicadScript("autoplace_schematic_fields", args);
      return {
        content: [
          {
            type: "text",
            text: result.success ? result.message : `Failed: ${result.message || "Unknown error"}`,
          },
        ],
      };
    },
  );
}