mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-02-01 20:23:23 +08:00
432 lines
15 KiB
TypeScript
432 lines
15 KiB
TypeScript
"use client"
|
|
|
|
import * as React from "react"
|
|
import {
|
|
ColumnDef,
|
|
ColumnFiltersState,
|
|
ColumnSizingState,
|
|
flexRender,
|
|
getCoreRowModel,
|
|
getFacetedRowModel,
|
|
getFacetedUniqueValues,
|
|
getFilteredRowModel,
|
|
getPaginationRowModel,
|
|
getSortedRowModel,
|
|
SortingState,
|
|
useReactTable,
|
|
VisibilityState,
|
|
} from "@tanstack/react-table"
|
|
import {
|
|
IconChevronDown,
|
|
IconChevronLeft,
|
|
IconChevronRight,
|
|
IconChevronsLeft,
|
|
IconChevronsRight,
|
|
IconLayoutColumns,
|
|
IconPlus,
|
|
IconDownload,
|
|
IconSearch,
|
|
IconLoader2,
|
|
} from "@tabler/icons-react"
|
|
import { Button } from "@/components/ui/button"
|
|
import {
|
|
DropdownMenu,
|
|
DropdownMenuCheckboxItem,
|
|
DropdownMenuContent,
|
|
DropdownMenuLabel,
|
|
DropdownMenuSeparator,
|
|
DropdownMenuItem,
|
|
DropdownMenuTrigger,
|
|
} from "@/components/ui/dropdown-menu"
|
|
import { Input } from "@/components/ui/input"
|
|
import { Label } from "@/components/ui/label"
|
|
import {
|
|
Select,
|
|
SelectContent,
|
|
SelectItem,
|
|
SelectTrigger,
|
|
SelectValue,
|
|
} from "@/components/ui/select"
|
|
import {
|
|
Table,
|
|
TableBody,
|
|
TableCell,
|
|
TableHead,
|
|
TableHeader,
|
|
TableRow,
|
|
} from "@/components/ui/table"
|
|
import type { Endpoint } from "@/types/endpoint.types"
|
|
|
|
interface EndpointsDataTableProps<TData extends { id: number | string }, TValue> {
|
|
columns: ColumnDef<TData, TValue>[]
|
|
data: TData[]
|
|
searchPlaceholder?: string
|
|
searchColumn?: string
|
|
searchValue?: string
|
|
onSearch?: (value: string) => void
|
|
isSearching?: boolean
|
|
onAddNew?: () => void
|
|
addButtonText?: string
|
|
onSelectionChange?: (selectedRows: TData[]) => void
|
|
pagination?: { pageIndex: number; pageSize: number }
|
|
onPaginationChange?: (pagination: { pageIndex: number; pageSize: number }) => void
|
|
totalCount?: number
|
|
totalPages?: number
|
|
onDownloadAll?: () => void
|
|
onDownloadSelected?: () => void
|
|
}
|
|
|
|
export function EndpointsDataTable<TData extends { id: number | string }, TValue>({
|
|
columns,
|
|
data,
|
|
searchPlaceholder = "搜索主机名...",
|
|
searchColumn = "url",
|
|
searchValue,
|
|
onSearch,
|
|
isSearching = false,
|
|
onAddNew,
|
|
addButtonText = "Add",
|
|
onSelectionChange,
|
|
pagination: externalPagination,
|
|
onPaginationChange,
|
|
totalCount,
|
|
totalPages,
|
|
onDownloadAll,
|
|
onDownloadSelected,
|
|
}: EndpointsDataTableProps<TData, TValue>) {
|
|
const [rowSelection, setRowSelection] = React.useState<Record<string, boolean>>({})
|
|
const [columnVisibility, setColumnVisibility] = React.useState<VisibilityState>({})
|
|
const [columnFilters, setColumnFilters] = React.useState<ColumnFiltersState>([])
|
|
const [columnSizing, setColumnSizing] = React.useState<ColumnSizingState>({})
|
|
const [sorting, setSorting] = React.useState<SortingState>([])
|
|
const [internalPagination, setInternalPagination] = React.useState<{ pageIndex: number, pageSize: number }>({
|
|
pageIndex: 0,
|
|
pageSize: 10,
|
|
})
|
|
|
|
// 本地搜索输入状态(只在回车或点击按钮时触发搜索)
|
|
const [localSearchValue, setLocalSearchValue] = React.useState(searchValue ?? "")
|
|
|
|
React.useEffect(() => {
|
|
setLocalSearchValue(searchValue ?? "")
|
|
}, [searchValue])
|
|
|
|
const handleSearchSubmit = () => {
|
|
if (onSearch) {
|
|
onSearch(localSearchValue)
|
|
} else {
|
|
table.getColumn(searchColumn)?.setFilterValue(localSearchValue)
|
|
}
|
|
}
|
|
|
|
const handleKeyDown = (e: React.KeyboardEvent<HTMLInputElement>) => {
|
|
if (e.key === 'Enter') {
|
|
handleSearchSubmit()
|
|
}
|
|
}
|
|
|
|
const pagination = externalPagination || internalPagination
|
|
|
|
const handlePaginationChange = React.useCallback((updaterOrValue: { pageIndex: number; pageSize: number } | ((prev: { pageIndex: number; pageSize: number }) => { pageIndex: number; pageSize: number })) => {
|
|
if (onPaginationChange) {
|
|
if (typeof updaterOrValue === 'function') {
|
|
const newPagination = updaterOrValue(pagination)
|
|
onPaginationChange(newPagination)
|
|
} else {
|
|
onPaginationChange(updaterOrValue)
|
|
}
|
|
} else {
|
|
setInternalPagination(updaterOrValue)
|
|
}
|
|
}, [onPaginationChange, pagination])
|
|
|
|
const table = useReactTable({
|
|
data,
|
|
columns,
|
|
state: {
|
|
sorting,
|
|
columnVisibility,
|
|
rowSelection,
|
|
columnFilters,
|
|
columnSizing,
|
|
pagination,
|
|
},
|
|
getRowId: (row) => row.id.toString(),
|
|
enableRowSelection: true,
|
|
enableColumnResizing: true,
|
|
columnResizeMode: 'onChange',
|
|
onRowSelectionChange: setRowSelection,
|
|
onSortingChange: setSorting,
|
|
onColumnFiltersChange: setColumnFilters,
|
|
onColumnVisibilityChange: setColumnVisibility,
|
|
onColumnSizingChange: setColumnSizing,
|
|
onPaginationChange: handlePaginationChange,
|
|
getCoreRowModel: getCoreRowModel(),
|
|
getFilteredRowModel: getFilteredRowModel(),
|
|
getPaginationRowModel: externalPagination ? undefined : getPaginationRowModel(),
|
|
getSortedRowModel: getSortedRowModel(),
|
|
getFacetedRowModel: getFacetedRowModel(),
|
|
getFacetedUniqueValues: getFacetedUniqueValues(),
|
|
manualPagination: !!externalPagination,
|
|
pageCount: totalPages,
|
|
})
|
|
|
|
React.useEffect(() => {
|
|
if (onSelectionChange) {
|
|
const selectedRows = table.getFilteredSelectedRowModel().rows.map(row => row.original)
|
|
onSelectionChange(selectedRows)
|
|
}
|
|
}, [rowSelection, onSelectionChange, table])
|
|
|
|
return (
|
|
<div className="w-full space-y-4">
|
|
<div className="flex items-center justify-between">
|
|
<div className="flex items-center space-x-2">
|
|
<Input
|
|
placeholder={searchPlaceholder}
|
|
value={localSearchValue}
|
|
onChange={(e) => setLocalSearchValue(e.target.value)}
|
|
onKeyDown={handleKeyDown}
|
|
className="h-8 max-w-sm"
|
|
/>
|
|
<Button variant="outline" size="sm" onClick={handleSearchSubmit} disabled={isSearching}>
|
|
{isSearching ? (
|
|
<IconLoader2 className="h-4 w-4 animate-spin" />
|
|
) : (
|
|
<IconSearch className="h-4 w-4" />
|
|
)}
|
|
</Button>
|
|
</div>
|
|
|
|
<div className="flex items-center space-x-2">
|
|
<DropdownMenu>
|
|
<DropdownMenuTrigger asChild>
|
|
<Button variant="outline" size="sm">
|
|
<IconLayoutColumns />
|
|
Columns
|
|
<IconChevronDown />
|
|
</Button>
|
|
</DropdownMenuTrigger>
|
|
<DropdownMenuContent align="end">
|
|
{table
|
|
.getAllColumns()
|
|
.filter(
|
|
(column) =>
|
|
typeof column.accessorFn !== "undefined" && column.getCanHide()
|
|
)
|
|
.map((column) => {
|
|
return (
|
|
<DropdownMenuCheckboxItem
|
|
key={column.id}
|
|
className="capitalize"
|
|
checked={column.getIsVisible()}
|
|
onCheckedChange={(value) => column.toggleVisibility(!!value)}
|
|
>
|
|
{column.id === "id" && "ID"}
|
|
{column.id === "url" && "URL"}
|
|
{column.id === "endpoint" && "Endpoint"}
|
|
{column.id === "method" && "Method"}
|
|
{column.id === "statusCode" && "Status"}
|
|
{column.id === "title" && "Title"}
|
|
{column.id === "contentLength" && "Size"}
|
|
{column.id === "contentType" && "Content Type"}
|
|
{column.id === "responseTime" && "Response time"}
|
|
{column.id === "tags" && "Tags"}
|
|
{column.id === "createdAt" && "Created At"}
|
|
{column.id === "updatedAt" && "Updated At"}
|
|
{!["id", "url", "endpoint", "method", "statusCode", "title", "contentLength", "contentType", "responseTime", "tags", "createdAt", "updatedAt"].includes(column.id) && column.id}
|
|
</DropdownMenuCheckboxItem>
|
|
)
|
|
})}
|
|
</DropdownMenuContent>
|
|
</DropdownMenu>
|
|
|
|
{(onDownloadAll || onDownloadSelected) && (
|
|
<DropdownMenu>
|
|
<DropdownMenuTrigger asChild>
|
|
<Button variant="outline" size="sm">
|
|
<IconDownload />
|
|
Download
|
|
<IconChevronDown />
|
|
</Button>
|
|
</DropdownMenuTrigger>
|
|
<DropdownMenuContent align="end" className="w-64">
|
|
<DropdownMenuLabel>Download Options</DropdownMenuLabel>
|
|
<DropdownMenuSeparator />
|
|
{onDownloadAll && (
|
|
<DropdownMenuItem onClick={onDownloadAll}>
|
|
<IconDownload className="h-4 w-4" />
|
|
Download All Endpoints
|
|
</DropdownMenuItem>
|
|
)}
|
|
{onDownloadSelected && (
|
|
<DropdownMenuItem
|
|
onClick={onDownloadSelected}
|
|
disabled={table.getFilteredSelectedRowModel().rows.length === 0}
|
|
>
|
|
<IconDownload className="h-4 w-4" />
|
|
Download Selected Endpoints
|
|
</DropdownMenuItem>
|
|
)}
|
|
</DropdownMenuContent>
|
|
</DropdownMenu>
|
|
)}
|
|
|
|
{onAddNew && (
|
|
<Button onClick={onAddNew} size="sm">
|
|
<IconPlus />
|
|
{addButtonText}
|
|
</Button>
|
|
)}
|
|
</div>
|
|
</div>
|
|
|
|
<div className="rounded-md border overflow-x-auto">
|
|
<Table style={{ minWidth: table.getCenterTotalSize() }}>
|
|
<TableHeader>
|
|
{table.getHeaderGroups().map((headerGroup) => (
|
|
<TableRow key={headerGroup.id}>
|
|
{headerGroup.headers.map((header) => {
|
|
return (
|
|
<TableHead
|
|
key={header.id}
|
|
colSpan={header.colSpan}
|
|
style={{ width: header.getSize() }}
|
|
className="relative group"
|
|
>
|
|
{header.isPlaceholder
|
|
? null
|
|
: flexRender(
|
|
header.column.columnDef.header,
|
|
header.getContext()
|
|
)}
|
|
{header.column.getCanResize() && (
|
|
<div
|
|
onMouseDown={header.getResizeHandler()}
|
|
onTouchStart={header.getResizeHandler()}
|
|
onDoubleClick={() => header.column.resetSize()}
|
|
className="absolute -right-2.5 top-0 h-full w-5 cursor-col-resize select-none touch-none bg-transparent flex justify-center"
|
|
>
|
|
<div className="w-1.5 h-full bg-transparent group-hover:bg-border" />
|
|
</div>
|
|
)}
|
|
</TableHead>
|
|
)
|
|
})}
|
|
</TableRow>
|
|
))}
|
|
</TableHeader>
|
|
|
|
<TableBody>
|
|
{table.getRowModel().rows?.length ? (
|
|
table.getRowModel().rows.map((row) => (
|
|
<TableRow
|
|
key={row.id}
|
|
data-state={row.getIsSelected() && "selected"}
|
|
>
|
|
{row.getVisibleCells().map((cell) => (
|
|
<TableCell
|
|
key={cell.id}
|
|
style={{ width: cell.column.getSize() }}
|
|
>
|
|
{flexRender(
|
|
cell.column.columnDef.cell,
|
|
cell.getContext()
|
|
)}
|
|
</TableCell>
|
|
))}
|
|
</TableRow>
|
|
))
|
|
) : (
|
|
<TableRow>
|
|
<TableCell
|
|
colSpan={columns.length}
|
|
className="h-24 text-center"
|
|
>
|
|
No results
|
|
</TableCell>
|
|
</TableRow>
|
|
)}
|
|
</TableBody>
|
|
</Table>
|
|
</div>
|
|
|
|
<div className="flex items-center justify-between px-2">
|
|
<div className="flex-1 text-sm text-muted-foreground">
|
|
{table.getFilteredSelectedRowModel().rows.length} of{" "}
|
|
{externalPagination ? totalCount : table.getFilteredRowModel().rows.length} row(s) selected
|
|
</div>
|
|
|
|
<div className="flex items-center space-x-6 lg:space-x-8">
|
|
<div className="flex items-center space-x-2">
|
|
<Label htmlFor="rows-per-page" className="text-sm font-medium">
|
|
Rows per page
|
|
</Label>
|
|
<Select
|
|
value={`${table.getState().pagination.pageSize}`}
|
|
onValueChange={(value) => {
|
|
table.setPageSize(Number(value))
|
|
}}
|
|
>
|
|
<SelectTrigger className="h-8 w-[90px]" id="rows-per-page">
|
|
<SelectValue placeholder={table.getState().pagination.pageSize} />
|
|
</SelectTrigger>
|
|
<SelectContent side="top">
|
|
{[10, 20, 50, 100, 200, 500, 1000].map((pageSize) => (
|
|
<SelectItem key={pageSize} value={`${pageSize}`}>
|
|
{pageSize}
|
|
</SelectItem>
|
|
))}
|
|
</SelectContent>
|
|
</Select>
|
|
</div>
|
|
|
|
<div className="flex w-[100px] items-center justify-center text-sm font-medium">
|
|
Page {table.getState().pagination.pageIndex + 1} of{" "}
|
|
{externalPagination ? totalPages : table.getPageCount()}
|
|
</div>
|
|
|
|
<div className="flex items-center space-x-2">
|
|
<Button
|
|
variant="outline"
|
|
className="hidden h-8 w-8 p-0 lg:flex"
|
|
onClick={() => table.setPageIndex(0)}
|
|
disabled={externalPagination ? pagination.pageIndex === 0 : !table.getCanPreviousPage()}
|
|
>
|
|
<span className="sr-only">Go to first page</span>
|
|
<IconChevronsLeft />
|
|
</Button>
|
|
<Button
|
|
variant="outline"
|
|
className="h-8 w-8 p-0"
|
|
onClick={() => table.previousPage()}
|
|
disabled={externalPagination ? pagination.pageIndex === 0 : !table.getCanPreviousPage()}
|
|
>
|
|
<span className="sr-only">Go to previous page</span>
|
|
<IconChevronLeft />
|
|
</Button>
|
|
<Button
|
|
variant="outline"
|
|
className="h-8 w-8 p-0"
|
|
onClick={() => table.nextPage()}
|
|
disabled={externalPagination ? pagination.pageIndex >= (totalPages || 1) - 1 : !table.getCanNextPage()}
|
|
>
|
|
<span className="sr-only">Go to next page</span>
|
|
<IconChevronRight />
|
|
</Button>
|
|
<Button
|
|
variant="outline"
|
|
className="hidden h-8 w-8 p-0 lg:flex"
|
|
onClick={() => table.setPageIndex((totalPages || table.getPageCount()) - 1)}
|
|
disabled={externalPagination ? pagination.pageIndex >= (totalPages || 1) - 1 : !table.getCanNextPage()}
|
|
>
|
|
<span className="sr-only">Go to last page</span>
|
|
<IconChevronsRight />
|
|
</Button>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|