import { Badge } from "~/ui/Badge";

type Tone = "danger" | "warning" | "neutral";

const TONES: Record<string, Tone> = {
  critical: "danger",
  high: "danger",
  error: "danger",
  moderate: "warning",
  medium: "warning",
  warning: "warning",
};

export function SeverityBadge({ severity }: { severity: string | null }) {
  if (!severity) return null;
  return <Badge tone={TONES[severity] ?? "neutral"}>{severity}</Badge>;
}