@atlaskit/editor-core 172.1.2 → 172.2.0

package/dist/cjs/plugins/table/commands/selection.js

@@ -13,15 +13,7 @@ var _cellSelection = require("@atlaskit/editor-tables/cell-selection");
 
 var _utils = require("@atlaskit/editor-tables/utils");
 
-var _gapCursorSelection = require("../../selection/gap-cursor-selection");
-
-var _commands = require("../../selection/commands");
-
-var _types = require("../../selection/types");
-
-var _pluginFactory = require("../../selection/plugin-factory");
-
-var _selection = require("../../../utils/selection");
+var _selection = require("@atlaskit/editor-common/selection");
 
 var TableSelectionDirection;
 exports.TableSelectionDirection = TableSelectionDirection;
@@ -31,205 +23,225 @@ exports.TableSelectionDirection = TableSelectionDirection;
   TableSelectionDirection["BottomToTop"] = "BottomToTop";
 })(TableSelectionDirection || (exports.TableSelectionDirection = TableSelectionDirection = {}));
 
-var arrowLeftFromTable = function arrowLeftFromTable(state, dispatch) {
-  var selection = state.selection;
-
-  if (selection instanceof _cellSelection.CellSelection) {
-    return arrowLeftFromCellSelection(selection)(state, dispatch);
-  } else if (selection instanceof _gapCursorSelection.GapCursorSelection) {
-    return arrowLeftFromGapCursor(selection)(state, dispatch);
-  } else if (selection instanceof _prosemirrorState.TextSelection) {
-    return arrowLeftFromText(selection)(state, dispatch);
-  }
+var arrowLeftFromTable = function arrowLeftFromTable(editorSelectionAPI) {
+  return function () {
+    return function (state, dispatch) {
+      var selection = state.selection;
+
+      if (selection instanceof _cellSelection.CellSelection) {
+        return arrowLeftFromCellSelection(editorSelectionAPI)(selection)(state, dispatch);
+      } else if (selection instanceof _selection.GapCursorSelection) {
+        return arrowLeftFromGapCursor(editorSelectionAPI)(selection)(state, dispatch);
+      } else if (selection instanceof _prosemirrorState.TextSelection) {
+        return arrowLeftFromText(editorSelectionAPI)(selection)(state, dispatch);
+      }
 
-  return false;
+      return false;
+    };
+  };
 };
 
 exports.arrowLeftFromTable = arrowLeftFromTable;
 
-var arrowRightFromTable = function arrowRightFromTable(state, dispatch) {
-  var selection = state.selection;
-
-  if (selection instanceof _cellSelection.CellSelection) {
-    return arrowRightFromCellSelection(selection)(state, dispatch);
-  } else if (selection instanceof _gapCursorSelection.GapCursorSelection) {
-    return arrowRightFromGapCursor(selection)(state, dispatch);
-  } else if (selection instanceof _prosemirrorState.TextSelection) {
-    return arrowRightFromText(selection)(state, dispatch);
-  }
+var arrowRightFromTable = function arrowRightFromTable(editorSelectionAPI) {
+  return function () {
+    return function (state, dispatch) {
+      var selection = state.selection;
+
+      if (selection instanceof _cellSelection.CellSelection) {
+        return arrowRightFromCellSelection(editorSelectionAPI)(selection)(state, dispatch);
+      } else if (selection instanceof _selection.GapCursorSelection) {
+        return arrowRightFromGapCursor(editorSelectionAPI)(selection)(state, dispatch);
+      } else if (selection instanceof _prosemirrorState.TextSelection) {
+        return arrowRightFromText(editorSelectionAPI)(selection)(state, dispatch);
+      }
 
-  return false;
+      return false;
+    };
+  };
 };
 
 exports.arrowRightFromTable = arrowRightFromTable;
 
-var arrowLeftFromCellSelection = function arrowLeftFromCellSelection(selection) {
-  return function (state, dispatch) {
-    if ((0, _utils.isTableSelected)(state.selection)) {
-      var _getSelectionPluginSt = (0, _pluginFactory.getPluginState)(state),
-          selectionRelativeToNode = _getSelectionPluginSt.selectionRelativeToNode;
-
-      if (selectionRelativeToNode === _types.RelativeSelectionPos.Start) {
-        // we have full table cell selection and want to set gap cursor selection before table
-        return setGapCursorBeforeTable()(state, dispatch);
-      } else if (selectionRelativeToNode === _types.RelativeSelectionPos.End) {
-        // we have full table cell selection and want to set selection at end of last cell
-        return setSelectionAtEndOfLastCell(selection)(state, dispatch);
-      } else if (selectionRelativeToNode === undefined) {
-        // we have full table cell selection and want to set selection at start of first cell
-        return setSelectionAtStartOfFirstCell(selection, _types.RelativeSelectionPos.Before)(state, dispatch);
+var arrowLeftFromCellSelection = function arrowLeftFromCellSelection(editorSelectionAPI) {
+  return function (selection) {
+    return function (state, dispatch) {
+      if ((0, _utils.isTableSelected)(state.selection) && editorSelectionAPI) {
+        var _editorSelectionAPI$g = editorSelectionAPI.getSelectionPluginState(state),
+            selectionRelativeToNode = _editorSelectionAPI$g.selectionRelativeToNode;
+
+        if (selectionRelativeToNode === _selection.RelativeSelectionPos.Start) {
+          // we have full table cell selection and want to set gap cursor selection before table
+          return setGapCursorBeforeTable(editorSelectionAPI)()(state, dispatch);
+        } else if (selectionRelativeToNode === _selection.RelativeSelectionPos.End) {
+          // we have full table cell selection and want to set selection at end of last cell
+          return setSelectionAtEndOfLastCell(editorSelectionAPI)(selection)(state, dispatch);
+        } else if (selectionRelativeToNode === undefined) {
+          // we have full table cell selection and want to set selection at start of first cell
+          return setSelectionAtStartOfFirstCell(editorSelectionAPI)(selection, _selection.RelativeSelectionPos.Before)(state, dispatch);
+        }
       }
-    }
 
-    return false;
+      return false;
+    };
   };
 };
 
-var arrowRightFromCellSelection = function arrowRightFromCellSelection(selection) {
-  return function (state, dispatch) {
-    if ((0, _utils.isTableSelected)(state.selection)) {
-      var _getSelectionPluginSt2 = (0, _pluginFactory.getPluginState)(state),
-          selectionRelativeToNode = _getSelectionPluginSt2.selectionRelativeToNode;
-
-      if (selectionRelativeToNode === _types.RelativeSelectionPos.Start) {
-        // we have full table cell selection and want to set selection at start of first cell
-        return setSelectionAtStartOfFirstCell(selection)(state, dispatch);
-      } else if (selectionRelativeToNode === _types.RelativeSelectionPos.End || selectionRelativeToNode === undefined) {
-        // we have full table cell selection and want to set gap cursor selection after table
-        return setGapCursorAfterTable()(state, dispatch);
+var arrowRightFromCellSelection = function arrowRightFromCellSelection(editorSelectionAPI) {
+  return function (selection) {
+    return function (state, dispatch) {
+      if ((0, _utils.isTableSelected)(state.selection) && editorSelectionAPI) {
+        var _editorSelectionAPI$g2 = editorSelectionAPI.getSelectionPluginState(state),
+            selectionRelativeToNode = _editorSelectionAPI$g2.selectionRelativeToNode;
+
+        if (selectionRelativeToNode === _selection.RelativeSelectionPos.Start) {
+          // we have full table cell selection and want to set selection at start of first cell
+          return setSelectionAtStartOfFirstCell(editorSelectionAPI)(selection)(state, dispatch);
+        } else if (selectionRelativeToNode === _selection.RelativeSelectionPos.End || selectionRelativeToNode === undefined) {
+          // we have full table cell selection and want to set gap cursor selection after table
+          return setGapCursorAfterTable(editorSelectionAPI)()(state, dispatch);
+        }
       }
-    }
 
-    return false;
+      return false;
+    };
   };
 };
 
-var arrowLeftFromGapCursor = function arrowLeftFromGapCursor(selection) {
-  return function (state, dispatch) {
-    var doc = state.doc;
-    var $from = selection.$from,
-        from = selection.from,
-        side = selection.side;
-
-    if (side === _gapCursorSelection.Side.RIGHT) {
-      if ($from.nodeBefore && $from.nodeBefore.type.name === 'table') {
-        // we have a gap cursor after a table node and want to set a full table cell selection
-        return selectFullTable({
-          node: $from.nodeBefore,
-          startPos: doc.resolve(from - 1).start($from.depth + 1),
-          dir: TableSelectionDirection.TopToBottom
-        })(state, dispatch);
+var arrowLeftFromGapCursor = function arrowLeftFromGapCursor(editorSelectionAPI) {
+  return function (selection) {
+    return function (state, dispatch) {
+      var doc = state.doc;
+      var $from = selection.$from,
+          from = selection.from,
+          side = selection.side;
+
+      if (side === _selection.Side.RIGHT) {
+        if ($from.nodeBefore && $from.nodeBefore.type.name === 'table') {
+          // we have a gap cursor after a table node and want to set a full table cell selection
+          return selectFullTable(editorSelectionAPI)({
+            node: $from.nodeBefore,
+            startPos: doc.resolve(from - 1).start($from.depth + 1),
+            dir: TableSelectionDirection.TopToBottom
+          })(state, dispatch);
+        }
+      } else if (side === _selection.Side.LEFT) {
+        var table = (0, _utils.findTable)(selection);
+
+        if (table && isSelectionAtStartOfTable($from, selection) && editorSelectionAPI) {
+          var _editorSelectionAPI$g3 = editorSelectionAPI.getSelectionPluginState(state),
+              selectionRelativeToNode = _editorSelectionAPI$g3.selectionRelativeToNode;
+
+          if (selectionRelativeToNode === _selection.RelativeSelectionPos.Before) {
+            // we have a gap cursor at start of first table cell and want to set a gap cursor selection before table
+            return setGapCursorBeforeTable(editorSelectionAPI)()(state, dispatch);
+          } else {
+            // we have a gap cursor at start of first table cell and want to set a full table cell selection
+            return selectFullTable(editorSelectionAPI)({
+              node: table.node,
+              startPos: table.start,
+              dir: TableSelectionDirection.BottomToTop
+            })(state, dispatch);
+          }
+        }
       }
-    } else if (side === _gapCursorSelection.Side.LEFT) {
-      var table = (0, _utils.findTable)(selection);
 
-      if (table && isSelectionAtStartOfTable($from, selection)) {
-        var _getSelectionPluginSt3 = (0, _pluginFactory.getPluginState)(state),
-            selectionRelativeToNode = _getSelectionPluginSt3.selectionRelativeToNode;
+      return false;
+    };
+  };
+};
+
+var arrowRightFromGapCursor = function arrowRightFromGapCursor(editorSelectionAPI) {
+  return function (selection) {
+    return function (state, dispatch) {
+      var $from = selection.$from,
+          from = selection.from,
+          $to = selection.$to,
+          side = selection.side;
+
+      if (side === _selection.Side.LEFT) {
+        if ($from.nodeAfter && $from.nodeAfter.type.name === 'table') {
+          // we have a gap cursor before a table node and want to set a full table cell selection
+          return selectFullTable(editorSelectionAPI)({
+            node: $from.nodeAfter,
+            startPos: from + 1,
+            dir: TableSelectionDirection.BottomToTop
+          })(state, dispatch);
+        }
+      } else if (side === _selection.Side.RIGHT) {
+        var table = (0, _utils.findTable)(selection);
 
-        if (selectionRelativeToNode === _types.RelativeSelectionPos.Before) {
-          // we have a gap cursor at start of first table cell and want to set a gap cursor selection before table
-          return setGapCursorBeforeTable()(state, dispatch);
-        } else {
-          // we have a gap cursor at start of first table cell and want to set a full table cell selection
-          return selectFullTable({
+        if (table && isSelectionAtEndOfTable($to, selection)) {
+          // we have a gap cursor at end of last table cell and want to set a full table cell selection
+          return selectFullTable(editorSelectionAPI)({
             node: table.node,
             startPos: table.start,
-            dir: TableSelectionDirection.BottomToTop
+            dir: TableSelectionDirection.TopToBottom
           })(state, dispatch);
         }
       }
-    }
 
-    return false;
+      return false;
+    };
   };
 };
 
-var arrowRightFromGapCursor = function arrowRightFromGapCursor(selection) {
-  return function (state, dispatch) {
-    var $from = selection.$from,
-        from = selection.from,
-        $to = selection.$to,
-        side = selection.side;
-
-    if (side === _gapCursorSelection.Side.LEFT) {
-      if ($from.nodeAfter && $from.nodeAfter.type.name === 'table') {
-        // we have a gap cursor before a table node and want to set a full table cell selection
-        return selectFullTable({
-          node: $from.nodeAfter,
-          startPos: from + 1,
-          dir: TableSelectionDirection.BottomToTop
-        })(state, dispatch);
-      }
-    } else if (side === _gapCursorSelection.Side.RIGHT) {
+var arrowLeftFromText = function arrowLeftFromText(editorSelectionAPI) {
+  return function (selection) {
+    return function (state, dispatch) {
       var table = (0, _utils.findTable)(selection);
 
-      if (table && isSelectionAtEndOfTable($to, selection)) {
-        // we have a gap cursor at end of last table cell and want to set a full table cell selection
-        return selectFullTable({
-          node: table.node,
-          startPos: table.start,
-          dir: TableSelectionDirection.TopToBottom
-        })(state, dispatch);
+      if (table) {
+        var $from = selection.$from;
+
+        if (isSelectionAtStartOfTable($from, selection) && $from.parent.type.name === 'paragraph' && $from.depth === table.depth + 3 && // + 3 for: row, cell & paragraph nodes
+        editorSelectionAPI) {
+          var _editorSelectionAPI$g4 = editorSelectionAPI.getSelectionPluginState(state),
+              selectionRelativeToNode = _editorSelectionAPI$g4.selectionRelativeToNode;
+
+          if (selectionRelativeToNode === _selection.RelativeSelectionPos.Before) {
+            // we have a text selection at start of first table cell, directly inside a top level paragraph,
+            // and want to set gap cursor selection before table
+            return setGapCursorBeforeTable(editorSelectionAPI)()(state, dispatch);
+          } else {
+            // we have a text selection at start of first table cell, directly inside a top level paragraph,
+            // and want to set a full table cell selection
+            return selectFullTable(editorSelectionAPI)({
+              node: table.node,
+              startPos: table.start,
+              dir: TableSelectionDirection.BottomToTop
+            })(state, dispatch);
+          }
+        }
       }
-    }
 
-    return false;
+      return false;
+    };
   };
 };
 
-var arrowLeftFromText = function arrowLeftFromText(selection) {
-  return function (state, dispatch) {
-    var table = (0, _utils.findTable)(selection);
-
-    if (table) {
-      var $from = selection.$from;
+var arrowRightFromText = function arrowRightFromText(editorSelectionAPI) {
+  return function (selection) {
+    return function (state, dispatch) {
+      var table = (0, _utils.findTable)(selection);
 
-      if (isSelectionAtStartOfTable($from, selection) && $from.parent.type.name === 'paragraph' && $from.depth === table.depth + 3 // + 3 for: row, cell & paragraph nodes
-      ) {
-        var _getSelectionPluginSt4 = (0, _pluginFactory.getPluginState)(state),
-            selectionRelativeToNode = _getSelectionPluginSt4.selectionRelativeToNode;
+      if (table) {
+        var $to = selection.$to;
 
-        if (selectionRelativeToNode === _types.RelativeSelectionPos.Before) {
-          // we have a text selection at start of first table cell, directly inside a top level paragraph,
-          // and want to set gap cursor selection before table
-          return setGapCursorBeforeTable()(state, dispatch);
-        } else {
-          // we have a text selection at start of first table cell, directly inside a top level paragraph,
+        if (isSelectionAtEndOfTable($to, selection) && $to.parent.type.name === 'paragraph' && $to.depth === table.depth + 3 // + 3 for: row, cell & paragraph nodes
+        ) {
+          // we have a text selection at end of last table cell, directly inside a top level paragraph,
           // and want to set a full table cell selection
-          return selectFullTable({
+          return selectFullTable(editorSelectionAPI)({
             node: table.node,
             startPos: table.start,
-            dir: TableSelectionDirection.BottomToTop
+            dir: TableSelectionDirection.TopToBottom
           })(state, dispatch);
         }
       }
-    }
-
-    return false;
-  };
-};
-
-var arrowRightFromText = function arrowRightFromText(selection) {
-  return function (state, dispatch) {
-    var table = (0, _utils.findTable)(selection);
-
-    if (table) {
-      var $to = selection.$to;
-
-      if (isSelectionAtEndOfTable($to, selection) && $to.parent.type.name === 'paragraph' && $to.depth === table.depth + 3 // + 3 for: row, cell & paragraph nodes
-      ) {
-        // we have a text selection at end of last table cell, directly inside a top level paragraph,
-        // and want to set a full table cell selection
-        return selectFullTable({
-          node: table.node,
-          startPos: table.start,
-          dir: TableSelectionDirection.TopToBottom
-        })(state, dispatch);
-      }
-    }
 
-    return false;
+      return false;
+    };
   };
 };
 /**
@@ -240,89 +252,161 @@ var arrowRightFromText = function arrowRightFromText(selection) {
  */
 
 
-var selectFullTable = function selectFullTable(_ref) {
-  var node = _ref.node,
-      startPos = _ref.startPos,
-      dir = _ref.dir;
-  return function (state, dispatch) {
-    var doc = state.doc;
-
-    var _TableMap$get = _tableMap.TableMap.get(node),
-        map = _TableMap$get.map;
+var selectFullTable = function selectFullTable(editorSelectionAPI) {
+  return function (_ref) {
+    var node = _ref.node,
+        startPos = _ref.startPos,
+        dir = _ref.dir;
+    return function (state, dispatch) {
+      var doc = state.doc;
+
+      var _TableMap$get = _tableMap.TableMap.get(node),
+          map = _TableMap$get.map;
+
+      var $firstCell = doc.resolve(startPos + map[0]);
+      var $lastCell = doc.resolve(startPos + map[map.length - 1]);
+      var fullTableSelection;
+      var selectionRelativeToNode;
+
+      if (dir === TableSelectionDirection.TopToBottom) {
+        fullTableSelection = new _cellSelection.CellSelection($firstCell, $lastCell);
+        selectionRelativeToNode = _selection.RelativeSelectionPos.End;
+      } else {
+        fullTableSelection = new _cellSelection.CellSelection($lastCell, $firstCell);
+        selectionRelativeToNode = _selection.RelativeSelectionPos.Start;
+      }
 
-    var $firstCell = doc.resolve(startPos + map[0]);
-    var $lastCell = doc.resolve(startPos + map[map.length - 1]);
-    var fullTableSelection;
-    var selectionRelativeToNode;
+      if (editorSelectionAPI) {
+        var tr = editorSelectionAPI.setSelectionRelativeToNode({
+          selectionRelativeToNode: selectionRelativeToNode,
+          selection: fullTableSelection
+        })(state);
 
-    if (dir === TableSelectionDirection.TopToBottom) {
-      fullTableSelection = new _cellSelection.CellSelection($firstCell, $lastCell);
-      selectionRelativeToNode = _types.RelativeSelectionPos.End;
-    } else {
-      fullTableSelection = new _cellSelection.CellSelection($lastCell, $firstCell);
-      selectionRelativeToNode = _types.RelativeSelectionPos.Start;
-    }
+        if (dispatch) {
+          dispatch(tr);
+          return true;
+        }
+      }
 
-    return (0, _commands.setSelectionRelativeToNode)(selectionRelativeToNode, fullTableSelection)(state, dispatch);
+      return false;
+    };
   };
 };
 
-var setSelectionAtStartOfFirstCell = function setSelectionAtStartOfFirstCell(selection, selectionRelativeToNode) {
-  return function (state, dispatch) {
-    var $anchorCell = selection.$anchorCell,
-        $headCell = selection.$headCell;
-    var $firstCell = $anchorCell.pos < $headCell.pos ? $anchorCell : $headCell;
-    var $firstPosInsideCell = state.doc.resolve($firstCell.pos + 1); // check if first pos should have a gap cursor, otherwise find closest text selection
+var setSelectionAtStartOfFirstCell = function setSelectionAtStartOfFirstCell(editorSelectionAPI) {
+  return function (selection, selectionRelativeToNode) {
+    return function (state, dispatch) {
+      var $anchorCell = selection.$anchorCell,
+          $headCell = selection.$headCell;
+      var $firstCell = $anchorCell.pos < $headCell.pos ? $anchorCell : $headCell;
+      var $firstPosInsideCell = state.doc.resolve($firstCell.pos + 1); // check if first pos should have a gap cursor, otherwise find closest text selection
+
+      var selectionAtStartOfCell = _selection.GapCursorSelection.valid($firstPosInsideCell) ? new _selection.GapCursorSelection($firstPosInsideCell, _selection.Side.LEFT) : _prosemirrorState.Selection.findFrom($firstPosInsideCell, 1);
+
+      if (editorSelectionAPI) {
+        var tr = editorSelectionAPI.setSelectionRelativeToNode({
+          selectionRelativeToNode: selectionRelativeToNode,
+          selection: selectionAtStartOfCell
+        })(state);
+
+        if (dispatch) {
+          dispatch(tr);
+          return true;
+        }
+      }
 
-    var selectionAtStartOfCell = _gapCursorSelection.GapCursorSelection.valid($firstPosInsideCell) ? new _gapCursorSelection.GapCursorSelection($firstPosInsideCell, _gapCursorSelection.Side.LEFT) : _prosemirrorState.Selection.findFrom($firstPosInsideCell, 1);
-    return (0, _commands.setSelectionRelativeToNode)(selectionRelativeToNode, selectionAtStartOfCell)(state, dispatch);
+      return false;
+    };
   };
 };
 
-var setSelectionAtEndOfLastCell = function setSelectionAtEndOfLastCell(selection, selectionRelativeToNode) {
-  return function (state, dispatch) {
-    var $anchorCell = selection.$anchorCell,
-        $headCell = selection.$headCell;
-    var $lastCell = $anchorCell.pos > $headCell.pos ? $anchorCell : $headCell;
-    var lastPosInsideCell = $lastCell.pos + ($lastCell.nodeAfter ? $lastCell.nodeAfter.content.size : 0) + 1;
-    var $lastPosInsideCell = state.doc.resolve(lastPosInsideCell); // check if last pos should have a gap cursor, otherwise find closest text selection
+var setSelectionAtEndOfLastCell = function setSelectionAtEndOfLastCell(editorSelectionAPI) {
+  return function (selection, selectionRelativeToNode) {
+    return function (state, dispatch) {
+      var $anchorCell = selection.$anchorCell,
+          $headCell = selection.$headCell;
+      var $lastCell = $anchorCell.pos > $headCell.pos ? $anchorCell : $headCell;
+      var lastPosInsideCell = $lastCell.pos + ($lastCell.nodeAfter ? $lastCell.nodeAfter.content.size : 0) + 1;
+      var $lastPosInsideCell = state.doc.resolve(lastPosInsideCell); // check if last pos should have a gap cursor, otherwise find closest text selection
+
+      var selectionAtEndOfCell = _selection.GapCursorSelection.valid($lastPosInsideCell) ? new _selection.GapCursorSelection($lastPosInsideCell, _selection.Side.RIGHT) : _prosemirrorState.Selection.findFrom($lastPosInsideCell, -1);
+
+      if (editorSelectionAPI) {
+        var tr = editorSelectionAPI.setSelectionRelativeToNode({
+          selectionRelativeToNode: selectionRelativeToNode,
+          selection: selectionAtEndOfCell
+        })(state);
+
+        if (dispatch) {
+          dispatch(tr);
+          return true;
+        }
+      }
 
-    var selectionAtEndOfCell = _gapCursorSelection.GapCursorSelection.valid($lastPosInsideCell) ? new _gapCursorSelection.GapCursorSelection($lastPosInsideCell, _gapCursorSelection.Side.RIGHT) : _prosemirrorState.Selection.findFrom($lastPosInsideCell, -1);
-    return (0, _commands.setSelectionRelativeToNode)(selectionRelativeToNode, selectionAtEndOfCell)(state, dispatch);
+      return false;
+    };
   };
 };
 
-var setGapCursorBeforeTable = function setGapCursorBeforeTable() {
-  return function (state, dispatch) {
-    var table = (0, _utils.findTable)(state.selection);
+var setGapCursorBeforeTable = function setGapCursorBeforeTable(editorSelectionAPI) {
+  return function () {
+    return function (state, dispatch) {
+      var table = (0, _utils.findTable)(state.selection);
+
+      if (table) {
+        var $beforeTablePos = state.doc.resolve(table.pos);
 
-    if (table) {
-      var $beforeTablePos = state.doc.resolve(table.pos);
+        if (_selection.GapCursorSelection.valid($beforeTablePos)) {
+          var selectionBeforeTable = new _selection.GapCursorSelection($beforeTablePos, _selection.Side.LEFT);
 
-      if (_gapCursorSelection.GapCursorSelection.valid($beforeTablePos)) {
-        var selectionBeforeTable = new _gapCursorSelection.GapCursorSelection($beforeTablePos, _gapCursorSelection.Side.LEFT);
-        return (0, _commands.setSelectionRelativeToNode)(undefined, selectionBeforeTable)(state, dispatch);
+          if (editorSelectionAPI) {
+            var tr = editorSelectionAPI.setSelectionRelativeToNode({
+              selectionRelativeToNode: undefined,
+              selection: selectionBeforeTable
+            })(state);
+
+            if (dispatch) {
+              dispatch(tr);
+              return true;
+            }
+          }
+        }
       }
-    }
 
-    return false;
+      return false;
+    };
   };
 };
 
-var setGapCursorAfterTable = function setGapCursorAfterTable() {
-  return function (state, dispatch) {
-    var table = (0, _utils.findTable)(state.selection);
+var setGapCursorAfterTable = function setGapCursorAfterTable(editorSelectionAPI) {
+  return function () {
+    return function (state, dispatch) {
+      var table = (0, _utils.findTable)(state.selection);
+
+      if (table) {
+        var $afterTablePos = state.doc.resolve(table.pos + table.node.nodeSize);
 
-    if (table) {
-      var $afterTablePos = state.doc.resolve(table.pos + table.node.nodeSize);
+        if (_selection.GapCursorSelection.valid($afterTablePos)) {
+          var selectionAfterTable = new _selection.GapCursorSelection($afterTablePos, _selection.Side.RIGHT);
 
-      if (_gapCursorSelection.GapCursorSelection.valid($afterTablePos)) {
-        var selectionAfterTable = new _gapCursorSelection.GapCursorSelection($afterTablePos, _gapCursorSelection.Side.RIGHT);
-        return (0, _commands.setSelectionRelativeToNode)(undefined, selectionAfterTable)(state, dispatch);
+          if (editorSelectionAPI) {
+            var tr = editorSelectionAPI.setSelectionRelativeToNode({
+              selectionRelativeToNode: undefined,
+              selection: selectionAfterTable
+            })(state);
+
+            if (dispatch) {
+              dispatch(tr);
+              return true;
+            }
+          }
+
+          return false;
+        }
       }
-    }
 
-    return false;
+      return false;
+    };
   };
 };