import type { Cell, Column, ColumnDef, Header, HeaderGroup, Row, SortingState, Table, } from "@tanstack/react-table"; import { flexRender, getCoreRowModel, getSortedRowModel, useReactTable, } from "@tanstack/react-table"; import { atom, useAtom } from "jotai"; import { ArrowDownIcon, ArrowUpIcon, ChevronsUpDownIcon } from "lucide-react"; import type { HTMLAttributes, ReactNode } from "react"; import { createContext, memo, useCallback, useContext } from "react"; import { Button } from "@/components/ui/button"; import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuTrigger, } from "@/components/ui/dropdown-menu"; import { TableBody as TableBodyRaw, TableCell as TableCellRaw, TableHeader as TableHeaderRaw, TableHead as TableHeadRaw, Table as TableRaw, TableRow as TableRowRaw, } from "@/components/ui/table"; import { cn } from "@/lib/utils"; export type { ColumnDef } from "@tanstack/react-table"; const sortingAtom = atom([]); export const TableContext = createContext<{ data: unknown[]; columns: ColumnDef[]; table: Table | null; }>({ data: [], columns: [], table: null, }); export type TableProviderProps = { columns: ColumnDef[]; data: TData[]; children: ReactNode; className?: string; }; export function TableProvider({ columns, data, children, className, }: TableProviderProps) { const [sorting, setSorting] = useAtom(sortingAtom); const table = useReactTable({ data, columns, getCoreRowModel: getCoreRowModel(), getSortedRowModel: getSortedRowModel(), onSortingChange: (updater) => { // @ts-expect-error updater is a function that returns a sorting object const newSorting = updater(sorting); setSorting(newSorting); }, state: { sorting, }, }); return ( {children} ); } export type TableHeadProps = { header: Header; className?: string; }; export const TableHead = memo(({ header, className }: TableHeadProps) => ( {header.isPlaceholder ? null : flexRender(header.column.columnDef.header, header.getContext())} )); TableHead.displayName = "TableHead"; export type TableHeaderGroupProps = { headerGroup: HeaderGroup; children: (props: { header: Header }) => ReactNode; }; export const TableHeaderGroup = ({ headerGroup, children, }: TableHeaderGroupProps) => ( {headerGroup.headers.map((header) => children({ header }))} ); export type TableHeaderProps = { className?: string; children: (props: { headerGroup: HeaderGroup }) => ReactNode; }; export const TableHeader = ({ className, children }: TableHeaderProps) => { const { table } = useContext(TableContext); return ( {table?.getHeaderGroups().map((headerGroup) => children({ headerGroup }))} ); }; export interface TableColumnHeaderProps extends HTMLAttributes { column: Column; title: string; } export function TableColumnHeader({ column, title, className, }: TableColumnHeaderProps) { // Extract inline event handlers to prevent unnecessary re-renders const handleSortAsc = useCallback(() => { column.toggleSorting(false); }, [column]); const handleSortDesc = useCallback(() => { column.toggleSorting(true); }, [column]); if (!column.getCanSort()) { return

{title}

; } return (

Asc Desc

); } export type TableCellProps = { cell: Cell; className?: string; }; export const TableCell = ({ cell, className }: TableCellProps) => ( {flexRender(cell.column.columnDef.cell, cell.getContext())} ); export type TableRowProps = { row: Row; children: (props: { cell: Cell }) => ReactNode; className?: string; }; export const TableRow = ({ row, children, className }: TableRowProps) => ( {row.getVisibleCells().map((cell) => children({ cell }))} ); export type TableBodyProps = { children: (props: { row: Row }) => ReactNode; className?: string; }; export const TableBody = ({ children, className }: TableBodyProps) => { const { columns, table } = useContext(TableContext); const rows = table?.getRowModel().rows; return ( {rows?.length ? ( rows.map((row) => children({ row })) ) : ( No results. )} ); };

table

Instalação

Dependências

Registry dependencies