From bb1f842d920c5960ea76e503a0696c6cd6691906 Mon Sep 17 00:00:00 2001
From: "Eric A. Miller" <emiller@cpan.org>
Date: Wed, 18 Sep 2013 23:35:28 -0400
Subject: [PATCH] add csv download option to device addresses

---
 .../Netdisco/Web/Plugin/Device/Addresses.pm   | 38 +++++++++++++++----
 Netdisco/share/views/ajax/device/addresses.tt |  1 +
 .../share/views/ajax/device/addresses_csv.tt  | 11 ++++++
 3 files changed, 43 insertions(+), 7 deletions(-)
 create mode 100644 Netdisco/share/views/ajax/device/addresses_csv.tt

diff --git a/Netdisco/lib/App/Netdisco/Web/Plugin/Device/Addresses.pm b/Netdisco/lib/App/Netdisco/Web/Plugin/Device/Addresses.pm
index 8fc175d2..3633b9b8 100644
--- a/Netdisco/lib/App/Netdisco/Web/Plugin/Device/Addresses.pm
+++ b/Netdisco/lib/App/Netdisco/Web/Plugin/Device/Addresses.pm
@@ -7,22 +7,46 @@ use Dancer::Plugin::Auth::Extensible;
 
 use App::Netdisco::Web::Plugin;
 
-register_device_tab({ tag => 'addresses', label => 'Addresses' });
+register_device_tab( { tag => 'addresses', label => 'Addresses' } );
 
 # device interface addresses
 ajax '/ajax/content/device/addresses' => require_login sub {
     my $q = param('q');
 
-    my $device = schema('netdisco')->resultset('Device')
-      ->search_for_device($q) or send_error('Bad device', 400);
+    my $device
+        = schema('netdisco')->resultset('Device')->search_for_device($q)
+        or send_error( 'Bad device', 400 );
 
-    my $set = $device->device_ips->search({}, {order_by => 'alias'});
+    my $set = $device->device_ips->search( {}, { order_by => 'alias' } );
     return unless $set->count;
 
     content_type('text/html');
-    template 'ajax/device/addresses.tt', {
-      results => $set,
-    }, { layout => undef };
+    template 'ajax/device/addresses.tt', { results => $set, },
+        { layout => undef };
+};
+
+get '/device/addresses' => require_login sub {
+    my $q      = param('q');
+    my $format = param('format');
+
+    my $device
+        = schema('netdisco')->resultset('Device')->search_for_device($q)
+        or send_error( 'Bad device', 400 );
+
+    my $set = $device->device_ips->search( {}, { order_by => 'alias' } );
+    return unless $set->count;
+
+    if ( $format eq 'csv' ) {
+
+        header( 'Content-Type' => 'text/comma-separated-values' );
+        header( 'Content-Disposition' =>
+                "attachment; filename=\"addresses.csv\"" );
+        template 'ajax/device/addresses_csv.tt', { results => $set, },
+            { layout => undef };
+    }
+    else {
+        return;
+    }
 };
 
 true;
diff --git a/Netdisco/share/views/ajax/device/addresses.tt b/Netdisco/share/views/ajax/device/addresses.tt
index 36231a3c..b6c46dfa 100644
--- a/Netdisco/share/views/ajax/device/addresses.tt
+++ b/Netdisco/share/views/ajax/device/addresses.tt
@@ -1,3 +1,4 @@
+[% INCLUDE "download_as.tt" %]
 <table class="table table-bordered table-condensed table-striped nd_floatinghead">
   <thead>
     <tr>
diff --git a/Netdisco/share/views/ajax/device/addresses_csv.tt b/Netdisco/share/views/ajax/device/addresses_csv.tt
new file mode 100644
index 00000000..6c155605
--- /dev/null
+++ b/Netdisco/share/views/ajax/device/addresses_csv.tt
@@ -0,0 +1,11 @@
+[% USE CSV -%]
+[% CSV.dump([ 'Address' 'DNS' 'Interface' 'Description' 'Prefix' ]) %]
+
+[% WHILE (row = results.next) %]
+  [% mylist = [] %]
+  [% FOREACH col IN [ row.alias row.dns row.port row.device_port.name row.subnet ] %]
+    [% mylist.push(col) %]
+  [% END %]
+  [% CSV.dump(mylist) %]
+
+[% END %]
\ No newline at end of file