From 0c81f212b60c94294f74915c94d2df37cdb89558 Mon Sep 17 00:00:00 2001
From: Kyle Drake <kyle@kyledrake.net>
Date: Tue, 29 Dec 2015 22:48:46 -0600
Subject: [PATCH] further improvements to site inspector

---
 app/admin.rb         |  1 +
 views/admin/site.erb | 15 ++++++++++++---
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/app/admin.rb b/app/admin.rb
index 2431f1cd..80f751d8 100644
--- a/app/admin.rb
+++ b/app/admin.rb
@@ -14,6 +14,7 @@ end
 get '/admin/site/:username' do |username|
   require_admin
   @site = Site[username: username]
+  not_found if @site.nil?
   @title = "Site Inspector - #{@site.username}"
   erb :'admin/site'
 end
diff --git a/views/admin/site.erb b/views/admin/site.erb
index 859ef90b..957d376a 100644
--- a/views/admin/site.erb
+++ b/views/admin/site.erb
@@ -7,7 +7,7 @@
 
 <div class="content misc-page single-Col txt-Center" style="padding-top: 20px;">
 
-  <div>
+  <div style="float: left;">
     <form action="/admin/banhammer" target="_blank" method="POST">
       <%== csrf_token_input_html %>
       <input type="hidden" name="username" value="<%= @site.username %>">
@@ -21,6 +21,15 @@
     </form>
   </div>
 
+  <div style="float: right;">
+    <form method="GET" onsubmit="window.location.href = '/admin/site/'+$('#sitename').val(); return false">
+      <input id="sitename" type="text" placeholder="sitename" autocomplete="off"><br>
+      <input type="submit" class="btn-Action" value="Lookup">
+    </form>
+  </div>
+
+  <div style="clear: both"></div>
+
   <h3>Site Pages</h3>
 
   <table>
@@ -49,7 +58,7 @@
             <button>Ham</button>
           </form>
 
-          <form action="/admin/site_files/train" target="_blank" method="POST" onsubmit="return confirm('Confirm spam train of <%= "#{@site.username}/#{site_file.path}" %>');">
+          <form action="/admin/site_files/train" target="_blank" method="POST">
             <%== csrf_token_input_html %>
             <input type="hidden" name="site_id" value="<%= @site.id %>">
             <input type="hidden" name="path" value="<%= site_file.path %>">
@@ -57,7 +66,7 @@
             <button>Spam</button>
           </form>
 
-          <form action="/admin/site_files/train" target="_blank" method="POST" onsubmit="return confirm('Confirm phishing train of <%= "#{@site.username}/#{site_file.path}" %>');">
+          <form action="/admin/site_files/train" target="_blank" method="POST">
             <%== csrf_token_input_html %>
             <input type="hidden" name="site_id" value="<%= @site.id %>">
             <input type="hidden" name="path" value="<%= site_file.path %>">