refactor code and add notificationHandler

Author: Hesammousavi

.gitignore

@@ -1,3 +1,4 @@
 node_modules
 package-lock.json
-.DS_Store
+.DS_Store
+.idea

README.md

@@ -109,7 +109,7 @@ const exampleNotification = require('./notifications/exampleNotification');
 
 app.get('/', (req, res) => {
   // send notificaiton
-  res.notify({ email : '[email protected]' } , exampleNotification() )
+  res.notify(exampleNotification()).to({ email : '[email protected]' })
   res.send('Hello World!')
 });
 ```

example/server.js

@@ -26,10 +26,10 @@ app.use(notification.register(notificationConfig))
 
 app.get('/', (req, res) => {
   // send notificaiton
-  res.notify({ email : '[email protected]' } , exampleNotification() )
+  res.notify(exampleNotification()).to({ email : '[email protected]' })
   res.send('Hello World!')
 })
 
 app.listen(port, () => {
   console.log(`Example app listening on port ${port}`)
-})
+})

src/index.js

@@ -1,4 +1,5 @@
 const MailChannel = require('./channels/mailChannel');
+const NotificationHandler = require("./notificationHandler");
 
 
 /**
@@ -7,45 +8,29 @@ const MailChannel = require('./channels/mailChannel');
  * @returns {function} return middleware
  */
 
-module.exports = ({ config , channels }) => {
-
-  if(typeof config != 'object')
-    throw new Error('you must set conifg as object')
-
-  /**
-   * LIST OF CHANNELS THAT NOTIFICATION CAN SEND
-   * And Merge Default Channels with Custom Channels
-   */
-   channels = {
-     mail : MailChannel,
-     ...channels
-   }
-
-  /**
-   * the user data for send notification
-   * @param notifiable
-   * @param notification
-   */
-  const notifyHandler = async (notifiable, notification) => {
-    // get channels that notification must send to them
-    let channelsMustBeNotify = notification.via();
-
-    // notify to notifiable with different channels
-    for (const channel of channelsMustBeNotify) {
-      await channelToNotify(channel , { notifiable, notification });
+module.exports = ({config, channels}) => {
+
+    if (typeof config != 'object')
+        throw new Error('you must set conifg as object')
+
+    /**
+     * LIST OF CHANNELS THAT NOTIFICATION CAN SEND
+     * And Merge Default Channels with Custom Channels
+     */
+    channels = {
+        mail: MailChannel,
+        ...channels
+    }
+
+    /**
+     * the user data for send notification
+     * @param notification
+     */
+    const notifyHandler = (notification) => (new NotificationHandler(config, notification)).setChannels(channels).handle()
+
+    return (req, res, next) => {// set notify handler
+        res.notify = notifyHandler;
+        next();
     }
-  }
-
-  const channelToNotify = async (channel , { notifiable, notification }) => {
-    // check channel exists in global channels list
-    if(channels[channel])
-      await (new channels[channel](config)).send(notifiable , notification);
-  }
-
-  return (req , res , next) => {
-    // set notify handler
-    res.notify = notifyHandler;
-    next();
-  }
 }
 

src/notificationHandler.js

@@ -0,0 +1,60 @@
+class NotificationHandler {
+    /**
+     * the notification class to send message with different channels
+     */
+    notification;
+
+
+    /**
+     * channels list that the system support
+     */
+    channels;
+
+    /**
+     *
+     * prepare list of channels and notifications must be sent
+     */
+    listOfNotificationMustSendWithChannels = {};
+
+
+    /**
+     * list of configs to get from register middleware of package
+     */
+    config;
+
+    constructor(config, notification) {
+        this.config = config;
+        this.notification = notification;
+    }
+
+
+    setChannels(channels) {
+        this.channels = channels;
+        return this;
+    }
+
+    handle() {
+        // get channels that notification must send to them
+        let channelsMustBeNotify = this.notification.via();
+
+        // notify to notifiable with different channels
+        for (const channel of channelsMustBeNotify) {
+            this.listOfNotificationMustSendWithChannels[channel] = this.notification;
+        }
+
+        return this;
+    }
+
+     async to(notifiable) {
+        for (const channel in this.listOfNotificationMustSendWithChannels) {
+            let notification = this.listOfNotificationMustSendWithChannels[channel];
+            // check channel exists in global channels list
+            if (this.channels[channel]) {
+                await (new this.channels[channel](this.config)).send(notifiable, notification);
+            }
+        }
+
+    }
+}
+
+module.exports = NotificationHandler;