import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import type { TorrentClawClient } from "../api-client.js";
import { ApiError } from "../api-client.js";

export function registerScanRequest(
  server: McpServer,
  client: TorrentClawClient,
): void {
  server.tool(
    "submit_scan_request",
    "Submit a torrent for audio/video quality analysis (codec, tracks, resolution, HDR). Use when the user wants to know the exact media specs of a torrent before downloading. Results are not instant — use get_scan_status to check progress. Rate limited to 5 requests per hour.",
    {
      info_hash: z
        .string()
        .regex(/^[a-fA-F0-9]{40}$/)
        .describe("40-character hex torrent info_hash to scan"),
      email: z
        .string()
        .email()
        .max(200)
        .describe("Email address for scan completion notification"),
    },
    async (params) => {
      try {
        const data = await client.submitScanRequest(
          params.info_hash.toLowerCase(),
          params.email,
        );
        return {
          content: [
            {
              type: "text",
              text: `Scan request submitted for ${params.info_hash.toLowerCase()}.\nStatus: ${data.status}\nUse get_scan_status(info_hash="${params.info_hash.toLowerCase()}") to check progress.`,
            },
          ],
        };
      } catch (error) {
        const message =
          error instanceof ApiError
            ? `TorrentClaw API error (${error.status}): ${error.message}`
            : `Request failed: ${error instanceof Error ? error.message : "Unknown error"}`;
        return { content: [{ type: "text", text: message }], isError: true };
      }
    },
  );

  server.tool(
    "get_scan_status",
    "Check the status of a torrent audio/video scan request. Returns the current scan status (pending, scanning, completed, failed). Use after submit_scan_request.",
    {
      info_hash: z
        .string()
        .regex(/^[a-fA-F0-9]{40}$/)
        .describe("40-character hex torrent info_hash to check"),
    },
    async (params) => {
      try {
        const data = await client.getScanStatus(params.info_hash.toLowerCase());
        const lines = [`Scan status for ${params.info_hash.toLowerCase()}:`];
        lines.push(`  Status: ${data.status}`);
        if (data.source) lines.push(`  Source: ${data.source}`);
        if (data.createdAt) lines.push(`  Submitted: ${data.createdAt}`);
        if (data.completedAt) lines.push(`  Completed: ${data.completedAt}`);
        return {
          content: [{ type: "text", text: lines.join("\n") }],
        };
      } catch (error) {
        const message =
          error instanceof ApiError
            ? `TorrentClaw API error (${error.status}): ${error.message}`
            : `Request failed: ${error instanceof Error ? error.message : "Unknown error"}`;
        return { content: [{ type: "text", text: message }], isError: true };
      }
    },
  );
}