rack-img-sizes 0.0.1 → 0.0.2

data/lib/rack/img_sizes.rb

@@ -11,29 +11,38 @@ module Rack
       @app = app
     end
 
-    def call(env)
+    def call(env); dup._call(env); end
+
+    def _call(env)
       @env = env
-      status, headers, response = @app.call(env)
+      @status, @headers, @body = @app.call(env)
+      return [@status, @headers, @body] unless html?
 
-      new_response = response.map do |part|
-        process_part(part)
-      end
+      new_response = Rack::Response.new([], @status, @headers)
+
+      @body.each { |part| new_response.write process_part(part) }
+      @body.close if @body.respond_to?(:close)
 
-      [status, headers, new_response]
+      new_response.finish
     end
 
     private
+    def html?; @headers['Content-Type'] =~ /html/; end
+
     def process_part(part)
-      doc = Nokogiri::HTML.parse(part)
+      doc = Nokogiri::HTML(part)
       doc.css("img").each do |img|
         old_img = img.dup.to_s
+
         if src = img["src"]
           uri = ::Rack::ImgSizes::URI.new(@env, src)
           width, height = *dimensions(uri.to_s)
+
           img.set_attribute("width" , width.to_s)
           img.set_attribute("height" , height.to_s)
 
-          part.sub!(old_img, img.to_s)
+          regexp = %r{(<img.*?src="#{Regexp.escape(src)}".*?>)}
+          part[regexp, 1] = img.to_s
         end
       end
       part

data/lib/rack/img_sizes/version.rb

@@ -1,5 +1,5 @@
 module Rack
-  module ImgSizes
-    VERSION = "0.0.1"
+  class ImgSizes
+    VERSION = "0.0.2"
   end
 end

data/test/integration_test.rb

@@ -27,8 +27,8 @@ class MiddlewareTest < Test::Unit::TestCase
       <article id="main" role="main">
         <header>
           <img src="logo.png">
-          <img src="/images/icons/adwords-24.gif">
-          <img src="http://www.google.com/images/logos/ps_logo.png">
+          <img src="/images/icons/adwords-24.gif"/>
+          <img src="http://www.google.com/images/logos/ps_logo.png" />
         </header>
       </article>
     </div>

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rack-img-sizes
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors: