1 files changed, 73 insertions, 0 deletions
diff --git a/lib/action_cable/server.rb b/lib/action_cable/server.rb
new file mode 100644
index 0000000000..2d80e96265
--- /dev/null
+++ b/lib/action_cable/server.rb
@@ -0,0 +1,73 @@
+require 'set'
+
+module ActionCable
+  class Server < Cramp::Websocket
+    on_start :initialize_subscriptions
+    on_data :received_data
+    on_finish :cleanup_subscriptions
+
+    class_attribute :registered_channels
+    self.registered_channels = Set.new
+
+    class << self
+      def register_channels(*channel_classes)
+        registered_channels.merge(channel_classes)
+      end
+    end
+
+    def initialize_subscriptions
+      @subscriptions = {}
+    end
+
+    def received_data(data)
+      data = ActiveSupport::JSON.decode data
+
+      case data['action']
+      when 'subscribe'
+        subscribe_channel(data)
+      when 'unsubscribe'
+        unsubscribe_channel(data)
+      when 'message'
+        process_message(data)
+      end
+    end
+
+    def cleanup_subscriptions
+      @subscriptions.each do |id, channel|
+        channel.unsubscribe
+      end
+    end
+
+    def publish(data)
+      render data
+    end
+
+    private
+      def subscribe_channel(data)
+        id_key = data['identifier']
+        id_options = ActiveSupport::JSON.decode(id_key).with_indifferent_access
+
+        if subscription = registered_channels.detect { |channel_klass| channel_klass.matches?(id_options) }
+          @subscriptions[id_key] = subscription.new(self, id_key, id_options)
+          @subscriptions[id_key].subscribe
+        else
+          # No channel found
+        end
+      end
+
+      def process_message(message)
+        id_key = message['identifier']
+
+        if @subscriptions[id_key]
+          @subscriptions[id_key].receive(ActiveSupport::JSON.decode message['data'])
+        end
+      end
+
+      def unsubscribe_channel(data)
+        id_key = data['identifier']
+        @subscriptions[id_key].unsubscribe
+        @subscriptions.delete(id_key)
+      end
+
+  end
+end