From f8d1c5b2c6f9351f300c596bc51d38d793b9e389 Mon Sep 17 00:00:00 2001
From: Igor Pashev <pashev.igor@gmail.com>
Date: Wed, 15 Jun 2016 15:44:25 +0800
Subject: Use HEAD /server/:server/processlist.json

To see which servers are really allowed by Sproxy
---
 src/Application.hs | 22 ++++++++++++++++------
 static/mywatch.js  | 40 ++++++++++++++++++++++++++++------------
 2 files changed, 44 insertions(+), 18 deletions(-)

diff --git a/src/Application.hs b/src/Application.hs
index 63aa258..899893e 100644
--- a/src/Application.hs
+++ b/src/Application.hs
@@ -18,14 +18,14 @@ import Data.Pool (Pool, withResource)
 import Data.Text.Lazy (Text)
 import Database.MySQL.Simple (Connection, query_)
 import GHC.Generics (Generic)
-import Network.HTTP.Types (notFound404)
+import Network.HTTP.Types (ok200, notFound404, StdMethod(HEAD))
 import Network.Wai (Application, Middleware)
 import Network.Wai.Middleware.RequestLogger (Destination(Handle),
   mkRequestLogger, RequestLoggerSettings(destination, outputFormat),
   OutputFormat(CustomOutputFormat))
 import Network.Wai.Middleware.Static (addBase, hasPrefix, staticPolicy, (>->))
 import System.IO (stderr)
-import Web.Scotty (ScottyM, ActionM, middleware, json, file, get,
+import Web.Scotty (ScottyM, ActionM, middleware, json, file, addroute, get,
   status, text, param, scottyApp)
 import qualified Data.HashMap.Lazy as HM
 
@@ -51,7 +51,10 @@ myProcess ps logger dataDir = do
   get "/index.html" $ file index_html
 
   get "/serverlist.json" $ json (sort $ HM.keys ps)
-  get "/server/:server/processlist.json" $ apiGetProcesses ps
+  get "/server/:server/processlist.json" $ apiGetProcessList ps
+
+  -- Used by client to see which servers are really allowed by Sproxy
+  addroute HEAD "/server/:server/processlist.json" $ apiCanProcessList ps
 
 data Process = Process {
     id      :: Int
@@ -65,12 +68,19 @@ data Process = Process {
 } deriving (Generic)
 instance ToJSON Process
 
-apiGetProcesses :: Pools -> ActionM ()
-apiGetProcesses ps = do
+apiCanProcessList :: Pools -> ActionM ()
+apiCanProcessList ps = do
+  server <- param "server"
+  case HM.lookup server ps of
+    Nothing -> status notFound404 >> text server
+    Just _  -> status ok200
+
+apiGetProcessList :: Pools -> ActionM ()
+apiGetProcessList ps = do
   server <- param "server"
   case HM.lookup server ps of
     Nothing -> status notFound404 >> text server
-    Just p -> do
+    Just p  -> do
       res <- withDB p $ \c ->
               query_ c "SELECT \
               \id, user, host, db, command, time, state, info \
diff --git a/static/mywatch.js b/static/mywatch.js
index c215a3c..7ea728e 100644
--- a/static/mywatch.js
+++ b/static/mywatch.js
@@ -56,20 +56,36 @@ $(function() {
     method: "GET",
     error: commonError,
     success: function(servers) {
+      var total = servers.length;
+      var available = [];
+      var checked = 0;
       $.each(servers, function(i, s) {
-        $('#serverList>ul')
-          .append('<li><a href="#">' + s +
-            '</a></li>')
-      });
-      $("#serverList a").on("click", function() {
-        var server = $(this).text();
-        $(this).parent().parent().find('.active').removeClass('active');
-        $(this).parent().addClass('active');
-        clearInterval(interval);
-        getProcessList(server);
-        interval = setInterval(getProcessList, 60 * 1000, server);
+        $.ajax({
+          url: "server/" + s + "/processlist.json",
+          method: "HEAD",
+          success: function() {
+            available.push(s);
+          },
+          complete: function() {
+            var menu = $('#serverList>ul');
+            checked++;
+            if (checked === total) {
+              $.each(available.sort(), function(i, s) {
+                menu.append('<li><a href="#">' + s + '</a></li>')
+              });
+              $("#serverList a").on("click", function() {
+                var server = $(this).text();
+                $(this).parent().parent().find('.active').removeClass('active');
+                $(this).parent().addClass('active');
+                clearInterval(interval);
+                getProcessList(server);
+                interval = setInterval(getProcessList, 60 * 1000, server);
+              });
+              info.hide();
+            }
+          }
+        });
       });
-      info.hide();
     }
   });
 });
-- 
cgit v1.2.3