Implement system notifications of connection state

2 files changed, 45 insertions, 0 deletions

diff --git a/app/app.js b/app/app.js
index 501bfac780..3171d69b76 100644
--- a/app/app.js
+++ b/app/app.js
@@ -11,6 +11,7 @@ import makeRoutes from './routes';
 import { log } from './lib/platform';
 import ReconnectionBackoff from './lib/reconnection-backoff';
 import { DaemonRpc } from './lib/daemon-rpc';
+import NotificationController from './notification-controller';
 import { setShutdownHandler } from './shutdown-handler';
 import { NoAccountError } from './errors';
 
@@ -30,6 +31,7 @@ import type { ConnectionState } from './redux/connection/reducers';
 import type { TrayIconType } from './tray-icon-controller';
 
 export default class AppRenderer {
+  _notificationController = new NotificationController();
   _daemonRpc: DaemonRpcProtocol = new DaemonRpc();
   _reconnectBackoff = new ReconnectionBackoff();
   _credentials: ?RpcCredentials;
@@ -523,6 +525,24 @@ export default class AppRenderer {
     }
 
     this._updateTrayIcon(connectionState);
+    this._showNotification(connectionState);
+  }
+
+  _showNotification(connectionState: ConnectionState) {
+    switch (connectionState) {
+      case 'connecting':
+        this._notificationController.show('Connecting');
+        break;
+      case 'connected':
+        this._notificationController.show('Secured');
+        break;
+      case 'disconnected':
+        this._notificationController.show('Unsecured');
+        break;
+      default:
+        log.error(`Unexpected ConnectionState: ${(connectionState: empty)}`);
+        return;
+    }
   }
 
   async _authenticate(sharedSecret: string) {
diff --git a/app/notification-controller.js b/app/notification-controller.js
new file mode 100644
index 0000000000..62d6381e72
--- /dev/null
+++ b/app/notification-controller.js
@@ -0,0 +1,25 @@
+// @flow
+import { remote } from 'electron';
+
+export default class NotificationController {
+  _activeNotification: ?Notification;
+
+  show(message: string) {
+    const lastNotification = this._activeNotification;
+    const newNotification = new Notification(remote.app.getName(), { body: message, silent: true });
+
+    this._activeNotification = newNotification;
+
+    newNotification.addEventListener('show', () => {
+      // If the notification is closed too soon, it might still get shown. If that happens, close()
+      // should be called again so that it is closed immediately.
+      if (this._activeNotification !== newNotification) {
+        newNotification.close();
+      }
+    });
+
+    if (lastNotification) {
+      lastNotification.close();
+    }
+  }
+}