@atlaskit/editor-plugin-breakout/dist/esm/editor-commands/set-breakout-mode.js

Breakout plugin for @atlaskit/editor-core

import { transferCodeBlockWrappedValue } from '@atlaskit/editor-common/code-block';
import { NodeSelection } from '@atlaskit/editor-prosemirror/state';
import { isExperimentEnabled } from '@atlaskit/platform-feature-experiments/is-experiment-enabled';
import { findSupportedNodeForBreakout } from '../pm-plugins/utils/find-breakout-node';
import { updateExpandedState } from '../pm-plugins/utils/single-player-expand';
export function setBreakoutMode(mode, isLivePage) {
  return function (state, dispatch) {
    var node = findSupportedNodeForBreakout(state.selection);
    if (!node) {
      return false;
    }
    var tr;
    if (isExperimentEnabled('platform_editor_lovability_resize_extensions')) {
      var marks = state.schema.marks.breakout.create({
        mode: mode
      }).addToSet(node.node.marks);
      tr = state.tr.setNodeMarkup(node.pos, node.node.type, node.node.attrs, marks);
    } else {
      tr = state.tr.setNodeMarkup(node.pos, node.node.type, node.node.attrs, [state.schema.marks.breakout.create({
        mode: mode
      })]);
    }
    if (node.node.type === state.schema.nodes.expand) {
      updateExpandedState(tr, node, isLivePage);
    } else if (node.node.type === state.schema.nodes.codeBlock) {
      var newNode = tr.doc.nodeAt(node.pos);
      var oldNode = node.node;
      if (newNode) {
        transferCodeBlockWrappedValue(oldNode, newNode);
      }
    }
    tr.setMeta('scrollIntoView', false);
    if (state.selection instanceof NodeSelection) {
      if (state.selection.$anchor.pos === node.pos) {
        tr.setSelection(NodeSelection.create(tr.doc, node.pos));
      }
    }
    if (dispatch) {
      dispatch(tr);
    }
    return true;
  };
}