Improve Table typing

.changeset/fluffy-pianos-end.md

@@ -0,0 +1,5 @@
+---
+'@itwin/itwinui-react': major
+---
+
+Table now has better type support. Users must now import types from `@itwin/itwinui-react/react-table` instead of from `@types/react-table`.

CONTRIBUTING.md

@@ -250,7 +250,7 @@ More examples can be found in the [style guide](./STYLEGUIDE.md).
 
 Each component has a corresponding jest test inside of its directory. Be sure to cover your added code with tests.
 
-Use `yarn test` to run the tests. Add `--watch` flag if you want tests to rerun after changes.
+Use `yarn test:unit` to run the unit tests. Run `yarn test:unit:watch` if you want unit tests to rerun after changes.
 
 We usually do not use `describe` block and our test case should start with 'should'.
 

apps/storybook/.gitignore

@@ -4,3 +4,6 @@ cypress-visual-screenshots/comparison
 cypress-visual-screenshots/diff
 cypress-visual-report
 cypress/screenshots
+
+# Compiled files
+dist

apps/storybook/package.json

@@ -34,7 +34,7 @@
     "vite": "^4.1.5"
   },
   "scripts": {
-    "build": "dotenv -v NODE_OPTIONS=\"--max-old-space-size=4096\" sb build",
+    "build": "tsc -p tsconfig.build.json && dotenv -v NODE_OPTIONS=\"--max-old-space-size=4096\" sb build",
     "dev": "sb dev -p 6006 --no-open --quiet",
     "lint": "eslint \"**/*.{ts,tsx}\" --max-warnings=0",
     "test": "node scripts/run-tests.js",

apps/storybook/src/InformationPanel.stories.tsx

@@ -4,7 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 import { Story, Meta } from '@storybook/react';
 import React from 'react';
-import { CellProps } from 'react-table';
+import { CellProps } from '@itwin/itwinui-react/react-table';
 import { action } from '@storybook/addon-actions';
 import { SvgEdit } from '@itwin/itwinui-icons-react';
 import {

apps/storybook/src/Table.stories.tsx

@@ -10,7 +10,8 @@ import type {
   Row,
   TableInstance,
   TableState,
-} from 'react-table';
+  TableOptions,
+} from '@itwin/itwinui-react/react-table';
 import {
   Checkbox,
   Code,
@@ -21,13 +22,10 @@ import {
   Table,
   Text,
   tableFilters,
-  TableFilterValue,
-  TableProps,
   Tooltip,
   DefaultCell,
   EditableCell,
   TablePaginator,
-  TablePaginatorRendererProps,
   ActionColumn,
   Anchor,
   SelectionColumn,
@@ -37,6 +35,10 @@ import {
   ProgressRadial,
   BaseFilter,
 } from '@itwin/itwinui-react';
+import type {
+  TableFilterValue,
+  TablePaginatorRendererProps,
+} from '@itwin/itwinui-react';
 import { useMemo, useState } from '@storybook/addons';
 import { action } from '@storybook/addon-actions';
 import {
@@ -47,6 +49,10 @@ import {
   SvgStatusSuccess,
   SvgStatusWarning,
 } from '@itwin/itwinui-icons-react';
+import { StoryFn } from '@storybook/react';
+
+type TableProps<T extends Record<string, unknown> = Record<string, unknown>> =
+  React.ComponentProps<typeof Table<T>>;
 
 export default {
   title: 'Core/Table',
@@ -172,7 +178,9 @@ export const SelectableMulti = () => {
         )}`,
       )(),
     [],
-  );
+  ) satisfies NonNullable<
+    TableProps<{ name: string; description: string }>['onSelect']
+  >;
 
   const onRowClick = useCallback(
     (event: React.MouseEvent, row: Row) =>
@@ -245,7 +253,9 @@ export const Sortable = () => {
   const onSort = useCallback(
     (state) => action(`Sort changed. Table state: ${JSON.stringify(state)}`)(),
     [],
-  );
+  ) satisfies NonNullable<
+    TableProps<{ name: string; description: string }>['onSort']
+  >;
 
   const columns = useMemo(
     () => [
@@ -440,7 +450,7 @@ export const Filters = () => {
     (
       filters: TableFilterValue<TableStoryDataType>[],
       state: TableState,
-      filteredData: Row<{ name: string; description: string }>[],
+      filteredData: Row<TableStoryDataType>[],
     ) => {
       // rowInfo is used due to JSON errors when displaying row data
       let rowInfo = '[';
@@ -456,7 +466,7 @@ export const Filters = () => {
       )();
     },
     [],
-  );
+  ) satisfies NonNullable<TableProps<TableStoryDataType>['onFilter']>;
 
   return (
     <Table
@@ -554,7 +564,12 @@ export const Expandable = () => {
         )}`,
       )(),
     [],
-  );
+  ) satisfies NonNullable<
+    TableProps<{
+      name: string;
+      description: string;
+    }>['onExpand']
+  >;
 
   const columns = useMemo(
     () => [
@@ -615,7 +630,12 @@ export const ExpandableSubrows = () => {
         )}`,
       )(),
     [],
-  );
+  ) satisfies NonNullable<
+    TableProps<{
+      name: string;
+      description: string;
+    }>['onExpand']
+  >;
 
   const columns = useMemo(
     () => [
@@ -826,7 +846,12 @@ export const RowInViewport = () => {
 
   const onRowInViewport = useCallback((rowData) => {
     action(`Row in view: ${JSON.stringify(rowData)}`)();
-  }, []);
+  }, []) satisfies NonNullable<
+    TableProps<{
+      name: string;
+      description: string;
+    }>['onRowInViewport']
+  >;
 
   return (
     <>
@@ -1052,7 +1077,7 @@ export const ControlledState = () => {
     subRows: DemoData[];
   };
 
-  const tableInstance = React.useRef<TableInstance>();
+  const tableInstance = React.useRef<TableInstance<DemoData>>();
   const [selectedRows, setSelectedRows] = useState<DemoData[]>([]);
   const [expandedRows, setExpandedRows] = useState<DemoData[]>([]);
 
@@ -1070,7 +1095,7 @@ export const ControlledState = () => {
       },
     ],
     [],
-  );
+  ) satisfies Column<DemoData>[];
 
   const data: DemoData[] = useMemo(
     () => [
@@ -1135,9 +1160,24 @@ export const ControlledState = () => {
         name: 'Row 2',
         description: 'Description 2',
         subRows: [
-          { name: 'Row 2.1', description: 'Description 2.1', subRows: [] },
-          { name: 'Row 2.2', description: 'Description 2.2', subRows: [] },
-          { name: 'Row 2.3', description: 'Description 2.3', subRows: [] },
+          {
+            id: '2.1',
+            name: 'Row 2.1',
+            description: 'Description 2.1',
+            subRows: [],
+          },
+          {
+            id: '2.2',
+            name: 'Row 2.2',
+            description: 'Description 2.2',
+            subRows: [],
+          },
+          {
+            id: '2.3',
+            name: 'Row 2.3',
+            description: 'Description 2.3',
+            subRows: [],
+          },
         ],
       },
       { id: '3', name: 'Row 3', description: 'Description 3', subRows: [] },
@@ -1180,18 +1220,28 @@ export const ControlledState = () => {
       <Table
         columns={columns}
         emptyTableContent='No data.'
-        stateReducer={useCallback((newState, action, prevState, instance) => {
-          tableInstance.current = instance;
-          return newState;
-        }, [])}
+        stateReducer={
+          useCallback((newState, action, prevState, instance) => {
+            tableInstance.current = instance;
+            return newState;
+          }, []) satisfies NonNullable<TableOptions<DemoData>['stateReducer']>
+        }
         isSelectable
-        onSelect={useCallback((selected) => {
-          setSelectedRows(selected ?? []);
-        }, [])}
-        onExpand={useCallback((expanded) => {
-          setExpandedRows(expanded);
-        }, [])}
-        getRowId={useCallback((rowData) => rowData.id, [])}
+        onSelect={
+          useCallback((selected) => {
+            setSelectedRows(selected ?? []);
+          }, []) satisfies NonNullable<TableProps<DemoData>['onSelect']>
+        }
+        onExpand={
+          useCallback((expanded) => {
+            setExpandedRows(expanded ?? []);
+          }, []) satisfies NonNullable<TableProps<DemoData>['onExpand']>
+        }
+        getRowId={
+          useCallback((rowData) => rowData.id, []) satisfies NonNullable<
+            TableOptions<DemoData>['getRowId']
+          >
+        }
         data={data}
       />
     </>
@@ -1462,9 +1512,9 @@ export const Full2 = () => {
     [isRowDisabled, menuItems],
   );
 
-  const rowProps = useCallback((row: Row<{ status: string | undefined }>) => {
+  const rowProps = useCallback((row: Row<TableStoryDataType>) => {
     return {
-      status: row.original.status,
+      status: row.original.status satisfies TableStoryDataType['status'],
     };
   }, []);
 
@@ -1590,7 +1640,7 @@ export const Localized = () => {
 };
 
 Localized.decorators = [
-  (Story) => (
+  (Story: StoryFn) => (
     <div style={{ height: '90vh' }}>
       <Story />
     </div>
@@ -1614,7 +1664,9 @@ export const Condensed = () => {
         )}`,
       )(),
     [],
-  );
+  ) satisfies NonNullable<
+    TableProps<{ name: string; description: string }>['onExpand']
+  >;
 
   const columns = useMemo(
     () => [
@@ -1864,7 +1916,7 @@ export const WithPaginator = () => {
 };
 
 WithPaginator.decorators = [
-  (Story) => (
+  (Story: StoryFn) => (
     <div style={{ height: '90vh' }}>
       <Story />
     </div>
@@ -2073,7 +2125,7 @@ export const WithManualPaginatorAndFilter = () => {
 };
 
 WithManualPaginatorAndFilter.decorators = [
-  (Story) => (
+  (Story: StoryFn) => (
     <div style={{ height: '90vh' }}>
       <Story />
     </div>
@@ -2231,7 +2283,7 @@ export const CustomFilter = () => {
 };
 
 CustomFilter.decorators = [
-  (Story) => (
+  (Story: StoryFn) => (
     <div style={{ height: '90vh' }}>
       <Story />
     </div>
@@ -2335,7 +2387,7 @@ export const ResizableColumns = () => {
   );
 
   const [columnResizeMode, setColumnResizeMode] =
-    React.useState<TableProps['columnResizeMode']>('fit');
+    React.useState<TableProps<TableStoryDataType>['columnResizeMode']>('fit');
 
   return (
     <>
@@ -2563,7 +2615,7 @@ export const HorizontalScroll = () => {
 };
 
 HorizontalScroll.decorators = [
-  (Story) => (
+  (Story: StoryFn) => (
     <div
       style={{
         height: '375px',
@@ -2904,7 +2956,7 @@ export const CustomizedColumns = () => {
         )}`,
       )(),
     [],
-  );
+  ) satisfies NonNullable<TableProps<(typeof data)[number]>['onExpand']>;
 
   const data = useMemo(
     () => [
@@ -2952,9 +3004,9 @@ export const CustomizedColumns = () => {
         Header: 'Name',
         accessor: 'name',
         cellRenderer: (props) => (
-          <DefaultCell
+          <DefaultCell<(typeof data)[number]>
             {...props}
-            isDisabled={(rowData) =>
+            isDisabled={(rowData: (typeof data)[number]) =>
               isCellDisabled(rowData) || isRowDisabled(rowData)
             }
           />
@@ -3000,6 +3052,11 @@ export const ColumnManager = () => {
     id: string;
     startDate: Date;
     endDate: Date;
+    price: string;
+    color: string;
+    stock: number;
+    rating: string;
+    location: string;
   };
 
   const columns = useMemo(
@@ -3314,7 +3371,7 @@ export const StickyColumns = () => {
 };
 
 StickyColumns.decorators = [
-  (Story) => (
+  (Story: StoryFn) => (
     <div
       style={{
         height: '375px',
@@ -3338,6 +3395,7 @@ export const StatusAndCellIcons = () => {
           props: CellRendererProps<{
             startIcon: JSX.Element;
             endIcon: JSX.Element;
+            isLoading?: boolean;
           }>,
         ) => (
           <DefaultCell

apps/storybook/tsconfig.build.json

@@ -0,0 +1,13 @@
+{
+  "$schema": "https://json.schemastore.org/tsconfig",
+  "extends": "./tsconfig.json",
+
+  // Temporarily testing only Table.stories.tsx.
+  // TODO: Fix all tsc errors in all stories and then include all stories.
+  "include": [],
+  "files": ["src/Table.stories.tsx"],
+
+  "compilerOptions": {
+    "outDir": "dist"
+  }
+}