import { z } from "zod";
import { type ToolMetadata, type InferSchema } from "xmcp";
import { ProFeature } from "../lib/localstack/license-checker";
import { CloudPodsApiClient } from "../lib/localstack/localstack.client";
import { ResponseBuilder } from "../core/response-builder";
import {
  runPreflights,
  requireAuthToken,
  requireLocalStackRunning,
  requireLocalStackCli,
  requireProFeature,
} from "../core/preflight";
import { withToolAnalytics } from "../core/analytics";

// Define the schema for tool parameters
export const schema = {
  action: z
    .enum(["save", "load", "delete"])
    .describe(
      "The Cloud Pods action to perform."
    ),

  pod_name: z
    .string()
    .refine((v) => v.trim().length > 0, {
      message: "pod_name must not be empty or whitespace",
    })
    .refine((v) => /^[A-Za-z0-9._-]{1,128}$/.test(v), {
      message:
        "pod_name may only contain letters, numbers, '.', '_' or '-' and be at most 128 characters",
    })
    .optional()
    .describe(
      "The name of the Cloud Pod. This is required for 'save', 'load', and 'delete' actions."
    ),
};

// Define tool metadata
export const metadata: ToolMetadata = {
  name: "localstack-cloud-pods",
  description: "Manages remote LocalStack Cloud Pods with following actions: save, load, delete",
  annotations: {
    title: "LocalStack Cloud Pods",
    readOnlyHint: false,
    destructiveHint: true,
    idempotentHint: false,
  },
};

export default async function localstackCloudPods({
  action,
  pod_name,
}: InferSchema<typeof schema>) {
  return withToolAnalytics("localstack-cloud-pods", { action, pod_name }, async () => {
    const preflightError = await runPreflights([
      requireAuthToken(),
      requireLocalStackRunning(),
      requireLocalStackCli(),
      requireProFeature(ProFeature.CLOUD_PODS),
    ]);
    if (preflightError) return preflightError;

    const client = new CloudPodsApiClient();

    switch (action) {
      case "save": {
        if (!pod_name || pod_name.trim() === "") {
          return ResponseBuilder.error(
            "Missing Required Parameter",
            "The `save` action requires the `pod_name` parameter to be specified."
          );
        }

        const result = await client.savePod(pod_name);
        if (!result.success) {
          if (result.statusCode === 409) {
            return ResponseBuilder.error(
              "Cloud Pods Error",
              `A Cloud Pod named '**${pod_name}**' already exists. Please choose a different name or delete the existing pod first.`
            );
          }
          return ResponseBuilder.error("Cloud Pods Error", result.message);
        }

        return ResponseBuilder.success(`Cloud Pod '**${pod_name}**' was saved successfully.`);
      }

      case "load": {
        if (!pod_name || pod_name.trim() === "") {
          return ResponseBuilder.error(
            "Missing Required Parameter",
            "The `load` action requires the `pod_name` parameter to be specified."
          );
        }

        const result = await client.loadPod(pod_name);
        if (!result.success) {
          if (result.statusCode === 404) {
            return ResponseBuilder.error(
              "Cloud Pods Error",
              `A Cloud Pod named '**${pod_name}**' could not be found.`
            );
          }
          return ResponseBuilder.error("Cloud Pods Error", result.message);
        }

        return ResponseBuilder.success(
          `Cloud Pod '**${pod_name}**' was loaded. Your LocalStack instance has been restored to this snapshot.`
        );
      }

      case "delete": {
        if (!pod_name || pod_name.trim() === "") {
          return ResponseBuilder.error(
            "Missing Required Parameter",
            "The `delete` action requires the `pod_name` parameter to be specified."
          );
        }

        const result = await client.deletePod(pod_name);
        if (!result.success) {
          if (result.statusCode === 404) {
            return ResponseBuilder.error(
              "Cloud Pods Error",
              `A Cloud Pod named '**${pod_name}**' could not be found.`
            );
          }
          return ResponseBuilder.error("Cloud Pods Error", result.message);
        }

        return ResponseBuilder.success(`Cloud Pod '**${pod_name}**' has been permanently deleted.`);
      }

      default:
        return ResponseBuilder.error("Unknown action", `Unsupported action: ${action}`);
    }
  });
}