"use client";

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

import { cn } from "@/lib/utils";

const badgeVariants = cva(
  "px-2.5 py-1 text-xs font-medium flex items-center gap-1 ring-1 ring-inset",
  {
    variants: {
      variant: {
        red: "bg-red-500 text-white ring-transparent",
        green: "bg-green-500 text-white dark:text-black ring-transparent",
        yellow: "bg-yellow-500 text-black ring-transparent",
        gray: "bg-gray-600 text-white ring-transparent",
        blue: "bg-blue-500 text-white ring-transparent",
        purple: "bg-purple-500 text-white ring-transparent",
        "red-subtle": `bg-red-100 dark:bg-red-500/20 text-red-600 dark:text-red-400
		ring-red-400 dark:ring-red-500/30`,
        "green-subtle": `bg-green-100 dark:bg-green-500/20 text-green-700 dark:text-green-400
		ring-green-400 dark:ring-green-500/30`,
        "yellow-subtle": `bg-yellow-100 dark:bg-yellow-500/20 text-yellow-700 dark:text-yellow-400
		ring-yellow-400 dark:ring-yellow-500/30`,
        "gray-subtle": `bg-gray-100 dark:bg-gray-500/20 text-gray-700 dark:text-gray-300
		ring-gray-400 dark:ring-gray-400/30`,
        "blue-subtle": `bg-blue-100 dark:bg-blue-500/20 text-blue-700 dark:text-blue-300
		ring-blue-400 dark:ring-blue-400/30`,
        "purple-subtle": `bg-purple-100 dark:bg-purple-500/20 text-purple-700 dark:text-purple-400
		ring-purple-400 dark:ring-purple-500/30`,
        custom: "",
      },
      allowIconOnly: {
        true: "max-md:px-1.5 max-md:py-1.5",
        false: null,
      },
      badgeRoundness: {
        full: "rounded-full",
        md: "rounded-md",
        custom: "",
      },
    },
    defaultVariants: {
      variant: "gray-subtle",
      allowIconOnly: false,
      badgeRoundness: "full",
    },
  }
);

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

function Badge({ className, variant, ...props }: BadgeProps) {
  return (
    <span className={cn(badgeVariants({ variant }), className)} {...props} />
  );
}

export { Badge, badgeVariants };