1 files changed, 114 insertions, 0 deletions
diff --git a/theme/javascript/core/search.js b/theme/javascript/core/search.js
new file mode 100755
index 0000000..4cbc3ed
--- /dev/null
+++ b/theme/javascript/core/search.js
@@ -0,0 +1,114 @@
+define([
+    "jQuery",
+    "lodash",
+    "lunr",
+    "utils/storage",
+    "core/state",
+    "core/sidebar"
+], function($, _, lunr, storage, state, sidebar) {
+    var index = null;
+
+    // Use a specific idnex
+    var useIndex = function(data) {
+        index = lunr.Index.load(data);
+    };
+
+    // Load complete index
+    var loadIndex = function() {
+        $.getJSON(state.basePath+"/search_index.json")
+        .then(useIndex);
+    };
+
+    // Search for a term
+    var search = function(q) {
+        if (!index) return;
+        var results = _.chain(index.search(q))
+        .map(function(result) {
+            var parts = result.ref.split("#")
+            return {
+                path: parts[0],
+                hash: parts[1]
+            }
+        })
+        .value();
+
+        return results;
+    };
+
+    // Toggle search bar
+    var toggleSearch = function(_state) {
+        if (state != null && isSearchOpen() == _state) return;
+
+        var $searchInput = $(".book-search input");
+        state.$book.toggleClass("with-search", _state);
+
+        // If search bar is open: focus input
+        if (isSearchOpen()) {
+            sidebar.toggle(true);
+            $searchInput.focus();
+        } else {
+            $searchInput.blur();
+            $searchInput.val("");
+            sidebar.filter(null);
+        }
+    };
+
+    // Return true if search bar is open
+    var isSearchOpen = function() {
+        return state.$book.hasClass("with-search");
+    };
+
+
+    var init = function() {
+        loadIndex();
+
+        // Toggle search
+        $(document).on("click", ".book-header .toggle-search", function(e) {
+            e.preventDefault();
+            toggleSearch();
+        });
+
+
+        // Type in search bar
+        $(document).on("keyup", ".book-search input", function(e) {
+            var key = (e.keyCode ? e.keyCode : e.which);
+            var q = $(this).val();
+
+            if (key == 27) {
+                e.preventDefault();
+                toggleSearch(false);
+                return;
+            }
+            if (q.length == 0) {
+                sidebar.filter(null);
+                storage.remove("keyword");
+            } else {
+                var results = search(q);
+                sidebar.filter(
+                    _.pluck(results, "path")
+                );
+                storage.set("keyword", q);
+            }
+        })
+
+    };
+
+    // filter sidebar menu with current search keyword
+    var recoverSearch = function() {
+        var keyword = storage.get("keyword", "");
+        if(keyword.length > 0) {
+            if(!isSearchOpen()){
+                toggleSearch();
+            }
+            sidebar.filter(_.pluck(search(keyword), "path"));
+        }
+        $(".book-search input").val(keyword);
+    };
+
+    return {
+        init: init,
+        search: search,
+        toggle: toggleSearch,
+        recover:recoverSearch
+    };
+});