chore: export

src/hooks/useTable.tsx

@@ -4,14 +4,35 @@ import type { TableRowSelection } from 'antd/es/table/interface';
 
 import { safeKey } from '@/hooks/useData';
 
-export default function useTableRowSelection() {
+type rowSelectionOptions = {
+  key: 'id' | string;
+  rowType: 'id' | 'row' | 'all';
+}
+export function useTableRowSelection(options: Partial<rowSelectionOptions> = {}) {
+  const { key = 'id', rowType = 'id' } = options;
   const [selectedRowKeys, setSelectedRowKeys] = useState<React.Key[]>([]);
   const [selectedRowIDs, setSelectedRowIDs] = useState<string[]>([]);
+  const [selectedRows, setSelectedRows] = useState<Record<string|symbol, any>[]>([]);
 
-  const onSelectChange = (newSelectedRowKeys: React.Key[], selectedRows: Record<string|symbol, any>) => {
-    const keys = selectedRows.map((i: any) => i[safeKey]);
-    setSelectedRowIDs(keys);
+  const onSelectChange = (newSelectedRowKeys: React.Key[], newSelectedRows: Record<string|symbol, any>[]) => {
+    const keys = newSelectedRows.map((i: any) => i[safeKey] || i[key]);
     setSelectedRowKeys(newSelectedRowKeys);
+    if (rowType === 'id') {
+      setSelectedRowIDs(keys);
+    }
+    if (rowType === 'row') {
+      setSelectedRows(newSelectedRows);
+    }
+    if (rowType === 'all') {
+      setSelectedRowIDs(keys);
+      setSelectedRows(newSelectedRows);
+    }
+  };
+
+  const rowReset = () => {
+    setSelectedRowKeys([]);
+    setSelectedRowIDs([]);
+    setSelectedRows([]);
   };
 
   const rowSelection: TableRowSelection<Record<string, any>> = {
@@ -24,14 +45,14 @@ export default function useTableRowSelection() {
     ],
   };
 
-  return { rowSelection, selectedRowIDs };
+  return { rowSelection, selectedRowIDs, selectedRows, rowReset };
 }
 
 export const TABLE_PAGINATION = {
   hideOnSinglePage: true,
   showSizeChanger: true,
   showQuickJumper: true,
-  defaultPageSize: 5,
+  defaultPageSize: 10,
   pageSizeOptions: [5, 10, 15, 20],
   showTotal: (total: number) => <span>Total {total} items</span>,
 };