import * as React from 'react';
import { Switch as SwitchPrimitive } from '@base-ui/react/switch';

import { classNames } from '../../lib/utils';

const Switch = React.forwardRef<
  HTMLSpanElement,
  SwitchPrimitive.Root.Props & {
    size?: 'sm' | 'default';
  }
>(function Switch({ className, size = 'default', ...props }, ref) {
  return (
    <SwitchPrimitive.Root
      data-slot="switch"
      data-size={size}
      className={classNames(
        'peer group/switch relative inline-flex shrink-0 items-center rounded-full border border-transparent transition-all outline-none after:absolute after:-inset-x-3 after:-inset-y-2 focus-visible:border-ring focus-visible:ring-3 focus-visible:ring-ring/70 dark:focus-visible:ring-ring/50 aria-invalid:border-destructive aria-invalid:ring-3 aria-invalid:ring-destructive/20 data-[size=default]:h-[18.4px] data-[size=default]:w-[32px] data-[size=sm]:h-[14px] data-[size=sm]:w-[24px] dark:aria-invalid:border-destructive/50 dark:aria-invalid:ring-destructive/40 data-checked:bg-primary data-unchecked:border-input data-unchecked:bg-input-fill dark:data-unchecked:border-transparent dark:data-unchecked:bg-input-fill/80 data-disabled:cursor-not-allowed data-disabled:opacity-50',
        className
      )}
      {...props}
      ref={ref}
    >
      <SwitchPrimitive.Thumb
        data-slot="switch-thumb"
        className="pointer-events-none block rounded-full border border-text-light bg-background ring-0 transition-transform dark:border-transparent group-data-[size=default]/switch:size-4 group-data-[size=sm]/switch:size-3 group-data-[size=default]/switch:data-checked:translate-x-[calc(100%-2px)] group-data-[size=sm]/switch:data-checked:translate-x-[calc(100%-2px)] dark:data-checked:bg-primary-foreground group-data-[size=default]/switch:data-unchecked:translate-x-0 group-data-[size=sm]/switch:data-unchecked:translate-x-0 dark:data-unchecked:bg-foreground"
      />
    </SwitchPrimitive.Root>
  );
});

export { Switch };