284 lines
		
	
	
		
			9.8 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
			
		
		
	
	
			284 lines
		
	
	
		
			9.8 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
package App::Netdisco::Web::Plugin::Device::Neighbors;
 | 
						||
 | 
						||
use Dancer ':syntax';
 | 
						||
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;
 | 
						||
 | 
						||
register_device_tab({ tag => 'netmap', label => 'Neighbors' });
 | 
						||
 | 
						||
ajax '/ajax/content/device/netmap' => require_login sub {
 | 
						||
    content_type('text/html');
 | 
						||
    template 'ajax/device/netmap.tt', {}, { layout => undef };
 | 
						||
};
 | 
						||
 | 
						||
ajax '/ajax/data/device/netmappositions' => require_login sub {
 | 
						||
    my $q = param('q');
 | 
						||
    my $qdev = schema('netdisco')->resultset('Device')
 | 
						||
      ->search_for_device($q) or send_error('Bad device', 400);
 | 
						||
 | 
						||
    my $p = param('positions') or send_error('Missing positions', 400);
 | 
						||
    my $positions = from_json($p) or send_error('Bad positions', 400);
 | 
						||
    send_error('Bad positions', 400) unless ref [] eq ref $positions;
 | 
						||
 | 
						||
    my $vlan = param('vlan');
 | 
						||
    undef $vlan if (defined $vlan and $vlan !~ m/^\d+$/);
 | 
						||
 | 
						||
    my $mapshow = param('mapshow');
 | 
						||
    return if !defined $mapshow or $mapshow !~ m/^(?:all|neighbors)$/;
 | 
						||
 | 
						||
    # list of groups selected by user and passed in param
 | 
						||
    my $hgroup = (ref [] eq ref param('hgroup') ? param('hgroup') : [param('hgroup')]);
 | 
						||
    # list of groups validated as real host groups and named host groups
 | 
						||
    my @hgrplist = List::MoreUtils::uniq
 | 
						||
                   grep { exists setting('host_group_displaynames')->{$_} }
 | 
						||
                   grep { exists setting('host_groups')->{$_} }
 | 
						||
                   grep { defined } @{ $hgroup };
 | 
						||
 | 
						||
    # list of locations selected by user and passed in param
 | 
						||
    my $lgroup = (ref [] eq ref param('lgroup') ? param('lgroup') : [param('lgroup')]);
 | 
						||
    my @lgrplist = List::MoreUtils::uniq grep { defined } @{ $lgroup };
 | 
						||
 | 
						||
    my %clean = ();
 | 
						||
    POSITION: foreach my $pos (@$positions) {
 | 
						||
      next unless ref {} eq ref $pos;
 | 
						||
      foreach my $k (qw/ID x y/) {
 | 
						||
        next POSITION unless exists $pos->{$k};
 | 
						||
        next POSITION unless $pos->{$k} =~ m/^[[:word:]\.-]+$/;
 | 
						||
      }
 | 
						||
      $clean{$pos->{ID}} = { x => $pos->{x}, y => $pos->{y} };
 | 
						||
    }
 | 
						||
    return unless scalar keys %clean;
 | 
						||
 | 
						||
    my $posrow = schema('netdisco')->resultset('NetmapPositions')->find({
 | 
						||
      device => (($mapshow eq 'neighbors') ? $qdev->ip : undef),
 | 
						||
      host_groups => \[ '= ?', [host_groups => [sort @hgrplist]] ],
 | 
						||
      locations   => \[ '= ?', [locations   => [sort @lgrplist]] ],
 | 
						||
      vlan => ($vlan || 0),
 | 
						||
    });
 | 
						||
 | 
						||
    if ($posrow) {
 | 
						||
      $posrow->update({ positions => to_json(\%clean) });
 | 
						||
    }
 | 
						||
    else {
 | 
						||
      schema('netdisco')->resultset('NetmapPositions')->create({
 | 
						||
        device => (($mapshow eq 'neighbors') ? $qdev->ip : undef),
 | 
						||
        host_groups => [sort @hgrplist],
 | 
						||
        locations   => [sort @lgrplist],
 | 
						||
        vlan => ($vlan || 0),
 | 
						||
        positions => to_json(\%clean),
 | 
						||
      });
 | 
						||
    }
 | 
						||
};
 | 
						||
 | 
						||
