3 files changed, 102 insertions, 0 deletions
diff --git a/packages/gitbook-plugin-lunr/src/actions.js b/packages/gitbook-plugin-lunr/src/actions.js
new file mode 100644
index 0000000..765fa2e
--- /dev/null
+++ b/packages/gitbook-plugin-lunr/src/actions.js
@@ -0,0 +1,43 @@
+const GitBook = require('gitbook-core');
+
+const TYPES = {
+    LOAD: 'lunr/load'
+};
+const INDEX_FILENAME = 'search_index.json';
+
+/**
+ * Load an index set
+ * @param {JSON} json
+ * @return {Action}
+ */
+function load(json) {
+    return { type: TYPES.LOAD, json };
+}
+
+/**
+ * Fetch an index
+ * @return {Action}
+ */
+function fetch() {
+    return (dispatch, getState) => {
+        const { lunr, file } = getState();
+        const { idx } = lunr;
+        const filePath = file.relative(INDEX_FILENAME);
+
+        if (idx) {
+            return GitBook.Promise.resolve();
+        }
+
+        return GitBook.Promise.resolve()
+        .then(() => {
+            return window.fetch(filePath);
+        })
+        .then(response => response.json())
+        .then(json => dispatch(load(json)));
+    };
+}
+
+module.exports = {
+    TYPES,
+    fetch
+};
diff --git a/packages/gitbook-plugin-lunr/src/index.js b/packages/gitbook-plugin-lunr/src/index.js
new file mode 100644
index 0000000..1135f51
--- /dev/null
+++ b/packages/gitbook-plugin-lunr/src/index.js
@@ -0,0 +1,28 @@
+const GitBook = require('gitbook-core');
+const reduce = require('./reducer');
+const actions = require('./actions');
+
+/**
+ * Search in the local index
+ * @param  {String} query
+ * @return {Promise<List>}
+ */
+function searchHandler(query, dispatch, getState) {
+    // Fetch the index if non loaded
+    return dispatch(actions.fetch())
+
+    // Execute the search
+    .then(() => {
+        const { idx, store } = getState().lunr;
+        const results = idx.search(query);
+
+        return results.map(({ref}) => store.get(ref).toJS());
+    });
+}
+
+module.exports = GitBook.createPlugin({
+    activate: (dispatch, getState, { Search }) => {
+        dispatch(Search.registerHandler('lunr', searchHandler));
+    },
+    reduce
+});
diff --git a/packages/gitbook-plugin-lunr/src/reducer.js b/packages/gitbook-plugin-lunr/src/reducer.js
new file mode 100644
index 0000000..7e317c4
--- /dev/null
+++ b/packages/gitbook-plugin-lunr/src/reducer.js
@@ -0,0 +1,31 @@
+const lunr = require('lunr');
+const GitBook = require('gitbook-core');
+const { Record } = GitBook.Immutable;
+
+const { TYPES } = require('./actions');
+
+/*
+    We store the lunr index an the document index in the store.
+ */
+
+const LunrState = Record({
+    idx: null,
+    store: {}
+});
+
+module.exports = GitBook.createReducer('lunr', (state, action) => {
+    state = state || LunrState();
+
+    switch (action.type) {
+
+    case TYPES.LOAD:
+        return state
+            .set('idx', lunr.Index.load(action.json.index))
+            .merge({
+                store: action.json.store
+            });
+
+    default:
+        return state;
+    }
+});