@dxos/react-ui-table 0.8.4-main.d05673bc65 → 0.8.4-main.d9fc60f731

package/src/components/Table/Table.tsx

@@ -2,457 +2,64 @@
 // Copyright 2024 DXOS.org
 //
 
-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,
-  useMemo,
-  useState,
-} from 'react';
+import { Atom } from '@effect-atom/atom-react';
+import { createContext } from '@radix-ui/react-context';
+import React, { type PropsWithChildren, type RefObject, forwardRef, useImperativeHandle, useRef } from 'react';
 
-import { type Type } from '@dxos/echo';
-import { log } from '@dxos/log';
-import { useAttention } from '@dxos/react-ui-attention';
-import {
-  type DxGridAxisMeta,
-  type DxGridElement,
-  type DxGridPlane,
-  type DxGridPlaneRange,
-  type DxGridPosition,
-  Grid,
-  type GridContentProps,
-  closestCell,
-  gridSeparatorBlockEnd,
-  gridSeparatorInlineEnd,
-} from '@dxos/react-ui-grid';
-import { DxEditRequest } from '@dxos/react-ui-grid';
-import { composableProps, mx } from '@dxos/ui-theme';
-import { type ComposableProps } from '@dxos/ui-types';
+import { type DxGridAxisMeta, type DxGridPosition } from '@dxos/react-ui-grid';
 
-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 { RowActionsMenu } from './RowActionsMenu';
+import { type InsertRowResult } from '../../model';
+import { TableContent } from './TableContent';
+import { TableToolbar } from './TableToolbar';
 
 const columnDefault = { grid: { minSize: 80, maxSize: 640 } };
 const rowDefault = { frozenRowsStart: { readonly: true, focusUnfurl: false } };
 const emptyColumnMeta = Atom.make<DxGridAxisMeta>({ grid: {} });
 
 //
-// Table.Root
+// Context
 //
 
-export type TableRootProps = PropsWithChildren;
-
-const TableRoot = ({ children }: TableRootProps) => {
-  return <>{children}</>;
+type TableContextValue = {
+  /** Mutable ref populated by Content so Root can expose the controller. */
+  controllerRef: RefObject<TableController>;
 };
 
+const [TableContextProvider, useTableContext] = createContext<TableContextValue>('Table');
+
 //
-// Table.Main
+// Controller
 //
 
-const emptyRowsAtom = Atom.make<unknown[]>([]);
-const emptyCellUpdateAtom = Atom.make<number>(0);
-
-export type TableController = {
+type TableController = {
   update?: (cell?: DxGridPosition) => void;
   handleInsertRowResult?: (insertRowResult?: InsertRowResult) => void;
 };
 
-export type TableMainProps<T extends Type.AnyEntity = Type.AnyEntity> = ComposableProps<HTMLDivElement> & {
-  schema?: T;
-  model?: TableModel;
-  presentation?: TablePresentation;
-  // 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],
-  );
-
-  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();
-        });
-      }
-    },
-    [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],
-  );
-
-  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);
-          } else {
-            onRowClick(cell);
-          }
-        }
-
-        if (model?.features.selection.enabled && model?.selection.selectionMode === 'single') {
-          model.selection.toggleSelectionForRowIndex(rowIndex);
-        }
-      }
-
-      if (!modals) {
-        return;
-      }
-
-      const buttonSelector = Object.values(tableButtons)
-        .map((button) => `button[${button.attr}]`)
-        .join(',');
-
-      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;
-        }
-
-        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?.toggleSort(data.fieldId);
-            break;
-          }
-          case 'saveDraftRow': {
-            handleSaveDraftRow();
-            break;
-          }
-        }
-        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;
-        }
-
-        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, modals, dxGrid],
-  );
-
-  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 handleKeyDown = useCallback<NonNullable<GridContentProps['onKeyDown']>>(
-    (event) => {
-      if (!model?.features.dataEditable) {
-        return;
-      }
-
-      const cell = closestCell(event.target);
-      if (!model || !cell) {
-        return;
-      }
+//
+// Root
+//
 
-      // Handle Meta+C (Copy) and Meta+V (Paste) commands
-      if (event.metaKey || event.ctrlKey) {
-        switch (event.key) {
-          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;
-          }
-        }
-      }
+const TABLE_ROOT_NAME = 'Table.Root';
 
-      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;
-        }
-      }
-    },
-    [model, dxGrid, handleSave],
-  );
+type TableRootProps = PropsWithChildren;
 
