"use client";

import type { SourceKind } from "@dci/core";
import { Archive, ExternalLink } from "lucide-react";
import { fmtDate } from "@/lib/format";
import { SOURCE_KIND_LABEL } from "@/lib/labels";
import { Badge } from "./badge";
import { cn } from "./cn";
import { Tooltip } from "./popover";

/**
 * Provenance / source-kind badge (OPERATOR · GOVERNMENT · UTILITY · REGULATORY · FILING · DATASET · NEWS · COMMUNITY…).
 * Primary kinds (operator, government, utility, filing, cloud provider, registry) get a filled dot; secondary kinds
 * (news, secondary, dataset, community) a hollow one. Hover / focus shows the source, published & retrieved dates and
 * whether an archived copy exists (documentId).
 */
export interface SourceBadgeProps {
  kind: SourceKind;
  name?: string | null;
  url?: string | null;
  publishedAt?: string | null;
  retrievedAt?: string | null;
  /** an archived document id means the page was captured and can be reprocessed */
  documentId?: string | null;
  size?: "xs" | "sm";
  /** show the source name next to the kind */
  withName?: boolean;
  className?: string;
}

const PRIMARY: ReadonlySet<SourceKind> = new Set<SourceKind>(["operator", "government", "utility", "filing", "cloud_provider", "registry"]);

/** Short upper-case kind label used in dense tables ("REGULATORY" for government filings when the source says so). */
export const SOURCE_KIND_CAPS: Record<SourceKind, string> = {
  operator: "OPERATOR",
  government: "GOVERNMENT",
  filing: "FILING",
  utility: "UTILITY",
  dataset: "DATASET",
  community: "COMMUNITY",
  secondary: "SECONDARY",
  news: "NEWS",
  cloud_provider: "CLOUD",
  registry: "REGISTRY",
};

export function SourceBadge({ kind, name, url, publishedAt, retrievedAt, documentId, size = "xs", withName, className }: SourceBadgeProps) {
  const primary = PRIMARY.has(kind);
  const inner = (
    <Badge tone="outline" size={size} className={cn("cursor-help", className)}>
      <span aria-hidden className={cn("inline-block size-[6px] rounded-full", primary ? "bg-ink-2" : "border-[1.5px] border-ink-3")} />
      <span className="tracking-[0.04em]">{SOURCE_KIND_CAPS[kind]}</span>
      {withName && name && <span className="max-w-[140px] truncate font-normal normal-case tracking-normal text-ink-3">{name}</span>}
    </Badge>
  );
  const details = (
    <div className="flex flex-col gap-0.5">
      <div className="font-medium">{name ?? SOURCE_KIND_LABEL[kind]}</div>
      <div className="opacity-80">
        {SOURCE_KIND_LABEL[kind]} · {primary ? "primary source" : "secondary source"}
      </div>
      {publishedAt && <div className="opacity-80">Published {fmtDate(publishedAt)}</div>}
      {retrievedAt && <div className="opacity-80">Retrieved {fmtDate(retrievedAt)}</div>}
      <div className="inline-flex items-center gap-1 opacity-80">
        <Archive className="size-3" aria-hidden /> {documentId ? "Archived copy on file" : "No archived copy recorded"}
      </div>
      {url && (
        <div className="inline-flex items-center gap-1 truncate opacity-80">
          <ExternalLink className="size-3" aria-hidden /> {safeHost(url)}
        </div>
      )}
    </div>
  );
  return <Tooltip content={details}>{inner}</Tooltip>;
}

function safeHost(url: string): string {
  try {
    return new URL(url).host.replace(/^www\./, "");
  } catch {
    return url;
  }
}