Action Cable channels merged

Also, proactively send notification counts on reconnection

Author: lmrodriguezr

app/channels/notification_counts_channel.rb

@@ -1,5 +1,12 @@
 class WebNotificationsChannel < ApplicationCable::Channel
   def subscribed
+    # Open streaming
     stream_for current_user
+
+    # Proactively send count on subscription
+    transmit({
+      kind: 'notification_count',
+      count: current_user.unseen_notifications.count
+    })
   end
 end

app/channels/web_notifications_channel.rb

@@ -6,6 +6,18 @@ const wn_struct = {
   },
 
   received(data) {
+    switch (data["kind"]) {
+      case "web_notification":
+        this.web_notification(data);
+        break;
+      case "notification_count":
+        this.notification_count(data);
+        break;
+    }
+  },
+
+  // Web Notifications (individual event)
+  web_notification(data) {
     if (Notification.permission === "granted") {
       this.browser_notification(data);
     }
@@ -33,6 +45,13 @@ const wn_struct = {
 
   toast_notification(data) {
     new_toast("", data["title"], ROOT_PATH + "alerts/" + data["alert"]);
+  },
+
+  // Notification Count (aggregated)
+  notification_count(data) {
+    var badge = $("#notification-count-badge");
+    badge.html(data["count"] == 0 ? "" : data["count"]);
+    badge.data("count", data["count"]);
   }
 };
 

app/javascript/channels/notification_counts_channel.js

@@ -25,6 +25,7 @@ def send_email_notification
   def broadcast_web_notification
     WebNotificationsChannel.broadcast_to(
       user,
+      kind: 'web_notification',
       title: title || action,
       tag: '%s:%s' % [notifiable.class.to_s, notifiable.id],
       alert: id
@@ -34,8 +35,10 @@ def broadcast_web_notification
   end
 
   def broadcast_notification_count
-    NotificationCountsChannel.broadcast_to(
-      user, count: user.unseen_notifications.count
+    WebNotificationsChannel.broadcast_to(
+      user,
+      kind: 'notification_count',
+      count: user.unseen_notifications.count
     )
   rescue => e
     Rails.logger.error e