From bf7e231ffea89afe6a2c5e1767d0c62ae9b0c09c Mon Sep 17 00:00:00 2001
From: Oliver Gorwits <oliver@cpan.org>
Date: Sun, 24 Dec 2017 10:09:26 +0000
Subject: [PATCH] node colors!

---
 .../Netdisco/Web/Plugin/Device/Neighbors.pm   | 26 +++++++++++--------
 share/views/ajax/device/netmap.tt             |  1 +
 2 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/lib/App/Netdisco/Web/Plugin/Device/Neighbors.pm b/lib/App/Netdisco/Web/Plugin/Device/Neighbors.pm
index 95bfaf43..d18265eb 100644
--- a/lib/App/Netdisco/Web/Plugin/Device/Neighbors.pm
+++ b/lib/App/Netdisco/Web/Plugin/Device/Neighbors.pm
@@ -5,6 +5,7 @@ use Dancer::Plugin::Ajax;
 use Dancer::Plugin::DBIC;
 use Dancer::Plugin::Auth::Extensible;
 
+use List::Util 'first';
 use List::MoreUtils ();
 use App::Netdisco::Util::Permission 'check_acl_only';
 use App::Netdisco::Web::Plugin;
@@ -117,20 +118,22 @@ ajax '/ajax/data/device/netmap' => require_login sub {
       '+select' => [\'row_number() over()'], '+as' => ['row_number'],
     });
 
-    NODE: while (my $device = $devices->next) {
+    # list of groups selected by user and passed in param
+    my $devgrp = (ref [] eq ref param('devgrp') ? param('devgrp') : [param('devgrp')]);
+    # list of groups validated as real host groups and named host groups
+    my @hgrplist = grep { exists setting('host_group_displaynames')->{$_} }
+                   grep { exists setting('host_groups')->{$_} }
+                   grep { defined } @{ $devgrp };
+
+    DEVICE: while (my $device = $devices->next) {
       # if in neighbors or vlan mode then use %ok_dev to filter
-      next NODE if (($mapshow eq 'neighbors') or $vlan)
+      next DEVICE if (($mapshow eq 'neighbors') or $vlan)
         and (not $ok_dev{$device->ip});
 
       # if in only mode then use ACLs to filter
-      if ($mapshow eq 'only') {
-        my $devgrp = (ref [] eq ref param('devgrp') ? param('devgrp') : [param('devgrp')]);
-        next NODE unless scalar grep { check_acl_only($device, $_) }
-                                grep { defined }
-                                 map { setting('host_groups')->{$_} }
-                                grep { defined }
-                                    @{ $devgrp };
-      }
+      my $first_hgrp =
+        first { check_acl_only($device, setting('host_groups')->{$_}) } @hgrplist;
+      next DEVICE if $mapshow eq 'only' and not $first_hgrp;
 
       $id_for{$device->ip} = $device->get_column('row_number');
       (my $name = ($device->dns || lc($device->name) || $device->ip)) =~ s/$domain$//;
@@ -138,7 +141,8 @@ ajax '/ajax/data/device/netmap' => require_login sub {
       $v3data{nodes}->{ ($device->get_column('row_number') - 1) } = {
         ID => $device->ip,
         SIZEVALUE => 3000,
-        (param('colorgroups') ? (COLORVALUE => 10) : ()),
+        (param('colorgroups') ?
+          (COLORVALUE => ($first_hgrp ? setting('host_group_displaynames')->{$first_hgrp} : 'Other')) : ()),
         LABEL => $name,
       };
 
diff --git a/share/views/ajax/device/netmap.tt b/share/views/ajax/device/netmap.tt
index 7a3afc87..57547d87 100644
--- a/share/views/ajax/device/netmap.tt
+++ b/share/views/ajax/device/netmap.tt
@@ -45,6 +45,7 @@ $.getJSON('[% uri_for('/ajax/data/device/netmap') %]?[% my_query %]', function(m
       .pinMode(true)
       .nodeEventToStopPinMode('none')
       .showLinkDirection(false)
+      .colorScheme('color20')
       // .preventLabelOverlappingOnForceEnd(true)
       // .zoomToFitOnForceEnd(true)
       .width( parseInt(jQuery('#netmap_pane').parent().css('width')) )