WIP

src/components/Debug.vue

@@ -1,14 +1,22 @@
 <template>
-  <pre class="json">{{ searchData }}</pre>
+  <div>
+    <p>Content of <code>@dynamic/vuepress-plugin-flexsearch/dictionary</code></p>
+    <pre class="json">{{ dictionaryData }}</pre>
+
+    <p>Content of <code>@dynamic/vuepress-plugin-flexsearch/data</code></p>
+    <pre class="json">{{ searchData }}</pre>
+  </div>
 </template>
 
 <script>
 import data from '@dynamic/vuepress-plugin-flexsearch/data';
+import dictionary from '@dynamic/vuepress-plugin-flexsearch/dictionary';
 
 export default {
   data() {
     return {
       searchData: data,
+      dictionaryData: dictionary,
     };
   },
 };

src/components/SearchBoxFlexSearch.vue

@@ -50,6 +50,7 @@
 <script>
 import { Document } from 'flexsearch'
 import data from '@dynamic/vuepress-plugin-flexsearch/data'
+import dictionary from '@dynamic/vuepress-plugin-flexsearch/dictionary'
 import Database from './database';
 import excerpt from './excerpt'
 import ngram from '../tokenizer/ngram'
@@ -97,13 +98,22 @@ export default {
      * @return {{title: string, path: string, excerpt: string}[]}
      */
     suggestions () {
-      const query = this.query.trim();
+      const queryTrimmed = this.query.trim();
 
-      if (query.length === 0) {
+      if (queryTrimmed.length === 0) {
         return [];
       }
 
       const localePath = this.$localePath;
+
+      let dictQueries = [];
+      Object.keys(dictionary).forEach((key) => {
+        if (queryTrimmed.indexOf(key) >= 0) {
+          dictQueries = dictQueries.concat(dictionary[key]);
+        }
+      });
+
+      const query = [queryTrimmed].concat(dictQueries).join(' ');
       const queryForSearch = ngram.createForSearch(query, FLEX_SEARCH_NGRAM_SIZE);
       const max = this.$site.themeConfig.searchMaxSuggestions || FLEX_SEARCH_MAX_SUGGESTIONS;
 

src/index.js

@@ -25,6 +25,8 @@ module.exports = (options, ctx) => {
     excerptTailText = ' ...',
     tokenizerType = 'kuromoji.default',
     ngramSize = 3,
+    /** @type {string[][]} */
+    dictionarySet = [],
   } = options;
 
   /**
@@ -38,6 +40,26 @@ module.exports = (options, ctx) => {
     }
   })(searchPaths);
 
+  /** @type {Map<string, Set<string>>} */
+  const dictMap = new Map();
+  dictionarySet.forEach((dictionaryItem) => {
+    for (let i = 0; i < dictionaryItem.length; i++) {
+      if (!dictMap.has(dictionaryItem[i])) {
+        dictMap.set(dictionaryItem[i], new Set());
+      }
+      for (let j = 0; j < dictionaryItem.length; j++) {
+        if (i !== j) {
+          dictMap.get(dictionaryItem[i]).add(dictionaryItem[j]);
+        }
+      }
+    }
+  });
+
+  const dictData = {};
+  dictMap.forEach((dictSet, key) => {
+    dictData[key] = Array.from(dictSet);
+  })
+
   return {
     define: {
       FLEX_SEARCH_HOTKEYS: searchHotKeys,
@@ -91,6 +113,7 @@ module.exports = (options, ctx) => {
 
         const pageData = await tokenizer.use(tokenizerType, page, {
           ngramSize: ngramSize,
+          dictMap: dictMap,
         });
 
         localeBasedPageData.get(localePath).set(page.key, {
@@ -116,6 +139,10 @@ module.exports = (options, ctx) => {
           name: 'vuepress-plugin-flexsearch/data.js',
           content: `export default ${JSON.stringify(flexSearchData, null, 2)}`,
         },
+        {
+          name: 'vuepress-plugin-flexsearch/dictionary.js',
+          content: `export default ${JSON.stringify(dictData, null, 2)}`,
+        }
       ];
     },
   };

src/tokenizer.js

@@ -14,6 +14,7 @@ const tokenizers = new Map();
 /**
  * @typedef {Object} TokenizerUseOption
  * @property {number} [ngramSize]
+ * @property {Map<string, Set<string>>} dictMap
  */
 
 /**
@@ -29,13 +30,22 @@ const use = async (type, page, option) => {
 
   const {
     ngramSize = 3,
+    dictMap,
   } = option || {};
 
   const tokenizer = tokenizers.get(type);
   const data = await tokenizer.create(page);
 
+  let dict = [];
+  const token = data.tokens.join('');
+  dictMap.forEach((dictSet, key) => {
+    if (token.indexOf(key) >= 0) {
+      dict = dict.concat(Array.from(dictSet));
+    }
+  });
+
   return {
-    dataForSearch: ngram.createForTokens(data.tokens, ngramSize),
+    dataForSearch: ngram.createForTokens(data.tokens.concat(dict), ngramSize),
     dataForExcerpt: data.excerpt,
   };
 };