import { Badge, type BadgeProps } from "@/components/ui/badge";

const VARIANT: Record<string, BadgeProps["variant"]> = {
  queued: "default",
  running: "info",
  completed: "success",
  failed: "danger",
  cancelled: "warning",
};

/** Status badge for crawl jobs (queued / running / completed / failed / cancelled). */
export function CrawlStatusBadge({ status, className }: { status: string; className?: string }) {
  return (
    <Badge variant={VARIANT[status] ?? "default"} dot className={className}>
      {status}
    </Badge>
  );
}

const PAGE_VARIANT: Record<string, BadgeProps["variant"]> = {
  success: "success",
  ok: "success",
  blocked: "warning",
  failed: "danger",
  error: "danger",
  timeout: "danger",
  skipped: "outline",
  pending: "default",
  queued: "default",
};

/** Status badge for individual crawled pages. */
export function CrawlPageStatusBadge({ status }: { status: string }) {
  return (
    <Badge variant={PAGE_VARIANT[status] ?? "default"} dot>
      {status}
    </Badge>
  );
}