"use client"; import { ChevronRight, File, Folder, FolderOpen } from "lucide-react"; import { AnimatePresence, motion } from "motion/react"; import { type ComponentProps, createContext, type HTMLAttributes, type ReactNode, useCallback, useContext, useId, useState, } from "react"; import { cn } from "@/lib/utils"; type TreeContextType = { expandedIds: Set; selectedIds: string[]; toggleExpanded: (nodeId: string) => void; handleSelection: (nodeId: string, ctrlKey: boolean) => void; showLines?: boolean; showIcons?: boolean; selectable?: boolean; multiSelect?: boolean; indent?: number; animateExpand?: boolean; }; const TreeContext = createContext(undefined); const useTree = () => { const context = useContext(TreeContext); if (!context) { throw new Error("Tree components must be used within a TreeProvider"); } return context; }; type TreeNodeContextType = { nodeId: string; level: number; isLast: boolean; parentPath: boolean[]; }; const TreeNodeContext = createContext( undefined ); const useTreeNode = () => { const context = useContext(TreeNodeContext); if (!context) { throw new Error("TreeNode components must be used within a TreeNode"); } return context; }; export type TreeProviderProps = { children: ReactNode; defaultExpandedIds?: string[]; showLines?: boolean; showIcons?: boolean; selectable?: boolean; multiSelect?: boolean; selectedIds?: string[]; onSelectionChange?: (selectedIds: string[]) => void; indent?: number; animateExpand?: boolean; className?: string; }; export const TreeProvider = ({ children, defaultExpandedIds = [], showLines = true, showIcons = true, selectable = true, multiSelect = false, selectedIds, onSelectionChange, indent = 20, animateExpand = true, className, }: TreeProviderProps) => { const [expandedIds, setExpandedIds] = useState>( new Set(defaultExpandedIds) ); const [internalSelectedIds, setInternalSelectedIds] = useState( selectedIds ?? [] ); const isControlled = selectedIds !== undefined && onSelectionChange !== undefined; const currentSelectedIds = isControlled ? selectedIds : internalSelectedIds; const toggleExpanded = useCallback((nodeId: string) => { setExpandedIds((prev) => { const newSet = new Set(prev); if (newSet.has(nodeId)) { newSet.delete(nodeId); } else { newSet.add(nodeId); } return newSet; }); }, []); const handleSelection = useCallback( (nodeId: string, ctrlKey = false) => { if (!selectable) { return; } let newSelection: string[]; if (multiSelect && ctrlKey) { newSelection = currentSelectedIds.includes(nodeId) ? currentSelectedIds.filter((id) => id !== nodeId) : [...currentSelectedIds, nodeId]; } else { newSelection = currentSelectedIds.includes(nodeId) ? [] : [nodeId]; } if (isControlled) { onSelectionChange?.(newSelection); } else { setInternalSelectedIds(newSelection); } }, [ selectable, multiSelect, currentSelectedIds, isControlled, onSelectionChange, ] ); return ( {children} ); }; export type TreeViewProps = HTMLAttributes; export const TreeView = ({ className, children, ...props }: TreeViewProps) => (

{children}

); export type TreeNodeProps = HTMLAttributes & { nodeId?: string; level?: number; isLast?: boolean; parentPath?: boolean[]; children?: ReactNode; }; export const TreeNode = ({ nodeId: providedNodeId, level = 0, isLast = false, parentPath = [], children, className, onClick, ...props }: TreeNodeProps) => { const generatedId = useId(); const nodeId = providedNodeId ?? generatedId; // Build the parent path - mark positions where the parent was the last child const currentPath = level === 0 ? [] : [...parentPath]; if (level > 0 && parentPath.length < level - 1) { // Fill in missing levels with false (not last) while (currentPath.length < level - 1) { currentPath.push(false); } } if (level > 0) { currentPath[level - 1] = isLast; } return (

{children}

); }; export type TreeNodeTriggerProps = ComponentProps; export const TreeNodeTrigger = ({ children, className, onClick, ...props }: TreeNodeTriggerProps) => { const { selectedIds, toggleExpanded, handleSelection, indent } = useTree(); const { nodeId, level } = useTreeNode(); const isSelected = selectedIds.includes(nodeId); return ( { toggleExpanded(nodeId); handleSelection(nodeId, e.ctrlKey || e.metaKey); onClick?.(e); }} style={{ paddingLeft: level * (indent ?? 0) + 8 }} whileTap={{ scale: 0.98, transition: { duration: 0.1 } }} {...props} > {children as ReactNode} ); }; export const TreeLines = () => { const { showLines, indent } = useTree(); const { level, isLast, parentPath } = useTreeNode(); if (!showLines || level === 0) { return null; } return (

{/* Render vertical lines for all parent levels */} {Array.from({ length: level }, (_, index) => { const shouldHideLine = parentPath[index] === true; if (shouldHideLine && index === level - 1) { return null; } return (

); })} {/* Horizontal connector line */}

{/* Vertical line to midpoint for last items */} {isLast && (

)}

); }; export type TreeNodeContentProps = ComponentProps & { hasChildren?: boolean; }; export const TreeNodeContent = ({ children, hasChildren = false, className, ...props }: TreeNodeContentProps) => { const { animateExpand, expandedIds } = useTree(); const { nodeId } = useTreeNode(); const isExpanded = expandedIds.has(nodeId); return ( {hasChildren && isExpanded && ( {children} )} ); }; export type TreeExpanderProps = ComponentProps & { hasChildren?: boolean; }; export const TreeExpander = ({ hasChildren = false, className, onClick, ...props }: TreeExpanderProps) => { const { expandedIds, toggleExpanded } = useTree(); const { nodeId } = useTreeNode(); const isExpanded = expandedIds.has(nodeId); if (!hasChildren) { return

; } return ( { e.stopPropagation(); toggleExpanded(nodeId); onClick?.(e); }} transition={{ duration: 0.2, ease: "easeInOut" }} {...props} > ); }; export type TreeIconProps = ComponentProps & { icon?: ReactNode; hasChildren?: boolean; }; export const TreeIcon = ({ icon, hasChildren = false, className, ...props }: TreeIconProps) => { const { showIcons, expandedIds } = useTree(); const { nodeId } = useTreeNode(); const isExpanded = expandedIds.has(nodeId); if (!showIcons) { return null; } const getDefaultIcon = () => hasChildren ? ( isExpanded ? ( ) : ( ) ) : ( ); return ( {icon || getDefaultIcon()} ); }; export type TreeLabelProps = HTMLAttributes; export const TreeLabel = ({ className, ...props }: TreeLabelProps) => ( );

tree

Instalação

Dependências