Move connection and subscriber code into their own classes

1 files changed, 26 insertions, 0 deletions

diff --git a/lib/assets/javascripts/cable/subscriber_manager.js.coffee b/lib/assets/javascripts/cable/subscriber_manager.js.coffee
new file mode 100644
index 0000000000..4f46efe817
--- /dev/null
+++ b/lib/assets/javascripts/cable/subscriber_manager.js.coffee
@@ -0,0 +1,26 @@
+class Cable.SubscriberManager
+  constructor: (@cable) ->
+    @subscribers = {}
+
+  add: (identifier, subscriber) ->
+    @subscribers[identifier] = subscriber
+    if @cable.sendCommand(identifier, "subscribe")
+      @notify(identifier, "onConnect")
+
+  reload: ->
+    for identifier in Object.keys(@subscribers)
+      if @cable.sendCommand(identifier, "subscribe")
+        @notify(identifier, "onConnect")
+
+  remove: (identifier) ->
+    if subscriber = @subscribers[identifier]
+      @cable.sendCommand(identifier, "unsubscribe")
+      delete @subscribers[identifier]
+
+  notifyAll: (event, args...) ->
+    for identifier in Object.keys(@subscribers)
+      @notify(identifier, event, args...)
+
+  notify: (identifier, event, args...) ->
+    if subscriber = @subscribers[identifier]
+      subscriber[event]?(args...)