bpmn-js/lib/features/di-ordering/BpmnDiOrdering.js

A bpmn 2.0 toolkit and web modeler

import { getDi } from '../../util/ModelUtil';

import {
  filter,
  forEach,
  map
} from 'min-dash';

import { selfAndAllChildren } from 'diagram-js/lib/util/Elements';

/**
 * @typedef {import('diagram-js/lib/core/EventBus').default} EventBus
 * @typedef {import('diagram-js/lib/core/Canvas').default} Canvas
 */

var HIGH_PRIORITY = 2000;

/**
 * @param {EventBus} eventBus
 * @param {Canvas} canvas
 */
export default function BpmnDiOrdering(eventBus, canvas) {

  eventBus.on('saveXML.start', HIGH_PRIORITY, orderDi);

  function orderDi() {
    var rootElements = canvas.getRootElements();

    forEach(rootElements, function(root) {
      var rootDi = getDi(root),
          elements,
          diElements;

      elements = selfAndAllChildren([ root ], false);

      // only bpmndi:Shape and bpmndi:Edge can be direct children of bpmndi:Plane
      elements = filter(elements, function(element) {
        return element !== root && !element.labelTarget;
      });

      diElements = map(elements, getDi);

      rootDi.set('planeElement', diElements);
    });
  }
}

BpmnDiOrdering.$inject = [ 'eventBus', 'canvas' ];