My First Clean Architecture

1 files changed, 34 insertions, 0 deletions

diff --git a/src/background/infrastructures/content-message-listener.js b/src/background/infrastructures/content-message-listener.js
new file mode 100644
index 0000000..a0ed66c
--- /dev/null
+++ b/src/background/infrastructures/content-message-listener.js
@@ -0,0 +1,34 @@
+import messages from '../../shared/messages';
+import CompletionsController from '../controllers/completions';
+
+export default class ContentMessageListener {
+  constructor() {
+    this.completionsController = new CompletionsController();
+  }
+
+  run() {
+    browser.runtime.onMessage.addListener((message, sender) => {
+      try {
+        return this.onMessage(message, sender);
+      } catch (e) {
+        return browser.tabs.sendMessage(sender.tab.id, {
+          type: messages.CONSOLE_SHOW_ERROR,
+          text: e.message,
+        });
+      }
+    });
+  }
+
+  onMessage(message) {
+    switch (message.type) {
+    case messages.CONSOLE_QUERY_COMPLETIONS:
+      return this.onConsoleQueryCompletions(message);
+    }
+  }
+
+  async onConsoleQueryCompletions(message) {
+    let completions =
+      await this.completionsController.getCompletions(message.text);
+    return Promise.resolve(completions.serialize());
+  }
+}