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

const badgeVariants = cva(
  "inline-flex items-center rounded-full px-2.5 py-0.5 text-xs font-semibold transition-colors",
  {
    variants: {
      variant: {
        default: "bg-primary/10 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",
        muted: "bg-muted-foreground/15 text-muted-foreground",
        accent: "bg-accent/15 text-accent",
        outline: "border border-border text-foreground",
      },
    },
    defaultVariants: { variant: "default" },
  }
);

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

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

// Helper qui mappe un statut métier à une variante de badge
export function statutBadgeVariant(statut: string): VariantProps<typeof badgeVariants>["variant"] {
  switch (statut) {
    case "DISPONIBLE":
      return "success";
    case "OCCUPEE":
      return "danger";
    case "NETTOYAGE":
      return "warning";
    case "MAINTENANCE":
      return "muted";
    case "CONFIRMEE":
      return "info";
    case "EN_ATTENTE":
      return "warning";
    case "ANNULEE":
      return "danger";
    case "TERMINEE":
      return "muted";
    case "PAYE":
      return "success";
    case "PARTIEL":
      return "warning";
    case "NON_PAYE":
      return "danger";
    default:
      return "default";
  }
}

export { Badge, badgeVariants };