@atlaskit/editor-plugin-show-diff 26.0.0 → 26.0.2

package/dist/es2019/pm-plugins/calculateDiff/calculateDiffDecorations.js

@@ -14,6 +14,7 @@ import { createDocMarginAnchorWidget } from '../decorations/createAnchorDecorati
 import { createBlockChangedDecoration } from '../decorations/createBlockChangedDecoration';
 import { createInlineChangedDecoration } from '../decorations/createInlineChangedDecoration';
 import { createNodeChangedDecorationWidget } from '../decorations/createNodeChangedDecorationWidget';
+import { createReplacedCellDecorations, getReplacedCells } from '../decorations/createReplacedCellDecorations';
 import { extractDiffDescriptors } from '../decorations/decorationKeys';
 import { extractContributorTags } from '../decorations/extractContributorTags';
 import { createDeletedLineBreakDecoration } from '../decorations/utils/createDeletedLineBreakWidget';
@@ -548,7 +549,11 @@ const calculateDiffDecorationsInner = ({
     // suppressed in the clean view. `hideDeletedDiffs` is load-bearing: when false the reviewer
     // asked to compare against the original, and the deleted side is the only place the
     // previous formatting is visible.
-    const isMarkOnly = shouldHideMarkOnlyDeletions && hideDeletedDiffs && change.deleted.length > 0 && isMarkOnlyChange({
+    //
+    // Not on an inverted diff: there the widget's side is the suggested content, which the clean
+    // view keeps. Suppressing it hides the suggestion along with the displayed side it replaces,
+    // e.g. a cell whose only change is its background, as the signature ignores attrs.
+    const isMarkOnly = shouldHideMarkOnlyDeletions && hideDeletedDiffs && !(isInverted && fg('platform_editor_ai_show_diff_patch_2')) && change.deleted.length > 0 && isMarkOnlyChange({
       change,
       originalDoc,
       newDoc: tr.doc
@@ -625,7 +630,35 @@ const calculateDiffDecorationsInner = ({
         hideDeletedDiffs,
         isInverted
       });
-      if (isSmartNodeLevel) {
+
+      // A whole-cell replacement keeps its cells in place, so it takes its own decorations
+      // instead of the generic inline and block ones.
+      const replacedCells = fg('platform_editor_ai_show_diff_patch_2') ? getReplacedCells({
+        slice: originalDoc.slice(change.fromA, change.toA),
+        doc: tr.doc,
+        fromB: change.fromB,
+        toB: change.toB
+      }) : [];
+      if (replacedCells.length > 0) {
+        decorations.push(...createReplacedCellDecorations({
+          replacedCells,
+          change,
+          doc: tr.doc,
+          attributionKey,
+          colorScheme: changeColorScheme,
+          hasDeletedWidget: willRenderDeletedWidget,
+          hideAddedDiffsUnderline,
+          isActive,
+          isDeletedWidgetBelow,
+          isInserted,
+          leftAnchorId,
+          reveal,
+          shouldHideDeleted,
+          showContributorTags,
+          showIndicators,
+          tagMountContext
+        }));
+      } else if (isSmartNodeLevel) {
         // For a large inserted table, skip the O(cells) inline decorations inside the
         // table (the cell overlays added below provide the highlight without the
         // expensive content re-render); leaf blocks outside the table still get theirs.
@@ -674,7 +707,7 @@ const calculateDiffDecorationsInner = ({
           tagMountContext
         }));
       }
-      decorations.push(...calculateNodesForBlockDecoration({
+      decorations.push(...(replacedCells.length > 0 ? [] : calculateNodesForBlockDecoration({
         // A deleted widget placed below anchors at the change's end, inside the very block
         // this would let the decoration claim.
         allowTrailingCloseTokens: !isDeletedWidgetBelow,
@@ -694,7 +727,7 @@ const calculateDiffDecorationsInner = ({
         intl,
         coarseTableCellsOnly: useCoarseTableDecoration,
         leftAnchorId
-      }));
+      })));
     }
     if (change.deleted.length > 0 && !isMarkOnly && !hasNoDeletedContent) {
       const shouldHideDeleted = shouldHideDeletedSide({

package/dist/es2019/pm-plugins/calculateDiff/diffBySteps.js

@@ -3,6 +3,7 @@ import { Mark } from '@atlaskit/editor-prosemirror/model';
 import { Mapping, ReplaceStep } from '@atlaskit/editor-prosemirror/transform';
 import { fg } from '@atlaskit/platform-feature-flags/fg';
 import { getAttributionKey } from '../decorations/colorSchemes/attributions';
+import { isCellAttrChanged } from '../decorations/utils/cellAttrs';
 import { buildCharsByOffset } from '../utils/charsByOffset';
 import { attrAwareTokenEncoder } from './attrAwareTokenEncoder';
 import { optimizeChanges } from './optimizeChanges';
@@ -17,6 +18,14 @@ const WHITESPACE_REGEX = /\s/u;
  * excluded: `groupDeletedColumnChanges` relies on a replaced table being one whole-table change.
  */
 const isGranularIsolatingNode = node => !!node.type.spec.isolating && node.type.spec.tableRole !== 'table';
+const CELL_TYPE_NAMES = ['tableCell', 'tableHeader'];
+
+/**
+ * Whether a table cell's replacement changes one of its diffable attributes, such as its background.
+ * Such a cell is diffed whole: it renders as a cell style plus a copy of its row, which already shows
+ * the cell's new content, so inline marks inside it would only repeat that.
+ */
+const isCellAttrChange = (replacedNode, replacingNode) => CELL_TYPE_NAMES.includes(replacedNode.type.name) && isCellAttrChanged(replacedNode, replacingNode);
 const mapPosition = (mapping, pos) => mapping.map(pos);
 
 /**
@@ -206,7 +215,8 @@ const mergeOverlappingByNewDocRange = changes => {
  * - The replaced slice has only one child
  * - The replacing slice has only one child
  * - Both children are the same textblock type, or (behind `platform_editor_ai_show_diff_patch_2`)
- *   the same isolating, non-table node type whose text content differs
+ *   the same isolating, non-table node type whose text content differs (for a table cell, only when
+ *   its diffable attributes are unchanged)
  * - The replaced slice and replacing slice have the same text content
  * - The replaced slice and replacing slice have the same child marks (if text content is equal)
  */
@@ -244,7 +254,7 @@ const shouldCheckGranularDiff = (step, before, from, to) => {
   // `hasSameChildMarks` only compares an isolating node's direct (block) children, so it cannot
   // vouch for one with unchanged text. Keep the existing whole-node fallback for that case.
   if (isIsolatingNode) {
-    return !isTextContentEqual;
+    return !isTextContentEqual && !(replacingNode && isCellAttrChange(replacedNode, replacingNode));
   }
   return !isTextContentEqual || isTextContentEqual && hasSameChildMarks(replacedNode, replacingNode);
 };

package/dist/es2019/pm-plugins/decorations/createNodeChangedDecorationWidget.js

@@ -10,6 +10,7 @@ import { clampAnchorPosIntoCell, createAnchorNameSpan, createLeftAnchorWidget }
 import { createChangedRowDecorationWidgets } from './createChangedRowDecorationWidgets';
 import { createContributorTagHost, resolveHoistedCodeBlockAnchor, unmountContributorTag } from './createContributorTagWidget';
 import { getAtomicInlineChangedAttrs } from './createInlineChangedDecoration';
+import { createReplacedCellWidgets } from './createReplacedCellDecorations';
 import { AnchorTypeKey, buildContributorTagDecorationSpec, buildDiffDecorationSpec, buildAnchorDecorationKey, scrollMarginTopValue } from './decorationKeys';
 import { absorbFirstChildMarginReset } from './utils/absorbFirstChildMarginReset';
 import { createDeletedLineBreakDecoration } from './utils/createDeletedLineBreakWidget';
@@ -214,6 +215,19 @@ export const createNodeChangedDecorationWidget = ({
     isInserted
   });
   if (tableDiffMode.kind === 'cells') {
+    if (fg('platform_editor_ai_show_diff_patch_2')) {
+      return createReplacedCellWidgets({
+        slice,
+        newDoc,
+        change,
+        nodeViewSerializer,
+        colorScheme,
+        intl,
+        isActive,
+        isInserted,
+        leftAnchorId
+      });
+    }
     return createTableCellContentWidgets({
       slice,
       newDoc,

package/dist/es2019/pm-plugins/decorations/createReplacedCellDecorations.js

@@ -0,0 +1,338 @@
+import { getDarkModeLCHColor } from '@atlaskit/adf-schema/get-dark-mode-lch-color';
+import { isRgb } from '@atlaskit/adf-schema/is-rgb';
+import { rgbToHex } from '@atlaskit/adf-schema/rgb-to-hex';
+import { getCellDomAttrs } from '@atlaskit/adf-schema/tableNodes';
+import { convertToInlineCss } from '@atlaskit/editor-common/css-helper';
+import { areNodesEqualIgnoreAttrs } from '@atlaskit/editor-common/utils/document';
+import { Decoration } from '@atlaskit/editor-prosemirror/view';
+import { createBlockChangedDecoration } from './createBlockChangedDecoration';
+import { createInlineChangedDecoration } from './createInlineChangedDecoration';
+import { buildDiffDecorationSpec } from './decorationKeys';
+import { isCellAttrChanged, isCellBackgroundChanged, normalizeCellBackground } from './utils/cellAttrs';
+import { injectInnerWrapper, wrapBlockNodeView } from './utils/wrapBlockNodeView';
+const CELL_TYPE_NAMES = ['tableCell', 'tableHeader'];
+/**
+ * Pairs each cell of a whole-cell replacement with the live cell it replaces, or `[]` when the slice
+ * is not a run of whole cells matching the live ones. Whole-cell replacement preserves cell count and
+ * order, so the nth live cell in the range matches the nth cell in the slice.
+ */
+export const getReplacedCells = ({
+  slice,
+  doc,
+  fromB,
+  toB
+}) => {
+  const otherCells = slice.content.content;
+  if (slice.openStart !== 0 || slice.openEnd !== 0 || otherCells.length === 0 || !otherCells.every(node => CELL_TYPE_NAMES.includes(node.type.name))) {
+    return [];
+  }
+  const liveCells = [];
+  doc.nodesBetween(fromB, toB, (node, pos) => {
+    if (CELL_TYPE_NAMES.includes(node.type.name)) {
+      liveCells.push({
+        node,
+        pos
+      });
+      return false;
+    }
+    return true;
+  });
+  if (liveCells.length !== otherCells.length) {
+    return [];
+  }
+  return otherCells.map((otherCell, index) => ({
+    otherCell,
+    liveCell: liveCells[index].node,
+    liveCellPos: liveCells[index].pos
+  }));
+};
+
+/**
+ * Whether two cells hold the same content, ignoring the cells' own attributes and `localId`s, which
+ * a suggestion can regenerate. Such a replacement only changed the cell (e.g. its background), so
+ * there is no content to diff.
+ */
+const isCellContentEqual = (left, right) => {
+  if (left.childCount !== right.childCount) {
+    return false;
+  }
+  for (let index = 0; index < left.childCount; index++) {
+    if (!areNodesEqualIgnoreAttrs(left.child(index), right.child(index), ['localId'])) {
+      return false;
+    }
+  }
+  return true;
+};
+
+/**
+ * Whether the cell itself changed — its type, or an attribute that represents a change such as its
+ * background or span — as opposed to only its content.
+ */
+const isCellChanged = (left, right) => left.type !== right.type || isCellAttrChanged(left, right);
+
+/**
+ * A custom (non-palette) cell colour as the schema renders it: inverted in dark mode so text stays
+ * readable, except for design tokens, which are already themed.
+ */
+const getCustomCellColor = background => {
+  const color = isRgb(background) && rgbToHex(background) || background;
+  if (color.startsWith('var(') || document.documentElement.getAttribute('data-color-mode') !== 'dark') {
+    return background;
+  }
+  return getDarkModeLCHColor(color);
+};
+
+/**
+ * The colour `cell` renders with. A palette colour is the editor's own themed variable for its colour
+ * name (e.g. for dark mode), the one `td[colorname]` cells use; a custom colour is the schema's own
+ * treatment of it; no colour is the editor's default.
+ */
+const getCellBackgroundColor = cell => {
+  const background = normalizeCellBackground(cell.attrs.background);
+  if (!background) {
+    return cell.type.name === 'tableHeader' ? "var(--ds-background-accent-gray-subtlest, #F0F1F2)" : "var(--ds-surface, #FFFFFF)";
+  }
+  const {
+    colorname
+  } = getCellDomAttrs(cell);
+  return colorname ? `var(--ak-editor-table-cell-background-${colorname.toLowerCase().split(' ').join('-')})` : getCustomCellColor(background);
+};
+
+/**
+ * A layer painting `color` over the cell's own background and under its content, so the cell reads
+ * as having the new background without any of the cell's own attributes being overridden: removing
+ * a decoration removes the attributes it set rather than restoring the ones they replaced.
+ */
+const createCellBackgroundLayer = color => {
+  const layer = document.createElement('div');
+  layer.setAttribute('aria-hidden', 'true');
+  layer.contentEditable = 'false';
+  layer.dataset.testid = 'show-diff-new-cell-background';
+  layer.setAttribute('style', convertToInlineCss({
+    position: 'absolute',
+    inset: 0,
+    zIndex: -1,
+    borderRadius: 'inherit',
+    pointerEvents: 'none',
+    backgroundColor: color
+  }));
+  return layer;
+};
+
+/** The other side's cell content, as blocks, for the top of the live cell. */
+const createCellContentWidget = ({
+  cell,
+  pos,
+  nodeViewSerializer,
+  colorScheme,
+  intl,
+  isActive,
+  isInserted,
+  leftAnchorId
+}) => {
+  const dom = document.createElement('div');
+  cell.content.forEach(block => {
+    const nodeView = nodeViewSerializer.tryCreateNodeView(block, 0, undefined, isInserted, colorScheme);
+    if (nodeView) {
+      wrapBlockNodeView({
+        dom,
+        nodeView,
+        targetNode: block,
+        colorScheme,
+        intl,
+        isActive,
+        isInserted
+      });
+      return;
+    }
+    const serialized = nodeViewSerializer.serializeNode(block);
+    if (serialized instanceof HTMLElement) {
+      dom.append(injectInnerWrapper({
+        node: serialized,
+        colorScheme,
+        isActive,
+        isInserted
+      }));
+    }
+  });
+  if (dom.childNodes.length === 0) {
+    return undefined;
+  }
+  dom.setAttribute('data-testid', isInserted ? 'show-diff-changed-decoration' : 'show-diff-deleted-decoration');
+
+  // Before the cell's own blocks, as a block of its own, so it never lands inside a paragraph the
+  // displayed side may hide.
+  return Decoration.widget(pos + 1, dom, {
+    ...buildDiffDecorationSpec({
+      colorScheme,
+      decorationType: 'widget',
+      diffId: crypto.randomUUID(),
+      leftAnchorId,
+      isActive,
+      isInserted,
+      side: -1
+    }),
+    marks: []
+  });
+};
+
+/**
+ * The other side of a whole-cell replacement, drawn inside the live cells so the table keeps its
+ * shape: each live cell takes the other cell's background when it differs, and its content when that
+ * differs. A cell whose content did not change has nothing to diff.
+ */
+export const createReplacedCellWidgets = ({
+  slice,
+  newDoc,
+  change,
+  nodeViewSerializer,
+  colorScheme,
+  intl,
+  isActive,
+  isInserted,
+  leftAnchorId
+}) => getReplacedCells({
+  slice,
+  doc: newDoc,
+  fromB: change.fromB,
+  toB: change.toB
+}).flatMap(({
+  otherCell,
+  liveCell,
+  liveCellPos
+}) => {
+  const decorations = [];
+  if (isInserted && isCellBackgroundChanged(otherCell, liveCell)) {
+    const cellEnd = liveCellPos + liveCell.nodeSize;
+    decorations.push(
+    // The cell becomes the layer's stacking context, so the layer sits above the cell's own
+    // background rather than behind it. Through `z-index` at the cell overlay's own level, not
+    // `isolation`: an isolated cell would trap its overlay below later cells, which then cover
+    // the overlay's outline where it overlaps their border. Properties the cell never sets, so
+    // removing them takes nothing of its own.
+    Decoration.node(liveCellPos, cellEnd, {
+      style: convertToInlineCss({
+        position: 'relative',
+        // The highest z-index a cell diff overlay takes (the rounded-corner variant).
+        zIndex: 2
+      })
+    }, {
+      key: `show-diff-new-cell-background-host-${liveCellPos}`
+    }),
+    // At the cell's end, beside its diff overlay, so the cell's first block stays its first child.
+    Decoration.widget(cellEnd - 1, createCellBackgroundLayer(getCellBackgroundColor(otherCell)), {
+      key: `show-diff-new-cell-background-${liveCellPos}`,
+      side: -1,
+      marks: [],
+      ignoreSelection: true
+    }));
+  }
+  if (!isCellContentEqual(otherCell, liveCell)) {
+    const contentWidget = createCellContentWidget({
+      cell: otherCell,
+      pos: liveCellPos,
+      nodeViewSerializer,
+      colorScheme,
+      intl,
+      isActive,
+      isInserted,
+      leftAnchorId
+    });
+    if (contentWidget) {
+      decorations.push(contentWidget);
+    }
+  }
+  return decorations;
+});
+
+/**
+ * The displayed side of a whole-cell replacement. A table cell is never removed on its own, only
+ * replaced in place, so a live cell is never hidden — hiding it would leave the row a cell short. It
+ * takes the changed-cell treatment only when the cell itself changed; its content is diffed, and
+ * hidden in the clean view, only when that changed. A cell whose content stayed the same keeps
+ * showing it.
+ */
+export const createReplacedCellDecorations = ({
+  replacedCells,
+  change,
+  doc,
+  attributionKey,
+  colorScheme,
+  hasDeletedWidget,
+  hideAddedDiffsUnderline,
+  isActive,
+  isDeletedWidgetBelow,
+  isInserted,
+  leftAnchorId,
+  reveal,
+  shouldHideDeleted,
+  showContributorTags,
+  showIndicators,
+  tagMountContext
+}) => replacedCells.flatMap(({
+  otherCell,
+  liveCell,
+  liveCellPos
+}) => {
+  const cellEnd = liveCellPos + liveCell.nodeSize;
+  const decorations = isCellChanged(otherCell, liveCell) ? createBlockChangedDecoration({
+    attributionKey,
+    change: {
+      from: liveCellPos,
+      to: cellEnd,
+      name: liveCell.type.name
+    },
+    colorScheme,
+    doc,
+    isActive,
+    // The cell stays either way, so it reads as changed, not removed.
+    isInserted: true,
+    leftAnchorId,
+    showContributorTags,
+    showIndicators,
+    tagMountContext
+  }) : [];
+  if (isCellContentEqual(otherCell, liveCell)) {
+    return decorations;
+  }
+  decorations.push(...createInlineChangedDecoration({
+    attributionKey,
+    change: {
+      ...change,
+      fromB: liveCellPos,
+      toB: cellEnd
+    },
+    colorScheme,
+    doc,
+    hasDeletedWidget,
+    hideAddedDiffsUnderline,
+    isActive,
+    isDeletedWidgetBelow,
+    isInserted,
+    leftAnchorId,
+    reveal,
+    shouldHideDeleted,
+    showContributorTags,
+    showIndicators,
+    tagMountContext
+  }));
+  if (shouldHideDeleted) {
+    liveCell.forEach((block, offset) => {
+      const blockPos = liveCellPos + 1 + offset;
+      decorations.push(...createBlockChangedDecoration({
+        change: {
+          from: blockPos,
+          to: blockPos + block.nodeSize,
+          name: block.type.name
+        },
+        colorScheme,
+        doc,
+        isActive,
+        isInserted,
+        leftAnchorId,
+        shouldHideDeleted
+      }));
+    });
+  }
+  return decorations;
+});

package/dist/es2019/pm-plugins/decorations/utils/cellAttrs.js

@@ -0,0 +1,19 @@
+import { getDiffableAttrNames } from './diffableAttrs';
+
+/**
+ * A cell background as it renders: absent, empty and `"null"` — the string agent payloads send for
+ * no background — all mean the cell has none.
+ */
+export const normalizeCellBackground = background => typeof background === 'string' && background !== '' && background !== 'null' ? background : null;
+
+/** Whether two cells render different backgrounds. */
+export const isCellBackgroundChanged = (left, right) => normalizeCellBackground(left.attrs.background) !== normalizeCellBackground(right.attrs.background);
+
+/**
+ * Whether two cells differ by an attribute that represents a change, such as their background or
+ * span, as opposed to only their content.
+ */
+export const isCellAttrChanged = (left, right) => {
+  var _getDiffableAttrNames;
+  return ((_getDiffableAttrNames = getDiffableAttrNames(left.type.name, left.attrs)) !== null && _getDiffableAttrNames !== void 0 ? _getDiffableAttrNames : []).some(name => name === 'background' ? isCellBackgroundChanged(left, right) : left.attrs[name] !== right.attrs[name]);
+};

package/dist/es2019/pm-plugins/scrollToDiff.js

@@ -2,6 +2,7 @@ import { tableMarginTop } from '@atlaskit/editor-common/styles/table';
 import { getStickyHeaderHeight } from '@atlaskit/editor-common/table/get-sticky-header-height';
 import { fg } from '@atlaskit/platform-feature-flags/fg';
 import { isDiffDecoration, SCROLL_TARGET_MARGIN_CSS_PROPERTY } from './decorations/decorationKeys';
+import { getWidgetElement } from './decorations/utils/getWidgetElement';
 
 /**
  * Extra space above the scrolled-to element so it does not sit flush under the
@@ -19,21 +20,21 @@ function updateScrollTargetMargin(view, targetPos, api) {
   const tableHeaderHeight = isLimitedModeEnabled ? undefined : getStickyHeaderHeight(view, targetPos);
   view.dom.style.setProperty(SCROLL_TARGET_MARGIN_CSS_PROPERTY, `${tableHeaderHeight ? tableHeaderHeight + tableMarginTop : 0}px`);
 }
-
-/**
- * Returns the resolved HTMLElement for a given DOM node, walking up to the
- * parent element if the node itself is not an Element (e.g. a text node).
- */
 function scrollToSelection(node) {
-  const element = node instanceof Element ? node : (node === null || node === void 0 ? void 0 : node.parentElement) instanceof Element ? node.parentElement : null;
+  // ProseMirror may return a text node; scroll its parent element instead.
+  const element = node instanceof Element ? node : node === null || node === void 0 ? void 0 : node.parentElement;
   if (!(element instanceof HTMLElement)) {
     return;
   }
+  const scrollOptions = {
+    behavior: 'smooth',
+    block: 'start'
+  };
+  if (fg('platform_editor_ai_suggestion_table_scroll_fix')) {
+    scrollOptions.inline = 'center';
+  }
   if (fg('platform_editor_ai_show_diff_patch_1')) {
-    element.scrollIntoView({
-      behavior: 'smooth',
-      block: 'start'
-    });
+    element.scrollIntoView(scrollOptions);
     return;
   }
 
@@ -41,14 +42,23 @@ function scrollToSelection(node) {
   const previousScrollMarginTop = element.style.scrollMarginTop;
   element.style.scrollMarginTop = `${SCROLL_TOP_MARGIN_PX}px`;
   try {
-    element.scrollIntoView({
-      behavior: 'smooth',
-      block: 'start'
-    });
+    element.scrollIntoView(scrollOptions);
   } finally {
     element.style.scrollMarginTop = previousScrollMarginTop;
   }
 }
+function resolveScrollTarget(view, target) {
+  var _view$domAtPos;
+  if (isDiffDecoration(target) && target.spec.decorationType === 'widget') {
+    return getWidgetElement(target);
+  }
+  if (isDiffDecoration(target) && fg('platform_editor_ai_show_diff_patch_2')) {
+    var _Array$from$find;
+    return (_Array$from$find = Array.from(view.dom.querySelectorAll('[data-diff-id]')).find(element => element.dataset.diffId === target.spec.diffId)) !== null && _Array$from$find !== void 0 ? _Array$from$find : view.nodeDOM(target.from);
+  }
+  const targetNode = view.nodeDOM(target.from);
+  return targetNode instanceof Element ? targetNode : (_view$domAtPos = view.domAtPos(target.from)) === null || _view$domAtPos === void 0 ? void 0 : _view$domAtPos.node;
+}
 
 /**
  * Schedules scrolling to the decoration at the given index after the next frame. Defaults to the
@@ -66,28 +76,27 @@ export const scrollToDecoration = (view, decorations, activeIndex = 0, api) => {
   // deleted-content widget rather than at the group's `from` position — see `scrollTarget` in
   // `getScrollableDecorations`. Only set under `platform_editor_ai_show_diff_patch_1`.
   const target = isDiffDecoration(decoration) && decoration.spec.scrollTarget || decoration;
-  let rafId = requestAnimationFrame(() => {
-    rafId = null;
+  let rafId = null;
+  let cancelled = false;
+  const scrollToTarget = () => {
+    if (view.isDestroyed) {
+      return;
+    }
     if (fg('platform_editor_ai_show_diff_patch_1')) {
       updateScrollTargetMargin(view, target.from, api);
     }
-    if (isDiffDecoration(target) && target.spec.decorationType === 'widget') {
-      var _target$type;
-      // @ts-expect-error - decoration.type is not typed public API
-      const widgetDom = target === null || target === void 0 ? void 0 : (_target$type = target.type) === null || _target$type === void 0 ? void 0 : _target$type.toDOM;
-      scrollToSelection(widgetDom);
-    } else if (isDiffDecoration(target) && fg('platform_editor_ai_show_diff_patch_2')) {
-      var _Array$from$find;
-      const targetNode = (_Array$from$find = Array.from(view.dom.querySelectorAll('[data-diff-id]')).find(element => element.dataset.diffId === target.spec.diffId)) !== null && _Array$from$find !== void 0 ? _Array$from$find : view.nodeDOM(target === null || target === void 0 ? void 0 : target.from);
-      scrollToSelection(targetNode);
-    } else {
-      var _view$domAtPos;
-      const targetNode = view.nodeDOM(target === null || target === void 0 ? void 0 : target.from);
-      const node = targetNode instanceof Element ? targetNode : (_view$domAtPos = view.domAtPos(target === null || target === void 0 ? void 0 : target.from)) === null || _view$domAtPos === void 0 ? void 0 : _view$domAtPos.node;
-      scrollToSelection(node);
+    scrollToSelection(resolveScrollTarget(view, target));
+  };
+
+  // Called from the plugin's post-DOM update hook; defer measurement until layout can settle.
+  rafId = requestAnimationFrame(() => {
+    rafId = null;
+    if (!cancelled) {
+      scrollToTarget();
     }
   });
   return () => {
+    cancelled = true;
     if (rafId !== null) {
       cancelAnimationFrame(rafId);
       rafId = null;

package/dist/esm/pm-plugins/calculateDiff/calculateDiffDecorations.js

@@ -22,6 +22,7 @@ import { createDocMarginAnchorWidget } from '../decorations/createAnchorDecorati
 import { createBlockChangedDecoration } from '../decorations/createBlockChangedDecoration';
 import { createInlineChangedDecoration } from '../decorations/createInlineChangedDecoration';
 import { createNodeChangedDecorationWidget } from '../decorations/createNodeChangedDecorationWidget';
+import { createReplacedCellDecorations, getReplacedCells } from '../decorations/createReplacedCellDecorations';
 import { extractDiffDescriptors } from '../decorations/decorationKeys';
 import { extractContributorTags } from '../decorations/extractContributorTags';
 import { createDeletedLineBreakDecoration } from '../decorations/utils/createDeletedLineBreakWidget';
@@ -596,7 +597,11 @@ var calculateDiffDecorationsInner = function calculateDiffDecorationsInner(_ref8
     // suppressed in the clean view. `hideDeletedDiffs` is load-bearing: when false the reviewer
     // asked to compare against the original, and the deleted side is the only place the
     // previous formatting is visible.
-    var isMarkOnly = shouldHideMarkOnlyDeletions && hideDeletedDiffs && change.deleted.length > 0 && isMarkOnlyChange({
+    //
+    // Not on an inverted diff: there the widget's side is the suggested content, which the clean
+    // view keeps. Suppressing it hides the suggestion along with the displayed side it replaces,
+    // e.g. a cell whose only change is its background, as the signature ignores attrs.
+    var isMarkOnly = shouldHideMarkOnlyDeletions && hideDeletedDiffs && !(isInverted && fg('platform_editor_ai_show_diff_patch_2')) && change.deleted.length > 0 && isMarkOnlyChange({
       change: change,
       originalDoc: originalDoc,
       newDoc: tr.doc
@@ -682,7 +687,35 @@ var calculateDiffDecorationsInner = function calculateDiffDecorationsInner(_ref8
         hideDeletedDiffs: hideDeletedDiffs,
         isInverted: isInverted
       });
-      if (isSmartNodeLevel) {
+
+      // A whole-cell replacement keeps its cells in place, so it takes its own decorations
+      // instead of the generic inline and block ones.
+      var replacedCells = fg('platform_editor_ai_show_diff_patch_2') ? getReplacedCells({
+        slice: originalDoc.slice(change.fromA, change.toA),
+        doc: tr.doc,
+        fromB: change.fromB,
+        toB: change.toB
+      }) : [];
+      if (replacedCells.length > 0) {
+        decorations.push.apply(decorations, _toConsumableArray(createReplacedCellDecorations({
+          replacedCells: replacedCells,
+          change: change,
+          doc: tr.doc,
+          attributionKey: attributionKey,
+          colorScheme: changeColorScheme,
+          hasDeletedWidget: willRenderDeletedWidget,
+          hideAddedDiffsUnderline: hideAddedDiffsUnderline,
+          isActive: isActive,
+          isDeletedWidgetBelow: isDeletedWidgetBelow,
+          isInserted: isInserted,
+          leftAnchorId: leftAnchorId,
+          reveal: reveal,
+          shouldHideDeleted: shouldHideDeleted,
+          showContributorTags: showContributorTags,
+          showIndicators: showIndicators,
+          tagMountContext: tagMountContext
+        })));
+      } else if (isSmartNodeLevel) {
         // For a large inserted table, skip the O(cells) inline decorations inside the
         // table (the cell overlays added below provide the highlight without the
         // expensive content re-render); leaf blocks outside the table still get theirs.
@@ -742,7 +775,7 @@ var calculateDiffDecorationsInner = function calculateDiffDecorationsInner(_ref8
           tagMountContext: tagMountContext
         })));
       }
-      decorations.push.apply(decorations, _toConsumableArray(calculateNodesForBlockDecoration({
+      decorations.push.apply(decorations, _toConsumableArray(replacedCells.length > 0 ? [] : calculateNodesForBlockDecoration({
         // A deleted widget placed below anchors at the change's end, inside the very block
         // this would let the decoration claim.
         allowTrailingCloseTokens: !isDeletedWidgetBelow,