@dxos/react-ui-table 0.8.4-main.c1de068 → 0.8.4-main.c85a9c8dae

package/src/components/Table/Table.tsx

@@ -2,409 +2,465 @@
 // Copyright 2024 DXOS.org
 //
 
-import { type Schema } from 'effect/Schema';
+import { Atom, RegistryContext, useAtomValue } from '@effect-atom/atom-react';
+import * as String from 'effect/String';
 import React, {
+  type JSX,
   type MouseEvent,
   type PropsWithChildren,
+  type Ref,
   type WheelEvent,
   forwardRef,
   useCallback,
+  useContext,
+  useEffect,
   useImperativeHandle,
-  useState,
   useMemo,
-  useEffect,
+  useState,
 } from 'react';
 
-import { type Client } from '@dxos/client';
-import { Filter } from '@dxos/echo';
-import { getValue } from '@dxos/echo-schema';
-import { invariant } from '@dxos/invariant';
-// TODO(wittjosiah): Remove dependency on react-client.
-import { getSpace } from '@dxos/react-client/echo';
-import { useTranslation } from '@dxos/react-ui';
+import { type Type } from '@dxos/echo';
+import { log } from '@dxos/log';
 import { useAttention } from '@dxos/react-ui-attention';
 import {
-  closestCell,
+  type DxGridAxisMeta,
   type DxGridElement,
-  type DxGridPosition,
-  type GridContentProps,
-  Grid,
   type DxGridPlane,
   type DxGridPlaneRange,
-  gridSeparatorInlineEnd,
+  type DxGridPosition,
+  Grid,
+  type GridContentProps,
+  closestCell,
   gridSeparatorBlockEnd,
+  gridSeparatorInlineEnd,
 } from '@dxos/react-ui-grid';
-import { mx } from '@dxos/react-ui-theme';
-import { isNotFalsy, safeParseInt } from '@dxos/util';
+import { DxEditRequest } from '@dxos/react-ui-grid';
+import { mx, useComposableProps } from '@dxos/ui-theme';
+import { type ComposableProps } from '@dxos/ui-types';
+
+import { type InsertRowResult, ModalController, type TableModel, type TablePresentation } from '../../model';
+import { tableButtons, tableControls } from '../../util';
+import { type OnCreateHandler, type TableCellEditorProps, TableValueEditor } from '../TableCellEditor';
+import { TableToolbar } from '../TableToolbar';
 
 import { ColumnActionsMenu } from './ColumnActionsMenu';
 import { ColumnSettings } from './ColumnSettings';
-import { CreateRefPanel } from './CreateRefPanel';
 import { RowActionsMenu } from './RowActionsMenu';
-import { ModalController, type TableModel, type TablePresentation } from '../../model';
-import { translationKey } from '../../translations';
-import { tableButtons, tableControls } from '../../util';
-import { createOption, TableValueEditor, type TableCellEditorProps } from '../TableCellEditor';
+
+const columnDefault = { grid: { minSize: 80, maxSize: 640 } };
+const rowDefault = { frozenRowsStart: { readonly: true, focusUnfurl: false } };
+const emptyColumnMeta = Atom.make<DxGridAxisMeta>({ grid: {} });
 
 //
 // Table.Root
 //
 
-export type TableRootProps = PropsWithChildren<{ role?: string }>;
+export type TableRootProps = PropsWithChildren;
 
