diff options
Diffstat (limited to 'packages/gitbook-plugin-lunr/src/index.js')
-rw-r--r-- | packages/gitbook-plugin-lunr/src/index.js | 28 |
1 files changed, 28 insertions, 0 deletions
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 +}); |