@beyondwork/docx-react-component 1.0.12 → 1.0.14

package/src/ui/WordReviewEditor.tsx

@@ -14,9 +14,14 @@ import type {
   CompatibilityReport,
   EditorError,
   EditorWarning,
+  FormattingAlignment,
   ExportDocxOptions,
+  InsertImageOptions,
+  InsertTableOptions,
   PersistedEditorSnapshot,
   RuntimeRenderSnapshot,
+  SearchOptions,
+  SearchResultSnapshot,
   SelectionSnapshot as PublicSelectionSnapshot,
   ExportResult,
   WordReviewEditorEvent,
@@ -27,8 +32,24 @@ import {
   createDetachedAnchor,
   createNodeAnchor,
   createRangeAnchor,
+  type TransactionMapping,
 } from "../core/selection/mapping.ts";
-import { createCanonicalDocumentId } from "../core/state/editor-state.ts";
+import {
+  applyFormattingOperationToDocument,
+  getFormattingStateFromRenderSnapshot,
+} from "../core/commands/formatting-commands.ts";
+import { insertImage as insertImageInDocument } from "../core/commands/image-commands.ts";
+import {
+  applyTableStructureOperation,
+} from "../core/commands/table-structure-commands.ts";
+import {
+  insertPageBreak as insertPageBreakInDocument,
+  insertTable as insertTableInDocument,
+} from "../core/commands/text-commands.ts";
+import {
+  createCanonicalDocumentId,
+  type SelectionSnapshot as InternalSelectionSnapshot,
+} from "../core/state/editor-state.ts";
 import {
   createDocumentRuntime,
   type DocumentRuntime,
@@ -39,7 +60,14 @@ import {
   hasValidPersistedSourcePackageDigest,
 } from "../io/source-package-provenance.ts";
 import { deriveCapabilities } from "../runtime/session-capabilities";
-import { TwProseMirrorSurface } from "../ui-tailwind/editor-surface/tw-prosemirror-surface";
+import {
+  createSearchExcerpt,
+  findSearchMatches,
+} from "../ui-tailwind/editor-surface/search-plugin";
+import {
+  TwProseMirrorSurface,
+  type TwProseMirrorSurfaceRef,
+} from "../ui-tailwind/editor-surface/tw-prosemirror-surface";
 import { TwReviewWorkspace } from "../ui-tailwind/tw-review-workspace";
 import type { ReviewRailTab } from "../ui-tailwind/review/tw-review-rail";
 import type { ViewMode } from "../ui-tailwind/toolbar/tw-toolbar";
@@ -103,6 +131,7 @@ type AccessibleRegionId = (typeof ACCESSIBLE_REGION_ORDER)[number];
 
 export function __createWordReviewEditorRefBridge(
   runtime: WordReviewEditorRuntime,
+  mountedSurface?: TwProseMirrorSurfaceRef | null,
 ): WordReviewEditorRef {
   return {
     focus: () => runtime.focus(),
@@ -125,6 +154,127 @@ export function __createWordReviewEditorRefBridge(
     getWarnings: () => runtime.getWarnings(),
     getComments: () => runtime.getRenderSnapshot().comments,
     getTrackedChanges: () => runtime.getRenderSnapshot().trackedChanges,
+    getFormattingState: () => getFormattingStateFromRenderSnapshot(runtime.getRenderSnapshot()),
+    toggleBold: () => {
+      applyRuntimeFormattingOperation(runtime, { type: "toggle", mark: "bold" });
+    },
+    toggleItalic: () => {
+      applyRuntimeFormattingOperation(runtime, { type: "toggle", mark: "italic" });
+    },
+    toggleUnderline: () => {
+      applyRuntimeFormattingOperation(runtime, { type: "toggle", mark: "underline" });
+    },
+    toggleStrikethrough: () => {
+      applyRuntimeFormattingOperation(runtime, {
+        type: "toggle",
+        mark: "strikethrough",
+      });
+    },
+    toggleSuperscript: () => {
+      applyRuntimeFormattingOperation(runtime, {
+        type: "toggle",
+        mark: "superscript",
+      });
+    },
+    toggleSubscript: () => {
+      applyRuntimeFormattingOperation(runtime, { type: "toggle", mark: "subscript" });
+    },
+    setFontFamily: (fontFamily) => {
+      applyRuntimeFormattingOperation(runtime, {
+        type: "set-font-family",
+        fontFamily,
+      });
+    },
+    setFontSize: (size) => {
+      applyRuntimeFormattingOperation(runtime, { type: "set-font-size", size });
+    },
+    setTextColor: (color) => {
+      applyRuntimeFormattingOperation(runtime, { type: "set-text-color", color });
+    },
+    setHighlightColor: (color) => {
+      applyRuntimeFormattingOperation(runtime, {
+        type: "set-highlight-color",
+        color,
+      });
+    },
+    setAlignment: (alignment) => {
+      applyRuntimeFormattingOperation(runtime, {
+        type: "set-alignment",
+        alignment,
+      });
+    },
+    indent: () => {
+      applyRuntimeFormattingOperation(runtime, { type: "indent" });
+    },
+    outdent: () => {
+      applyRuntimeFormattingOperation(runtime, { type: "outdent" });
+    },
+    insertPageBreak: () => {
+      applyRuntimeInsertPageBreak(runtime);
+    },
+    insertTable: (options) => {
+      applyRuntimeInsertTable(runtime, options);
+    },
+    insertImage: (options) => {
+      applyRuntimeInsertImage(runtime, options);
+    },
+    addRowBefore: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "add-row-before",
+      });
+    },
+    addRowAfter: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "add-row-after",
+      });
+    },
+    addColumnBefore: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "add-column-before",
+      });
+    },
+    addColumnAfter: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "add-column-after",
+      });
+    },
+    deleteRow: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "delete-row",
+      });
+    },
+    deleteColumn: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "delete-column",
+      });
+    },
+    deleteTable: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "delete-table",
+      });
+    },
+    mergeCells: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "merge-cells",
+      });
+    },
+    splitCell: () => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "split-cell",
+      });
+    },
+    setCellBackground: (color) => {
+      applyRuntimeTableStructureOperation(runtime, mountedSurface ?? null, {
+        type: "set-cell-background",
+        color,
+      });
+    },
+    search: (query, options) =>
+      mountedSurface?.search(query, options) ??
+      searchSnapshotSurface(runtime.getRenderSnapshot(), query, options),
+    clearSearch: () => {
+      mountedSurface?.clearSearch();
+    },
     scrollToRevision: (revisionId: string) => {
       const revision = runtime.getRenderSnapshot().trackedChanges.revisions.find(
         (r) => r.revisionId === revisionId,
@@ -312,6 +462,7 @@ export const WordReviewEditor = forwardRef<WordReviewEditorRef, WordReviewEditor
     const [showTrackedChanges, setShowTrackedChanges] = useState(false);
     const [activeRevisionId, setActiveRevisionId] = useState<string | undefined>();
     const runtimeRef = useRef<WordReviewEditorRuntime | null>(null);
+    const surfaceRef = useRef<TwProseMirrorSurfaceRef | null>(null);
     const shellRef = useRef<HTMLDivElement | null>(null);
     const lastAnnouncedErrorIdRef = useRef<string | null>(null);
     const autosaveTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
@@ -547,6 +698,146 @@ export const WordReviewEditor = forwardRef<WordReviewEditorRef, WordReviewEditor
         getWarnings: () => activeRuntime.getWarnings(),
         getComments: () => activeRuntime.getRenderSnapshot().comments,
         getTrackedChanges: () => activeRuntime.getRenderSnapshot().trackedChanges,
+        getFormattingState: () =>
+          getFormattingStateFromRenderSnapshot(activeRuntime.getRenderSnapshot()),
+        toggleBold: () => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "toggle",
+            mark: "bold",
+          });
+        },
+        toggleItalic: () => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "toggle",
+            mark: "italic",
+          });
+        },
+        toggleUnderline: () => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "toggle",
+            mark: "underline",
+          });
+        },
+        toggleStrikethrough: () => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "toggle",
+            mark: "strikethrough",
+          });
+        },
+        toggleSuperscript: () => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "toggle",
+            mark: "superscript",
+          });
+        },
+        toggleSubscript: () => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "toggle",
+            mark: "subscript",
+          });
+        },
+        setFontFamily: (fontFamily) => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "set-font-family",
+            fontFamily,
+          });
+        },
+        setFontSize: (size) => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "set-font-size",
+            size,
+          });
+        },
+        setTextColor: (color) => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "set-text-color",
+            color,
+          });
+        },
+        setHighlightColor: (color) => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "set-highlight-color",
+            color,
+          });
+        },
+        setAlignment: (alignment) => {
+          applyRuntimeFormattingOperation(activeRuntime, {
+            type: "set-alignment",
+            alignment,
+          });
+        },
+        indent: () => {
+          applyRuntimeFormattingOperation(activeRuntime, { type: "indent" });
+        },
+        outdent: () => {
+          applyRuntimeFormattingOperation(activeRuntime, { type: "outdent" });
+        },
+        insertPageBreak: () => {
+          applyRuntimeInsertPageBreak(activeRuntime);
+        },
+        insertTable: (options) => {
+          applyRuntimeInsertTable(activeRuntime, options);
+        },
+        insertImage: (options) => {
+          applyRuntimeInsertImage(activeRuntime, options);
+        },
+        addRowBefore: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "add-row-before",
+          });
+        },
+        addRowAfter: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "add-row-after",
+          });
+        },
+        addColumnBefore: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "add-column-before",
+          });
+        },
+        addColumnAfter: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "add-column-after",
+          });
+        },
+        deleteRow: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "delete-row",
+          });
+        },
+        deleteColumn: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "delete-column",
+          });
+        },
+        deleteTable: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "delete-table",
+          });
+        },
+        mergeCells: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "merge-cells",
+          });
+        },
+        splitCell: () => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "split-cell",
+          });
+        },
+        setCellBackground: (color) => {
+          applyRuntimeTableStructureOperation(activeRuntime, surfaceRef.current, {
+            type: "set-cell-background",
+            color,
+          });
+        },
+        search: (query, options) =>
+          surfaceRef.current?.search(query, options) ??
+          searchSnapshotSurface(activeRuntime.getRenderSnapshot(), query, options),
+        clearSearch: () => {
+          surfaceRef.current?.clearSearch();
+        },
         scrollToRevision: (revisionId: string) => {
           const revision = activeRuntime.getRenderSnapshot().trackedChanges.revisions.find(
             (r) => r.revisionId === revisionId,
@@ -667,12 +958,16 @@ export const WordReviewEditor = forwardRef<WordReviewEditorRef, WordReviewEditor
     }
 
     function addReviewComment(): void {
-      activeRuntime.addComment({
-        anchor: snapshot.selection.activeRange,
-        body: "New review comment",
-        authorId: currentUser.userId,
-      });
-      setActiveRailTab("comments");
+      try {
+        activeRuntime.addComment({
+          anchor: snapshot.selection.activeRange,
+          body: "New review comment",
+          authorId: currentUser.userId,
+        });
+        setActiveRailTab("comments");
+      } catch {
+        // Runtime already emitted a concrete export-safety error for invalid anchors.
+      }
     }
 
     function exportCurrentDocument(): void {
@@ -700,6 +995,10 @@ export const WordReviewEditor = forwardRef<WordReviewEditorRef, WordReviewEditor
       ? derivedCapabilities
       : { ...derivedCapabilities, reviewRailVisible: false };
     const diagnosticsModeMessage = getDiagnosticsModeMessage(loadError ?? snapshot.fatalError);
+    const addCommentDisabledReason =
+      !capabilities.canAddComment && !snapshot.selection.isCollapsed
+        ? "Select text within one paragraph to add a DOCX comment."
+        : undefined;
     const accessibilityInstructionsId = `${documentId}-accessibility-instructions`;
     const accessibilityStatusId = `${documentId}-accessibility-status`;
     const accessibilityAlertId = `${documentId}-accessibility-alert`;
@@ -856,12 +1155,14 @@ export const WordReviewEditor = forwardRef<WordReviewEditorRef, WordReviewEditor
           activeRevisionId={activeRevisionId}
           showTrackedChanges={showTrackedChanges}
           selectionPreview={selectionPreview}
+          addCommentDisabledReason={addCommentDisabledReason}
           onViewModeChange={setViewMode}
           onActiveRailTabChange={setActiveRailTab}
           onShowTrackedChangesChange={setShowTrackedChanges}
           {...reviewCallbacks}
           document={
             <TwProseMirrorSurface
+              ref={surfaceRef}
               currentUser={currentUser}
               snapshot={snapshot}
               reviewMode={reviewMode}
@@ -885,6 +1186,231 @@ export const WordReviewEditor = forwardRef<WordReviewEditorRef, WordReviewEditor
   },
 );
 
+function applyRuntimeFormattingOperation(
+  runtime: WordReviewEditorRuntime,
+  operation:
+    | { type: "toggle"; mark: "bold" | "italic" | "underline" | "strikethrough" | "superscript" | "subscript" }
+    | { type: "set-font-family"; fontFamily: string | null }
+    | { type: "set-font-size"; size: number | null }
+    | { type: "set-text-color"; color: string | null }
+    | { type: "set-highlight-color"; color: string | null }
+    | { type: "set-alignment"; alignment: FormattingAlignment }
+    | { type: "indent" }
+    | { type: "outdent" },
+): void {
+  const snapshot = runtime.getRenderSnapshot();
+  if (!snapshot.isReady || snapshot.readOnly || snapshot.fatalError) {
+    return;
+  }
+
+  const result = applyFormattingOperationToDocument(
+    runtime.getPersistedSnapshot().canonicalDocument,
+    snapshot,
+    operation,
+  );
+  if (!result.changed) {
+    return;
+  }
+
+  runtime.dispatch({
+    type: "document.replace",
+    document: result.document,
+    selection: toRuntimeSelectionSnapshot(result.selection),
+    origin: {
+      source: "api",
+      timestamp: new Date().toISOString(),
+    },
+  });
+}
+
+function applyRuntimeInsertPageBreak(runtime: WordReviewEditorRuntime): void {
+  const snapshot = runtime.getRenderSnapshot();
+  if (!canApplyRuntimeMutation(snapshot)) {
+    return;
+  }
+
+  const timestamp = new Date().toISOString();
+  const result = insertPageBreakInDocument(
+    runtime.getPersistedSnapshot().canonicalDocument,
+    toRuntimeSelectionSnapshot(snapshot.selection),
+    { timestamp },
+  );
+  dispatchRuntimeDocumentMutation(runtime, result, timestamp);
+}
+
+function applyRuntimeInsertTable(
+  runtime: WordReviewEditorRuntime,
+  options: InsertTableOptions,
+): void {
+  const snapshot = runtime.getRenderSnapshot();
+  if (!canApplyRuntimeMutation(snapshot)) {
+    return;
+  }
+
+  const timestamp = new Date().toISOString();
+  const result = insertTableInDocument(
+    runtime.getPersistedSnapshot().canonicalDocument,
+    toRuntimeSelectionSnapshot(snapshot.selection),
+    options,
+    { timestamp },
+  );
+  dispatchRuntimeDocumentMutation(runtime, result, timestamp);
+}
+
+function applyRuntimeInsertImage(
+  runtime: WordReviewEditorRuntime,
+  options: InsertImageOptions,
+): void {
+  const snapshot = runtime.getRenderSnapshot();
+  if (!canApplyRuntimeMutation(snapshot)) {
+    return;
+  }
+
+  const timestamp = new Date().toISOString();
+  try {
+    const result = insertImageInDocument(
+      runtime.getPersistedSnapshot().canonicalDocument,
+      toRuntimeSelectionSnapshot(snapshot.selection),
+      options.data,
+      options.mimeType,
+      options.width,
+      options.height,
+      {
+        timestamp,
+        altText: options.altText,
+      },
+    );
+    dispatchRuntimeDocumentMutation(runtime, {
+      changed: true,
+      document: result.document,
+      selection: result.selection,
+      mapping: result.mapping,
+    }, timestamp);
+  } catch {
+    return;
+  }
+}
+
+function applyRuntimeTableStructureOperation(
+  runtime: WordReviewEditorRuntime,
+  mountedSurface: TwProseMirrorSurfaceRef | null | undefined,
+  operation:
+    | { type: "add-row-before" }
+    | { type: "add-row-after" }
+    | { type: "add-column-before" }
+    | { type: "add-column-after" }
+    | { type: "delete-row" }
+    | { type: "delete-column" }
+    | { type: "delete-table" }
+    | { type: "merge-cells" }
+    | { type: "split-cell" }
+    | { type: "set-cell-background"; color: string },
+): void {
+  const snapshot = runtime.getRenderSnapshot();
+  if (!canApplyRuntimeMutation(snapshot)) {
+    return;
+  }
+
+  const timestamp = new Date().toISOString();
+  const result = applyTableStructureOperation(
+    runtime.getPersistedSnapshot().canonicalDocument,
+    snapshot,
+    mountedSurface?.getTableSelection() ?? null,
+    operation,
+  );
+  dispatchRuntimeDocumentMutation(runtime, result, timestamp);
+}
+
+function canApplyRuntimeMutation(snapshot: RuntimeRenderSnapshot): boolean {
+  return snapshot.isReady && !snapshot.readOnly && !snapshot.fatalError;
+}
+
+function dispatchRuntimeDocumentMutation(
+  runtime: WordReviewEditorRuntime,
+  result: {
+    changed: boolean;
+    document: PersistedEditorSnapshot["canonicalDocument"];
+    selection: InternalSelectionSnapshot;
+    mapping?: TransactionMapping;
+  },
+  timestamp: string,
+): void {
+  if (!result.changed) {
+    return;
+  }
+
+  runtime.dispatch({
+    type: "document.replace",
+    document: {
+      ...result.document,
+      updatedAt: timestamp,
+    },
+    mapping: result.mapping,
+    selection: result.selection,
+    origin: {
+      source: "api",
+      timestamp,
+    },
+  });
+}
+
+function searchSnapshotSurface(
+  snapshot: RuntimeRenderSnapshot,
+  query: string,
+  options: SearchOptions = {},
+): SearchResultSnapshot[] {
+  const normalizedQuery = query.trim();
+  if (!normalizedQuery || !snapshot.surface) {
+    return [];
+  }
+
+  const rawResults = findSearchMatches(
+    snapshot.surface.plainText,
+    normalizedQuery,
+    options,
+  ).slice(0, options.limit ?? Number.POSITIVE_INFINITY);
+  const activeResultIndex = getActiveSearchResultIndex(rawResults, snapshot.selection);
+
+  return rawResults.map((result, index) => ({
+    resultId: `search-result-${index}`,
+    anchor: {
+      kind: "range",
+      from: result.from,
+      to: result.to,
+      assoc: {
+        start: -1,
+        end: 1,
+      },
+    },
+    excerpt: createSearchExcerpt(
+      snapshot.surface?.plainText ?? "",
+      result.from,
+      result.to,
+    ),
+    isActive: index === activeResultIndex,
+  }));
+}
+
+function getActiveSearchResultIndex(
+  results: Array<{ from: number; to: number }>,
+  selection: PublicSelectionSnapshot,
+): number {
+  if (results.length === 0) {
+    return -1;
+  }
+
+  const selectionFrom = Math.min(selection.anchor, selection.head);
+  const selectionTo = Math.max(selection.anchor, selection.head);
+  const activeIndex = results.findIndex((result) => {
+    if (selectionFrom === selectionTo) {
+      return selectionFrom >= result.from && selectionFrom <= result.to;
+    }
+    return selectionFrom < result.to && selectionTo > result.from;
+  });
+
+  return activeIndex >= 0 ? activeIndex : 0;
+}
+
 function applyRegionAttributes(shell: HTMLElement): void {
   const toolbar = shell.querySelector<HTMLElement>("header");
   if (toolbar) {

package/src/ui-tailwind/chrome/tw-selection-toolbar.tsx

@@ -5,6 +5,8 @@ import { MessageSquare } from "lucide-react";
 export interface TwSelectionToolbarProps {
   selectionPreview: string;
   readOnly: boolean;
+  canAddComment?: boolean;
+  disabledReason?: string;
   onAddComment?: () => void;
 }
 
@@ -12,6 +14,10 @@ const focusRingClass =
   "focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-accent focus-visible:ring-offset-2 focus-visible:ring-offset-canvas";
 
 export function TwSelectionToolbar(props: TwSelectionToolbarProps) {
+  const addCommentDisabled = props.readOnly || props.canAddComment === false;
+  const tooltipLabel = addCommentDisabled
+    ? props.disabledReason ?? "Select text within one paragraph to add a DOCX comment"
+    : "Add comment";
   return (
     <div className="mb-6 inline-flex items-center gap-1 rounded-lg bg-canvas shadow-lg ring-1 ring-border p-1">
       <Tooltip.Root>
@@ -19,7 +25,7 @@ export function TwSelectionToolbar(props: TwSelectionToolbarProps) {
           <button
             type="button"
             aria-label="Comment"
-            disabled={props.readOnly}
+            disabled={addCommentDisabled}
             onClick={props.onAddComment}
             className={`inline-flex h-7 w-7 items-center justify-center rounded-md transition-colors text-accent hover:bg-accent-soft disabled:opacity-30 disabled:cursor-not-allowed ${focusRingClass}`}
           >
@@ -31,7 +37,7 @@ export function TwSelectionToolbar(props: TwSelectionToolbarProps) {
             className="rounded-md bg-primary px-2 py-1 text-xs text-white shadow-md z-50"
             sideOffset={6}
           >
-            Add comment
+            {tooltipLabel}
           </Tooltip.Content>
         </Tooltip.Portal>
       </Tooltip.Root>

package/src/ui-tailwind/editor-surface/pm-state-from-snapshot.ts

@@ -1,5 +1,5 @@
 import { Fragment, type Node as PMNode } from "prosemirror-model";
-import { EditorState, type Plugin, TextSelection } from "prosemirror-state";
+import { EditorState, type Plugin, Selection, TextSelection } from "prosemirror-state";
 
 import type {
   EditorSurfaceSnapshot,
@@ -43,12 +43,13 @@ export function createPMStateFromSnapshot(
     positionMap.pmDocSize - 1,
   );
 
-  let pmSelection: TextSelection;
+  let pmSelection: Selection;
   try {
-    pmSelection = TextSelection.create(doc, pmAnchor, pmHead);
+    pmSelection = TextSelection.between(doc.resolve(pmAnchor), doc.resolve(pmHead));
   } catch {
-    // If the position is invalid (e.g., inside an atom), fall back to start
-    pmSelection = TextSelection.create(doc, 1);
+    // If the mapped runtime selection is invalid or lands in a non-text block,
+    // let ProseMirror choose the nearest valid starting selection.
+    pmSelection = Selection.atStart(doc);
   }
 
   const state = EditorState.create({
@@ -258,6 +259,7 @@ function buildTable(
             rowspan: cell.rowspan,
             gridSpan: cell.gridSpan,
             verticalMerge: cell.verticalMerge,
+            backgroundColor: cell.backgroundColor ?? null,
           },
           Fragment.from(cellContent),
         ),