Badge

import * as React from "react";
import { cva, type VariantProps } from "class-variance-authority";
import { cn } from "./lib/cn";

const badgeVariants = cva(
  // same 6px radius as the rest of the library — a pill/rounded-full
  // shape is the common default elsewhere, but that breaks the "one
  // radius everywhere" rule, so it's deliberately not used here
  "inline-flex w-fit items-center gap-1 whitespace-nowrap rounded-md font-medium",
  {
    variants: {
      variant: {
        default: "bg-muted text-muted-foreground",
        primary: "bg-primary/15 text-primary",
        success: "bg-success/15 text-success",
        danger: "bg-danger/15 text-danger",
        warning: "bg-warning/15 text-warning",
        info: "bg-info/15 text-info",
      },
      size: {
        sm: "h-5 px-1.5 text-[0.6875rem]",
        md: "h-6 px-2 text-xs",
      },
    },
    defaultVariants: {
      variant: "default",
      size: "sm",
    },
  },
);

export interface BadgeProps
  extends React.HTMLAttributes<HTMLSpanElement>,
    VariantProps<typeof badgeVariants> {}

export const Badge = React.forwardRef<HTMLSpanElement, BadgeProps>(
  ({ className, variant, size, ...props }, ref) => {
    return <span ref={ref} className={cn(badgeVariants({ variant, size }), className)} {...props} />;
  },
);
Badge.displayName = "Badge";