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

BlockMenu plugin for @atlaskit/editor-core

import { getDefaultCodeBlockAttrs } from '@atlaskit/editor-common/code-block';
import { breakoutResizableNodes, getBreakoutResizableNodes } from '@atlaskit/editor-common/utils';
import { isExperimentEnabled } from '@atlaskit/platform-feature-experiments/is-experiment-enabled';
import { expValEquals } from '@atlaskit/tmp-editor-statsig/exp-val-equals';
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 const wrapStep = (nodes, context) => {
  const {
    schema,
    targetNodeTypeName,
    fromNode
  } = context;

  // When wrapping into expand, convert any expand children to nestedExpand
  // since expand cannot be a direct child of expand
  let processedNodes = nodes;
  if (targetNodeTypeName === 'expand') {
    processedNodes = nodes.map(node => {
      if (node.type.name === 'expand') {
        const nestedExpandNode = convertExpandToNestedExpand(node, schema);
        return nestedExpandNode !== null && nestedExpandNode !== void 0 ? nestedExpandNode : node;
      }
      return node;
    });
  }
  const targetNodeType = schema.nodes[targetNodeTypeName];
  const isExpandType = targetNodeTypeName === 'expand' || targetNodeTypeName === 'nestedExpand';
  const isCodeBlock = targetNodeTypeName === 'codeBlock';
  const nodeAttrs = isExpandType ? {
    localId: crypto.randomUUID()
  } : isCodeBlock ? getDefaultCodeBlockAttrs() : {};
  let breakoutResizableNodesList = [];
  if (isExperimentEnabled('platform_editor_lovability_resize_extensions')) {
    breakoutResizableNodesList = getBreakoutResizableNodes();
    // rule is NOT a supported transform source/target
    breakoutResizableNodesList = breakoutResizableNodesList.filter(node => node !== 'rule');
  } else {
    breakoutResizableNodesList = expValEquals('platform_editor_lovability_resize_dividers_panels', 'isEnabled', true) ? [...breakoutResizableNodes, 'panel'] : breakoutResizableNodes;
  }
  const sourceSupportsBreakout = breakoutResizableNodesList.includes(fromNode.type.name);
  const targetSupportsBreakout = breakoutResizableNodesList.includes(targetNodeType.name);
  const shouldPreserveBreakout = sourceSupportsBreakout && targetSupportsBreakout;
  let marks;
  if (shouldPreserveBreakout) {
    const breakoutMark = fromNode.marks.find(mark => mark.type.name === 'breakout');
    if (breakoutMark) {
      marks = [breakoutMark];
    }
  }
  const outputNode = targetNodeType.createAndFill(nodeAttrs, removeDisallowedMarks(processedNodes, schema.nodes[targetNodeTypeName]), marks);
  return outputNode ? [outputNode] : nodes;
};