Move learnSpelling and forgetLearnedSpelling to appActions

Auditors: @bridiver

app/spellChecker.js

@@ -0,0 +1,14 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+const electron = require('electron')
+const session = electron.session
+
+module.exports.addWord = (word) => {
+  session.defaultSession.spellChecker.addWord(word)
+}
+
+module.exports.removeWord = (word) => {
+  session.defaultSession.spellChecker.removeWord(word)
+}

js/actions/appActions.js

@@ -1482,6 +1482,20 @@ const appActions = {
       tag,
       editKey
     })
+  },
+
+  learnSpelling: function (word) {
+    dispatch({
+      actionType: appConstants.APP_LEARN_SPELLING,
+      word
+    })
+  },
+
+  forgetLearnedSpelling: function (word) {
+    dispatch({
+      actionType: appConstants.APP_FORGET_LEARNED_SPELLING,
+      word
+    })
   }
 }
 

js/constants/appConstants.js

@@ -139,7 +139,9 @@ const appConstants = {
   APP_SWIPE_LEFT: _,
   APP_SWIPE_RIGHT: _,
   APP_ADD_BOOKMARK: _,
-  APP_EDIT_BOOKMARK: _
+  APP_EDIT_BOOKMARK: _,
+  APP_LEARN_SPELLING: _,
+  APP_FORGET_LEARNED_SPELLING: _
 }
 
 module.exports = mapValuesByKeys(appConstants)

js/constants/messages.js

@@ -76,7 +76,6 @@ const messages = {
   GO_FORWARD: _,
   RELOAD: _,
   DETACH: _,
-  IS_MISSPELLED: _, /** @arg {string} word, the word to check */
   PASSWORD_DETAILS_UPDATED: _, /** @arg {Object} passwords app state */
   PASSWORD_SITE_DETAILS_UPDATED: _, /** @arg {Object} passwords app state */
   // Init

js/contextMenus.js

@@ -560,19 +560,17 @@ function getMisspelledSuggestions (selection, isMisspelled, suggestions) {
     if (isMisspelled) {
       template.push({
         label: locale.translation('learnSpelling'),
-        click: (item, focusedWindow) => {
-          if (focusedWindow) {
-            focusedWindow.webContents.addWord(selection)
-          }
+        click: () => {
+          appActions.learnSpelling(selection)
+          webviewActions.replace(selection)
         }
       }, CommonMenu.separatorMenuItem)
     } else {
       template.push({
         label: locale.translation('forgetLearnedSpelling'),
-        click: (item, focusedWindow) => {
-          if (focusedWindow) {
-            focusedWindow.webContents.removeWord(selection)
-          }
+        click: () => {
+          appActions.forgetLearnedSpelling(selection)
+          webviewActions.replace(selection)
         }
       }, CommonMenu.separatorMenuItem)
     }
@@ -972,11 +970,10 @@ function mainTemplateInit (nodeProps, frame, tab) {
   if (isInputField) {
     let misspelledSuggestions = []
     if (nodeProps.misspelledWord) {
-      if (nodeProps.misspelledWord === 'match-brave-custom-dictionary' && !nodeProps.dictionarySuggestions.length) {
-        misspelledSuggestions = getMisspelledSuggestions(nodeProps.selectionText, false, nodeProps.dictionarySuggestions)
-      } else {
-        misspelledSuggestions = getMisspelledSuggestions(nodeProps.selectionText, true, nodeProps.dictionarySuggestions)
-      }
+      misspelledSuggestions = getMisspelledSuggestions(nodeProps.selectionText, true, nodeProps.dictionarySuggestions)
+    } else if (nodeProps.properties.hasOwnProperty('learnedSpelling') &&
+               nodeProps.properties['learnedSpelling'] === nodeProps.selectionText) {
+      misspelledSuggestions = getMisspelledSuggestions(nodeProps.selectionText, false, nodeProps.dictionarySuggestions)
     }
 
     const editableItems = getEditableItems(nodeProps.selectionText, nodeProps.editFlags, true)

js/stores/appStore.js

@@ -28,6 +28,7 @@ const diff = require('immutablediff')
 const debounce = require('../lib/debounce')
 const path = require('path')
 const autofill = require('../../app/autofill')
+const spellChecker = require('../../app/spellChecker')
 const nativeImage = require('../../app/nativeImage')
 const filtering = require('../../app/filtering')
 const basicAuth = require('../../app/browser/basicAuth')
@@ -904,6 +905,16 @@ const handleAppAction = (action) => {
     case appConstants.APP_SWIPE_RIGHT:
       appState = appState.set('swipeRightPercent', action.percent)
       break
+    case appConstants.APP_LEARN_SPELLING:
+      if (typeof action.word === 'string') {
+        spellChecker.addWord(action.word)
+      }
+      break
+    case appConstants.APP_FORGET_LEARNED_SPELLING:
+      if (typeof action.word === 'string') {
+        spellChecker.removeWord(action.word)
+      }
+      break
     default:
   }