1 files changed, 43 insertions, 0 deletions
diff --git a/sdc-workflow-designer-ui/src/main/frontend/src/features/version/composition/custom-modeler/custom/CustomContextPadProvider.js b/sdc-workflow-designer-ui/src/main/frontend/src/features/version/composition/custom-modeler/custom/CustomContextPadProvider.js
new file mode 100644
index 00000000..0f2ba528
--- /dev/null
+++ b/sdc-workflow-designer-ui/src/main/frontend/src/features/version/composition/custom-modeler/custom/CustomContextPadProvider.js
@@ -0,0 +1,43 @@
+import inherits from 'inherits';
+
+import ContextPadProvider from 'bpmn-js/lib/features/context-pad/ContextPadProvider';
+
+import { isAny } from 'bpmn-js/lib/features/modeling/util/ModelingUtil';
+
+import { assign, bind } from 'min-dash';
+
+export default function CustomContextPadProvider(injector, connect, translate) {
+    injector.invoke(ContextPadProvider, this);
+
+    var cached = bind(this.getContextPadEntries, this);
+
+    this.getContextPadEntries = function(element) {
+        var actions = cached(element);
+
+        var businessObject = element.businessObject;
+
+        function startConnect(event, element, autoActivate) {
+            connect.start(event, element, autoActivate);
+        }
+
+        if (isAny(businessObject, ['custom:triangle', 'custom:circle'])) {
+            assign(actions, {
+                connect: {
+                    group: 'connect',
+                    className: 'bpmn-icon-connection-multi',
+                    title: translate('Connect using custom connection'),
+                    action: {
+                        click: startConnect,
+                        dragstart: startConnect
+                    }
+                }
+            });
+        }
+
+        return actions;
+    };
+}
+
+inherits(CustomContextPadProvider, ContextPadProvider);
+
+CustomContextPadProvider.$inject = ['injector', 'connect', 'translate'];