From 0f608f3a15edf004e8e2785e1aff066e022d51a3 Mon Sep 17 00:00:00 2001
From: Jeremy Stretch <jstretch@digitalocean.com>
Date: Mon, 17 Jul 2017 13:19:25 -0400
Subject: [PATCH] Added device config view

---
 netbox/dcim/urls.py                          |  1 +
 netbox/dcim/views.py                         | 18 ++++++-
 netbox/templates/dcim/device_config.html     | 53 ++++++++++++++++++++
 netbox/templates/dcim/device_status.html     |  2 +-
 netbox/templates/dcim/inc/device_header.html |  5 +-
 5 files changed, 74 insertions(+), 5 deletions(-)
 create mode 100644 netbox/templates/dcim/device_config.html

diff --git a/netbox/dcim/urls.py b/netbox/dcim/urls.py
index 6adbc4dae..172f634fb 100644
--- a/netbox/dcim/urls.py
+++ b/netbox/dcim/urls.py
@@ -124,6 +124,7 @@ urlpatterns = [
     url(r'^devices/(?P<pk>\d+)/inventory/$', views.DeviceInventoryView.as_view(), name='device_inventory'),
     url(r'^devices/(?P<pk>\d+)/status/$', views.DeviceStatusView.as_view(), name='device_status'),
     url(r'^devices/(?P<pk>\d+)/lldp-neighbors/$', views.DeviceLLDPNeighborsView.as_view(), name='device_lldp_neighbors'),
+    url(r'^devices/(?P<pk>\d+)/config/$', views.DeviceConfigView.as_view(), name='device_config'),
     url(r'^devices/(?P<pk>\d+)/add-secret/$', secret_add, name='device_addsecret'),
     url(r'^devices/(?P<device>\d+)/services/assign/$', ServiceCreateView.as_view(), name='service_assign'),
     url(r'^devices/(?P<object_id>\d+)/images/add/$', ImageAttachmentEditView.as_view(), name='device_add_image', kwargs={'model': Device}),
diff --git a/netbox/dcim/views.py b/netbox/dcim/views.py
index fdadc679d..ea07138d5 100644
--- a/netbox/dcim/views.py
+++ b/netbox/dcim/views.py
@@ -921,7 +921,8 @@ class DeviceInventoryView(View):
         })
 
 
-class DeviceStatusView(View):
+class DeviceStatusView(PermissionRequiredMixin, View):
+    permission_required = 'dcim.napalm_read'
 
     def get(self, request, pk):
 
@@ -932,7 +933,8 @@ class DeviceStatusView(View):
         })
 
 
-class DeviceLLDPNeighborsView(View):
+class DeviceLLDPNeighborsView(PermissionRequiredMixin, View):
+    permission_required = 'dcim.napalm_read'
 
     def get(self, request, pk):
 
@@ -951,6 +953,18 @@ class DeviceLLDPNeighborsView(View):
         })
 
 
+class DeviceConfigView(PermissionRequiredMixin, View):
+    permission_required = 'dcim.napalm_read'
+
+    def get(self, request, pk):
+
+        device = get_object_or_404(Device, pk=pk)
+
+        return render(request, 'dcim/device_config.html', {
+            'device': device,
+        })
+
+
 class DeviceCreateView(PermissionRequiredMixin, ObjectEditView):
     permission_required = 'dcim.add_device'
     model = Device
