Handling snake case terms in search results (#1182)

Author: jwarwick

assets/js/search.js

@@ -122,21 +122,37 @@ function titleExtractor (document) {
 
  if (type === 'function' || type === 'callback' || type === 'type') {
  var modFun = title.replace(/\/\d+/, '')
- var modOrFun = modFun.replace('.', ' ')
+ var modOrFun = modFun.replace(/\./g, ' ')
  var parts = title.split('.')
  title = title + ' ' + modFun + ' ' + modOrFun + ' ' + parts[parts.length - 1]
  }
 
  return title
 }
 
+function snakeCaseSplitter (builder) {
+ function snakeCaseFunction (token) {
+ var snakeTokens = token.toString().split("_").map(function (str) {
+ return token.clone().update(function () { return str })
+ })
+ if (snakeTokens.length > 1) {
+ snakeTokens.push(token)
+ }
+ return snakeTokens
+ }
+
+ lunr.Pipeline.registerFunction(snakeCaseFunction, 'snakeCaseSplitter')
+ builder.pipeline.before(lunr.stemmer, snakeCaseFunction)
+}
+
 function createIndex () {
  return lunr(function () {
  this.ref('ref')
  this.field('title', {boost: 3, extractor: titleExtractor})
  this.field('doc')
  this.metadataWhitelist = ['position']
  this.pipeline.remove(lunr.stopWordFilter)
+ this.use(snakeCaseSplitter)
 
  searchNodes.forEach(function (doc) {
  this.add(doc)