-const TableRoot = ({ children, role = 'article' }: TableRootProps) => {
-  return (
-    <div
-      role='none'
-      className={mx(
-        'relative !border-separator [&_.dx-grid]:max-is-[--dx-grid-content-inline-size] [&_.dx-grid]:max-bs-[--dx-grid-content-block-size]',
-        role === 'card--popover' && 'popover-card-height',
-        role === 'section' && 'attention-surface',
-        role === 'card--intrinsic' && '[&_.dx-grid]:bs-[--dx-grid-content-block-size]',
-        ['card--popover', 'section', 'card--extrinsic'].includes(role) && 'overflow-hidden',
-        ['article', 'slide'].includes(role) && 'flex flex-col [&_.dx-grid]:grow [&_.dx-grid]:bs-0',
-      )}
-    >
-      {children}
-    </div>
-  );
+const TableRoot = ({ children }: TableRootProps) => {
+  return <>{children}</>;
 };
 
 //
 // Table.Main
 //
 
+const emptyRowsAtom = Atom.make<unknown[]>([]);
+const emptyCellUpdateAtom = Atom.make<number>(0);
+
 export type TableController = {
   update?: (cell?: DxGridPosition) => void;
+  handleInsertRowResult?: (insertRowResult?: InsertRowResult) => void;
 };
 
-export type TableMainProps = {
+export type TableMainProps<T extends Type.AnyEntity = Type.AnyEntity> = ComposableProps<{
+  schema?: T;
   model?: TableModel;
   presentation?: TablePresentation;
-  schema?: Schema.AnyNoContext;
-  client?: Client;
-  // TODO(burdon): Rename since attention isn't a useful concept here? Standardize across other components. Pass property into useAttention.
+  // TODO(burdon): Factor out attention.
   ignoreAttention?: boolean;
+  onCreate?: OnCreateHandler;
   onRowClick?: (row: any) => void;
-};
+  testId?: string;
+}>;
+
+const TableMainInner = <T extends Type.AnyEntity = Type.AnyEntity>(
+  { schema, model, presentation, ignoreAttention, onCreate, onRowClick, testId, ...props }: TableMainProps<T>,
+  forwardedRef: Ref<TableController>,
+) => {
+  const registry = useContext(RegistryContext);
+  const [dxGrid, setDxGrid] = useState<DxGridElement | null>(null);
+  const { hasAttention } = useAttention(model?.id ?? 'table');
+  const modals = useMemo(() => new ModalController(registry), [registry]);
+  const columnMeta = useAtomValue(model?.columnMeta ?? emptyColumnMeta);
+  // Subscribe to rows atom to trigger re-render when rows change.
+  const rows = useAtomValue(model?.rows ?? emptyRowsAtom);
+  // Subscribe to cell update counter to trigger re-render when individual cells change.
+  const cellUpdateCounter = useAtomValue(model?.cellUpdate ?? emptyCellUpdateAtom);
+  // Derive column count from columnMeta (reactive) with fallback.
+  const columnCount = Object.keys(columnMeta.grid).length || model?.projection.getFields().length || 0;
+
+  const draftRowCount = model?.getDraftRowCount() ?? 0;
+
+  const frozen = useMemo(() => {
+    const noActionColumn =
+      model?.features.dataEditable === false &&
+      model?.features.schemaEditable === false &&
+      model.rowActions.length === 0;
+
+    return {
+      frozenRowsStart: 1,
+      frozenRowsEnd: Math.max(1, draftRowCount),
+      frozenColsStart: model?.features.selection.enabled ? 1 : 0,
+      frozenColsEnd: noActionColumn ? 0 : 1,
+    };
+  }, [model, draftRowCount]);
+
+  const getCells = useCallback<NonNullable<GridContentProps['getCells']>>(
+    (range: DxGridPlaneRange, plane: DxGridPlane) => presentation?.getCells(range, plane) ?? {},
+    [presentation],
+  );
 
-const TableMain = forwardRef<TableController, TableMainProps>(
-  ({ model, presentation, ignoreAttention, schema, client, onRowClick }, forwardedRef) => {
-    const [dxGrid, setDxGrid] = useState<DxGridElement | null>(null);
-    const { hasAttention } = useAttention(model?.id ?? 'table');
-    const { t } = useTranslation(translationKey);
-    const modals = useMemo(() => new ModalController(), []);
-
-    const draftRowCount = model?.getDraftRowCount() ?? 0;
-
-    const frozen = useMemo(() => {
-      const noActionColumn =
-        model?.features.dataEditable === false &&
-        model?.features.schemaEditable === false &&
-        model.rowActions.length === 0;
-
-      return {
-        frozenRowsStart: 1,
-        frozenRowsEnd: draftRowCount,
-        frozenColsStart: model?.features.selection.enabled ? 1 : 0,
-        frozenColsEnd: noActionColumn ? 0 : 1,
-      };
-    }, [model, draftRowCount]);
-
-    const getCells = useCallback<NonNullable<GridContentProps['getCells']>>(
-      (range: DxGridPlaneRange, plane: DxGridPlane) => presentation?.getCells(range, plane) ?? {},
-      [presentation],
-    );
-
-    useEffect(() => {
-      if (!presentation || !dxGrid) {
-        return;
+  useEffect(() => {
+    if (!dxGrid || !presentation) {
+      return;
+    }
+
+    dxGrid.getCells = getCells;
+  }, [dxGrid, presentation, getCells]);
+
+  // Trigger grid update when rows or cell values change.
+  useEffect(() => {
+    if (!dxGrid) {
+      return;
+    }
+
+    dxGrid.updateCells(true);
+    dxGrid.requestUpdate();
+  }, [dxGrid, rows, cellUpdateCounter]);
+
+  const handleInsertRowResult = useCallback(
+    (insertResult?: InsertRowResult) => {
+      if (insertResult === 'draft') {
+        requestAnimationFrame(() => {
+          dxGrid?.setFocus({ plane: 'frozenRowsEnd', col: 0, row: 0 });
+          dxGrid?.refocus();
+        });
+      } else {
+        requestAnimationFrame(() => {
+          dxGrid?.setFocus({ plane: 'grid', col: 0, row: model ? model.getRowCount() - 1 : 0 });
+          dxGrid?.refocus();
+        });
       }
-      dxGrid.getCells = getCells;
-    }, [presentation, dxGrid, getCells]);
-
-    /**
-     * Provides an external controller that can be called to repaint the table.
-     */
-    useImperativeHandle<TableController, TableController>(forwardedRef, () => {
-      if (!presentation || !dxGrid) {
-        return {};
+    },
+    [model, dxGrid],
+  );
+
+  const handleSave = useCallback(() => {
+    dxGrid?.updateCells(true);
+    dxGrid?.requestUpdate();
+  }, [dxGrid]);
+
+  const handleSaveDraftRow = useCallback(
+    (rowIndex = 0) => {
+      if (model && dxGrid) {
+        const didCommitSuccessfully = model.commitDraftRow(rowIndex);
+        if (didCommitSuccessfully) {
+          requestAnimationFrame(() => {
+            dxGrid.scrollToEndRow();
+          });
+        }
       }
+    },
+    [model, dxGrid],
+  );
 
-      return {
-        update: (cell) => {
-          if (cell) {
-            dxGrid.updateIfWithinBounds(cell, true);
-          } else {
-            dxGrid.updateCells(true);
-            dxGrid.requestUpdate();
-          }
-        },
-      };
-    }, [presentation, dxGrid]);
-
-    const handleGridClick = useCallback(
-      (event: MouseEvent) => {
-        const rowIndex = safeParseInt((event.target as HTMLElement).closest('[aria-rowindex]')?.ariaRowIndex ?? '');
-        if (rowIndex != null) {
-          if (onRowClick) {
+  const handleGridClick = useCallback(
+    (event: MouseEvent) => {
+      const cell = closestCell(event.target as HTMLElement);
+      if (cell) {
+        const { row: rowIndex, plane } = cell;
+        if (onRowClick) {
+          if (plane === 'grid') {
             const row = model?.getRowAt(rowIndex);
             row && onRowClick(row);
-          }
-
-          if (model?.features.selection.enabled && model?.selection.selectionMode === 'single') {
-            model.selection.toggleSelectionForRowIndex(rowIndex);
+          } else {
+            onRowClick(cell);
           }
         }
 
-        if (!modals) {
-          return;
+        if (model?.features.selection.enabled && model?.selection.selectionMode === 'single') {
+          model.selection.toggleSelectionForRowIndex(rowIndex);
         }
+      }
 
-        const buttonSelector = Object.values(tableButtons)
-          .map((button) => `button[${button.attr}]`)
-          .join(',');
+      if (!modals) {
+        return;
+      }
 
-        const tableButtonElement = (event.target as HTMLElement).closest(buttonSelector) as HTMLElement;
-        if (tableButtonElement) {
-          const button = Object.values(tableButtons).find((btn) => tableButtonElement.hasAttribute(btn.attr));
-          if (!button) {
-            return;
-          }
+      const buttonSelector = Object.values(tableButtons)
+        .map((button) => `button[${button.attr}]`)
+        .join(',');
 
-          modals.setTrigger(tableButtonElement);
-          const data = button.getData(tableButtonElement);
-          switch (data.type) {
-            case 'rowMenu': {
-              modals.showRowMenu(data.rowIndex);
-              break;
-            }
-            case 'columnSettings': {
-              modals.showColumnMenu(data.fieldId);
-              break;
-            }
-            case 'newColumn': {
-              modals.showColumnCreator();
-              break;
-            }
-            case 'sort': {
-              model?.sorting?.toggleSort(data.fieldId);
-              break;
-            }
-            case 'saveDraftRow': {
-              if (model) {
-                const didCommitSuccessfully = model.commitDraftRow(data.rowIndex);
-                if (dxGrid && didCommitSuccessfully) {
-                  requestAnimationFrame(() => {
-                    dxGrid.scrollToEndRow();
-                  });
-                }
-              }
-              break;
-            }
-          }
+      const tableButtonElement = (event.target as HTMLElement).closest(buttonSelector) as HTMLElement;
+      if (tableButtonElement) {
+        const button = Object.values(tableButtons).find((btn) => tableButtonElement.hasAttribute(btn.attr));
+        if (!button) {
           return;
         }
 
-        const controlSelector = Object.values(tableControls)
-          .map((control) => `input[${control.attr}]`)
-          .join(',');
-
-        const controlElement = (event.target as HTMLElement).closest(controlSelector) as HTMLElement;
-        if (controlElement) {
-          const control = Object.values(tableControls).find((ctrl) => controlElement.hasAttribute(ctrl.attr));
-          if (!control) {
-            return;
+        modals.setTrigger(tableButtonElement);
+        const data = button.getData(tableButtonElement);
+        switch (data.type) {
+          case 'rowMenu': {
+            modals.showRowMenu(data.rowIndex);
+            break;
           }
-
-          const data = control.getData(controlElement);
-          switch (data.type) {
-            case 'checkbox': {
-              if (data.header) {
-                model?.selection.setSelection(model.selection.allRowsSeleted.value ? 'none' : 'all');
-              } else {
-                model?.selection.toggleSelectionForRowIndex(data.rowIndex);
-              }
-              break;
-            }
-            case 'switch': {
-              if (model) {
-                model.updateCellData({ row: data.rowIndex, col: data.colIndex }, (value) => !value);
-              }
-              break;
-            }
+          case 'columnSettings': {
+            modals.showColumnMenu(data.fieldId);
+            break;
+          }
+          case 'newColumn': {
+            modals.showColumnCreator();
+            break;
+          }
+          case 'sort': {
+            model?.toggleSort(data.fieldId);
+            break;
+          }
+          case 'saveDraftRow': {
+            handleSaveDraftRow();
+            break;
           }
         }
-      },
-      [model, modals, dxGrid],
-    );
+        return;
+      }
 
-    const handleFocus = useCallback<NonNullable<TableCellEditorProps['onFocus']>>(
-      (increment, delta) => {
-        if (dxGrid) {
-          dxGrid.refocus(increment, delta);
-        }
-      },
-      [dxGrid],
-    );
+      const controlSelector = Object.values(tableControls)
+        .map((control) => `input[${control.attr}]`)
+        .join(',');
 
-    const handleEnter = useCallback<NonNullable<TableCellEditorProps['onEnter']>>(
-      (cell) => {
-        if (!model?.features.dataEditable) {
+      const controlElement = (event.target as HTMLElement).closest(controlSelector) as HTMLElement;
+      if (controlElement) {
+        const control = Object.values(tableControls).find((ctrl) => controlElement.hasAttribute(ctrl.attr));
+        if (!control) {
           return;
         }
 
-        // TODO(burdon): Insert row only if bottom row isn't completely blank already.
-        if (model && cell.row === model.getRowCount() - 1) {
-          model.insertRow();
-          if (dxGrid) {
-            requestAnimationFrame(() => {
-              dxGrid?.scrollToRow(cell.row + 1);
-              dxGrid?.refocus('row', 1);
-            });
+        const data = control.getData(controlElement);
+        switch (data.type) {
+          case 'checkbox': {
+            if (data.header) {
+              model?.selection.setSelection(model.selection.allRowsSelected ? 'none' : 'all');
+            } else {
+              model?.selection.toggleSelectionForRowIndex(data.rowIndex);
+            }
+            break;
+          }
+          case 'switch': {
+            if (model) {
+              model.updateCellData({ row: data.rowIndex, col: data.colIndex }, (value) => !value);
+            }
+            break;
           }
         }
-      },
-      [model, dxGrid],
-    );
+      }
+    },
+    [model, modals, dxGrid],
+  );
 
-    const handleKeyDown = useCallback<NonNullable<GridContentProps['onKeyDown']>>(
-      (event) => {
-        if (!model?.features.dataEditable) {
-          return;
+  const handleFocus = useCallback<NonNullable<TableCellEditorProps['onFocus']>>(
+    (increment, delta, cell) => {
+      if (dxGrid && model) {
+        if (cell?.plane === 'grid' && cell?.row >= model.getRowCount() - 1 && increment !== 'col') {
+          handleInsertRowResult(draftRowCount < 1 ? model.insertRow() : 'final');
+        } else if (cell?.plane === 'frozenRowsEnd' && increment === 'row') {
+          handleSaveDraftRow(cell.row);
+          handleInsertRowResult(model.insertRow());
+        } else {
+          dxGrid.refocus(increment, delta);
         }
+      }
+    },
+    [dxGrid, model],
+  );
 
-        const cell = closestCell(event.target);
-        if (!model || !cell) {
-          return;
-        }
+  const handleKeyDown = useCallback<NonNullable<GridContentProps['onKeyDown']>>(
+    (event) => {
+      if (!model?.features.dataEditable) {
+        return;
+      }
+
+      const cell = closestCell(event.target);
+      if (!model || !cell) {
+        return;
+      }
 
+      // Handle Meta+C (Copy) and Meta+V (Paste) commands
+      if (event.metaKey || event.ctrlKey) {
         switch (event.key) {
-          case 'Backspace':
-          case 'Delete': {
-            model.setCellData(cell, undefined);
+          case 'c': {
+            // Copy focused cell's text content to clipboard
+            try {
+              const cellData = model.getCellData(cell);
+              const textContent = cellData?.toString() ?? '';
+              void navigator.clipboard.writeText(textContent);
+              event.preventDefault();
+            } catch (error) {
+              log.catch(error);
+            }
+            break;
+          }
+          case 'v': {
+            // Paste clipboard content to focused cell
+            event.preventDefault();
+            void navigator.clipboard.readText().then((clipboardText) => {
+              try {
+                // Attempt to set the cell's content to clipboard content
+                model.setCellData(cell, String.trim(clipboardText).replace(/[\n\r]+/, ' '));
+                handleSave();
+              } catch {
+                // If validation fails, emit a DxEditRequest event with initialContent from clipboard
+                // TODO(thure): Should `dx-grid` expose a method like this?
+                const cellElement = (event.target as HTMLElement).closest(
+                  '[data-dx-grid-action="cell"]',
+                ) as HTMLElement;
+                if (cellElement) {
+                  const rect = cellElement.getBoundingClientRect();
+                  const editRequest = new DxEditRequest({
+                    cellIndex: `${cell.plane},${cell.col},${cell.row}`,
+                    cellBox: {
+                      insetInlineStart: rect.left,
+                      insetBlockStart: rect.top,
+                      inlineSize: rect.width,
+                      blockSize: rect.height,
+                    },
+                    cellElement,
+                    initialContent: clipboardText,
+                  });
+                  cellElement.dispatchEvent(editRequest);
+                }
+              }
+            });
             break;
           }
         }
-      },
-      [model],
-    );
-
-    const handleAxisResize = useCallback<NonNullable<GridContentProps['onAxisResize']>>(
-      (event) => {
-        if (event.axis === 'col') {
-          const columnIndex = parseInt(event.index, 10);
-          model?.setColumnWidth(columnIndex, event.size);
-        }
-      },
-      [model],
-    );
+      }
 
-    const handleWheel = useCallback(
-      (event: WheelEvent) => {
-        if (!ignoreAttention && !hasAttention) {
-          event.stopPropagation();
+      switch (event.key) {
+        case 'Backspace':
+        case 'Delete': {
+          try {
+            model.setCellData(cell, undefined);
+            event.preventDefault();
+            handleSave();
+          } catch {
+            // Delete results in a validation error; don’t prevent default so dx-grid can emit an edit request.
+          }
+          break;
         }
-      },
-      [hasAttention, ignoreAttention],
-    );
+      }
+    },
+    [model, dxGrid, handleSave],
+  );
 
-    const handleNewColumn = useCallback(() => {
-      const columns = model?.getColumnCount();
-      if (dxGrid && columns) {
-        dxGrid.scrollToColumn(columns - 1);
+  const handleAxisResize = useCallback<NonNullable<GridContentProps['onAxisResize']>>(
+    (event) => {
+      if (event.axis === 'col') {
+        const columnIndex = parseInt(event.index, 10);
+        model?.setColumnWidth(columnIndex, event.size);
       }
-    }, [model, dxGrid]);
-
-    // TODO(burdon): Factor out?
-    // TODO(burdon): Generalize to handle other value types (e.g., enums).
-    const handleQuery = useCallback<NonNullable<TableCellEditorProps['onQuery']>>(
-      async ({ field, props }, text) => {
-        if (model && props.referenceSchema && field.referencePath) {
-          const space = getSpace(model.view);
-          invariant(space);
-
-          let schema;
-          if (client) {
-            schema = client.graph.schemaRegistry.getSchema(props.referenceSchema);
-          }
-          if (!schema) {
-            schema = space.db.schemaRegistry.getSchema(props.referenceSchema);
-          }
+    },
+    [model],
+  );
 
-          if (schema) {
-            const { objects } = await space.db.query(Filter.type(schema)).run();
-            const options = objects
-              .map((obj) => {
-                const value = getValue(obj, field.referencePath!);
-                if (!value || typeof value !== 'string') {
-                  return undefined;
-                }
+  const handleWheel = useCallback(
+    (event: WheelEvent) => {
+      if (!ignoreAttention && !hasAttention) {
+        event.stopPropagation();
+      }
+    },
+    [hasAttention, ignoreAttention],
+  );
 
-                return {
-                  label: value,
-                  data: obj,
-                };
-              })
-              .filter(isNotFalsy);
-
-            return [
-              ...options,
-              {
-                label: t('create new object label', { text }),
-                data: createOption(text),
-              },
-            ];
-          }
-        }
+  const handleNewColumn = useCallback(() => {
+    const columns = model?.getColumnCount();
+    if (dxGrid && columns) {
+      dxGrid.scrollToColumn(columns - 1);
+    }
+  }, [model, dxGrid]);
+
+  /**
+   * Provides an external controller that can be called to repaint the table.
+   */
+  useImperativeHandle<TableController, TableController>(forwardedRef, () => {
+    if (!dxGrid || !presentation) {
+      return {};
+    }
 
-        return [];
+    return {
+      update: (cell) => {
+        if (cell) {
+          dxGrid.updateIfWithinBounds(cell, true);
+        } else {
+          dxGrid.updateCells(true);
+          dxGrid.requestUpdate();
+        }
       },
-      [model, client, t],
-    );
+      handleInsertRowResult,
+    };
+  }, [dxGrid, model, presentation]);
 
-    const handleSave = useCallback(() => {
-      dxGrid?.updateCells(true);
-    }, [dxGrid]);
-
-    if (!model || !modals) {
-      return <span role='none' className='attention-surface' />;
-    }
+  if (!model || !modals) {
+    return <span role='none' className='dx-attention-surface' />;
+  }
 
-    return (
+  const { className, ...rest } = useComposableProps(props);
+  return (
+    <div {...rest} role='none' className={mx('dx-container relative', className)}>
       <Grid.Root id={model.id ?? 'table-grid'}>
-        <TableValueEditor
+        <TableValueEditor<T>
           model={model}
           modals={modals}
           schema={schema}
-          onEnter={handleEnter}
           onFocus={handleFocus}
-          onQuery={handleQuery}
           onSave={handleSave}
+          onCreate={onCreate}
         />
         <Grid.Content
-          className={mx('[--dx-grid-base:var(--baseSurface)]', gridSeparatorInlineEnd, gridSeparatorBlockEnd)}
+          className={mx('[--dx-grid-base:var(--base-surface)]', gridSeparatorInlineEnd, gridSeparatorBlockEnd)}
           frozen={frozen}
-          columns={model.columnMeta.value}
-          limitRows={model.getRowCount() ?? 0}
-          limitColumns={model.projection.fields.length}
+          columns={columnMeta}
+          columnDefault={columnDefault}
+          rowDefault={rowDefault}
+          limitRows={rows.length}
+          limitColumns={columnCount}
           overscroll='trap'
           onAxisResize={handleAxisResize}
           onClick={handleGridClick}
-          onKeyDown={handleKeyDown}
+          onKeyDownCapture={handleKeyDown}
           onWheelCapture={handleWheel}
+          {...(testId && { 'data-testid': testId })}
           ref={setDxGrid}
         />
         <RowActionsMenu model={model} modals={modals} />
         <ColumnActionsMenu model={model} modals={modals} />
         <ColumnSettings model={model} modals={modals} onNewColumn={handleNewColumn} />
-        <CreateRefPanel model={model} modals={modals} />
       </Grid.Root>
-    );
-  },
-);
+    </div>
+  );
+};
+
+const TableMain = forwardRef(TableMainInner) as <T extends Type.AnyEntity = Type.AnyEntity>(
+  props: TableMainProps<T> & { ref?: Ref<TableController> },
+) => JSX.Element;
+
+//
+// Table
+//
 
 export const Table = {
   Root: TableRoot,
   Main: TableMain,
+  Toolbar: TableToolbar,
 };