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

468 lines
14 KiB
TypeScript

'use client';
import * as React from 'react';
import { DropdownMenu as DropdownMenuPrimitive } from 'radix-ui';
import { Check, ChevronRight, Circle } from 'lucide-react';
import {
AnimatePresence,
motion,
type HTMLMotionProps,
type Transition,
} from 'motion/react';
import { cn } from '@workspace/ui/lib/utils';
import {
MotionHighlight,
MotionHighlightItem,
} from '@/registry/effects/motion-highlight';
type DropdownMenuContextType = {
isOpen: boolean;
highlightTransition: Transition;
animateOnHover: boolean;
};
const DropdownMenuContext = React.createContext<
DropdownMenuContextType | undefined
>(undefined);
const useDropdownMenu = (): DropdownMenuContextType => {
const context = React.useContext(DropdownMenuContext);
if (!context) {
throw new Error('useDropdownMenu must be used within a DropdownMenu');
}
return context;
};
type DropdownMenuProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Root
> & {
transition?: Transition;
animateOnHover?: boolean;
};
function DropdownMenu({
children,
transition = { type: 'spring', stiffness: 350, damping: 35 },
animateOnHover = true,
...props
}: DropdownMenuProps) {
const [isOpen, setIsOpen] = React.useState(
props?.open ?? props?.defaultOpen ?? false,
);
React.useEffect(() => {
if (props?.open !== undefined) setIsOpen(props.open);
}, [props?.open]);
const handleOpenChange = React.useCallback(
(open: boolean) => {
setIsOpen(open);
props.onOpenChange?.(open);
},
[props],
);
return (
<DropdownMenuContext.Provider
value={{ isOpen, highlightTransition: transition, animateOnHover }}
>
<DropdownMenuPrimitive.Root
data-slot="dropdown-menu"
{...props}
onOpenChange={handleOpenChange}
>
{children}
</DropdownMenuPrimitive.Root>
</DropdownMenuContext.Provider>
);
}
type DropdownMenuTriggerProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Trigger
>;
function DropdownMenuTrigger(props: DropdownMenuTriggerProps) {
return (
<DropdownMenuPrimitive.Trigger
data-slot="dropdown-menu-trigger"
{...props}
/>
);
}
type DropdownMenuGroupProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Group
>;
function DropdownMenuGroup(props: DropdownMenuGroupProps) {
return (
<DropdownMenuPrimitive.Group data-slot="dropdown-menu-group" {...props} />
);
}
type DropdownMenuPortalProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Portal
>;
function DropdownMenuPortal(props: DropdownMenuPortalProps) {
return (
<DropdownMenuPrimitive.Portal data-slot="dropdown-menu-portal" {...props} />
);
}
type DropdownMenuSubProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Sub
>;
function DropdownMenuSub(props: DropdownMenuSubProps) {
return <DropdownMenuPrimitive.Sub data-slot="dropdown-menu-sub" {...props} />;
}
type DropdownMenuRadioGroupProps = React.ComponentProps<
typeof DropdownMenuPrimitive.RadioGroup
>;
function DropdownMenuRadioGroup(props: DropdownMenuRadioGroupProps) {
return (
<DropdownMenuPrimitive.RadioGroup
data-slot="dropdown-menu-radio-group"
{...props}
/>
);
}
type DropdownMenuSubTriggerProps = React.ComponentProps<
typeof DropdownMenuPrimitive.SubTrigger
> & {
inset?: boolean;
};
function DropdownMenuSubTrigger({
className,
children,
inset,
disabled,
...props
}: DropdownMenuSubTriggerProps) {
return (
<MotionHighlightItem disabled={disabled}>
<DropdownMenuPrimitive.SubTrigger {...props} disabled={disabled} asChild>
<motion.div
data-slot="dropdown-menu-sub-trigger"
data-inset={inset}
data-disabled={disabled}
whileTap={{ scale: 0.95 }}
className={cn(
"[&:not([data-highlight])]:focus:bg-accent focus:text-accent-foreground [&:not([data-highlight])]:data-[state=open]:bg-accent data-[state=open]:text-accent-foreground data-[state=open]:[&_[data-chevron]]:rotate-90 [&_[data-chevron]]:transition-transform [&_[data-chevron]]:duration-150 [&_[data-chevron]]:ease-in-out [&_svg:not([class*='text-'])]:text-muted-foreground relative z-[1] flex cursor-default select-none items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0",
inset && 'pl-8',
className,
)}
>
{children}
<ChevronRight data-chevron className="ml-auto" />
</motion.div>
</DropdownMenuPrimitive.SubTrigger>
</MotionHighlightItem>
);
}
type DropdownMenuSubContentProps = React.ComponentProps<
typeof DropdownMenuPrimitive.SubContent
>;
function DropdownMenuSubContent({
className,
...props
}: DropdownMenuSubContentProps) {
return (
<DropdownMenuPrimitive.SubContent
data-slot="dropdown-menu-sub-content"
className={cn(
'z-50 min-w-[8rem] overflow-hidden rounded-md border bg-popover p-1 text-popover-foreground shadow-lg data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 origin-[--radix-dropdown-menu-content-transform-origin]',
className,
)}
{...props}
/>
);
}
type DropdownMenuContentProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Content
> &
HTMLMotionProps<'div'> & {
transition?: Transition;
};
function DropdownMenuContent({
className,
children,
sideOffset = 4,
transition = { duration: 0.2 },
...props
}: DropdownMenuContentProps) {
const { isOpen, highlightTransition, animateOnHover } = useDropdownMenu();
return (
<AnimatePresence>
{isOpen && (
<DropdownMenuPrimitive.Portal
forceMount
data-slot="dropdown-menu-portal"
>
<DropdownMenuPrimitive.Content
sideOffset={sideOffset}
asChild
{...props}
>
<motion.div
key="dropdown-menu-content"
data-slot="dropdown-menu-content"
className={cn(
'z-50 max-h-[var(--radix-dropdown-menu-content-available-height)] min-w-[8rem] overflow-y-auto overflow-x-hidden rounded-md border bg-popover p-1 text-popover-foreground shadow-md data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2 origin-[--radix-dropdown-menu-content-transform-origin]',
className,
)}
initial={{
opacity: 0,
scale: 0.95,
}}
animate={{
opacity: 1,
scale: 1,
}}
exit={{
opacity: 0,
scale: 0.95,
}}
transition={transition}
style={{ willChange: 'opacity, transform' }}
{...props}
>
<MotionHighlight
hover
className="rounded-sm"
controlledItems
transition={highlightTransition}
enabled={animateOnHover}
>
{children}
</MotionHighlight>
</motion.div>
</DropdownMenuPrimitive.Content>
</DropdownMenuPrimitive.Portal>
)}
</AnimatePresence>
);
}
type DropdownMenuItemProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Item
> & {
inset?: boolean;
variant?: 'default' | 'destructive';
};
function DropdownMenuItem({
className,
children,
inset,
disabled,
variant = 'default',
...props
}: DropdownMenuItemProps) {
return (
<MotionHighlightItem
activeClassName={
variant === 'default'
? 'bg-accent'
: 'bg-destructive/10 dark:bg-destructive/20'
}
disabled={disabled}
>
<DropdownMenuPrimitive.Item {...props} disabled={disabled} asChild>
<motion.div
data-slot="dropdown-menu-item"
data-inset={inset}
data-variant={variant}
data-disabled={disabled}
whileTap={{ scale: 0.95 }}
className={cn(
"[&:not([data-highlight])]:focus:bg-accent focus:text-accent-foreground data-[variant=destructive]:text-destructive [&:not([data-highlight])]:data-[variant=destructive]:focus:bg-destructive/10 dark:[&:not([data-highlight])]:data-[variant=destructive]:focus:bg-destructive/20 data-[variant=destructive]:focus:text-destructive data-[variant=destructive]:*:[svg]:!text-destructive [&_svg:not([class*='text-'])]:text-muted-foreground relative z-[1] flex cursor-default select-none items-center gap-2 rounded-sm px-2 py-1.5 text-sm outline-none transition-colors focus-visible:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50 [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0",
inset && 'pl-8',
className,
)}
>
{children}
</motion.div>
</DropdownMenuPrimitive.Item>
</MotionHighlightItem>
);
}
type DropdownMenuCheckboxItemProps = React.ComponentProps<
typeof DropdownMenuPrimitive.CheckboxItem
>;
function DropdownMenuCheckboxItem({
className,
children,
checked,
disabled,
...props
}: DropdownMenuCheckboxItemProps) {
return (
<MotionHighlightItem disabled={disabled}>
<DropdownMenuPrimitive.CheckboxItem
{...props}
checked={checked}
disabled={disabled}
asChild
>
<motion.div
data-slot="dropdown-menu-checkbox-item"
data-disabled={disabled}
whileTap={{ scale: 0.95 }}
className={cn(
"[&:not([data-highlight])]:focus:bg-accent focus:text-accent-foreground relative flex cursor-default items-center gap-2 rounded-sm py-1.5 pr-2 pl-8 text-sm outline-hidden select-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
className,
)}
>
<span className="absolute left-2 flex size-3.5 items-center justify-center">
<DropdownMenuPrimitive.ItemIndicator data-slot="dropdown-menu-checkbox-item-indicator">
<Check className="size-4" />
</DropdownMenuPrimitive.ItemIndicator>
</span>
{children}
</motion.div>
</DropdownMenuPrimitive.CheckboxItem>
</MotionHighlightItem>
);
}
type DropdownMenuRadioItemProps = React.ComponentProps<
typeof DropdownMenuPrimitive.RadioItem
>;
function DropdownMenuRadioItem({
className,
children,
disabled,
...props
}: DropdownMenuRadioItemProps) {
return (
<MotionHighlightItem disabled={disabled}>
<DropdownMenuPrimitive.RadioItem {...props} disabled={disabled} asChild>
<motion.div
data-slot="dropdown-menu-radio-item"
data-disabled={disabled}
whileTap={{ scale: 0.95 }}
className={cn(
"[&:not([data-highlight])]:focus:bg-accent focus:text-accent-foreground relative flex cursor-default items-center gap-2 rounded-sm py-1.5 pr-2 pl-8 text-sm outline-hidden select-none data-[disabled]:pointer-events-none data-[disabled]:opacity-50 [&_svg]:pointer-events-none [&_svg]:shrink-0 [&_svg:not([class*='size-'])]:size-4",
className,
)}
>
<span className="pointer-events-none absolute left-2 flex size-3.5 items-center justify-center">
<DropdownMenuPrimitive.ItemIndicator data-slot="dropdown-menu-radio-item-indicator">
<Circle className="size-2 fill-current" />
</DropdownMenuPrimitive.ItemIndicator>
</span>
{children}
</motion.div>
</DropdownMenuPrimitive.RadioItem>
</MotionHighlightItem>
);
}
type DropdownMenuLabelProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Label
> & {
inset?: boolean;
};
function DropdownMenuLabel({
className,
inset,
...props
}: DropdownMenuLabelProps) {
return (
<DropdownMenuPrimitive.Label
data-slot="dropdown-menu-label"
data-inset={inset}
className={cn(
'px-2 py-1.5 text-sm font-semibold',
inset && 'pl-8',
className,
)}
{...props}
/>
);
}
type DropdownMenuSeparatorProps = React.ComponentProps<
typeof DropdownMenuPrimitive.Separator
>;
function DropdownMenuSeparator({
className,
...props
}: DropdownMenuSeparatorProps) {
return (
<DropdownMenuPrimitive.Separator
data-slot="dropdown-menu-separator"
className={cn('-mx-1 my-1 h-px bg-border', className)}
{...props}
/>
);
}
type DropdownMenuShortcutProps = React.ComponentProps<'span'>;
function DropdownMenuShortcut({
className,
...props
}: DropdownMenuShortcutProps) {
return (
<span
data-slot="dropdown-menu-shortcut"
className={cn(
'text-muted-foreground ml-auto text-xs tracking-widest',
className,
)}
{...props}
/>
);
}
export {
DropdownMenu,
DropdownMenuTrigger,
DropdownMenuContent,
DropdownMenuItem,
DropdownMenuCheckboxItem,
DropdownMenuRadioItem,
DropdownMenuLabel,
DropdownMenuSeparator,
DropdownMenuShortcut,
DropdownMenuGroup,
DropdownMenuPortal,
DropdownMenuSub,
DropdownMenuSubContent,
DropdownMenuSubTrigger,
DropdownMenuRadioGroup,
type DropdownMenuProps,
type DropdownMenuTriggerProps,
type DropdownMenuContentProps,
type DropdownMenuItemProps,
type DropdownMenuCheckboxItemProps,
type DropdownMenuRadioItemProps,
type DropdownMenuLabelProps,
type DropdownMenuSeparatorProps,
type DropdownMenuShortcutProps,
type DropdownMenuGroupProps,
type DropdownMenuPortalProps,
type DropdownMenuSubProps,
type DropdownMenuSubContentProps,
type DropdownMenuSubTriggerProps,
type DropdownMenuRadioGroupProps,
};