diff --git a/netbox/templates/dcim/device_config.html b/netbox/templates/dcim/device_config.html
new file mode 100644
index 000000000..b62ff0211
--- /dev/null
+++ b/netbox/templates/dcim/device_config.html
@@ -0,0 +1,53 @@
+{% extends '_base.html' %}
+{% load staticfiles %}
+
+{% block title %}{{ device }} - Config{% endblock %}
+
+{% block content %}
+    {% include 'inc/ajax_loader.html' %}
+    {% include 'dcim/inc/device_header.html' with active_tab='config' %}
+    <div class="row">
+        <div class="col-md-10 col-md-offset-1">
+            <div class="panel panel-default">
+                <div class="panel-heading"><strong>Device Configuration</strong></div>
+                <div class="panel-body">
+                    <ul class="nav nav-tabs" role="tablist">
+                        <li role="presentation" class="active"><a href="#running" aria-controls="running" role="tab" data-toggle="tab">Running</a></li>
+                        <li role="presentation"><a href="#startup" aria-controls="startup" role="tab" data-toggle="tab">Startup</a></li>
+                        <li role="presentation"><a href="#candidate" aria-controls="candidate" role="tab" data-toggle="tab">Candidate</a></li>
+                    </ul>
+                    <div class="tab-content">
+                        <div role="tabpanel" class="tab-pane active" id="running">
+                            <pre id="running_config"></pre>
+                        </div>
+                        <div role="tabpanel" class="tab-pane" id="startup">
+                            <pre id="startup_config"></pre>
+                        </div>
+                        <div role="tabpanel" class="tab-pane" id="candidate">
+                            <pre id="candidate_config"></pre>
+                        </div>
+                    </div>
+                </div>
+            </div>
+        </div>
+    </div>
+{% endblock %}
+
+{% block javascript %}
+<script type="text/javascript">
+$(document).ready(function() {
+    $.ajax({
+        url: "{% url 'dcim-api:device-napalm' pk=device.pk %}?method=get_config",
+        dataType: 'json',
+        success: function(json) {
+            $('#running_config').html($.trim(json['get_config']['running']));
+            $('#startup_config').html($.trim(json['get_config']['startup']));
+            $('#candidate_config').html($.trim(json['get_config']['candidate']));
+        },
+        error: function(xhr) {
+            alert(xhr.responseText);
+        }
+    });
+});
+</script>
+{% endblock %}
diff --git a/netbox/templates/dcim/device_status.html b/netbox/templates/dcim/device_status.html
index 5e5dab168..4d3c9ba78 100644
--- a/netbox/templates/dcim/device_status.html
+++ b/netbox/templates/dcim/device_status.html
@@ -1,7 +1,7 @@
 {% extends '_base.html' %}
 {% load staticfiles %}
 
-{% block title %}{{ device }} - NAPALM{% endblock %}
+{% block title %}{{ device }} - Status{% endblock %}
 
 {% block content %}
     {% include 'inc/ajax_loader.html' %}
diff --git a/netbox/templates/dcim/inc/device_header.html b/netbox/templates/dcim/inc/device_header.html
index 8a807873a..b9eb7fb19 100644
--- a/netbox/templates/dcim/inc/device_header.html
+++ b/netbox/templates/dcim/inc/device_header.html
@@ -45,8 +45,9 @@
 <ul class="nav nav-tabs" style="margin-bottom: 20px">
     <li role="presentation"{% if active_tab == 'info' %} class="active"{% endif %}><a href="{% url 'dcim:device' pk=device.pk %}">Info</a></li>
     <li role="presentation"{% if active_tab == 'inventory' %} class="active"{% endif %}><a href="{% url 'dcim:device_inventory' pk=device.pk %}">Inventory</a></li>
-    <li role="presentation"{% if active_tab == 'status' %} class="active"{% endif %}><a href="{% url 'dcim:device_status' pk=device.pk %}">Status</a></li>
-    {% if device.status == 1 and device.platform.rpc_client and device.primary_ip %}
+    {% if perms.dcim.napalm_read and device.status == 1 and device.platform.napalm_driver and device.primary_ip %}
+        <li role="presentation"{% if active_tab == 'status' %} class="active"{% endif %}><a href="{% url 'dcim:device_status' pk=device.pk %}">Status</a></li>
         <li role="presentation"{% if active_tab == 'lldp-neighbors' %} class="active"{% endif %}><a href="{% url 'dcim:device_lldp_neighbors' pk=device.pk %}">LLDP Neighbors</a></li>
+        <li role="presentation"{% if active_tab == 'config' %} class="active"{% endif %}><a href="{% url 'dcim:device_config' pk=device.pk %}">Configuration</a></li>
     {% endif %}
 </ul>