lib/angelo/base.rb in angelo-0.1.24 vs lib/angelo/base.rb in angelo-0.2.0

@@ -3,11 +3,11 @@
   class Base
     include ParamsParser
     include Celluloid::Logger
 
     extend Forwardable
-    def_delegators :@responder, :content_type, :headers, :redirect, :request
+    def_delegators :@responder, :content_type, :headers, :redirect, :request, :transfer_encoding
     def_delegators :@klass, :websockets, :sses, :sse_event, :sse_message
 
     @@addr = DEFAULT_ADDR
     @@port = DEFAULT_PORT
 
@@ -285,11 +285,11 @@
       halt 200, data
     end
 
     def eventsource &block
       headers SSE_HEADER
-      async :handle_event_source, responder.connection.detach.socket, block
+      async :handle_event_source, EventSource.new(responder.connection.detach.socket), block
       halt 200, :sse
     end
 
     def report_errors?
       @@report_errors
@@ -301,9 +301,46 @@
 
     def filter which
       fs = self.class.filters[which][:default]
       fs += self.class.filters[which][request.path] if self.class.filters[which][request.path]
       fs.each {|f| f.bind(self).call}
+    end
+
+    def chunked_response &block
+      transfer_encoding :chunked
+      ChunkedResponse.new &block
+    end
+
+    class EventSource
+      extend Forwardable
+
+      def_delegators :@socket, :close, :closed?, :<<, :write
+      attr_reader :socket
+
+      def initialize socket
+        @socket = socket
+      end
+
+      def event name, data = nil
+        @socket.write Base.sse_event(name, data)
+      end
+
+      def message data
+        @socket.write Base.sse_message(data)
+      end
+
+    end
+
+    class ChunkedResponse
+
+      def initialize &block
+        @chunker = block
+      end
+
+      def each &block
+        @chunker[block]
+      end
+
     end
 
   end
 
 end