@atlaskit/editor-common/dist/es2019/transforms/expand.js

A package that contains common classes and components for editor and renderer

import { Fragment, Slice } from '@atlaskit/editor-prosemirror/model';
import { findParentNodeOfType, findSelectedNodeOfType } from '@atlaskit/editor-prosemirror/utils';
import { mapChildren } from '../utils/slice';
export const findExpand = (state, selection) => {
  const {
    expand,
    nestedExpand
  } = state.schema.nodes;
  return findSelectedNodeOfType([expand, nestedExpand])(selection || state.selection) || findParentNodeOfType([expand, nestedExpand])(selection || state.selection);
};

// If the top level is a single expand, and the expand is not
// a part of copied content, then return unwrap contents.
// This is needed for handling content copied from expand.
// https://product-fabric.atlassian.net/browse/ED-9146
export const transformSliceToRemoveOpenExpand = (slice, schema) => {
  if (slice.openStart > 1 && slice.openEnd > 1 && slice.content.childCount === 1 && slice.content.firstChild && slice.content.firstChild.type === schema.nodes.expand) {
    return new Slice(slice.content.firstChild.content, slice.openStart - 1, slice.openEnd - 1);
  }
  return slice;
};
export const transformSliceToRemoveOpenNestedExpand = (slice, schema) => {
  if (slice.openStart > 1 && slice.openEnd > 1 && slice.content.childCount === 1 && slice.content.firstChild && slice.content.firstChild.type === schema.nodes.nestedExpand) {
    return new Slice(slice.content.firstChild.content, slice.openStart - 1, slice.openEnd - 1);
  }
  return slice;
};
export const transformSliceNestedExpandToExpand = (slice, schema) => {
  const {
    expand,
    nestedExpand
  } = schema.nodes;
  const children = [];
  mapChildren(slice.content, node => {
    if (node.type === nestedExpand) {
      children.push(expand.createChecked(node.attrs, node.content, node.marks));
    } else {
      children.push(node);
    }
  });
  return new Slice(Fragment.fromArray(children), slice.openStart, slice.openEnd);
};
export const transformSliceExpandToNestedExpand = slice => {
  const children = [];
  try {
    mapChildren(slice.content, currentNode => {
      const {
        expand,
        nestedExpand
      } = currentNode.type.schema.nodes;
      if (currentNode.type === expand) {
        const nestedExpandNode = nestedExpand.createChecked(currentNode.attrs, currentNode.content, currentNode.marks);
        if (nestedExpandNode) {
          children.push(nestedExpandNode);
        }
      } else {
        children.push(currentNode);
      }
    });
  } catch (e) {
    // Will throw error if unable to convert expand to nested expand.
    // Example: expand containing a table being converted to nested expand containing table.
    return null;
  }
  return new Slice(Fragment.fromArray(children), slice.openStart, slice.openEnd);
};