@milkdown/preset-gfm 6.5.2 → 6.5.3

package/src/table/plugin/cell-selection.ts

@@ -1,19 +1,22 @@
 /* Copyright 2021, Milkdown by Mirone. */
 
-import { Fragment, Node, ResolvedPos, Slice } from '@milkdown/prose/model';
+import type { Node, ResolvedPos } from '@milkdown/prose/model'
+import { Fragment, Slice } from '@milkdown/prose/model'
+import type {
+  EditorState,
+  Transaction,
+} from '@milkdown/prose/state'
 import {
-    EditorState,
-    NodeSelection,
-    Selection,
-    SelectionRange,
-    TextSelection,
-    Transaction,
-} from '@milkdown/prose/state';
-import { Mappable } from '@milkdown/prose/transform';
-import { Decoration, DecorationSet } from '@milkdown/prose/view';
-
-import { TableMap } from './table-map';
-import { inSameTable, pointsAtCell, removeColSpan, setAttr } from './util';
+  NodeSelection,
+  Selection,
+  SelectionRange,
+  TextSelection,
+} from '@milkdown/prose/state'
+import type { Mappable } from '@milkdown/prose/transform'
+import { Decoration, DecorationSet } from '@milkdown/prose/view'
+
+import { TableMap } from './table-map'
+import { inSameTable, pointsAtCell, removeColSpan, setAttr } from './util'
 
 // ::- A [`Selection`](http://prosemirror.net/docs/ref/#state.Selection)
 // subclass that represents a cell selection spanning part of a table.
