@atlaskit/editor-plugin-block-controls/dist/esm/editor-commands/move-node-with-block-menu.js

Block controls plugin for @atlaskit/editor-core

import _defineProperty from "@babel/runtime/helpers/defineProperty";
function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
import { INPUT_METHOD } from '@atlaskit/editor-common/analytics';
import { DIRECTION } from '@atlaskit/editor-common/types';
import { editorExperiment } from '@atlaskit/tmp-editor-statsig/experiments';
import { key } from '../pm-plugins/main';
import { mapPreservedSelection } from '../pm-plugins/utils/selection';
import { moveNode } from './move-node';
import { canMoveNodeUpOrDown, getNodeBoundsFromSelection } from './utils/move-node-utils';
var getSelectionToIndex = function getSelectionToIndex(fromIndex, $to, depth) {
  var toIndex = $to.index(depth);
  var toIndexAfter = $to.indexAfter(depth);

  // If $to is at the start of a node (indexAfter === index), don't include that node
  // This occurs when the preserved selection is outside of inline positions at node boundaries
  if (toIndexAfter === toIndex && toIndex > fromIndex) {
    return toIndex - 1;
  }
  return toIndex;
};
export var moveNodeWithBlockMenu = function moveNodeWithBlockMenu(api, direction) {
  return function (_ref) {
    var _api$blockControls$sh;
    var tr = _ref.tr;
    if (!editorExperiment('platform_editor_block_menu', true)) {
      return tr;
    }
    var preservedSelection = api === null || api === void 0 || (_api$blockControls$sh = api.blockControls.sharedState.currentState()) === null || _api$blockControls$sh === void 0 ? void 0 : _api$blockControls$sh.preservedSelection;
    var selection = preservedSelection !== null && preservedSelection !== void 0 ? preservedSelection : tr.selection;

    // Use getNodeBoundsFromSelection to get the proper block boundaries
    var nodeBounds = getNodeBoundsFromSelection(selection);
    if (!nodeBounds) {
      return tr;
    }
    var currentNodePos = nodeBounds.from;
    var $from = selection.$from,
      $to = selection.$to;
    var depth = tr.doc.resolve(currentNodePos).depth;
    var fromIndex = $from.index(depth);
    var trAfterNodeMove = tr;
    if (direction === DIRECTION.UP) {
      if (fromIndex > 0) {
        var moveToPos = $from.posAtIndex(fromIndex - 1, depth);
        trAfterNodeMove = moveNode(api)(currentNodePos, moveToPos, INPUT_METHOD.BLOCK_MENU)({
          tr: tr
        }) || tr;
      }
    } else {
      // selectionToIndex is the index of the last node in the selection
      var selectionToIndex = getSelectionToIndex(fromIndex, $to, depth);
      // Adding 2 so we jump over the next node to the position after it
      var moveToIndex = selectionToIndex + 2;
      if (moveToIndex <= $to.node(depth).childCount) {
        var _moveToPos = $to.posAtIndex(moveToIndex, depth);
        trAfterNodeMove = moveNode(api)(currentNodePos, _moveToPos, INPUT_METHOD.BLOCK_MENU)({
          tr: tr
        }) || tr;
      }
    }

    // map selection
    var mappedSelection = mapPreservedSelection(selection, trAfterNodeMove);
    if (mappedSelection) {
      trAfterNodeMove.setSelection(mappedSelection);
    }

    //recalculate canMoveUp and canMoveDown
    var _canMoveNodeUpOrDown = canMoveNodeUpOrDown(trAfterNodeMove),
      moveUp = _canMoveNodeUpOrDown.moveUp,
      moveDown = _canMoveNodeUpOrDown.moveDown;
    var currentMeta = trAfterNodeMove.getMeta(key);
    var newMeta = _objectSpread(_objectSpread({}, currentMeta), {
      toggleMenu: {
        moveUp: moveUp,
        moveDown: moveDown
      }
    });
    return trAfterNodeMove.setMeta(key, newMeta);
  };
};