import { Button as ButtonPrimitive } from "@base-ui/react/button"
import { cva, type VariantProps } from "class-variance-authority"
import { cn } from "./utils.ts"

const buttonVariants = cva("pmx-button pmx-ui-button", {
  variants: {
    variant: {
      default: "pmx-ui-button-default",
      outline: "pmx-ui-button-outline",
      ghost: "pmx-ui-button-ghost",
      destructive: "pmx-ui-button-destructive",
    },
    size: {
      default: "pmx-ui-button-size-default",
      sm: "pmx-ui-button-size-sm",
      icon: "pmx-ui-button-size-icon",
      "icon-sm": "pmx-ui-button-size-icon-sm",
    },
  },
  defaultVariants: { variant: "default", size: "default" },
})

export function Button({
  className,
  variant = "default",
  size = "default",
  ...props
}: ButtonPrimitive.Props & VariantProps<typeof buttonVariants>) {
  return (
    <ButtonPrimitive
      data-slot="button"
      data-variant={
        {
          default: "primary",
          outline: "outline",
          ghost: "ghost",
          destructive: "danger",
        }[variant ?? "default"]
      }
      data-shape={size === "icon" || size === "icon-sm" ? "icon" : undefined}
      data-size={size === "sm" || size === "icon-sm" ? "sm" : undefined}
      className={cn(buttonVariants({ variant, size, className }))}
      {...props}
    />
  )
}

export { buttonVariants }