@tanstack/table-core 9.0.0-alpha.4 → 9.0.0-alpha.41

package/src/features/row-selection/rowSelectionFeature.ts

@@ -0,0 +1,164 @@
+import {
+  assignPrototypeAPIs,
+  assignTableAPIs,
+  makeStateUpdater,
+} from '../../utils'
+import {
+  getDefaultRowSelectionState,
+  row_getCanMultiSelect,
+  row_getCanSelect,
+  row_getCanSelectSubRows,
+  row_getIsAllSubRowsSelected,
+  row_getIsSelected,
+  row_getIsSomeSelected,
+  row_getToggleSelectedHandler,
+  row_toggleSelected,
+  table_getFilteredSelectedRowModel,
+  table_getGroupedSelectedRowModel,
+  table_getIsAllPageRowsSelected,
+  table_getIsAllRowsSelected,
+  table_getIsSomePageRowsSelected,
+  table_getIsSomeRowsSelected,
+  table_getPreSelectedRowModel,
+  table_getSelectedRowModel,
+  table_getToggleAllPageRowsSelectedHandler,
+  table_getToggleAllRowsSelectedHandler,
+  table_resetRowSelection,
+  table_setRowSelection,
+  table_toggleAllPageRowsSelected,
+  table_toggleAllRowsSelected,
+} from './rowSelectionFeature.utils'
+import type { RowData } from '../../types/type-utils'
+import type { TableFeature, TableFeatures } from '../../types/TableFeatures'
+// import type {
+//   Row_RowSelection,
+//   TableOptions_RowSelection,
+//   TableState_RowSelection,
+//   Table_RowSelection,
+// } from './rowSelectionFeature.types'
+
+export interface RowSelectionFeatureConstructors<
+  TFeatures extends TableFeatures,
+  TData extends RowData,
+> {
+  // Row: Row_RowSelection
+  // Table: Table_RowSelection<TFeatures, TData>
+  // TableOptions: TableOptions_RowSelection<TFeatures, TData>
+  // TableState: TableState_RowSelection
+}
+
+export function constructRowSelectionFeature<
+  TFeatures extends TableFeatures,
+  TData extends RowData,
+>(): TableFeature<RowSelectionFeatureConstructors<TFeatures, TData>> {
+  return {
+    getInitialState: (initialState) => {
+      return {
+        rowSelection: getDefaultRowSelectionState(),
+        ...initialState,
+      }
+    },
+
+    getDefaultTableOptions: (table) => {
+      return {
+        onRowSelectionChange: makeStateUpdater('rowSelection', table),
+        enableRowSelection: true,
+        enableMultiRowSelection: true,
+        enableSubRowSelection: true,
+      }
+    },
+
+    assignRowPrototype: (prototype, table) => {
+      assignPrototypeAPIs('rowSelectionFeature', prototype, table, {
+        row_toggleSelected: {
+          fn: (row, value, opts) => row_toggleSelected(row, value, opts),
+        },
+        row_getIsSelected: {
+          fn: (row) => row_getIsSelected(row),
+        },
+        row_getIsSomeSelected: {
+          fn: (row) => row_getIsSomeSelected(row),
+        },
+        row_getIsAllSubRowsSelected: {
+          fn: (row) => row_getIsAllSubRowsSelected(row),
+        },
+        row_getCanSelect: {
+          fn: (row) => row_getCanSelect(row),
+        },
+        row_getCanSelectSubRows: {
+          fn: (row) => row_getCanSelectSubRows(row),
+        },
+        row_getCanMultiSelect: {
+          fn: (row) => row_getCanMultiSelect(row),
+        },
+        row_getToggleSelectedHandler: {
+          fn: (row) => row_getToggleSelectedHandler(row),
+        },
+      })
+    },
+
+    constructTableAPIs: (table) => {
+      assignTableAPIs('rowSelectionFeature', table, {
+        table_setRowSelection: {
+          fn: (updater) => table_setRowSelection(table, updater),
+        },
+        table_resetRowSelection: {
+          fn: (defaultState) => table_resetRowSelection(table, defaultState),
+        },
+        table_toggleAllRowsSelected: {
+          fn: (value) => table_toggleAllRowsSelected(table, value),
+        },
+        table_toggleAllPageRowsSelected: {
+          fn: (value) => table_toggleAllPageRowsSelected(table, value),
+        },
+        table_getPreSelectedRowModel: {
+          fn: () => table_getPreSelectedRowModel(table),
+        },
+        table_getSelectedRowModel: {
+          fn: () => table_getSelectedRowModel(table),
+          memoDeps: () => [
+            table.atoms.rowSelection?.get(),
+            table.getCoreRowModel(),
+          ],
+        },
+        table_getFilteredSelectedRowModel: {
+          fn: () => table_getFilteredSelectedRowModel(table),
+          memoDeps: () => [
+            table.atoms.rowSelection?.get(),
+            table.getFilteredRowModel(),
+          ],
+        },
+        table_getGroupedSelectedRowModel: {
+          fn: () => table_getGroupedSelectedRowModel(table),
+          memoDeps: () => [
+            table.atoms.rowSelection?.get(),
+            table.getSortedRowModel(),
+          ],
+        },
+        table_getIsAllRowsSelected: {
+          fn: () => table_getIsAllRowsSelected(table),
+        },
+        table_getIsAllPageRowsSelected: {
+          fn: () => table_getIsAllPageRowsSelected(table),
+        },
+        table_getIsSomeRowsSelected: {
+          fn: () => table_getIsSomeRowsSelected(table),
+        },
+        table_getIsSomePageRowsSelected: {
+          fn: () => table_getIsSomePageRowsSelected(table),
+        },
+        table_getToggleAllRowsSelectedHandler: {
+          fn: () => table_getToggleAllRowsSelectedHandler(table),
+        },
+        table_getToggleAllPageRowsSelectedHandler: {
+          fn: () => table_getToggleAllPageRowsSelectedHandler(table),
+        },
+      })
+    },
+  }
+}
+
+/**
+ * The Row Selection feature adds row selection state and APIs to the table and row objects.
+ */
+export const rowSelectionFeature = constructRowSelectionFeature()

