import { Badge, Button, Group, Paper, Stack, Text } from "@mantine/core";
import type {
  CreateMessageRequestParams,
  CreateMessageResult,
} from "@modelcontextprotocol/client";
import type { PendingRequestOrigin } from "@inspector/core/mcp/types.js";
import { MrtrOriginNote } from "../../elements/MrtrOriginNote/MrtrOriginNote";

export interface InlineSamplingRequestProps {
  request: CreateMessageRequestParams;
  queuePosition: string;
  draftResult?: CreateMessageResult;
  /**
   * How the request reached the Inspector — a legacy server→client request or a
   * modern MRTR `input_required` round. Drives the era-accurate note; defaults
   * to `"server-request"` (legacy, note hidden). (#1704)
   */
  origin?: PendingRequestOrigin;
  onAutoRespond: () => void;
  onEditAndSend: () => void;
  onReject: () => void;
  onViewDetails: () => void;
}

const RequestContainer = Paper.withProps({
  p: "md",
  withBorder: true,
});

const QueueLabel = Text.withProps({
  size: "xs",
  c: "dimmed",
});

const PreviewText = Text.withProps({
  size: "sm",
  c: "dimmed",
  lineClamp: 2,
});

const DraftPreview = Text.withProps({
  size: "sm",
  lineClamp: 2,
});

const DetailButton = Button.withProps({
  variant: "subtle",
  size: "xs",
});

const ActionsRow = Group.withProps({
  justify: "flex-end",
  gap: "xs",
});

const CompactButton = Button.withProps({
  size: "xs",
  variant: "light",
});

const RejectButton = Button.withProps({
  size: "xs",
  variant: "light",
  color: "red",
});

function extractPreview(request: CreateMessageRequestParams): string {
  const lastMessage = request.messages[request.messages.length - 1];
  if (!lastMessage) return "";
  const content = lastMessage.content;
  if (Array.isArray(content)) {
    const textBlock = content.find((b) => b.type === "text");
    return textBlock && "text" in textBlock ? textBlock.text : "";
  }
  return content.type === "text" ? content.text : `[${content.type}]`;
}

function extractDraftPreview(draft: CreateMessageResult): string {
  switch (draft.content.type) {
    case "text":
      return draft.content.text;
    case "image":
      return "[Image content]";
    case "audio":
      return "[Audio content]";
  }
}

function extractModelHints(
  request: CreateMessageRequestParams,
): string[] | undefined {
  const hints = request.modelPreferences?.hints;
  if (!hints || hints.length === 0) return undefined;
  return hints.map((h) => h.name).filter(Boolean) as string[];
}

function formatModelHints(hints: string[]): string {
  return `Model hints: ${hints.join(", ")}`;
}

export function InlineSamplingRequest({
  request,
  queuePosition,
  draftResult,
  origin = "server-request",
  onAutoRespond,
  onEditAndSend,
  onReject,
  onViewDetails,
}: InlineSamplingRequestProps) {
  const modelHints = extractModelHints(request);
  const messagePreview = extractPreview(request);
  const draftPreview = draftResult ? extractDraftPreview(draftResult) : null;

  return (
    <RequestContainer>
      <Stack gap="sm">
        <Group justify="space-between">
          <Badge color="blue">sampling/createMessage</Badge>
          <QueueLabel>{queuePosition}</QueueLabel>
        </Group>

        <MrtrOriginNote origin={origin} />

        {modelHints && <Text size="sm">{formatModelHints(modelHints)}</Text>}

        <PreviewText>{messagePreview}</PreviewText>

        <DetailButton onClick={onViewDetails}>View Details</DetailButton>

        {draftPreview !== null && <DraftPreview>{draftPreview}</DraftPreview>}

        <ActionsRow>
          <CompactButton onClick={onAutoRespond}>Auto-respond</CompactButton>
          <CompactButton onClick={onEditAndSend}>Edit &amp; Send</CompactButton>
          <RejectButton onClick={onReject}>Reject</RejectButton>
        </ActionsRow>
      </Stack>
    </RequestContainer>
  );
}