-  const handleAxisResize = useCallback<NonNullable<GridContentProps['onAxisResize']>>(
-    (event) => {
-      if (event.axis === 'col') {
-        const columnIndex = parseInt(event.index, 10);
-        model?.setColumnWidth(columnIndex, event.size);
-      }
-    },
-    [model],
-  );
+const TableRoot = forwardRef<TableController, TableRootProps>(({ children }, forwardedRef) => {
+  const controllerRef = useRef<TableController>({});
 
-  const handleWheel = useCallback(
-    (event: WheelEvent) => {
-      if (!ignoreAttention && !hasAttention) {
-        event.stopPropagation();
-      }
-    },
-    [hasAttention, ignoreAttention],
+  useImperativeHandle(
+    forwardedRef,
+    () => ({
+      update: (cell) => controllerRef.current?.update?.(cell),
+      handleInsertRowResult: (result) => controllerRef.current?.handleInsertRowResult?.(result),
+    }),
+    [],
   );
 
-  const handleNewColumn = useCallback(() => {
-    const columns = model?.getColumnCount();
-    if (dxGrid && columns) {
-      dxGrid.scrollToColumn(columns - 1);
-    }
-  }, [model, dxGrid]);
+  return <TableContextProvider controllerRef={controllerRef}>{children}</TableContextProvider>;
+});
 
-  /**
-   * Provides an external controller that can be called to repaint the table.
-   */
-  useImperativeHandle<TableController, TableController>(forwardedRef, () => {
-    if (!dxGrid || !presentation) {
-      return {};
-    }
-
-    return {
-      update: (cell) => {
-        if (cell) {
-          dxGrid.updateIfWithinBounds(cell, true);
-        } else {
-          dxGrid.updateCells(true);
-          dxGrid.requestUpdate();
-        }
-      },
-      handleInsertRowResult,
-    };
-  }, [dxGrid, model, presentation]);
-
-  if (!model || !modals) {
-    return <span role='none' className='dx-attention-surface' />;
-  }
-
-  return (
-    <div {...composableProps(props, { role: 'none', className: 'dx-container relative' })}>
-      <Grid.Root id={model.id ?? 'table-grid'}>
-        <TableValueEditor<T>
-          model={model}
-          modals={modals}
-          schema={schema}
-          onFocus={handleFocus}
-          onSave={handleSave}
-          onCreate={onCreate}
-        />
-        <Grid.Content
-          className={mx('[--dx-grid-base:var(--base-surface)]', gridSeparatorInlineEnd, gridSeparatorBlockEnd)}
-          frozen={frozen}
-          columns={columnMeta}
-          columnDefault={columnDefault}
-          rowDefault={rowDefault}
-          limitRows={rows.length}
-          limitColumns={columnCount}
-          overscroll='trap'
-          onAxisResize={handleAxisResize}
-          onClick={handleGridClick}
-          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} />
-      </Grid.Root>
-    </div>
-  );
-};
-
-const TableMain = forwardRef(TableMainInner) as <T extends Type.AnyEntity = Type.AnyEntity>(
-  props: TableMainProps<T> & { ref?: Ref<TableController> },
-) => JSX.Element;
+TableRoot.displayName = TABLE_ROOT_NAME;
 
 //
 // Table
@@ -460,6 +67,10 @@ const TableMain = forwardRef(TableMainInner) as <T extends Type.AnyEntity = Type
 
 export const Table = {
   Root: TableRoot,
-  Main: TableMain,
   Toolbar: TableToolbar,
+  Content: TableContent,
 };
+
+export type { TableController, TableRootProps };
+
+export { useTableContext };