1 files changed, 16 insertions, 9 deletions
diff --git a/src/console/index.js b/src/console/index.js
index b07d8e7..895fcc2 100644
--- a/src/console/index.js
+++ b/src/console/index.js
@@ -17,18 +17,25 @@ window.addEventListener('load', () => {
   consoleComponent = new ConsoleComponent(document.body, store);
 });
 
-store.subscribe(() => {
-  completionComponent.update();
-  consoleComponent.update();
-});
-
-browser.runtime.onMessage.addListener((action) => {
-  switch (action.type) {
+const onMessage = (message) => {
+  switch (message.type) {
   case messages.CONSOLE_SHOW_COMMAND:
-    return store.dispatch(consoleActions.showCommand(action.command));
+    return store.dispatch(consoleActions.showCommand(message.command));
   case messages.CONSOLE_SHOW_ERROR:
-    return store.dispatch(consoleActions.showError(action.text));
+    return store.dispatch(consoleActions.showError(message.text));
+  case messages.CONSOLE_SHOW_INFO:
+    return store.dispatch(consoleActions.showInfo(message.text));
   case messages.CONSOLE_HIDE_COMMAND:
     return store.dispatch(consoleActions.hideCommand());
   }
+};
+
+store.subscribe(() => {
+  completionComponent.update();
+  consoleComponent.update();
 });
+
+browser.runtime.onMessage.addListener(onMessage);
+window.addEventListener('message', (message) => {
+  onMessage(JSON.parse(message.data));
+}, false);