Allow users to add custom words to dictionary

app/menu-context.json

@@ -1,4 +1,11 @@
 [
+    {
+        "labelKey": "menu-add-to-dictionary",
+        "command": "addToDict"
+    },
+    {
+        "type": "separator"
+    },
     {
         "labelKey": "menu-undo",
         "accelerator": "CmdOrCtrl+Z",

app/renderer/abr-document.js

@@ -31,6 +31,9 @@ function AbrDocument () {
     // Start with an empty table of contents
     this.toc = [];
 
+    // Custom words
+    this.words = [];
+
     // Run this in a background thread
     var worker = cp.fork(__dirname + "/toc-worker.js");
 
@@ -87,6 +90,18 @@ function AbrDocument () {
             // Spellchecker init
             if (config.spellchecker.active) {
                 that.setDictionary(config.spellchecker.language, config.spellchecker.src);
+
+                // load custom words
+                var words = config.spellchecker["custom-words"];
+                if (words != null) {
+                    that.words = words;
+                    words.forEach(function (word) {
+                        // just for safety
+                        if (word != "") {
+                            spellchecker.add(word)
+                        }
+                    });
+                }
             }
 
             // Editor font-size
@@ -773,6 +788,15 @@ AbrDocument.prototype = {
     setDictionary: function (lang, path) {
         if (lang) {
             spellchecker.setDictionary(lang, path);
+
+            // reload custom words
+            this.words.forEach(function (word) {
+                // just for safety
+                if (word != "") {
+                    spellchecker.add(word)
+                }
+            });
+
             // Refresh CodeMirror highlight + enable spelling
             this.cm.setOption("mode", "abr-spellcheck-on");
             this.setConfig("spellchecker:active", true);
@@ -789,6 +813,18 @@ AbrDocument.prototype = {
         return spellchecker.isMisspelled;
     },
 
+    // Adds the given word to our custom spellcheck-ok words
+    addSpelledWord: function (word) {
+        // add to in-memory spellchecker
+        spellchecker.add(word);
+
+        // add to config file
+        if (this.words.indexOf(word) == -1) {
+            this.words.push(word)
+            this.setConfig("spellchecker:custom-words", this.words)
+        }
+    },
+
     // Scale text
     setFontSize: function (size) {
         var min = 8,

app/renderer/commands.js

@@ -90,6 +90,62 @@ var commands = {
         document.execCommand("paste");
     },
 
+    addToDict: function(win, abrDoc, cm) {
+        var pos = cm.doc.getCursor();
+        var line = cm.doc.getLine(pos.line);
+
+        // split all words on the currently-selected line
+        var wordDelimiters = "!\"#$%&()*+,-./:;<=>?@[\\]^_`{|}~ \t",
+            initialChar = 0,
+            word = "",
+            words = [];
+
+        for (var i = 0; i < line.length; i++) {
+            var ch = line[i];
+
+            if (wordDelimiters.includes(ch)) {
+                if (word != "") {
+                    words.push([initialChar, word.replace(/[’ʼ]/g, "'")]);
+                    word = "";
+                }
+                initialChar = i;
+            } else {
+                word += ch;
+            }
+        }
+        if (word != "") {
+            words.push([initialChar, word.replace(/[’ʼ]/g, "'")]);
+        }
+
+        // if there are no words on this line, return
+        if (words.length < 1) {
+            return
+        }
+
+        // find which specific word which is selected
+        var selectedWord = words[0][1];
+
+        for(var i = 0; i < words.length; i++) {
+            var newWordStart = words[i][0],
+                newWord = words[i][1];
+
+            if (newWordStart < pos.ch) {
+                selectedWord = newWord;
+            }
+        }
+
+        // confirm the selected word is misspelled
+        var isMisspelled = abrDoc.getSpellcheckFunc()(selectedWord)
+
+        if (isMisspelled) {
+            abrDoc.addSpelledWord(selectedWord);
+
+            // refresh overlays so user sees the word get marked as spelled fine
+            cm.setOption("mode", "abr-spellcheck-off");
+            cm.setOption("mode", "abr-spellcheck-on");
+        }
+    },
+
     find: function(win, abrDoc, cm) {
         $(".CodeMirror-dialog").remove(); // FIXME: error when double key
         cm.execCommand("clearSearch");

default/lang/en.json

@@ -1,4 +1,5 @@
 {
+    "menu-add-to-dictionary": "Add to Dictionary",
     "menu-undo": "Undo",
     "menu-redo": "Redo",
     "menu-cut": "Cut",