From e8ad84276f28ae681dda3a0c04e33d9c2a0217e1 Mon Sep 17 00:00:00 2001
From: Sean McArthur <sean.monstar@gmail.com>
Date: Mon, 19 Sep 2011 18:14:32 -0500
Subject: [PATCH] adding a /services/admin page

This page requires superuser access, and only provides buttons which
trigger hardcoded tasks to be run, such as indexing all the Packages again.
---
 apps/base/templates/admin.html | 30 ++++++++++++++++++++++++++++++
 apps/base/urls.py              |  1 +
 apps/base/views.py             | 26 +++++++++++++++++++++++---
 media/jetpack/css/UI.Forms.css |  3 ++-
 4 files changed, 56 insertions(+), 4 deletions(-)
 create mode 100644 apps/base/templates/admin.html

diff --git a/apps/base/templates/admin.html b/apps/base/templates/admin.html
new file mode 100644
index 00000000..d937588c
--- /dev/null
+++ b/apps/base/templates/admin.html
@@ -0,0 +1,30 @@
+{% extends 'base.html' %}
+
+{% block title %}Mozilla FlightDeck Admin Area{% endblock title %}
+ 
+{% block head_prejs %}
+<style type="text/css">
+    .UI_Forms .UI_Form_Actions li {
+        float:none;
+        display:inline-block;
+    }
+</style>
+{% endblock %}
+
+{% block app_content %}
+<form method="post" action="" class="UI_Forms">
+<h2 class="UI_Heading" style="padding-top:3em;">Search</h2>
+{% if message %}
+    <p><strong>{{message}}</strong></p>
+{% endif %}
+    {% csrf_token %}
+    <ul class="UI_Form_Actions">
+        <li>
+            <button name="action" value="setup_mapping" type="submit">Setup Mapping</button>
+        </li>
+        <li>
+            <button name="action" value="index_all" type="submit">Index All</button>
+        </li>
+</form>
+
+{% endblock %}
diff --git a/apps/base/urls.py b/apps/base/urls.py
index 6e7e5ff7..7f5f9214 100644
--- a/apps/base/urls.py
+++ b/apps/base/urls.py
@@ -6,4 +6,5 @@ urlpatterns = patterns('',
         name='graphite'),
     url('^services/monitor$', 'base.views.monitor', name='monitor'),
     url('^services/settings$', 'base.views.site_settings', name='settings'),
+    url('^services/admin$', 'base.views.admin', name='admin_triggers'),
 )
diff --git a/apps/base/views.py b/apps/base/views.py
index e8acbb5d..093917b2 100644
--- a/apps/base/views.py
+++ b/apps/base/views.py
@@ -10,10 +10,11 @@ from django.shortcuts import render_to_response
 from django.template import RequestContext, loader
 from django.views.debug import get_safe_settings
 
-from jetpack.models import Package, SDK
-import base.tasks
-from base.models import CeleryResponse
 from elasticutils import get_es
+import base.tasks
+from search.cron import index_all, setup_mapping
+from jetpack.models import Package, SDK
+from base.models import CeleryResponse
 
 log = commonware.log.getLogger('f.monitor')
 
@@ -44,6 +45,25 @@ def site_settings(request):
         {'settings': safe},
         context_instance=RequestContext(request))
 
+@user_passes_test(lambda u: u.is_superuser)
+def admin(request):
+    msg = ''
+    if request.method == 'POST':
+        action = request.POST.get('action')
+        log_msg = '[admin] %s by %s (id: %d)'
+        if action == 'setup_mapping':
+            msg = 'setup_mapping triggered'
+            log.info(log_msg % (msg, request.user, request.user.pk))
+            setup_mapping()
+        elif action == 'index_all':
+            msg = 'index_all triggered'
+            log.info(log_msg % (msg , request.user, request.user.pk))
+            index_all()
+
+    return render_to_response('admin.html', {
+            'message': msg
+        }, context_instance=RequestContext(request))
+
 
 def monitor(request):
     status = True
diff --git a/media/jetpack/css/UI.Forms.css b/media/jetpack/css/UI.Forms.css
index eb10b889..b4af48ef 100644
--- a/media/jetpack/css/UI.Forms.css
+++ b/media/jetpack/css/UI.Forms.css
@@ -111,7 +111,8 @@ authors:
 			-moz-border-radius: 6px;
 		}
 
-			.UI_Forms .UI_Form_Actions li input {
+			.UI_Forms .UI_Form_Actions li input,
+            .UI_Forms .UI_Form_Actions li button {
 				background: -moz-linear-gradient(270deg, #C5F3AE, #96CE79);
 				border-top: solid 1px #fff;
 				display: block;