1 files changed, 99 insertions, 0 deletions
diff --git a/workflow-designer-ui/src/main/frontend/src/features/version/composition/custom-modeler/index.js b/workflow-designer-ui/src/main/frontend/src/features/version/composition/custom-modeler/index.js
new file mode 100644
index 00000000..86fbff6a
--- /dev/null
+++ b/workflow-designer-ui/src/main/frontend/src/features/version/composition/custom-modeler/index.js
@@ -0,0 +1,99 @@
+import Modeler from 'bpmn-js/lib/Modeler';
+
+import { assign, isArray } from 'min-dash';
+
+import inherits from 'inherits';
+
+import CustomModule from './custom';
+
+export default function CustomModeler(options) {
+    Modeler.call(this, options);
+
+    this._customElements = [];
+}
+
+inherits(CustomModeler, Modeler);
+
+CustomModeler.prototype._modules = [].concat(CustomModeler.prototype._modules, [
+    CustomModule
+]);
+
+/**
+ * Add a single custom element to the underlying diagram
+ *
+ * @param {Object} customElement
+ */
+CustomModeler.prototype._addCustomShape = function(customElement) {
+    this._customElements.push(customElement);
+
+    var canvas = this.get('canvas'),
+        elementFactory = this.get('elementFactory');
+
+    var customAttrs = assign({ businessObject: customElement }, customElement);
+
+    var customShape = elementFactory.create('shape', customAttrs);
+
+    return canvas.addShape(customShape);
+};
+
+CustomModeler.prototype._addCustomConnection = function(customElement) {
+    this._customElements.push(customElement);
+
+    var canvas = this.get('canvas'),
+        elementFactory = this.get('elementFactory'),
+        elementRegistry = this.get('elementRegistry');
+
+    var customAttrs = assign({ businessObject: customElement }, customElement);
+
+    var connection = elementFactory.create(
+        'connection',
+        assign(customAttrs, {
+            source: elementRegistry.get(customElement.source),
+            target: elementRegistry.get(customElement.target)
+        }),
+        elementRegistry.get(customElement.source).parent
+    );
+
+    return canvas.addConnection(connection);
+};
+
+/**
+ * Add a number of custom elements and connections to the underlying diagram.
+ *
+ * @param {Array<Object>} customElements
+ */
+CustomModeler.prototype.addCustomElements = function(customElements) {
+    if (!isArray(customElements)) {
+        throw new Error('argument must be an array');
+    }
+
+    var shapes = [],
+        connections = [];
+
+    customElements.forEach(function(customElement) {
+        if (isCustomConnection(customElement)) {
+            connections.push(customElement);
+        } else {
+            shapes.push(customElement);
+        }
+    });
+
+    // add shapes before connections so that connections
+    // can already rely on the shapes being part of the diagram
+    shapes.forEach(this._addCustomShape, this);
+
+    connections.forEach(this._addCustomConnection, this);
+};
+
+/**
+ * Get custom elements with their current status.
+ *
+ * @return {Array<Object>} custom elements on the diagram
+ */
+CustomModeler.prototype.getCustomElements = function() {
+    return this._customElements;
+};
+
+function isCustomConnection(element) {
+    return element.type === 'custom:connection';
+}