import { cn } from "@aur-ui/ui/lib/utils";
import { cva, type VariantProps } from "class-variance-authority";
import { Slot } from "radix-ui";
import type * as React from "react";

/**
 * Badge — Aur status pill.
 *
 * Two orthogonal axes mirror the Button hierarchy: `variant` picks the
 * semantic colour, `tone` picks soft (tinted wash, the default) or solid
 * (filled, white text). Defaults preserve the original soft rendering so
 * existing call sites are untouched.
 */
const badgeVariants = cva(
  "inline-flex shrink-0 items-center rounded-full font-medium whitespace-nowrap [&_svg]:pointer-events-none",
  {
    variants: {
      variant: {
        brand: "",
        neutral: "",
        positive: "",
        warning: "",
        danger: "",
      },
      tone: {
        soft: "",
        solid: "text-white",
      },
      size: {
        default:
          "gap-1 px-2 py-0.5 text-xs [&_svg:not([class*='size-'])]:size-3",
        sm: "gap-1 px-1.5 py-px text-2xs [&_svg:not([class*='size-'])]:size-2.5",
      },
    },
    compoundVariants: [
      { variant: "brand", tone: "soft", class: "bg-brand-soft text-brand" },
      { variant: "neutral", tone: "soft", class: "bg-ink-wash text-ink-500" },
      {
        variant: "positive",
        tone: "soft",
        class: "bg-positive-soft text-positive",
      },
      {
        variant: "warning",
        tone: "soft",
        class: "bg-warning-soft text-warning",
      },
      { variant: "danger", tone: "soft", class: "bg-danger-soft text-danger" },
      { variant: "brand", tone: "solid", class: "bg-brand" },
      // ink-700 flips light in dark mode, so the label uses text-surface
      // (dark-on-light there, white-on-dark in light mode) instead of white.
      {
        variant: "neutral",
        tone: "solid",
        class: "bg-ink-700 text-surface",
      },
      { variant: "positive", tone: "solid", class: "bg-positive" },
      { variant: "warning", tone: "solid", class: "bg-warning" },
      { variant: "danger", tone: "solid", class: "bg-danger" },
    ],
    defaultVariants: {
      variant: "brand",
      tone: "soft",
      size: "default",
    },
  },
);

function Badge({
  className,
  variant = "brand",
  tone = "soft",
  size = "default",
  dot = false,
  asChild = false,
  children,
  ...props
}: React.ComponentProps<"span"> &
  VariantProps<typeof badgeVariants> & {
    /** Render a leading status dot tinted to match the badge text color. */
    dot?: boolean;
    asChild?: boolean;
  }) {
  const Comp = asChild ? Slot.Root : "span";

  return (
    <Comp
      data-slot="badge"
      data-variant={variant}
      data-tone={tone}
      data-size={size}
      className={cn(badgeVariants({ variant, tone, size }), className)}
      {...props}
    >
      {dot ? (
        <span
          aria-hidden
          data-slot="badge-dot"
          className="size-1.5 shrink-0 rounded-full bg-current"
        />
      ) : null}
      {children}
    </Comp>
  );
}

const badgeCountVariants = cva(
  "inline-flex h-4 min-w-4 shrink-0 items-center justify-center rounded-full px-1 text-2xs font-medium text-white tabular-nums leading-none",
  {
    variants: {
      variant: {
        brand: "bg-brand",
        neutral: "bg-ink-700 text-surface",
        positive: "bg-positive",
        warning: "bg-warning",
        danger: "bg-danger",
      },
    },
    defaultVariants: {
      variant: "brand",
    },
  },
);

export type BadgeCountProps = React.ComponentProps<"span"> &
  VariantProps<typeof badgeCountVariants> & {
    count: number;
    /** Values above this render as "{max}+". */
    max?: number;
    /** Render the badge even when count is 0 (hidden by default). */
    showZero?: boolean;
  };

/**
 * Numeric counter badge — round for a single digit, stretches naturally for
 * more. Solid-toned; hidden at zero unless `showZero`.
 */
function BadgeCount({
  className,
  variant = "brand",
  count,
  max = 99,
  showZero = false,
  ...props
}: BadgeCountProps) {
  if (count === 0 && !showZero) return null;

  return (
    <span
      data-slot="badge-count"
      data-variant={variant}
      className={cn(badgeCountVariants({ variant }), className)}
      {...props}
    >
      {count > max ? `${max}+` : count}
    </span>
  );
}

export { Badge, BadgeCount, badgeCountVariants, badgeVariants };