@atlaskit/editor-plugin-block-menu/dist/cjs/editor-actions/isTransformToTargetDisabled.js

BlockMenu plugin for @atlaskit/editor-core

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.isTransformToTargetDisabled = exports.canParentContainNodeType = void 0;
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
var _utils = require("@atlaskit/editor-common/selection/utils");
var _model = require("@atlaskit/editor-prosemirror/model");
var _isExperimentEnabled = require("@atlaskit/platform-feature-experiments/is-experiment-enabled");
var _transform = require("../editor-commands/transform-node-utils/transform");
var _types = require("../editor-commands/transform-node-utils/types");
var _utils2 = require("../editor-commands/transform-node-utils/utils");
var _transformSource = require("./transformSource");
function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t.return || t.return(); } finally { if (u) throw o; } } }; }
function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
var getTargetMarks = function getTargetMarks(schema, currentMarks, marksToAdd, marksToRemove) {
  if (!marksToAdd && !marksToRemove) {
    return undefined;
  }
  var targetMarks = currentMarks;
  var _iterator = _createForOfIteratorHelper(marksToRemove !== null && marksToRemove !== void 0 ? marksToRemove : []),
    _step;
  try {
    for (_iterator.s(); !(_step = _iterator.n()).done;) {
      var _name = _step.value;
      var _markType = schema.marks[_name];
      if (_markType) {
        targetMarks = _markType.removeFromSet(targetMarks);
      }
    }
  } catch (err) {
    _iterator.e(err);
  } finally {
    _iterator.f();
  }
  for (var _i = 0, _Object$entries = Object.entries(marksToAdd !== null && marksToAdd !== void 0 ? marksToAdd : {}); _i < _Object$entries.length; _i++) {
    var _Object$entries$_i = (0, _slicedToArray2.default)(_Object$entries[_i], 2),
      name = _Object$entries$_i[0],
      attrs = _Object$entries$_i[1];
    var markType = schema.marks[name];
    if (markType) {
      targetMarks = markType.create(attrs).addToSet(markType.removeFromSet(targetMarks));
    }
  }
  return targetMarks;
};
var canParentContainNodeType = exports.canParentContainNodeType = function canParentContainNodeType(schema, selectedNodeTypeName, parentNode, nodeTypeName, nodeTypeAttrs, selectedNode, marksToAdd, marksToRemove) {
  var adjustedNodeTypeName = (0, _utils2.getTargetNodeTypeNameInContext)(nodeTypeName, true, parentNode, schema);
  if (!adjustedNodeTypeName) {
    return false;
  }
  var nodeType = schema.nodes[adjustedNodeTypeName];
  var content = null;
  var nodesThatCantBeNestedInNestedExpand = ['blockCard', 'embedCard', 'table'];
  if (nodesThatCantBeNestedInNestedExpand.includes(selectedNodeTypeName) && (adjustedNodeTypeName === 'expand' || adjustedNodeTypeName === 'nestedExpand')) {
    var node = schema.nodes[selectedNodeTypeName];
    content = node.createAndFill();
  }
  if ((0, _isExperimentEnabled.isExperimentEnabled)('platform_editor_block_menu_small_text') && selectedNode && (marksToAdd || marksToRemove)) {
    var targetMarks = getTargetMarks(schema, selectedNode.marks, marksToAdd, marksToRemove);
    var targetNode = nodeType.createAndFill(nodeTypeAttrs, content, targetMarks);
    return Boolean(targetNode && parentNode.type.validContent(_model.Fragment.from(targetNode)));
  }
  return parentNode.type.validContent(_model.Fragment.from(nodeType.createAndFill(nodeTypeAttrs, content)));
};
var isHeadingToHeadingTransformEnabled = function isHeadingToHeadingTransformEnabled(selectedNode, targetNodeTypeAttrs) {
  var _selectedNode$attrs;
  var selectedLevel = (_selectedNode$attrs = selectedNode.attrs) === null || _selectedNode$attrs === void 0 ? void 0 : _selectedNode$attrs.level;
  var targetLevel = targetNodeTypeAttrs === null || targetNodeTypeAttrs === void 0 ? void 0 : targetNodeTypeAttrs.level;
  if (selectedLevel === undefined || targetLevel === undefined) {
    return false;
  }
  return selectedLevel !== targetLevel;
};
var isTransformEnabledForNode = function isTransformEnabledForNode(node, targetNodeTypeName, targetNodeTypeAttrs, marksToAdd, marksToRemove, isNested, parent, schema) {
  var selectedNodeTypeName = (0, _types.toNodeTypeValue)(node.type.name);
  if (!selectedNodeTypeName) {
    return false;
  }
  if (selectedNodeTypeName === targetNodeTypeName) {
    var removesTargetMarks = marksToRemove === null || marksToRemove === void 0 ? void 0 : marksToRemove.some(function (name) {
      var markType = schema.marks[name];
      return markType ? Boolean(markType.isInSet(node.marks)) : false;
    });
    var addsOrReplacesTargetMarks = Object.entries(marksToAdd !== null && marksToAdd !== void 0 ? marksToAdd : {}).some(function (_ref) {
      var _markType$isInSet;
      var _ref2 = (0, _slicedToArray2.default)(_ref, 2),
        name = _ref2[0],
        attrs = _ref2[1];
      var markType = schema.marks[name];
      return markType ? !((_markType$isInSet = markType.isInSet(node.marks)) !== null && _markType$isInSet !== void 0 && _markType$isInSet.eq(markType.create(attrs))) : false;
    });
    if (removesTargetMarks || addsOrReplacesTargetMarks) {
      if (isNested && !canParentContainNodeType(schema, selectedNodeTypeName, parent, targetNodeTypeName, targetNodeTypeAttrs, node, marksToAdd, marksToRemove)) {
        return false;
      }
      return true;
    }
  }
  var isDisabledByStepsConfig = (0, _transform.isTransformDisabledBasedOnStepsConfig)(selectedNodeTypeName, targetNodeTypeName);
  if (isDisabledByStepsConfig) {
    return false;
  }
  if (selectedNodeTypeName === 'heading' && targetNodeTypeName === 'heading') {
    return isHeadingToHeadingTransformEnabled(node, targetNodeTypeAttrs);
  }
  if (isNested && !canParentContainNodeType(schema, selectedNodeTypeName, parent, targetNodeTypeName, targetNodeTypeAttrs, node, marksToAdd, marksToRemove)) {
    return false;
  }
  return true;
};
var isTransformToTargetDisabled = exports.isTransformToTargetDisabled = function isTransformToTargetDisabled(_ref3) {
  var selection = _ref3.selection,
    targetNodeTypeName = _ref3.targetNodeTypeName,
    targetNodeTypeAttrs = _ref3.targetNodeTypeAttrs,
    targetNodeMarkChanges = _ref3.targetNodeMarkChanges,
    transformRegistry = _ref3.transformRegistry;
  var _expandSelectionToBlo = (0, _utils.expandSelectionToBlockRange)(selection),
    range = _expandSelectionToBlo.range;
  if (!range) {
    return false;
  }
  var sourceNode = (0, _transformSource.getSingleTransformSourceNode)(selection, range);
  if ((0, _transformSource.isExtensionTransformSource)(sourceNode)) {
    var _transformRegistry$re;
    return (transformRegistry === null || transformRegistry === void 0 || (_transformRegistry$re = transformRegistry.resolve({
      source: sourceNode.toJSON(),
      targetTypeName: targetNodeTypeName
    })) === null || _transformRegistry$re === void 0 ? void 0 : _transformRegistry$re.status) !== 'supported';
  }
  var selectedNodes = (0, _utils2.getBlockNodesInRange)(range);
  var parent = range.parent;
  var isNested = range.depth >= 1;
  var schema = selection.$from.doc.type.schema;
  var _ref4 = targetNodeMarkChanges !== null && targetNodeMarkChanges !== void 0 ? targetNodeMarkChanges : {},
    marksToAdd = _ref4.marksToAdd,
    marksToRemove = _ref4.marksToRemove;
  var supportedTargetNodeTypeName = (0, _types.toNodeTypeValue)(targetNodeTypeName);
  if (!supportedTargetNodeTypeName) {
    return true;
  }
  var isEnabledForAnyNode = selectedNodes.some(function (node) {
    return isTransformEnabledForNode(node, supportedTargetNodeTypeName, targetNodeTypeAttrs, marksToAdd, marksToRemove, isNested, parent, schema);
  });
  return !isEnabledForAnyNode;
};