From 9b9985bd4ba810d5a4e985100c68c544df9bccaf Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Sat, 7 May 2016 13:43:32 -0700
Subject: Improvements

---
 ext/fg/js/api.js    | 4 ++--
 ext/fg/js/client.js | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

(limited to 'ext/fg/js')

diff --git a/ext/fg/js/api.js b/ext/fg/js/api.js
index 9026c8d3..e13c1421 100644
--- a/ext/fg/js/api.js
+++ b/ext/fg/js/api.js
@@ -45,6 +45,6 @@ function bgGetState(callback) {
     bgSendMessage('getState', null, callback);
 }
 
-function bgCanAddNotes(definitions, callback) {
-    bgSendMessage('canAddNotes', definitions, callback);
+function bgCanAddNotes(definitions, modes, callback) {
+    bgSendMessage('canAddNotes', {definitions, modes}, callback);
 }
diff --git a/ext/fg/js/client.js b/ext/fg/js/client.js
index b4c1c2d4..4b34ef9a 100644
--- a/ext/fg/js/client.js
+++ b/ext/fg/js/client.js
@@ -115,7 +115,7 @@ class Client {
                         this.definitions = definitions;
                         this.showPopup(range, content);
 
-                        bgCanAddNotes(definitions, (states) => {
+                        bgCanAddNotes(definitions, ['vocabExp', 'vocabReading'], (states) => {
                             if (states !== null) {
                                 states.forEach((state, index) => this.popup.sendMessage(
                                     'setActionState',
@@ -149,7 +149,7 @@ class Client {
                     this.definitions = definitions;
                     this.popup.setContent(content, definitions);
 
-                    bgCanAddNotes(definitions, (states) => {
+                    bgCanAddNotes(definitions, ['kanji'], (states) => {
                         if (states !== null) {
                             states.forEach((state, index) => this.popup.sendMessage(
                                 'setActionState',
-- 
cgit v1.2.3