"use client"

import * as React from "react"
import { CheckIcon } from "@radix-ui/react-icons"
import * as RadioGroupPrimitive from "@radix-ui/react-radio-group"

import { cn } from "../../design-lib/utils"
import { Label } from "./label"

const RadioGroup = React.forwardRef<
  React.ElementRef<typeof RadioGroupPrimitive.Root>,
  React.ComponentPropsWithoutRef<typeof RadioGroupPrimitive.Root>
>(({ className, ...props }, ref) => {
  return (
    <RadioGroupPrimitive.Root
      className={cn("grid gap-2", className)}
      {...props}
      ref={ref}
    />
  )
})
RadioGroup.displayName = RadioGroupPrimitive.Root.displayName

const RadioGroupItem = React.forwardRef<
  React.ElementRef<typeof RadioGroupPrimitive.Item>,
  React.ComponentPropsWithoutRef<typeof RadioGroupPrimitive.Item>
>(({ className, ...props }, ref) => {
  return (
    <RadioGroupPrimitive.Item
      ref={ref}
      className={cn(
        "aspect-square h-4 w-4 rounded-full border text-primary shadow-medium focus:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:cursor-not-allowed disabled:opacity-disabled",
        className
      )}
      {...props}
    >
      <RadioGroupPrimitive.Indicator className="flex items-center justify-center">
        <CheckIcon className="h-3.5 w-3.5 fill-primary" />
      </RadioGroupPrimitive.Indicator>
    </RadioGroupPrimitive.Item>
  )
})
RadioGroupItem.displayName = RadioGroupPrimitive.Item.displayName

const SquareRadioGroupItem = React.forwardRef<
  React.ElementRef<typeof Label>,
  React.ComponentPropsWithoutRef<typeof Label> & {
    id: string
    value: string
  }
>(({ className, children, id, value, ...props }, ref) => {
  return (
    <Label
      ref={ref}
      className={cn(
        "relative my-2 flex h-full items-center rounded-md border bg-background px-4  py-2 transition-colors duration-200 hover:bg-accent hover:text-accent-foreground  disabled:opacity-disabled    [&:has([data-state=checked])]:border-success-light [&:has([data-state=checked])]:bg-success-lighter [&:has([data-state=checked])]:shadow-medium"
      )}
      {...props}
    >
      <div className={cn("flex grow", className)}>{children}</div>
      <div className="flex h-full items-center px-2">
        <RadioGroupItem value={value} id={id} />
      </div>
    </Label>
  )
})
SquareRadioGroupItem.displayName = "SquareRadioGroupItem"

export { RadioGroup, RadioGroupItem, SquareRadioGroupItem }