From f2071a761fb549091880ed6fd490e16de43f84cb Mon Sep 17 00:00:00 2001
From: Neeraj <neeraj.viman@gmail.com>
Date: Mon, 16 Mar 2015 22:34:49 +0530
Subject: [PATCH] Frontend: Enable global search

---
 client.go         |   2 +-
 static/index.html |   7 ++-
 static/js/app.js  | 137 ++++++++++++++++++++++++++++++++++++++++------
 utils.go          |   2 +-
 4 files changed, 127 insertions(+), 21 deletions(-)

diff --git a/client.go b/client.go
index 06ff83f..da2cbdc 100644
--- a/client.go
+++ b/client.go
@@ -42,7 +42,7 @@ func NewClientFromURL(url string) (*Client, error) {
 		return nil, err
 	}
 
-	user, host := getHostUser(url)
+	user, host := getHostUserFromConnString(url)
 
 	return &Client{db: db, host: host, user: user}, nil
 }
diff --git a/static/index.html b/static/index.html
index cd086da..c407dca 100644
--- a/static/index.html
+++ b/static/index.html
@@ -30,8 +30,11 @@
         <div class="wrap">
           <div class="title">Database <i id="refresh-list" class="fa fa-refresh"></i></div>
           <div class="search">
-            <div class="form-group">
-              <input type="text" class="form-control" placeholder="Search for...">
+            <div class="input-group">
+              <input type="text" class="form-control" id="txtSearch" placeholder="Search for...">
+              <span class="input-group-btn">
+                <button class="btn btn-default" id="btnSearchDB" type="button">Go!</button>
+              </span>
             </div><!-- /input-group -->
           </div>
           <div id="database-tree">
diff --git a/static/js/app.js b/static/js/app.js
index 1f2c6da..bc6c97a 100644
--- a/static/js/app.js
+++ b/static/js/app.js
@@ -118,6 +118,9 @@ function getViewDefiniton(dbName, viewName, cb) {
 function closeConnection(cb) {
   apiCall("delete", "/disconnect", {}, cb);
 }
+function apiSearchDatabase(query, cb) {
+  apiCall("get", "/search/" + query, {}, cb);
+}
 
 
 var fnGetSelectedTable = function() {
@@ -298,7 +301,6 @@ var showAlterDBPopup = function(nodeName) {
       }
     });
   });
-
 };
 
 var showDropDBPopup = function(dbName) {
@@ -1118,7 +1120,7 @@ function initEditor(editorId, editorData) {
     exec: function(editor) {
       runQuery(editor);
     }
-    }, {
+  }, {
     name: "explain_query",
     bindKey: {
       win: "Ctrl-E",
@@ -1362,20 +1364,20 @@ $(document).ready(function() {
     apiCall("post", "/connect", {
       url: url
     }, function(resp) {
-        button.prop("disabled", false).text("Connect");
+      button.prop("disabled", false).text("Connect");
 
-        if (resp.error) {
-          connected = false;
-          $("#connection_error").text(resp.error).show();
-        } else {
-          connected = true;
-          $("#connection_window").hide();
-          loadDatabases();
-          $("#main").show();
-          //
-          fnSetConnectionInfo();
-        }
-      });
+      if (resp.error) {
+        connected = false;
+        $("#connection_error").text(resp.error).show();
+      } else {
+        connected = true;
+        $("#connection_window").hide();
+        loadDatabases();
+        $("#main").show();
+        //
+        fnSetConnectionInfo();
+      }
+    });
   });
 
   $('#lnkAddQueryTab').on('click', function(e) {
@@ -1468,7 +1470,7 @@ $(document).ready(function() {
           }
         });
         break;
-      }
+        }
       case "FUNC": {
         editFunction(dbName, procName, procDef, function(data) {
           if (data.error) {
@@ -1490,7 +1492,7 @@ $(document).ready(function() {
           }
         });
         break;
-      }
+        }
     }
 
 
@@ -1500,6 +1502,107 @@ $(document).ready(function() {
     loadDatabases();
   });
 
+  $('#btnSearchDB').on('click', function(e) {
+    var searchQuery = $.trim($('#txtSearch').val());
+
+    apiSearchDatabase(searchQuery, function(data) {
+      if (data.error) {
+        return;
+      }
+
+      var objData = {};
+      var arrDB = [];
+
+      data.rows.forEach(function(val) {
+        var dbName = val[1];
+
+        var objDBObject = {};
+        objDBObject.name = val[0];
+        objDBObject.type = val[2];
+
+        var a = objData[dbName];
+
+        if (typeof (a) === "undefined") {
+          objData[dbName] = [];
+          arrDB.push(dbName);
+        }
+
+        objData[dbName].push(objDBObject);
+      });
+
+      var objTree = [];
+      arrDB.forEach(function(dbName) {
+        var dbTable = {
+          label: 'Table',
+          type: 'tbl-holder',
+          children: []
+        };
+        var dbProc = {
+          label: 'Procedure',
+          children: [],
+          type: 'sp-holder'
+        };
+        var dbFunc = {
+          label: 'Function',
+          children: [],
+          type: 'fn-holder'
+        };
+        var dbView = {
+          label: 'View',
+          children: [],
+          type: 'vw-holder'
+        };
+
+
+        objData[dbName].forEach(function(val) {
+          if (val.type === 'TBL') {
+            dbTable.children.push({
+              label: val.name,
+              type: 'table'
+            });
+          } else if (val.type === 'PROC') {
+            dbProc.children.push({
+              label: val.name,
+              type: 'procedure'
+            });
+          } else if (val.type === 'FUNC') {
+            dbFunc.children.push({
+              label: val.name,
+              type: 'function'
+            });
+          }
+        });
+
+        objTree.push({
+          label: dbName,
+          type: 'database',
+          children: [dbTable, dbProc, dbFunc]
+        });
+      });
+      console.log(objTree);
+
+      $tree.tree('loadData', objTree);
+
+      //Open all nodes
+      //https://github.com/mbraak/jqTree/issues/206
+      var tree = $tree.tree('getTree');
+
+      tree.iterate(
+        function(node, level) {
+          if (!node.hasChildren()) {
+            // This will open the folder
+            $tree.tree('selectNode', node);
+            return false;
+          }
+
+          return true;
+        }
+      );
+
+      $tree.tree('selectNode', null);
+    });
+  });
+
   initModals();
 
   initEditor("custom_query");
diff --git a/utils.go b/utils.go
index 22da316..0e1bb93 100644
--- a/utils.go
+++ b/utils.go
@@ -28,7 +28,7 @@ func splice(s string, idx int, rem int, sAdd string) string {
 	return (s[0:idx] + sAdd + s[(idx+rem):len(s)])
 }
 
-func getHostUser(url string) (string, string) {
+func getHostUserFromConnString(url string) (string, string) {
 	colonIndx := strings.Index(url, ":")
 	userName := url[0:colonIndx]