feat(selection-visuals): add marker to related elements

Author: Niklas Kiefer

lib/features/selection/SelectionVisuals.js

@@ -2,6 +2,10 @@ import {
   forEach
 } from 'min-dash';
 
+import {
+  getType
+} from 'lib/util/Elements';
+
 var MARKER_HOVER = 'hover',
     MARKER_SELECTED = 'selected';
 
@@ -22,12 +26,29 @@ export default function SelectionVisuals(events, canvas, selection, styles) {
 
   this._multiSelectionBox = null;
 
+  function applyToRelatedElements(e, cls, fn) {
+
+    // connections -> mark related labels
+    if (getType(e) === 'connection') {
+      forEach(e.labels, function(label) {
+        fn(label, cls);
+      });
+    }
+
+    // label -> mark related connection
+    if (e.labelTarget) {
+      fn(e.labelTarget, cls);
+    }
+  }
+
   function addMarker(e, cls) {
     canvas.addMarker(e, cls);
+    applyToRelatedElements(e, cls, canvas.addMarker.bind(canvas));
   }
 
   function removeMarker(e, cls) {
     canvas.removeMarker(e, cls);
+    applyToRelatedElements(e, cls, canvas.removeMarker.bind(canvas));
   }
 
   events.on('element.hover', function(event) {