rack-dev-mark 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: df36ab5b5e659caeee87e9392be1626d6bc60834
-  data.tar.gz: f152c7f27e311c166858de3403a0dd56e074910c
+  metadata.gz: fff2b9a8075587a3cd9e845a0c894cdb3ba57aca
+  data.tar.gz: 740888ac9204d53a0956cec0379c9c401f06b538
 SHA512:
-  metadata.gz: 3287bb471bdf78d02ed55bb92a365f90042041a773db600ce2a429a656fb5e6a8bb462ca53a0a356a8976bb6009d40251537134c0ecf1751d8cc3e752ced7276
-  data.tar.gz: 8618e72dfd9a95ed37ccae14038ab616e80673187c40cd8a6d24043492230163685638ee0e6844a22a6a417ba6503cedf868adb076a666a2425f67241ee4c431
+  metadata.gz: 975b82a47ccc2761bdb6015d8982a929db6cc8f299ac7d32bd5d0ba0e39ade48d848958622ede56463defffc347046137e2c7ed1284bd92fd45d2bae641651f3
+  data.tar.gz: 5987c26f01630eaf4dd057d415facb884887f1799d3d41d50bbcdcf1bc44957844b14ea0dea25f1cbba6ac0ea694123d7b029d5039cb0223417ef5c86bafe190

data/lib/rack/dev-mark/middleware.rb

@@ -16,11 +16,17 @@ module Rack
         headers = HeaderHash.new(headers)
 
         if headers['Content-Type'].to_s =~ %r{^text/html;}i
-          begin
-            response = insert_dev_marks(response)
-          rescue Rack::DevMark::Exception => e
-            $stderr.write "Failed to insert dev marks: #{e.message}\n"
+          new_body = ''
+          response.each do |b|
+            begin
+              new_body << insert_dev_marks(b)
+            rescue Rack::DevMark::Exception => e
+              $stderr.write "Failed to insert dev marks: #{e.message}\n"
+            end
           end
+          response.close if response.respond_to?(:close)
+          headers['Content-Length'] &&= bytesize(new_body).to_s
+          response = [new_body]
         end
 
         [status, headers, response]
@@ -28,14 +34,10 @@ module Rack
 
       private
 
-      def insert_dev_marks(response)
-        body = ""
-        response.each do |r|
-          body.concat r.to_s
-        end
+      def insert_dev_marks(body)
         body = @title.insert_into(body, Rack::DevMark.env, Rack::DevMark.revision)
         body = @theme.insert_into(body, Rack::DevMark.env, Rack::DevMark.revision)
-        [body]
+        body
       end
     end
   end

data/lib/rack/dev-mark/version.rb

@@ -1,5 +1,5 @@
 module Rack
   module DevMark
-    VERSION = '0.1.0'
+    VERSION = '0.1.2'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rack-dev-mark
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Daisuke Taniwaki
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-06-02 00:00:00.000000000 Z
+date: 2014-06-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rack