bpmn-js/lib/features/distribute-elements/BpmnDistributeElements.js

import inherits from 'inherits-browser';

import RuleProvider from 'diagram-js/lib/features/rules/RuleProvider';
import { getParents } from 'diagram-js/lib/util/Elements';

import {
  filter
} from 'min-dash';

import {
  isAny
} from '../modeling/util/ModelingUtil';


/**
 * Registers element exclude filters for elements that
 * currently do not support distribution.
 */
export default function BpmnDistributeElements(distributeElements, eventBus, rules) {
  RuleProvider.call(this, eventBus);
}

BpmnDistributeElements.$inject = [ 'distributeElements', 'eventBus', 'rules' ];

inherits(BpmnDistributeElements, RuleProvider);

BpmnDistributeElements.prototype.init = function() {
  this.addRule('elements.distribute', function(context) {
    var elements = context.elements;

    elements = filter(elements, function(element) {
      var cannotDistribute = isAny(element, [
        'bpmn:Association',
        'bpmn:BoundaryEvent',
        'bpmn:DataInputAssociation',
        'bpmn:DataOutputAssociation',
        'bpmn:Lane',
        'bpmn:MessageFlow',
        'bpmn:SequenceFlow',
        'bpmn:TextAnnotation'
      ]);

      return !(element.labelTarget || cannotDistribute);
    });

    // filter out elements which are children of any of the selected elements
    elements = getParents(elements);

    if (elements.length < 3) {
      return false;
    }

    return elements;
  });
};