1 files changed, 21 insertions, 20 deletions

diff --git a/src/webview/notifications.js b/src/webview/notifications.js
index 97ce9d69b..4055b10de 100644
--- a/src/webview/notifications.js
+++ b/src/webview/notifications.js
@@ -1,45 +1,46 @@
 const { ipcRenderer } = require('electron');
 const uuidV1 = require('uuid/v1');
-// const FranzNotificationStore = [];
 
 class Notification {
+  static permission = 'granted';
+
   constructor(title = '', options = {}) {
     this.title = title;
     this.options = options;
     this.notificationId = uuidV1();
-    this.onclick = () => {};
 
-    ipcRenderer.sendToHost('notification', {
+    ipcRenderer.sendToHost('notification', this.onNotify({
       notificationId: this.notificationId,
       title,
       options,
-    });
+    }));
 
-    ipcRenderer.on(`notification-onclick:${this.notificationId}`, () => {
+    ipcRenderer.once(`notification-onclick:${this.notificationId}`, () => {
       this.onclick();
     });
   }
-}
 
-Notification.permission = 'granted';
+  static requestPermission(cb = null) {
+    if (!cb) {
+      return new Promise((resolve) => {
+        resolve(Notification.permission);
+      });
+    }
 
-Notification.requestPermission = (cb = null) => {
-  console.log(this);
-  if (!cb) {
-    return new Promise((resolve) => {
-      resolve(Notification.permission);
-    });
+    if (typeof (cb) === 'function') {
+      return cb(Notification.permission);
+    }
+
+    return Notification.permission;
   }
 
-  if (typeof (cb) === 'function') {
-    return cb(Notification.permission);
+  onNotify(data) {
+    return data;
   }
 
-  return Notification.permission;
-};
+  onClick() {}
 
-Notification.close = () => {
-  // no implementation yet
-};
+  close() {}
+}
 
 window.Notification = Notification;