devbar 0.0.1 → 0.0.2

data/lib/devbar/version.rb

@@ -1,3 +1,3 @@
 module Devbar
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/devbar.rb

@@ -7,15 +7,22 @@ module Rack
     end
 
     def call(env)
-      status, headers, response = @app.call(env)
+      status, headers, body  = @app.call(env)
+      return [status, headers, response] unless headers['Content-Type'] =~ /html/
 
-      if headers["Content-Type"].include? "text/html"
-        full_body = response.body
-        full_body.sub! /<\/body>/, bar + "</body>"
-        response.body = full_body
+      body.each do |part|
+        if part =~ /<\/body>/
+          part.sub!(/<\/body>/, "#{bar}</body>")
+
+          if headers['Content-Length']
+            headers['Content-Length'] = (headers['Content-Length'].to_i + bar.length).to_s
+          end
+
+          break
+        end
       end
 
-      [status, headers, response]
+      [status, headers, body]
     end
 
     def bar

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: devbar
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-03-13 00:00:00.000000000 Z
+date: 2013-03-14 00:00:00.000000000 Z
 dependencies: []
 description: ! ' Rack middleware to output a bar showing environment '
 email: