rack_iphone_web_app 0.0.3 → 0.0.4

data/VERSION

@@ -1 +1 @@
-0.0.3
+0.0.4

data/lib/rack_iphone_web_app.rb

@@ -33,6 +33,7 @@ module Rack
               localStorage.setItem('__cookie__', ''+document.cookie);
             }
           },1000);
+
         }
       })()
       </script>
@@ -45,24 +46,17 @@ module Rack
     
     def call(env)
       if iphone_web_app?(env)
-        status, headers, body = @app.call(env)
-        
         if new_session?(env)
           [200,{'Content-Length' => code(true).length.to_s, 'Content-Type' => 'text/html'}, code(true)]
         else
+          status, headers, body = @app.call(env)
           # Put in patch code
+          response = Rack::Response.new([], status, headers)
           body.each do |part|
-            if part =~ /<\/head>/
-              part.sub!(/<\/head>/, "#{code}</head>")
-            
-              if headers['Content-Length']
-                headers['Content-Length'] = (headers['Content-Length'].to_i + code.length).to_s
-              end
-            
-              break
-            end  
+            part.gsub!(/<\/head>/, "#{code}</head>")
+            response.write(part)
           end
-          [status, headers, body]
+          response.finish
         end
       else
         @app.call(env)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: rack_iphone_web_app
 version: !ruby/object:Gem::Version 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Jim Hoskins