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

Block controls plugin for @atlaskit/editor-core

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';
const getSelectionToIndex = (fromIndex, $to, depth) => {
  const toIndex = $to.index(depth);
  const 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 const moveNodeWithBlockMenu = (api, direction) => {
  return ({
    tr
  }) => {
    var _api$blockControls$sh;
    if (!editorExperiment('platform_editor_block_menu', true)) {
      return tr;
    }
    const preservedSelection = api === null || api === void 0 ? void 0 : (_api$blockControls$sh = api.blockControls.sharedState.currentState()) === null || _api$blockControls$sh === void 0 ? void 0 : _api$blockControls$sh.preservedSelection;
    const selection = preservedSelection !== null && preservedSelection !== void 0 ? preservedSelection : tr.selection;

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

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

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