lookupDictionary.js に単語を suggest させる - KBDAHOLIC

もっと使いやすくしたものを CodeRepos に commit しました ( http://d.hatena.ne.jp/janus_wel/20080927/1222486920 ) 。

:eijiro を常用してるんだけどスペルを間違えて何もヒットしなくてしょぼーんとなることが多いので suggest させるようにしてみた。使ったのは Firefox spell check 用の辞書。いろいろ見てみたんだけどこれが一番楽に扱えたので。あとちょっとデカい変更なのでここに書いとく。いやそのまま CodeRepos に commit したらみんな戸惑いそうな気がしたので。

とりあえず準備として以下のことをすると :eijiro や :goo などの各種コマンドで suggest が効くようになるはず。

lookupDictionary.js に以下のパッチをあてる。
https://addons.mozilla.org/ja/firefox/browse/type:3 にある中から使いたいものを選んでインストール。
g:lookupDictionary_dictionary に使いたい辞書の名前をセットする。使える辞書は :availabledictionaries コマンドで表示される。
- 一応デフォルトは 'en-US' ( https://addons.mozilla.org/ja/firefox/addon/3497 ) にしてるけど私は英国紳士なのだよという方は English (British) ( https://addons.mozilla.org/ja/firefox/addon/3366 ) をインストールして 'en-GB' にしてくださいな。
- lookupDictionary.js のサイト情報定義と合わせれば辞書がある言語すべてに対して有効。 Arabia とか Xhosa とか Zulu とか。うしろふたつはどこの言葉か知らないけど。

--- lookupDictionary.js.orig    2008-09-26 17:03:27.546875000 +0900
+++ lookupDictionary.js    2008-09-26 16:50:17.750000000 +0900
@@ -4,11 +4,12 @@
  * @description    Lookup words from Web dictionaries, and show the results in the bottom of the window
  * @description-ja Web上の辞書を引いた結果をコマンドライン・バッファへ出力します
  * @author teramako teramako@gmail.com
- * @version 0.2
+ * @version 0.3
  * ==/VimperatorPlugin==
  */
 (function(){
-[{
+
+const SITE_DEFINITION = [{
     names: ['eiji[ro]'],
     url: 'http://eow.alc.co.jp/%s/UTF-8/',
     shortHelp: 'SPACE ALC (英辞郎 on the Web)',
@@ -31,7 +32,91 @@
     url: 'http://www.answers.com/%s',
     shortHelp: 'Answers.com(英英辞書)',
     xpath: 'id("firstDs")'
-}].forEach(function(dictionary){
+}];
+
+// class definition
+function SpellChecker() {
+    this.initialize.apply(this, arguments);
+}
+SpellChecker.prototype = {
+    initialize: function () {
+        const MYSPELL  = "@mozilla.org/spellchecker/myspell;1";
+        const HUNSPELL = "@mozilla.org/spellchecker/hunspell;1";
+        const ENGINE   = "@mozilla.org/spellchecker/engine;1";
+
+        var spellclass = MYSPELL;
+        if (HUNSPELL in Components.classes)
+            spellclass = HUNSPELL;
+        if (ENGINE in Components.classes)
+            spellclass = ENGINE;
+
+        this.engine = Components.classes[spellclass]
+                      .createInstance(Components.interfaces.mozISpellCheckingEngine);
+    },
+
+    /**
+     * @return {Array}
+     */
+    getDictionaryList: function () {
+        var dictionaries = {};
+        this.engine.getDictionaryList(dictionaries, {});
+        return dictionaries.value;
+    },
+
+    /**
+     * @return {String}
+     */
+    dictionary: function () {
+        var dict;
+        try { dict = this.engine.dictionary; }
+        catch (e) {}
+        return dict ? dict : null;
+    },
+
+    /**
+     * @param {String} dict
+     */
+    setDictionary: function (dict) {
+        this.engine.dictionary = dict;
+    },
+
+    /**
+     * @param {Boolean} isBeginningWith
+     */
+    setBeginningWith: function (isBeginningWith) {
+        this.isBeginningWith = isBeginningWith;
+    },
+
+    /**
+     * @param {String} spell
+     * @return {Boolean}
+     */
+    check: function (spell) {
+        return this.engine.check(spell);
+    },
+
+    /**
+     * @param {String} spell
+     * @return {Array}
+     */
+    suggest: function (spell) {
+        var suggestions = {};
+        this.engine.suggest(spell, suggestions, {});
+        suggestions = suggestions.value;
+
+        if (this.isBeginningWith) {
+            suggestions = suggestions.filter( function (cand) {
+                return (cand.toLowerCase().indexOf(spell) === 0);
+            });
+        }
+
+        return suggestions;
+    },
+};
+
+var spellChecker = buildSpellChecker();
+
+SITE_DEFINITION.forEach(function(dictionary){
     liberator.commands.addUserCommand(
         dictionary.names,
         dictionary.shortHelp,
@@ -59,9 +144,31 @@
                 var xs = new XMLSerializer();
                 liberator.echo('<base href="' + url + '"/>' + xs.serializeToString( result ), true);
             }, dictionary.encode ? dictionary.encode : 'UTF-8');
-        },{}
+        },
+        {
+            completer: function (arg) {
+                if (!spellChecker.dictionary()) return [0, []];
+
+                var suggestions = spellChecker.suggest(arg);
+                var candidates = [];
+                for (var i=0, max=suggestions.length ; i<max ; ++i) {
+                    candidates.push([suggestions[i], 'suggest']);
+                }
+
+                if (!spellChecker.check(arg)) {
+                    candidates.unshift(['', 'not exist']);
+                }
+                return [0, candidates];
+            },
+        }
     );
 });
+liberator.commands.addUserCommand(
+    ['availabledictionaries'],
+    'display available dictionaries',
+    function () { liberator.echo('available dictionaries: ' + spellChecker.getDictionaryList()); },
+    {}
+);
 /**
  * @param {String} url
  * @param {Function} callback
@@ -115,6 +222,28 @@
     }
     return result;
 }
+
+/**
+ * @return {Object}
+ */
+function buildSpellChecker() {
+    var enable = liberator.globalVariables.lookupDictionary_enableSuggestion;
+    enable = (enable === undefined) ? true : !!parseInt(enable, 10);
+    if (!enable) return;
+
+    var spellChecker = new SpellChecker();
+    var dict = liberator.globalVariables.lookupDictionary_dictionary || 'en-US';
+    var dictionaries = spellChecker.getDictionaryList()
+    for (var i=0, max=dictionaries.length ; i<max ; ++i) {
+        if (dictionaries[i] === dict) spellChecker.setDictionary(dict);
+    }
+
+    var isBeginningWith = liberator.globalVariables.lookupDictionary_beginningWith
+    isBeginningWith = (isBeginningWith === undefined) ? false : !!parseInt(isBeginningWith, 10);
+    spellChecker.setBeginningWith(isBeginningWith);
+
+    return spellChecker;
+}
 })();
 
 // vim: fdm=marker sw=4 ts=4 et: