Fix to only add known elements

README.md

@@ -4,6 +4,8 @@
 
 This plugin serves a better overview which properties are set per shape.
 
+![Plugin-Example](./plugin-example.png)
+
 ## How to install
 
 Put this directory into the plugins directory of the Camunda Modeler and you're ready to go.

client/ClientPlugin.js

@@ -44,7 +44,7 @@ function ClientPlugin(eventBus, overlays, elementRegistry, editorActions) {
         }
     });
 
-    function removeShape(element){
+    function removeShape(element) {
         var elementObject = elementOverlays[element.id];
         for (var overlay in elementObject) {
             overlays.remove(elementObject[overlay]);
@@ -96,7 +96,7 @@ function ClientPlugin(eventBus, overlays, elementRegistry, editorActions) {
             var background = '';
             var location = 'right';
             var sort = 0;
-            var key;
+            var key = '';
 
             switch (extensions[extension].$type) {
                 case 'camunda:ExecutionListener':
@@ -189,13 +189,15 @@ function ClientPlugin(eventBus, overlays, elementRegistry, editorActions) {
                     });
                 }
             } else {
-                badges.push({
-                    badgeKey: key,
-                    badgeSort: sort,
-                    badgeType: type,
-                    badgeBackground: background,
-                    badgeLocation: location
-                });
+                if (key !== '') {
+                    badges.push({
+                        badgeKey: key,
+                        badgeSort: sort,
+                        badgeType: type,
+                        badgeBackground: background,
+                        badgeLocation: location
+                    });
+                }
             }
 
 

client/client-bundle.js

@@ -45,7 +45,7 @@ function ClientPlugin(eventBus, overlays, elementRegistry, editorActions) {
         }
     });
 
-    function removeShape(element){
+    function removeShape(element) {
         var elementObject = elementOverlays[element.id];
         for (var overlay in elementObject) {
             overlays.remove(elementObject[overlay]);
@@ -97,7 +97,7 @@ function ClientPlugin(eventBus, overlays, elementRegistry, editorActions) {
             var background = '';
             var location = 'right';
             var sort = 0;
-            var key;
+            var key = '';
 
             switch (extensions[extension].$type) {
                 case 'camunda:ExecutionListener':
@@ -190,13 +190,15 @@ function ClientPlugin(eventBus, overlays, elementRegistry, editorActions) {
                     });
                 }
             } else {
-                badges.push({
-                    badgeKey: key,
-                    badgeSort: sort,
-                    badgeType: type,
-                    badgeBackground: background,
-                    badgeLocation: location
-                });
+                if (key !== '') {
+                    badges.push({
+                        badgeKey: key,
+                        badgeSort: sort,
+                        badgeType: type,
+                        badgeBackground: background,
+                        badgeLocation: location
+                    });
+                }
             }