2 files changed, 21 insertions, 0 deletions
diff --git a/src/background/infrastructures/content-message-client.js b/src/background/infrastructures/content-message-client.js
index b3b37b4..d659560 100644
--- a/src/background/infrastructures/content-message-client.js
+++ b/src/background/infrastructures/content-message-client.js
@@ -9,4 +9,17 @@ export default class ContentMessageClient {
       });
     }
   }
+
+  async getAddonEnabled(tabId) {
+    let { enabled } = await browser.tabs.sendMessage(tabId, {
+      type: messages.ADDON_ENABLED_QUERY,
+    });
+    return enabled;
+  }
+
+  toggleAddonEnabled(tabId) {
+    return browser.tabs.sendMessage(tabId, {
+      type: messages.ADDON_TOGGLE_ENABLED,
+    });
+  }
 }
diff --git a/src/background/infrastructures/content-message-listener.js b/src/background/infrastructures/content-message-listener.js
index 00dba51..8a361c1 100644
--- a/src/background/infrastructures/content-message-listener.js
+++ b/src/background/infrastructures/content-message-listener.js
@@ -2,12 +2,14 @@ import messages from '../../shared/messages';
 import CompletionsController from '../controllers/completions';
 import SettingController from '../controllers/setting';
 import FindController from '../controllers/find';
+import AddonEnabledController from '../controllers/addon-enabled';
 
 export default class ContentMessageListener {
   constructor() {
     this.settingController = new SettingController();
     this.completionsController = new CompletionsController();
     this.findController = new FindController();
+    this.addonEnabledController = new AddonEnabledController();
   }
 
   run() {
@@ -35,6 +37,8 @@ export default class ContentMessageListener {
       return this.onFindGetKeyword();
     case messages.FIND_SET_KEYWORD:
       return this.onFindSetKeyword(message.keyword);
+    case messages.ADDON_ENABLED_RESPONSE:
+      return this.onAddonEnabledResponse(message.enabled);
     }
   }
 
@@ -58,4 +62,8 @@ export default class ContentMessageListener {
   onFindSetKeyword(keyword) {
     return this.findController.setKeyword(keyword);
   }
+
+  onAddonEnabledResponse(enabled) {
+    return this.addonEnabledController.indicate(enabled);
+  }
 }