# copied from SNMP::Info to avoid introducing dependency to web frontend
 | 
						||
sub munge_highspeed {
 | 
						||
    my $speed = shift;
 | 
						||
    my $fmt   = "%d Mbps";
 | 
						||
 | 
						||
    if ( $speed > 9999999 ) {
 | 
						||
        $fmt = "%d Tbps";
 | 
						||
        $speed /= 1000000;
 | 
						||
    }
 | 
						||
    elsif ( $speed > 999999 ) {
 | 
						||
        $fmt = "%.1f Tbps";
 | 
						||
        $speed /= 1000000.0;
 | 
						||
    }
 | 
						||
    elsif ( $speed > 9999 ) {
 | 
						||
        $fmt = "%d Gbps";
 | 
						||
        $speed /= 1000;
 | 
						||
    }
 | 
						||
    elsif ( $speed > 999 ) {
 | 
						||
        $fmt = "%.1f Gbps";
 | 
						||
        $speed /= 1000.0;
 | 
						||
    }
 | 
						||
    return sprintf( $fmt, $speed );
 | 
						||
}
 | 
						||
 | 
						||
sub to_speed {
 | 
						||
  my $speed = shift or return '';
 | 
						||
  ($speed = munge_highspeed($speed / 1_000_000)) =~ s/(?:\.0 |bps$)//g;
 | 
						||
  return $speed;
 | 
						||
}
 | 
						||
 | 
						||
sub make_node_infostring {
 | 
						||
  my $node = shift or return '';
 | 
						||
  my $fmt = ('<b>%s</b> is %s <b>%s %s</b><br>running <b>%s %s</b><br>Serial: <b>%s</b><br>'
 | 
						||
    .'Uptime: <b>%s</b><br>Location: <b>%s</b><br>Contact: <b>%s</b>');
 | 
						||
  return sprintf $fmt, $node->ip,
 | 
						||
    ((($node->vendor || '') =~ m/^[aeiou]/i) ? 'an' : 'a'),
 | 
						||
    ucfirst($node->vendor || ''),
 | 
						||
    map {defined $_ ? $_ : ''}
 | 
						||
    map {$node->$_}
 | 
						||
        (qw/model os os_ver serial uptime_age location contact/);
 | 
						||
}
 | 
						||
 | 
						||
sub make_link_infostring {
 | 
						||
  my $link = shift or return '';
 | 
						||
 | 
						||
  my $domain = quotemeta( setting('domain_suffix') || '' );
 | 
						||
  (my $left_name = lc($link->{left_dns} || $link->{left_name} || $link->{left_ip})) =~ s/$domain$//;
 | 
						||
  (my $right_name = lc($link->{right_dns} || $link->{right_name} || $link->{right_ip})) =~ s/$domain$//;
 | 
						||
 | 
						||
  if ($link->{aggports} == 1) {
 | 
						||
    return sprintf '<b>%s:%s</b> (%s)<br><b>%s:%s</b> (%s)',
 | 
						||
      $left_name, $link->{left_port}->[0],
 | 
						||
      ($link->{left_descr}->[0] || 'no description'),
 | 
						||
      $right_name, $link->{right_port}->[0],
 | 
						||
      ($link->{right_descr}->[0] || 'no description');
 | 
						||
  }
 | 
						||
  else {
 | 
						||
    return sprintf '<b>%s:(%s)</b><br><b>%s:(%s)</b>',
 | 
						||
      $left_name, join(',', @{$link->{left_port}}),
 | 
						||
      $right_name, join(',', @{$link->{right_port}});
 | 
						||
  }
 | 
						||
}
 | 
						||
 | 
						||
