2025-08-20 04:12:49 -06:00

116 lines
3.5 KiB
TypeScript

'use client';
import * as React from 'react';
import {
Switch as SwitchPrimitive,
type SwitchProps as SwitchPrimitiveProps,
} from '@headlessui/react';
import { motion, type HTMLMotionProps } from 'motion/react';
import { cn } from '@workspace/ui/lib/utils';
type SwitchProps<TTag extends React.ElementType = typeof motion.button> =
SwitchPrimitiveProps<TTag> &
Omit<HTMLMotionProps<'button'>, 'children'> & {
leftIcon?: React.ReactNode;
rightIcon?: React.ReactNode;
thumbIcon?: React.ReactNode;
onCheckedChange?: (checked: boolean) => void;
as?: TTag;
};
function Switch({
className,
leftIcon,
rightIcon,
thumbIcon,
onChange,
as = motion.button,
...props
}: SwitchProps) {
const [isChecked, setIsChecked] = React.useState(
props.checked ?? props.defaultChecked ?? false,
);
const [isTapped, setIsTapped] = React.useState(false);
React.useEffect(() => {
setIsChecked(props.checked ?? props.defaultChecked ?? false);
}, [props.checked, props.defaultChecked]);
const handleChange = React.useCallback(
(checked: boolean) => {
setIsChecked(checked);
onChange?.(checked);
},
[onChange],
);
return (
<SwitchPrimitive
data-slot="switch"
checked={isChecked}
onChange={handleChange}
className={cn(
'relative flex p-[3px] h-6 w-10 shrink-0 cursor-pointer items-center rounded-full transition-colors focus:outline-none focus-visible:ring-2 focus-visible:ring-ring focus-visible:ring-offset-2 focus-visible:ring-offset-background disabled:cursor-not-allowed disabled:opacity-50 data-[checked]:bg-primary bg-input data-[checked]:justify-end justify-start',
className,
)}
as={as}
whileTap="tap"
initial={false}
onTapStart={() => setIsTapped(true)}
onTapCancel={() => setIsTapped(false)}
onTap={() => setIsTapped(false)}
{...props}
>
{leftIcon && (
<motion.div
data-slot="switch-left-icon"
animate={
isChecked ? { scale: 1, opacity: 1 } : { scale: 0, opacity: 0 }
}
transition={{ type: 'spring', bounce: 0 }}
className="absolute [&_svg]:size-3 left-1 top-1/2 -translate-y-1/2 dark:text-neutral-500 text-neutral-400"
>
{typeof leftIcon !== 'string' ? leftIcon : null}
</motion.div>
)}
{rightIcon && (
<motion.div
data-slot="switch-right-icon"
animate={
isChecked ? { scale: 0, opacity: 0 } : { scale: 1, opacity: 1 }
}
transition={{ type: 'spring', bounce: 0 }}
className="absolute [&_svg]:size-3 right-1 top-1/2 -translate-y-1/2 dark:text-neutral-400 text-neutral-500"
>
{typeof rightIcon !== 'string' ? rightIcon : null}
</motion.div>
)}
<motion.span
data-slot="switch-thumb"
whileTap="tab"
className={cn(
'relative z-[1] [&_svg]:size-3 flex items-center justify-center rounded-full bg-background shadow-lg ring-0 dark:text-neutral-400 text-neutral-500',
)}
layout
transition={{ type: 'spring', stiffness: 300, damping: 25 }}
style={{
width: 18,
height: 18,
}}
animate={
isTapped
? { width: 21, transition: { duration: 0.1 } }
: { width: 18, transition: { duration: 0.1 } }
}
>
{thumbIcon && typeof thumbIcon !== 'string' ? thumbIcon : null}
</motion.span>
</SwitchPrimitive>
);
}
export { Switch, type SwitchProps };