package/src/features/row-selection/rowSelectionFeature.types.ts

@@ -0,0 +1,143 @@
+import type { OnChangeFn, RowData, Updater } from '../../types/type-utils'
+import type { TableFeatures } from '../../types/TableFeatures'
+import type { RowModel } from '../../core/row-models/coreRowModelsFeature.types'
+import type { Row } from '../../types/Row'
+
+export type RowSelectionState = Record<string, boolean | undefined>
+
+export interface TableState_RowSelection {
+  rowSelection: RowSelectionState
+}
+
+export interface TableOptions_RowSelection<
+  TFeatures extends TableFeatures,
+  TData extends RowData,
+> {
+  /**
+   * - Enables/disables multiple row selection for all rows in the table OR
+   * - A function that given a row, returns whether to enable/disable multiple row selection for that row's children/grandchildren
+   */
+  enableMultiRowSelection?: boolean | ((row: Row<TFeatures, TData>) => boolean)
+  /**
+   * - Enables/disables row selection for all rows in the table OR
+   * - A function that given a row, returns whether to enable/disable row selection for that row
+   */
+  enableRowSelection?: boolean | ((row: Row<TFeatures, TData>) => boolean)
+  /**
+   * Enables/disables automatic sub-row selection when a parent row is selected, or a function that enables/disables automatic sub-row selection for each row.
+   * (Use in combination with expanding or grouping features)
+   */
+  enableSubRowSelection?: boolean | ((row: Row<TFeatures, TData>) => boolean)
+  /**
+   * If provided, this function will be called with an `updaterFn` when `state.rowSelection` changes. This overrides the default internal state management, so you will need to persist the state change either fully or partially outside of the table.
+   */
+  onRowSelectionChange?: OnChangeFn<RowSelectionState>
+  // enableGroupingRowSelection?:
+  //   | boolean
+  //   | ((
+  //       row: Row<TFeatures, TData>
+  //     ) => boolean)
+  // isAdditiveSelectEvent?: (e: unknown) => boolean
+  // isInclusiveSelectEvent?: (e: unknown) => boolean
+  // selectRowsFn?: (
+  //   table: Table<TFeatures, TData>,
+  //   rowModel: RowModel<TFeatures, TData>
+  // ) => RowModel<TFeatures, TData>
+}
+
+export interface Row_RowSelection {
+  /**
+   * Returns whether or not the row can multi-select.
+   */
+  getCanMultiSelect: () => boolean
+  /**
+   * Returns whether or not the row can be selected.
+   */
+  getCanSelect: () => boolean
+  /**
+   * Returns whether or not the row can select sub rows automatically when the parent row is selected.
+   */
+  getCanSelectSubRows: () => boolean
+  /**
+   * Returns whether or not all of the row's sub rows are selected.
+   */
+  getIsAllSubRowsSelected: () => boolean
+  /**
+   * Returns whether or not the row is selected.
+   */
+  getIsSelected: () => boolean
+  /**
+   * Returns whether or not some of the row's sub rows are selected.
+   */
+  getIsSomeSelected: () => boolean
+  /**
+   * Returns a handler that can be used to toggle the row.
+   */
+  getToggleSelectedHandler: () => (event: unknown) => void
+  /**
+   * Selects/deselects the row.
+   */
+  toggleSelected: (value?: boolean, opts?: { selectChildren?: boolean }) => void
+}
+
+export interface Table_RowSelection<
+  TFeatures extends TableFeatures,
+  TData extends RowData,
+> {
+  /**
+   * Returns the row model of all rows that are selected after filtering has been applied.
+   */
+  getFilteredSelectedRowModel: () => RowModel<TFeatures, TData>
+  /**
+   * Returns the row model of all rows that are selected after grouping has been applied.
+   */
+  getGroupedSelectedRowModel: () => RowModel<TFeatures, TData>
+  /**
+   * Returns whether or not all rows on the current page are selected.
+   */
+  getIsAllPageRowsSelected: () => boolean
+  /**
+   * Returns whether or not all rows in the table are selected.
+   */
+  getIsAllRowsSelected: () => boolean
+  /**
+   * Returns whether or not any rows on the current page are selected.
+   */
+  getIsSomePageRowsSelected: () => boolean
+  /**
+   * Returns whether or not any rows in the table are selected.
+   */
+  getIsSomeRowsSelected: () => boolean
+  /**
+   * Returns the core row model of all rows before row selection has been applied.
+   */
+  getPreSelectedRowModel: () => RowModel<TFeatures, TData>
+  /**
+   * Returns the row model of all rows that are selected.
+   */
+  getSelectedRowModel: () => RowModel<TFeatures, TData>
+  /**
+   * Returns a handler that can be used to toggle all rows on the current page.
+   */
+  getToggleAllPageRowsSelectedHandler: () => (event: unknown) => void
+  /**
+   * Returns a handler that can be used to toggle all rows in the table.
+   */
+  getToggleAllRowsSelectedHandler: () => (event: unknown) => void
+  /**
+   * Resets the **rowSelection** state to the `initialState.rowSelection`, or `true` can be passed to force a default blank state reset to `{}`.
+   */
+  resetRowSelection: (defaultState?: boolean) => void
+  /**
+   * Sets or updates the `state.rowSelection` state.
+   */
+  setRowSelection: (updater: Updater<RowSelectionState>) => void
+  /**
+   * Selects/deselects all rows on the current page.
+   */
+  toggleAllPageRowsSelected: (value?: boolean) => void
+  /**
+   * Selects/deselects all rows in the table.
+   */
+  toggleAllRowsSelected: (value?: boolean) => void
+}