import type { McpDeploymentState } from "@archestra/shared";
import { z } from "zod";

export type K8sRuntimeStatus =
  | "not_initialized"
  | "initializing"
  | "running"
  | "error"
  | "stopped";

export interface K8sDeploymentStatusSummary {
  state: McpDeploymentState;
  message: string;
  error: string | null;
  serverName: string;
  deploymentName: string | null;
  namespace: string;
  restartCount?: number;
  podAge?: string;
  podName?: string;
}

export interface K8sRuntimeStatusSummary {
  status: K8sRuntimeStatus;
  mcpServers: Record<string, K8sDeploymentStatusSummary>;
}

const AvailableToolAnalysisSchema = z.object({
  status: z.enum(["completed", "awaiting_ollama_model", "error"]),
  error: z.string().nullable(),
  is_read: z.boolean().nullable(),
  is_write: z.boolean().nullable(),
});

export const AvailableToolSchema = z.object({
  id: z.string(),
  name: z.string(),
  description: z.string(),
  inputSchema: z.any().optional(),
  mcpServerId: z.string(),
  mcpServerName: z.string(),
  analysis: AvailableToolAnalysisSchema,
});

export type AvailableTool = z.infer<typeof AvailableToolSchema>;

export const McpServerContainerLogsSchema = z.object({
  logs: z.string(),
  containerName: z.string(),
  command: z.string(),
  namespace: z.string(),
});

export type McpServerContainerLogs = z.infer<
  typeof McpServerContainerLogsSchema
>;
