"use client"

import * as React from "react"
import * as TogglePrimitive from "@radix-ui/react-toggle"
import styles from "./Toggle.module.css"
import buttonStyles from "./Button.module.css"

export const Toggle = React.forwardRef<
  React.ElementRef<typeof TogglePrimitive.Root>,
  React.ComponentPropsWithoutRef<typeof TogglePrimitive.Root> &
    {
      variant?: "default" | "outline";
      size?: "md" | "sm" | "lg";
    }
>(({ className, variant = "default", size = "md", ...props }, ref) => (
  <TogglePrimitive.Root
    ref={ref}
    className={`${buttonStyles.button} ${styles.toggle} ${styles["toggle-variant-" + variant]} ${styles["toggle-size-" + size]} ${className ?? ""}`}
    {...props}
  />
))

Toggle.displayName = TogglePrimitive.Root.displayName