implement ip/host search for node, and vendor column

Netdisco/lib/Netdisco/DB/Result/NodeIp.pm

@@ -39,15 +39,25 @@ __PACKAGE__->set_primary_key("mac", "ip");
 # Created by DBIx::Class::Schema::Loader v0.07015 @ 2012-01-07 14:20:02
 # DO NOT MODIFY THIS OR ANYTHING ABOVE! md5sum:9+CuvuVWH88WxAf6IBij8g
 
-__PACKAGE__->has_many( nodeips => 'Netdisco::DB::Result::NodeIp',
+__PACKAGE__->belongs_to( oui => 'Netdisco::DB::Result::Oui',
+    sub {
+        my $args = shift;
+        return {
+            "$args->{foreign_alias}.oui" =>
+              { '=' => \"substring(cast($args->{self_alias}.mac as varchar) for 8)" }
+        };
+    }
+);
+
+__PACKAGE__->has_many( node_ips => 'Netdisco::DB::Result::NodeIp',
   { 'foreign.mac' => 'self.mac' } );
 __PACKAGE__->has_many( nodes => 'Netdisco::DB::Result::Node',
   { 'foreign.mac' => 'self.mac' } );
 
-sub tidy_nodeips {
+sub ip_aliases {
     my ($row, $archive) = @_;
 
-    return $row->nodeips(
+    return $row->node_ips(
       {
         ip  => { '!=' => $row->ip },
         ($archive ? () : (active => 1)),
@@ -64,7 +74,7 @@ sub tidy_nodeips {
     );
 }
 
-sub tidy_nodes {
+sub node_sightings {
     my ($row, $archive) = @_;
 
     return $row->nodes(

Netdisco/lib/Netdisco/DB/ResultSet/NodeIp.pm

@@ -1,24 +1,46 @@
 package Netdisco::DB::ResultSet::NodeIp;
 use base 'DBIx::Class::ResultSet';
 
+my $search_attr = {
+    order_by => {'-desc' => 'time_last'},
+    columns => [qw/ mac ip dns active oui.company /],
+    '+select' => [
+      \"to_char(time_first, 'YYYY-MM-DD HH24:MI')",
+      \"to_char(time_last, 'YYYY-MM-DD HH24:MI')",
+    ],
+    '+as' => [qw/ time_first time_last /],
+    join => 'oui'
+};
+
 sub by_ip {
-    my ($set, $ip, $archive) = @_;
+    my ($set, $archive, $ip) = @_;
     return $set unless $ip;
 
+    my $op = '=';
+    if ('NetAddr::IP::Lite' eq ref $ip) {
+        $op = '<<=' if $ip->num > 1;
+        $ip = $ip->cidr;
+    }
+
     return $set->search(
       {
-        ip => $ip,
+        ip => { $op => $ip },
         ($archive ? () : (active => 1)),
       },
+      $search_attr,
+    );
+}
+
+sub by_name {
+    my ($set, $archive, $name) = @_;
+    return $set unless $name;
+
+    return $set->search(
       {
-        order_by => {'-desc' => 'time_last'},
+        dns => { '-ilike' => $name },
-        columns => [qw/ mac ip dns active /],
+        ($archive ? () : (active => 1)),
-        '+select' => [
-          \"to_char(time_first, 'YYYY-MM-DD HH24:MI')",
-          \"to_char(time_last, 'YYYY-MM-DD HH24:MI')",
-        ],
-        '+as' => [qw/ time_first time_last /],
       },
+      $search_attr,
     );
 }
 

Netdisco/lib/Netdisco/Web.pm

@@ -3,7 +3,11 @@ package Netdisco::Web;
 use Dancer ':syntax';
 use Dancer::Plugin::Ajax;
 use Dancer::Plugin::DBIC;
+
 use Digest::MD5 ();
+use Socket6 ();
+use NetAddr::IP::Lite ':lower';
+use Regexp::Common 'net';
 
 hook 'before' => sub {
     if (! session('user') && request->path !~ m{^/login}) {
@@ -11,6 +15,16 @@ hook 'before' => sub {
         #var(requested_path => request->path);
         #request->path_info('/');
     }
+
+    # set up default search options for each type
+    if (not param('tab') or param('tab') ne 'node') {
+        params->{'stamps'} = 'checked';
+    }
+
+    # set up query string defaults for templates
+    var('query_defaults' => { map { ($_ => "tab=$_") } qw/node/ });
+    var('query_defaults')->{node} .= "\&$_=". (param($_) || '')
+      for qw/stamps vendor archived partial/;
 };
 
 get '/' => sub {
@@ -26,15 +40,35 @@ ajax '/ajax/content/search/:thing' => sub {
 ajax '/ajax/content/search/node' => sub {
     my $node = param('q');
     return unless $node;
+    my $set;
 
-    my $set = schema('netdisco')->resultset('NodeIp')
+    # if mac
-      ->by_ip($node, param('archived'));
+      # search on mac
-    return unless $set->count;
+    # try to make ip
+      # search on ip
+    # text search for node dns
 
-    content_type('text/html');
+    if ($node =~ m/^$RE{net}{MAC}$/) {
-    template 'ajax/node.tt', {
+    }
-      results => $set,
+    else {
-    }, { layout => undef };
+        if (my $ip = NetAddr::IP::Lite->new($node)) {
+            # by_ip() will extract cidr notation if necessary
+            $set = schema('netdisco')->resultset('NodeIp')
+              ->by_ip(param('archived'), $ip);
+            return unless $set->count;
+        }
+        else {
+            $node = "\%$node\%" if param('partial');
+            $set = schema('netdisco')->resultset('NodeIp')
+              ->by_name(param('archived'), $node);
+            return unless $set->count;
+        }
+
+        content_type('text/html');
+        template 'ajax/node.tt', {
+          results => $set,
+        }, { layout => undef };
+    }
 };
 
 # devices carrying vlan xxx
@@ -48,27 +82,10 @@ ajax '/ajax/content/search/vlan' => sub {
     content_type('text/html');
     template 'ajax/vlan.tt', {
       results => $set,
-      columns => [
-        { key => 'dns', label => 'Device' },
-        { key => 'vlan.description', label => 'Description' },
-        { key => 'model', label => 'Model' },
-        { key => 'os', label => 'OS' },
-        { key => 'vendor', label => 'Vendor' },
-      ],
-      hyperlink => sub {
-          my $row = shift;
-          return '/device?q='. $row->ip .'&vlan='. $vlan;
-      },
     }, { layout => undef };
 };
 
 get '/search' => sub {
-    # set up default search options for each type
-    if (not param('tab') or param('tab') ne 'node') {
-        params->{'stamps'} = 'checked';
-        params->{'vendor'} = 'checked';
-    }
-
     my $q = param('q');
     if ($q and not param('tab')) {
         # pick most likely tab for initial results

Netdisco/views/ajax/node.tt

@@ -2,6 +2,9 @@
   <thead>
     <tr>
       <th>MAC</th>
+      [% IF params.vendor %]
+      <th>Vendor</th>
+      [% END %]
       <th>Match</th>
       <th>Device or Node</th>
       [% IF params.stamps %]
@@ -13,10 +16,14 @@
   </tbody>
     [% WHILE (row = results.next) %]
     <tr>
-      <td><a class="nd_linkcell" href="/search?tab=node&stamps=on&vendor=on&q=[% row.mac | uri %]">[% row.mac %]</a></td>
+      <td><a class="nd_linkcell"
+        href="/search?[% vars.query_defaults.node %]&q=[% row.mac | uri %]">[% row.mac %]</a></td>
+      [% IF params.vendor %]
+      <td>[% row.oui.company %]</td>
+      [% END %]
       <td>IP &rarr; MAC</td>
-      <td><a href="/search?tab=node&stamps=on&vendor=on&q=[% row.ip | uri %]">[% row.ip %]</a>
+      <td><a href="/search?[% vars.query_defaults.node %]&q=[% row.ip | uri %]">[% row.ip %]</a>
-        [% ' (' _ row.dns _ ')' IF row.dns %]
+        [% ' (' _ row.dns.remove(settings.domain_suffix) _ ')' IF row.dns %]
         [% ' <span class="label warning">a</span>' IF NOT row.active %]
       </td>
       [% IF params.stamps %]
@@ -24,12 +31,15 @@
       <td>[% row.time_last %]</td>
       [% END %]
     </tr>
-    [% FOREACH node IN row.tidy_nodes(params.archived) %]
+    [% FOREACH node IN row.node_sightings(params.archived) %]
     <tr>
       <td>&nbsp;</td>
+      [% IF params.vendor %]
+      <td>&nbsp;</td>
+      [% END %]
       <td>Switch Port</td>
       <td><a href="/device?q=[% node.switch | url %]&port=[% node.port | url %]">[% node.switch %] [ [% node.port %] ]</a>
-        [% ' (' _ node.device.dns _ ')' IF node.device.dns %]
+        [% ' (' _ node.device.dns.remove(settings.domain_suffix) _ ')' IF node.device.dns %]
         [% ' <span class="label warning">a</span>' IF NOT node.active %]
       </td>
       [% IF params.stamps %]
@@ -38,12 +48,15 @@
       [% END %]
     </tr>
     [% END %]
-    [% FOREACH nodeip IN row.tidy_nodeips(params.archived) %]
+    [% FOREACH nodeip IN row.ip_aliases(params.archived) %]
     <tr>
       <td>&nbsp;</td>
+      [% IF params.vendor %]
+      <td>&nbsp;</td>
+      [% END %]
       <td>MAC &rarr; IP</td>
-      <td><a href="/search?tab=node&stamps=on&vendor=on&q=[% nodeip.ip | uri %]">[% nodeip.ip %]</a>
+      <td><a href="/search?[% vars.query_defaults.node %]&q=[% nodeip.ip | uri %]">[% nodeip.ip %]</a>
-        [% ' (' _ nodeip.dns _ ')' IF nodeip.dns %]
+        [% ' (' _ nodeip.dns.remove(settings.domain_suffix) _ ')' IF nodeip.dns %]
         [% ' <span class="label warning">a</span>' IF NOT nodeip.active %]
       </td>
       [% IF params.stamps %]

Netdisco/views/ajax/vlan.tt

@@ -1,21 +1,26 @@
 <table class="bordered-table condensed-table zebra-striped">
   <thead>
     <tr>
-      [% FOREACH col IN columns %]
+      <th>Device</th>
-      <th>[% col.label %]</th>
+      <th>Description</th>
-      [% END %]
+      <th>Model</th>
+      <th>OS</th>
+      <th>Vendor</th>
     </tr>
   </thead>
   </tbody>
     [% WHILE (row = results.next) %]
     <tr>
-      [% FOREACH col IN columns %]
+      <td><a class="nd_linkcell"
-        [% SET val = row %]
+        href="/device?q=[% row.ip %]&vlan=[% params.q %]">[% row.dns %]</a></td>
-        [% FOREACH method IN col.key.split('\.') %]
+      <td><a class="nd_linkcell nd_stealthlink"
-          [% SET val = val.$method %]
+        href="/device?q=[% row.ip %]&vlan=[% params.q %]">[% row.vlan.description %]</a></td>
-        [% END %]
+      <td><a class="nd_linkcell nd_stealthlink"
-        <td><a class="nd_linkcell nd_stealthlink" href="[% hyperlink(row) %]">[% val %]</a></td>
+        href="/device?q=[% row.ip %]&vlan=[% params.q %]">[% row.model %]</a></td>
-      [% END %]
+      <td><a class="nd_linkcell nd_stealthlink"
+        href="/device?q=[% row.ip %]&vlan=[% params.q %]">[% row.os %]</a></td>
+      <td><a class="nd_linkcell nd_stealthlink"
+        href="/device?q=[% row.ip %]&vlan=[% params.q %]">[% row.vendor %]</a></td>
     </tr>
     [% END %]
   </tbody>

Netdisco/views/inc/search/node.tt

@@ -24,3 +24,11 @@
                 <span class="nd_searchcheckbox uneditable-input">Archived Data <span class="nd_legendlabel label warning">a</span></span>
               </label>
             </div>
+            <div class="clearfix input-prepend">
+              <label class="add-on">
+                <input type="checkbox" id="partial" name="partial"[% ' checked="checked"' IF params.partial %]/>
+              </label>
+              <label for="partial">
+                <span class="nd_searchcheckbox uneditable-input">Partial Name</span>
+              </label>
+            </div>