@atlaskit/editor-plugin-block-menu/dist/esm/editor-commands/transform-node-utils/steps/wrapStep.js

BlockMenu plugin for @atlaskit/editor-core

import { getDefaultCodeBlockAttrs } from '@atlaskit/editor-common/code-block';
import { breakoutResizableNodes } from '@atlaskit/editor-common/utils';
import { removeDisallowedMarks } from '../marks';
import { convertExpandToNestedExpand } from '../utils';

/**
 * Wraps nodes into the target container type.
 * When wrapping into expand, any expand children are converted to nestedExpand
 * since expand cannot be a direct child of expand.
 *
 * Preserves breakout marks when both source and target nodes support resizing
 * (codeBlock, expand, layoutSection).
 */
export var wrapStep = function wrapStep(nodes, context) {
  var schema = context.schema,
    targetNodeTypeName = context.targetNodeTypeName,
    fromNode = context.fromNode;

  // When wrapping into expand, convert any expand children to nestedExpand
  // since expand cannot be a direct child of expand
  var processedNodes = nodes;
  if (targetNodeTypeName === 'expand') {
    processedNodes = nodes.map(function (node) {
      if (node.type.name === 'expand') {
        var nestedExpandNode = convertExpandToNestedExpand(node, schema);
        return nestedExpandNode !== null && nestedExpandNode !== void 0 ? nestedExpandNode : node;
      }
      return node;
    });
  }
  var targetNodeType = schema.nodes[targetNodeTypeName];
  var isExpandType = targetNodeTypeName === 'expand' || targetNodeTypeName === 'nestedExpand';
  var isCodeBlock = targetNodeTypeName === 'codeBlock';
  var nodeAttrs = isExpandType ? {
    localId: crypto.randomUUID()
  } : isCodeBlock ? getDefaultCodeBlockAttrs() : {};
  var sourceSupportsBreakout = breakoutResizableNodes.includes(fromNode.type.name);
  var targetSupportsBreakout = breakoutResizableNodes.includes(targetNodeType.name);
  var shouldPreserveBreakout = sourceSupportsBreakout && targetSupportsBreakout;
  var marks;
  if (shouldPreserveBreakout) {
    var breakoutMark = fromNode.marks.find(function (mark) {
      return mark.type.name === 'breakout';
    });
    if (breakoutMark) {
      marks = [breakoutMark];
    }
  }
  var outputNode = targetNodeType.createAndFill(nodeAttrs, removeDisallowedMarks(processedNodes, schema.nodes[targetNodeTypeName]), marks);
  return outputNode ? [outputNode] : nodes;
};