ajax '/ajax/data/device/netmap' => require_login sub {
 | 
						||
    my $q = param('q');
 | 
						||
    my $qdev = schema('netdisco')->resultset('Device')
 | 
						||
      ->search_for_device($q) or send_error('Bad device', 400);
 | 
						||
 | 
						||
    my $vlan = param('vlan');
 | 
						||
    undef $vlan if (defined $vlan and $vlan !~ m/^\d+$/);
 | 
						||
 | 
						||
    my $colorby = (param('colorby') || 'speed');
 | 
						||
    my $mapshow = (param('mapshow') || 'neighbors');
 | 
						||
    $mapshow = 'neighbors' if $mapshow !~ m/^(?:all|neighbors)$/;
 | 
						||
    $mapshow = 'all' unless $qdev->in_storage;
 | 
						||
 | 
						||
    # list of groups selected by user and passed in param
 | 
						||
    my $hgroup = (ref [] eq ref param('hgroup') ? param('hgroup') : [param('hgroup')]);
 | 
						||
    # list of groups validated as real host groups and named host groups
 | 
						||
    my @hgrplist = List::MoreUtils::uniq
 | 
						||
                   grep { exists setting('host_group_displaynames')->{$_} }
 | 
						||
                   grep { exists setting('host_groups')->{$_} }
 | 
						||
                   grep { defined } @{ $hgroup };
 | 
						||
 | 
						||
    # list of locations selected by user and passed in param
 | 
						||
    my $lgroup = (ref [] eq ref param('lgroup') ? param('lgroup') : [param('lgroup')]);
 | 
						||
    my @lgrplist = List::MoreUtils::uniq grep { defined } @{ $lgroup };
 | 
						||
 | 
						||
    my %ok_dev = ();
 | 
						||
    my %logvals = ();
 | 
						||
    my %metadata = ();
 | 
						||
    my %data = ( nodes => [], links => [] );
 | 
						||
    my $domain = quotemeta( setting('domain_suffix') || '' );
 | 
						||
 | 
						||
    # LINKS
 | 
						||
 | 
						||
    my $links = schema('netdisco')->resultset('Virtual::DeviceLinks')->search({
 | 
						||
      ($mapshow eq 'neighbors' ? ( -or => [
 | 
						||
          { left_ip  => $qdev->ip },
 | 
						||
          { right_ip => $qdev->ip },
 | 
						||
      ]) : ())
 | 
						||
    }, { result_class => 'DBIx::Class::ResultClass::HashRefInflator' });
 | 
						||
 | 
						||
    if ($vlan) {
 | 
						||
        $links = $links->search({
 | 
						||
          -or => [
 | 
						||
            { 'left_vlans.vlan' => $vlan },
 | 
						||
            { 'right_vlans.vlan' => $vlan },
 | 
						||
          ],
 | 
						||
        }, {
 | 
						||
          join => [qw/left_vlans right_vlans/],
 | 
						||
        });
 | 
						||
    }
 | 
						||
 | 
						||
    while (my $link = $links->next) {
 | 
						||
      push @{$data{'links'}}, {
 | 
						||
        FROMID => $link->{left_ip},
 | 
						||
        TOID   => $link->{right_ip},
 | 
						||
        INFOSTRING => make_link_infostring($link),
 | 
						||
        SPEED  => to_speed($link->{aggspeed}),
 | 
						||
      };
 | 
						||
 | 
						||
      ++$ok_dev{$link->{left_ip}};
 | 
						||
      ++$ok_dev{$link->{right_ip}};
 | 
						||
    }
 | 
						||
 | 
						||
    # DEVICES (NODES)
 | 
						||
 | 
						||
    my $posrow = schema('netdisco')->resultset('NetmapPositions')->find({
 | 
						||
      device => (($mapshow eq 'neighbors') ? $qdev->ip : undef),
 | 
						||
      host_groups => \[ '= ?', [host_groups => [sort @hgrplist]] ],
 | 
						||
      locations   => \[ '= ?', [locations   => [sort @lgrplist]] ],
 | 
						||
      vlan => ($vlan || 0),
 | 
						||
    });
 | 
						||
    my $pos_for = from_json( $posrow ? $posrow->positions : '{}' );
 | 
						||
 | 
						||
    my $devices = schema('netdisco')->resultset('Device')->search({}, {
 | 
						||
      '+select' => [\'floor(log(throughput.total))'], '+as' => ['log'],
 | 
						||
      join => 'throughput',
 | 
						||
    })->with_times;
 | 
						||
 | 
						||
    DEVICE: while (my $device = $devices->next) {
 | 
						||
      # if in neighbors or vlan mode then use %ok_dev to filter
 | 
						||
      next DEVICE if (($mapshow eq 'neighbors') or $vlan)
 | 
						||
        and (not $ok_dev{$device->ip});
 | 
						||
 | 
						||
      # if location picked then filter
 | 
						||
      next DEVICE if ((scalar @lgrplist) and ((!defined $device->location)
 | 
						||
        or (0 == scalar grep {$_ eq $device->location} @lgrplist)));
 | 
						||
 | 
						||
      # if host groups picked then use ACLs to filter
 | 
						||
      my $first_hgrp =
 | 
						||
        first { check_acl_only($device, setting('host_groups')->{$_}) } @hgrplist;
 | 
						||
      next DEVICE if ((scalar @hgrplist) and (not $first_hgrp));
 | 
						||
 | 
						||
      # now reset first_hgroup to be the group matching the device, if any
 | 
						||
      $first_hgrp = first { check_acl_only($device, setting('host_groups')->{$_}) }
 | 
						||
                          keys %{ setting('host_group_displaynames') || {} };
 | 
						||
 | 
						||
      ++$logvals{ $device->get_column('log') || 1 };
 | 
						||
      (my $name = lc($device->dns || $device->name || $device->ip)) =~ s/$domain$//;
 | 
						||
 | 
						||
      my %color_lkp = (
 | 
						||
        speed => (($device->get_column('log') || 1) * 1000),
 | 
						||
        hgroup => ($first_hgrp ?
 | 
						||
          setting('host_group_displaynames')->{$first_hgrp} : 'Other'),
 | 
						||
        lgroup => ($device->location || 'Other'),
 | 
						||
      );
 | 
						||
 | 
						||
      my $node = {
 | 
						||
        ID => $device->ip,
 | 
						||
        SIZEVALUE => (param('dynamicsize') ? $color_lkp{speed} : 3000),
 | 
						||
        ((exists $color_lkp{$colorby}) ? (COLORVALUE => $color_lkp{$colorby}) : ()),
 | 
						||
        LABEL => (param('showips') ? ($device->ip .' '. $name) : $name),
 | 
						||
        ORIG_LABEL => $name,
 | 
						||
        INFOSTRING => make_node_infostring($device),
 | 
						||
        LINK => uri_for('/device', {
 | 
						||
          tab => 'netmap',
 | 
						||
          q => $device->ip,
 | 
						||
          firstsearch => 'on',
 | 
						||
        })->path_query,
 | 
						||
      };
 | 
						||
 | 
						||
      if (exists $pos_for->{$device->ip}) {
 | 
						||
        $node->{'fixed'} = 1;
 | 
						||
        $node->{'x'} = $pos_for->{$device->ip}->{'x'};
 | 
						||
        $node->{'y'} = $pos_for->{$device->ip}->{'y'};
 | 
						||
      }
 | 
						||
      else {
 | 
						||
        ++$metadata{'newnodes'};
 | 
						||
      }
 | 
						||
 | 
						||
      push @{$data{'nodes'}}, $node;
 | 
						||
      $metadata{'centernode'} = $device->ip
 | 
						||
        if $qdev and $qdev->in_storage and $device->ip eq $qdev->ip;
 | 
						||
    }
 | 
						||
 | 
						||
    # to help get a sensible range of node sizes
 | 
						||
    $metadata{'numsizes'} = scalar keys %logvals;
 | 
						||
 | 
						||
    content_type('application/json');
 | 
						||
    to_json({ data => \%data, %metadata });
 | 
						||
};
 | 
						||
 | 
						||
true;
 |