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

BlockMenu plugin for @atlaskit/editor-core

import { Fragment } from '@atlaskit/editor-prosemirror/model';
import { isListWithIndentation } from '../nodeChecks';
const extractNestedLists = (node, schema) => {
  const items = [];
  const paragraph = schema.nodes.paragraph;
  const itemTypes = [schema.nodes.listItem, schema.nodes.taskItem, schema.nodes.decisionItem];
  const extract = currentNode => {
    currentNode.forEach(child => {
      if (itemTypes.some(type => child.type === type)) {
        const contentWithoutNestedLists = [];
        const nestedLists = [];

        // Check if this item type expects inline content (taskItem, decisionItem)
        // vs block content (listItem) based on the schema definition
        const isInlineItem = child.type.inlineContent;
        child.forEach(grandChild => {
          if (isListWithIndentation(grandChild.type.name, schema)) {
            nestedLists.push(grandChild);
          } else if (grandChild.isInline) {
            // For taskItem/decisionItem, keep text as-is (they support inline content)
            // For listItem, wrap text in paragraph (they require block content)
            if (isInlineItem) {
              contentWithoutNestedLists.push(grandChild);
            } else {
              contentWithoutNestedLists.push(paragraph.createAndFill({}, grandChild));
            }
          } else {
            contentWithoutNestedLists.push(grandChild);
          }
        });
        items.push(child.copy(Fragment.from(contentWithoutNestedLists)));
        nestedLists.forEach(nestedList => {
          extract(nestedList);
        });
      } else if (isListWithIndentation(child.type.name, schema)) {
        extract(child);
      }
    });
  };
  extract(node);
  return items;
};

/**
 * Given an array of nodes, returns an array with the flattened children of any list node
 * to it's first ancestor list, maintaining document order.
 *
 * @example
 * Input (deeply nested):
 * - bulletList
 *   - listItem "1"
 *     - bulletList
 *       - listItem "1.1"
 *         - bulletList
 *           - listItem "1.1.1"
 *       - listItem "1.2"
 *   - listItem "2"
 *
 * Output:
 * - bulletList
 *   - listItem "1"
 *   - listItem "1.1"
 *   - listItem "1.1.1"
 *   - listItem "1.2"
 *   - listItem "2"
 *
 * @param nodes
 * @param context
 * @returns
 */
export const flattenListStep = (nodes, context) => {
  return nodes.map(node => {
    if (isListWithIndentation(node.type.name, context.schema)) {
      return node.copy(Fragment.from(extractNestedLists(node, context.schema)));
    }
    return node;
  });
};