diagram-js/lib/features/modeling/cmd/DeleteElementsHandler.js

A toolbox for displaying and modifying diagrams on the web

import { forEach } from 'min-dash';

/**
 * @typedef {import('../../../core/ElementRegistry').default} ElementRegistry
 * @typedef {import('../Modeling').default} Modeling
 */

/**
 * @param {Modeling} modeling
 * @param {ElementRegistry} elementRegistry
 */
export default function DeleteElementsHandler(modeling, elementRegistry) {
  this._modeling = modeling;
  this._elementRegistry = elementRegistry;
}

DeleteElementsHandler.$inject = [
  'modeling',
  'elementRegistry'
];


DeleteElementsHandler.prototype.postExecute = function(context) {

  var modeling = this._modeling,
      elementRegistry = this._elementRegistry,
      elements = context.elements;

  forEach(elements, function(element) {

    // element may have been removed with previous
    // remove operations already (e.g. in case of nesting)
    if (!elementRegistry.get(element.id)) {
      return;
    }

    if (element.waypoints) {
      modeling.removeConnection(element);
    } else {
      modeling.removeShape(element);
    }
  });
};