@@ -21,332 +24,358 @@ import { inSameTable, pointsAtCell, removeColSpan, setAttr } from './util';
 // selects across cells, and will be drawn by giving selected cells a
 // `selectedCell` CSS class.
 export class CellSelection extends Selection {
-    // :: (ResolvedPos, ?ResolvedPos)
-    // A table selection is identified by its anchor and head cells. The
-    // positions given to this constructor should point _before_ two
-    // cells in the same table. They may be the same, to select a single
-    // cell.
-    constructor(public $anchorCell: ResolvedPos, public $headCell = $anchorCell) {
-        const table = $anchorCell.node(-1),
-            map = TableMap.get(table),
-            start = $anchorCell.start(-1);
-        const rect = map.rectBetween($anchorCell.pos - start, $headCell.pos - start);
-        const doc = $anchorCell.node(0);
-        const cells = map.cellsInRect(rect).filter((p) => p != $headCell.pos - start);
-        // Make the head cell the first range, so that it counts as the
-        // primary part of the selection
-        cells.unshift($headCell.pos - start);
-        const ranges = cells.map((pos) => {
-            const cell = table.nodeAt(pos) as Node,
-                from = pos + start + 1;
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            return new (SelectionRange as any)(doc.resolve(from), doc.resolve(from + cell.content.size));
-        });
-        super(ranges[0].$from, ranges[0].$to, ranges);
-        // :: ResolvedPos
-        // A resolved position pointing _in front of_ the anchor cell (the one
-        // that doesn't move when extending the selection).
-        this.$anchorCell = $anchorCell;
-        // :: ResolvedPos
-        // A resolved position pointing in front of the head cell (the one
-        // moves when extending the selection).
-        this.$headCell = $headCell;
+  // :: (ResolvedPos, ?ResolvedPos)
+  // A table selection is identified by its anchor and head cells. The
+  // positions given to this constructor should point _before_ two
+  // cells in the same table. They may be the same, to select a single
+  // cell.
+  constructor(public $anchorCell: ResolvedPos, public $headCell = $anchorCell) {
+    const table = $anchorCell.node(-1)
+    const map = TableMap.get(table)
+    const start = $anchorCell.start(-1)
+    const rect = map.rectBetween($anchorCell.pos - start, $headCell.pos - start)
+    const doc = $anchorCell.node(0)
+    const cells = map.cellsInRect(rect).filter(p => p != $headCell.pos - start)
+    // Make the head cell the first range, so that it counts as the
+    // primary part of the selection
+    cells.unshift($headCell.pos - start)
+    const ranges = cells.map((pos) => {
+      const cell = table.nodeAt(pos) as Node
+      const from = pos + start + 1
+
+      return new (SelectionRange as any)(doc.resolve(from), doc.resolve(from + cell.content.size))
+    })
+    super(ranges[0].$from, ranges[0].$to, ranges)
+    // :: ResolvedPos
+    // A resolved position pointing _in front of_ the anchor cell (the one
+    // that doesn't move when extending the selection).
+    this.$anchorCell = $anchorCell
+    // :: ResolvedPos
+    // A resolved position pointing in front of the head cell (the one
+    // moves when extending the selection).
+    this.$headCell = $headCell
+  }
+
+  map(doc: Node, mapping: Mappable): Selection {
+    const $anchorCell = doc.resolve(mapping.map(this.$anchorCell.pos))
+    const $headCell = doc.resolve(mapping.map(this.$headCell.pos))
+    if (pointsAtCell($anchorCell) && pointsAtCell($headCell) && inSameTable($anchorCell, $headCell)) {
+      const tableChanged = this.$anchorCell.node(-1) != $anchorCell.node(-1)
+      if (tableChanged && this.isRowSelection())
+        return CellSelection.rowSelection($anchorCell, $headCell)
+      else if (tableChanged && this.isColSelection())
+        return CellSelection.colSelection($anchorCell, $headCell)
+      else return new CellSelection($anchorCell, $headCell)
     }
+    return TextSelection.between($anchorCell, $headCell)
+  }
 
-    map(doc: Node, mapping: Mappable): Selection {
-        const $anchorCell = doc.resolve(mapping.map(this.$anchorCell.pos));
-        const $headCell = doc.resolve(mapping.map(this.$headCell.pos));
-        if (pointsAtCell($anchorCell) && pointsAtCell($headCell) && inSameTable($anchorCell, $headCell)) {
-            const tableChanged = this.$anchorCell.node(-1) != $anchorCell.node(-1);
-            if (tableChanged && this.isRowSelection()) return CellSelection.rowSelection($anchorCell, $headCell);
-            else if (tableChanged && this.isColSelection()) return CellSelection.colSelection($anchorCell, $headCell);
-            else return new CellSelection($anchorCell, $headCell);
+  // :: () → Slice
+  // Returns a rectangular slice of table rows containing the selected
+  // cells.
+  override content(): Slice {
+    const table = this.$anchorCell.node(-1)
+    const map = TableMap.get(table)
+    const start = this.$anchorCell.start(-1)
+    const rect = map.rectBetween(this.$anchorCell.pos - start, this.$headCell.pos - start)
+    const seen: Record<number, boolean> = {}
+    const rows = []
+    for (let row = rect.top; row < rect.bottom; row++) {
+      const rowContent = []
+      for (let index = row * map.width + rect.left, col = rect.left; col < rect.right; col++, index++) {
+        const pos = map.map[index] as number
+        if (!seen[pos]) {
+          seen[pos] = true
+          const cellRect = map.findCell(pos)
+          let cell = table.nodeAt(pos) as Node
+          const extraLeft = rect.left - cellRect.left
+          const extraRight = cellRect.right - rect.right
+          if (extraLeft > 0 || extraRight > 0) {
+            let attrs = cell.attrs
+            if (extraLeft > 0)
+              attrs = removeColSpan(attrs, 0, extraLeft)
+            if (extraRight > 0)
+              attrs = removeColSpan(attrs, attrs.colspan - extraRight, extraRight)
+            if (cellRect.left < rect.left)
+              cell = cell.type.createAndFill(attrs) as Node
+            else cell = cell.type.create(attrs, cell.content)
+          }
+          if (cellRect.top < rect.top || cellRect.bottom > rect.bottom) {
+            const attrs = setAttr(
+              cell.attrs,
+              'rowspan',
+              Math.min(cellRect.bottom, rect.bottom) - Math.max(cellRect.top, rect.top),
+            )
+            if (cellRect.top < rect.top)
+              cell = cell.type.createAndFill(attrs) as Node
+            else cell = cell.type.create(attrs, cell.content)
+          }
+          rowContent.push(cell)
         }
-        return TextSelection.between($anchorCell, $headCell);
+      }
+      rows.push(table.child(row).copy(Fragment.from(rowContent)))
     }
 
-    // :: () → Slice
-    // Returns a rectangular slice of table rows containing the selected
-    // cells.
-    override content(): Slice {
-        const table = this.$anchorCell.node(-1),
-            map = TableMap.get(table),
-            start = this.$anchorCell.start(-1);
-        const rect = map.rectBetween(this.$anchorCell.pos - start, this.$headCell.pos - start);
-        const seen: Record<number, boolean> = {},
-            rows = [];
-        for (let row = rect.top; row < rect.bottom; row++) {
-            const rowContent = [];
-            for (let index = row * map.width + rect.left, col = rect.left; col < rect.right; col++, index++) {
-                const pos = map.map[index] as number;
-                if (!seen[pos]) {
-                    seen[pos] = true;
-                    const cellRect = map.findCell(pos);
-                    let cell = table.nodeAt(pos) as Node;
-                    const extraLeft = rect.left - cellRect.left,
-                        extraRight = cellRect.right - rect.right;
-                    if (extraLeft > 0 || extraRight > 0) {
-                        let attrs = cell.attrs;
-                        if (extraLeft > 0) attrs = removeColSpan(attrs, 0, extraLeft);
-                        if (extraRight > 0) attrs = removeColSpan(attrs, attrs['colspan'] - extraRight, extraRight);
-                        if (cellRect.left < rect.left) cell = cell.type.createAndFill(attrs) as Node;
-                        else cell = cell.type.create(attrs, cell.content);
-                    }
-                    if (cellRect.top < rect.top || cellRect.bottom > rect.bottom) {
-                        const attrs = setAttr(
-                            cell.attrs,
-                            'rowspan',
-                            Math.min(cellRect.bottom, rect.bottom) - Math.max(cellRect.top, rect.top),
-                        );
-                        if (cellRect.top < rect.top) cell = cell.type.createAndFill(attrs) as Node;
-                        else cell = cell.type.create(attrs, cell.content);
-                    }
-                    rowContent.push(cell);
-                }
-            }
-            rows.push(table.child(row).copy(Fragment.from(rowContent)));
-        }
+    const fragment = this.isColSelection() && this.isRowSelection() ? table : rows
+    return new Slice(Fragment.from(fragment), 1, 1)
+  }
 
-        const fragment = this.isColSelection() && this.isRowSelection() ? table : rows;
-        return new Slice(Fragment.from(fragment), 1, 1);
+  override replace(tr: Transaction, content = Slice.empty) {
+    const mapFrom = tr.steps.length
+    const ranges = this.ranges
+    for (let i = 0; i < ranges.length; i++) {
+      const { $from, $to } = ranges[i] as SelectionRange
+      const mapping = tr.mapping.slice(mapFrom)
+      tr.replace(mapping.map($from.pos), mapping.map($to.pos), i ? Slice.empty : content)
     }
+    const sel = Selection.findFrom(tr.doc.resolve(tr.mapping.slice(mapFrom).map(this.to)), -1)
+    if (sel)
+      tr.setSelection(sel)
+  }
 
-    override replace(tr: Transaction, content = Slice.empty) {
-        const mapFrom = tr.steps.length,
-            ranges = this.ranges;
-        for (let i = 0; i < ranges.length; i++) {
-            const { $from, $to } = ranges[i] as SelectionRange,
-                mapping = tr.mapping.slice(mapFrom);
-            tr.replace(mapping.map($from.pos), mapping.map($to.pos), i ? Slice.empty : content);
-        }
-        const sel = Selection.findFrom(tr.doc.resolve(tr.mapping.slice(mapFrom).map(this.to)), -1);
-        if (sel) tr.setSelection(sel);
-    }
+  override replaceWith(tr: Transaction, node: Node) {
+    this.replace(tr, new Slice(Fragment.from(node), 0, 0))
+  }
 
-    override replaceWith(tr: Transaction, node: Node) {
-        this.replace(tr, new Slice(Fragment.from(node), 0, 0));
-    }
+  forEachCell(f: (node: Node, index: number) => void) {
+    const table = this.$anchorCell.node(-1)
+    const map = TableMap.get(table)
+    const start = this.$anchorCell.start(-1)
+    const cells = map.cellsInRect(map.rectBetween(this.$anchorCell.pos - start, this.$headCell.pos - start))
+    for (let i = 0; i < cells.length; i++)
+      f(table.nodeAt(cells[i] as number) as Node, start + (cells[i] as number))
+  }
 
-    forEachCell(f: (node: Node, index: number) => void) {
-        const table = this.$anchorCell.node(-1),
-            map = TableMap.get(table),
-            start = this.$anchorCell.start(-1);
-        const cells = map.cellsInRect(map.rectBetween(this.$anchorCell.pos - start, this.$headCell.pos - start));
-        for (let i = 0; i < cells.length; i++)
-            f(table.nodeAt(cells[i] as number) as Node, start + (cells[i] as number));
-    }
+  // :: () → bool
+  // True if this selection goes all the way from the top to the
+  // bottom of the table.
+  isColSelection() {
+    const anchorTop = this.$anchorCell.index(-1)
+    const headTop = this.$headCell.index(-1)
+    if (Math.min(anchorTop, headTop) > 0)
+      return false
+    const anchorBot = anchorTop + (this.$anchorCell.nodeAfter as Node).attrs.rowspan
+    const headBot = headTop + (this.$headCell.nodeAfter as Node).attrs.rowspan
+    return Math.max(anchorBot, headBot) == this.$headCell.node(-1).childCount
+  }
 
-    // :: () → bool
-    // True if this selection goes all the way from the top to the
-    // bottom of the table.
-    isColSelection() {
-        const anchorTop = this.$anchorCell.index(-1),
-            headTop = this.$headCell.index(-1);
-        if (Math.min(anchorTop, headTop) > 0) return false;
-        const anchorBot = anchorTop + (this.$anchorCell.nodeAfter as Node).attrs['rowspan'],
-            headBot = headTop + (this.$headCell.nodeAfter as Node).attrs['rowspan'];
-        return Math.max(anchorBot, headBot) == this.$headCell.node(-1).childCount;
+  // :: (ResolvedPos, ?ResolvedPos) → CellSelection
+  // Returns the smallest column selection that covers the given anchor
+  // and head cell.
+  static colSelection($anchorCell: ResolvedPos, $headCell = $anchorCell) {
+    const map = TableMap.get($anchorCell.node(-1))
+    const start = $anchorCell.start(-1)
+    const anchorRect = map.findCell($anchorCell.pos - start)
+    const headRect = map.findCell($headCell.pos - start)
+    const doc = $anchorCell.node(0)
+    if (anchorRect.top <= headRect.top) {
+      if (anchorRect.top > 0) {
+        const left = map.map[anchorRect.left] as number
+        $anchorCell = doc.resolve(start + left)
+      }
+      if (headRect.bottom < map.height) {
+        const pos = map.map[map.width * (map.height - 1) + headRect.right - 1] as number
+        $headCell = doc.resolve(start + pos)
+      }
     }
-
-    // :: (ResolvedPos, ?ResolvedPos) → CellSelection
-    // Returns the smallest column selection that covers the given anchor
-    // and head cell.
-    static colSelection($anchorCell: ResolvedPos, $headCell = $anchorCell) {
-        const map = TableMap.get($anchorCell.node(-1)),
-            start = $anchorCell.start(-1);
-        const anchorRect = map.findCell($anchorCell.pos - start),
-            headRect = map.findCell($headCell.pos - start);
-        const doc = $anchorCell.node(0);
-        if (anchorRect.top <= headRect.top) {
-            if (anchorRect.top > 0) {
-                const left = map.map[anchorRect.left] as number;
-                $anchorCell = doc.resolve(start + left);
-            }
-            if (headRect.bottom < map.height) {
-                const pos = map.map[map.width * (map.height - 1) + headRect.right - 1] as number;
-                $headCell = doc.resolve(start + pos);
-            }
-        } else {
-            if (headRect.top > 0) {
-                const left = map.map[anchorRect.left] as number;
-                $headCell = doc.resolve(start + left);
-            }
-            if (anchorRect.bottom < map.height) {
-                const pos = map.map[map.width * (map.height - 1) + anchorRect.right - 1] as number;
-                $anchorCell = doc.resolve(start + pos);
-            }
-        }
-        return new CellSelection($anchorCell, $headCell);
+    else {
+      if (headRect.top > 0) {
+        const left = map.map[anchorRect.left] as number
+        $headCell = doc.resolve(start + left)
+      }
+      if (anchorRect.bottom < map.height) {
+        const pos = map.map[map.width * (map.height - 1) + anchorRect.right - 1] as number
+        $anchorCell = doc.resolve(start + pos)
+      }
     }
+    return new CellSelection($anchorCell, $headCell)
+  }
 
-    // :: () → bool
-    // True if this selection goes all the way from the left to the
-    // right of the table.
-    isRowSelection() {
-        const map = TableMap.get(this.$anchorCell.node(-1)),
-            start = this.$anchorCell.start(-1);
-        const anchorLeft = map.colCount(this.$anchorCell.pos - start),
-            headLeft = map.colCount(this.$headCell.pos - start);
-        if (Math.min(anchorLeft, headLeft) > 0) return false;
-        const anchorRight = anchorLeft + (this.$anchorCell.nodeAfter as Node).attrs['colspan'],
-            headRight = headLeft + (this.$headCell.nodeAfter as Node).attrs['colspan'];
-        return Math.max(anchorRight, headRight) == map.width;
-    }
+  // :: () → bool
+  // True if this selection goes all the way from the left to the
+  // right of the table.
+  isRowSelection() {
+    const map = TableMap.get(this.$anchorCell.node(-1))
+    const start = this.$anchorCell.start(-1)
+    const anchorLeft = map.colCount(this.$anchorCell.pos - start)
+    const headLeft = map.colCount(this.$headCell.pos - start)
+    if (Math.min(anchorLeft, headLeft) > 0)
+      return false
+    const anchorRight = anchorLeft + (this.$anchorCell.nodeAfter as Node).attrs.colspan
+    const headRight = headLeft + (this.$headCell.nodeAfter as Node).attrs.colspan
+    return Math.max(anchorRight, headRight) == map.width
+  }
 
-    eq(other: Selection): boolean {
-        return (
-            other instanceof CellSelection &&
-            other.$anchorCell.pos == this.$anchorCell.pos &&
-            other.$headCell.pos == this.$headCell.pos
-        );
-    }
+  eq(other: Selection): boolean {
+    return (
+      other instanceof CellSelection
+            && other.$anchorCell.pos == this.$anchorCell.pos
+            && other.$headCell.pos == this.$headCell.pos
+    )
+  }
 
-    // :: (ResolvedPos, ?ResolvedPos) → CellSelection
-    // Returns the smallest row selection that covers the given anchor
-    // and head cell.
-    static rowSelection($anchorCell: ResolvedPos, $headCell = $anchorCell) {
-        const map = TableMap.get($anchorCell.node(-1)),
-            start = $anchorCell.start(-1);
-        const anchorRect = map.findCell($anchorCell.pos - start),
-            headRect = map.findCell($headCell.pos - start);
-        const doc = $anchorCell.node(0);
-        if (anchorRect.left <= headRect.left) {
-            if (anchorRect.left > 0) {
-                const pos = map.map[anchorRect.top * map.width] as number;
-                $anchorCell = doc.resolve(start + pos);
-            }
-            if (headRect.right < map.width) {
-                const pos = map.map[map.width * (headRect.top + 1) - 1] as number;
-                $headCell = doc.resolve(start + pos);
-            }
-        } else {
-            if (headRect.left > 0) {
-                const pos = map.map[headRect.top * map.width] as number;
-                $headCell = doc.resolve(start + pos);
-            }
-            if (anchorRect.right < map.width) {
-                const pos = map.map[map.width * (anchorRect.top + 1) - 1] as number;
-
-                $anchorCell = doc.resolve(start + pos);
-            }
-        }
-        return new CellSelection($anchorCell, $headCell);
+  // :: (ResolvedPos, ?ResolvedPos) → CellSelection
+  // Returns the smallest row selection that covers the given anchor
+  // and head cell.
+  static rowSelection($anchorCell: ResolvedPos, $headCell = $anchorCell) {
+    const map = TableMap.get($anchorCell.node(-1))
+    const start = $anchorCell.start(-1)
+    const anchorRect = map.findCell($anchorCell.pos - start)
+    const headRect = map.findCell($headCell.pos - start)
+    const doc = $anchorCell.node(0)
+    if (anchorRect.left <= headRect.left) {
+      if (anchorRect.left > 0) {
+        const pos = map.map[anchorRect.top * map.width] as number
+        $anchorCell = doc.resolve(start + pos)
+      }
+      if (headRect.right < map.width) {
+        const pos = map.map[map.width * (headRect.top + 1) - 1] as number
+        $headCell = doc.resolve(start + pos)
+      }
     }
+    else {
+      if (headRect.left > 0) {
+        const pos = map.map[headRect.top * map.width] as number
+        $headCell = doc.resolve(start + pos)
+      }
+      if (anchorRect.right < map.width) {
+        const pos = map.map[map.width * (anchorRect.top + 1) - 1] as number
 
-    toJSON() {
-        return {
-            type: 'cell',
-            anchor: this.$anchorCell.pos,
-            head: this.$headCell.pos,
-        };
+        $anchorCell = doc.resolve(start + pos)
+      }
     }
+    return new CellSelection($anchorCell, $headCell)
+  }
 
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    static override fromJSON(doc: Node, json: any) {
-        return new CellSelection(doc.resolve(json.anchor), doc.resolve(json.head));
+  toJSON() {
+    return {
+      type: 'cell',
+      anchor: this.$anchorCell.pos,
+      head: this.$headCell.pos,
     }
+  }
 
-    // :: (Node, number, ?number) → CellSelection
-    static create(doc: Node, anchorCell: number, headCell = anchorCell) {
-        return new CellSelection(doc.resolve(anchorCell), doc.resolve(headCell));
-    }
+  static override fromJSON(doc: Node, json: any) {
+    return new CellSelection(doc.resolve(json.anchor), doc.resolve(json.head))
+  }
 
-    override getBookmark() {
-        return new CellBookmark(this.$anchorCell.pos, this.$headCell.pos);
-    }
+  // :: (Node, number, ?number) → CellSelection
+  static create(doc: Node, anchorCell: number, headCell = anchorCell) {
+    return new CellSelection(doc.resolve(anchorCell), doc.resolve(headCell))
+  }
+
+  override getBookmark() {
+    return new CellBookmark(this.$anchorCell.pos, this.$headCell.pos)
+  }
 }
 
-CellSelection.prototype.visible = false;
+CellSelection.prototype.visible = false
 
-Selection.jsonID('cell', CellSelection);
+Selection.jsonID('cell', CellSelection)
 
 class CellBookmark {
-    constructor(public anchor: number, public head: number) {
-        this.anchor = anchor;
-        this.head = head;
-    }
-    map(mapping: Mappable) {
-        return new CellBookmark(mapping.map(this.anchor), mapping.map(this.head));
-    }
-    resolve(doc: Node): Selection {
-        const $anchorCell = doc.resolve(this.anchor),
-            $headCell = doc.resolve(this.head);
-        if (
-            $anchorCell.parent.type.spec['tableRole'] == 'row' &&
-            $headCell.parent.type.spec['tableRole'] == 'row' &&
-            $anchorCell.index() < $anchorCell.parent.childCount &&
-            $headCell.index() < $headCell.parent.childCount &&
-            inSameTable($anchorCell, $headCell)
-        )
-            return new CellSelection($anchorCell, $headCell);
-        else return Selection.near($headCell, 1);
-    }
+  constructor(public anchor: number, public head: number) {
+    this.anchor = anchor
+    this.head = head
+  }
+
+  map(mapping: Mappable) {
+    return new CellBookmark(mapping.map(this.anchor), mapping.map(this.head))
+  }
+
+  resolve(doc: Node): Selection {
+    const $anchorCell = doc.resolve(this.anchor)
+    const $headCell = doc.resolve(this.head)
+    if (
+      $anchorCell.parent.type.spec.tableRole == 'row'
+            && $headCell.parent.type.spec.tableRole == 'row'
+            && $anchorCell.index() < $anchorCell.parent.childCount
+            && $headCell.index() < $headCell.parent.childCount
+            && inSameTable($anchorCell, $headCell)
+    )
+      return new CellSelection($anchorCell, $headCell)
+    else return Selection.near($headCell, 1)
+  }
 }
 
 export function drawCellSelection(state: EditorState) {
-    if (!(state.selection instanceof CellSelection)) return null;
-    const cells: Decoration[] = [];
-    state.selection.forEachCell((node, pos) => {
-        cells.push(Decoration.node(pos, pos + node.nodeSize, { class: 'selectedCell' }));
-    });
-    return DecorationSet.create(state.doc, cells);
+  if (!(state.selection instanceof CellSelection))
+    return null
+  const cells: Decoration[] = []
+  state.selection.forEachCell((node, pos) => {
+    cells.push(Decoration.node(pos, pos + node.nodeSize, { class: 'selectedCell' }))
+  })
+  return DecorationSet.create(state.doc, cells)
 }
 
 function isCellBoundarySelection({ $from, $to }: Selection) {
-    if ($from.pos == $to.pos || $from.pos < $from.pos - 6) return false; // Cheap elimination
-    let afterFrom = $from.pos,
-        beforeTo = $to.pos,
-        depth = $from.depth;
-    for (; depth >= 0; depth--, afterFrom++) if ($from.after(depth + 1) < $from.end(depth)) break;
-    for (let d = $to.depth; d >= 0; d--, beforeTo--) if ($to.before(d + 1) > $to.start(d)) break;
-    return afterFrom == beforeTo && /row|table/.test($from.node(depth).type.spec['tableRole']);
+  if ($from.pos == $to.pos || $from.pos < $from.pos - 6)
+    return false // Cheap elimination
+  let afterFrom = $from.pos
+  let beforeTo = $to.pos
+  let depth = $from.depth
+  for (; depth >= 0; depth--, afterFrom++) {
+    if ($from.after(depth + 1) < $from.end(depth))
+      break
+  }
+  for (let d = $to.depth; d >= 0; d--, beforeTo--) {
+    if ($to.before(d + 1) > $to.start(d))
+      break
+  }
+  return afterFrom == beforeTo && /row|table/.test($from.node(depth).type.spec.tableRole)
 }
 
 function isTextSelectionAcrossCells({ $from, $to }: Selection) {
-    let fromCellBoundaryNode;
-    let toCellBoundaryNode;
-
-    for (let i = $from.depth; i > 0; i--) {
-        const node = $from.node(i);
-        if (node.type.spec['tableRole'] === 'cell' || node.type.spec['tableRole'] === 'header_cell') {
-            fromCellBoundaryNode = node;
-            break;
-        }
+  let fromCellBoundaryNode
+  let toCellBoundaryNode
+
+  for (let i = $from.depth; i > 0; i--) {
+    const node = $from.node(i)
+    if (node.type.spec.tableRole === 'cell' || node.type.spec.tableRole === 'header_cell') {
+      fromCellBoundaryNode = node
+      break
     }
+  }
 
-    for (let i = $to.depth; i > 0; i--) {
-        const node = $to.node(i);
-        if (node.type.spec['tableRole'] === 'cell' || node.type.spec['tableRole'] === 'header_cell') {
-            toCellBoundaryNode = node;
-            break;
-        }
+  for (let i = $to.depth; i > 0; i--) {
+    const node = $to.node(i)
+    if (node.type.spec.tableRole === 'cell' || node.type.spec.tableRole === 'header_cell') {
+      toCellBoundaryNode = node
+      break
     }
+  }
 
-    return fromCellBoundaryNode !== toCellBoundaryNode && $to.parentOffset === 0;
+  return fromCellBoundaryNode !== toCellBoundaryNode && $to.parentOffset === 0
 }
 
 export function normalizeSelection(state: EditorState, tr: Transaction | undefined, allowTableNodeSelection: boolean) {
-    const sel = (tr || state).selection,
-        doc = (tr || state).doc;
-    let normalize, role;
-    if (sel instanceof NodeSelection && (role = sel.node.type.spec['tableRole'])) {
-        if (role == 'cell' || role == 'header_cell') {
-            normalize = CellSelection.create(doc, sel.from);
-        } else if (role == 'row') {
-            const $cell = doc.resolve(sel.from + 1);
-            normalize = CellSelection.rowSelection($cell, $cell);
-        } else if (!allowTableNodeSelection) {
-            const map = TableMap.get(sel.node),
-                start = sel.from + 1;
-            const pos = map.map[map.width * map.height - 1] as number;
-            const lastCell = start + pos;
-            normalize = CellSelection.create(doc, start + 1, lastCell);
-        }
-    } else if (sel instanceof TextSelection && isCellBoundarySelection(sel)) {
-        normalize = TextSelection.create(doc, sel.from);
-    } else if (sel instanceof TextSelection && isTextSelectionAcrossCells(sel)) {
-        normalize = TextSelection.create(doc, sel.$from.start(), sel.$from.end());
+  const sel = (tr || state).selection
+  const doc = (tr || state).doc
+  let normalize, role
+  // eslint-disable-next-line no-cond-assign
+  if (sel instanceof NodeSelection && (role = sel.node.type.spec.tableRole)) {
+    if (role == 'cell' || role == 'header_cell') {
+      normalize = CellSelection.create(doc, sel.from)
+    }
+    else if (role == 'row') {
+      const $cell = doc.resolve(sel.from + 1)
+      normalize = CellSelection.rowSelection($cell, $cell)
+    }
+    else if (!allowTableNodeSelection) {
+      const map = TableMap.get(sel.node)
+      const start = sel.from + 1
+      const pos = map.map[map.width * map.height - 1] as number
+      const lastCell = start + pos
+      normalize = CellSelection.create(doc, start + 1, lastCell)
     }
-    if (normalize) (tr || (tr = state.tr)).setSelection(normalize);
-    return tr;
+  }
+  else if (sel instanceof TextSelection && isCellBoundarySelection(sel)) {
+    normalize = TextSelection.create(doc, sel.from)
+  }
+  else if (sel instanceof TextSelection && isTextSelectionAcrossCells(sel)) {
+    normalize = TextSelection.create(doc, sel.$from.start(), sel.$from.end())
+  }
+  if (normalize)
+    (tr || (tr = state.tr)).setSelection(normalize)
+  return tr
 }