From 5eadb4d73dc5384509efeea3b9584ded19956c7b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jos=C3=A9=20Valim?= <jose.valim@gmail.com>
Date: Fri, 20 May 2011 22:20:51 +0200
Subject: Temporarily ship with ContentLength middleware.

---
 railties/lib/rails/application.rb         |  2 +-
 railties/lib/rails/rack.rb                |  8 +++----
 railties/lib/rails/rack/content_length.rb | 38 +++++++++++++++++++++++++++++++
 railties/lib/rails/rack/static.rb         |  5 ----
 4 files changed, 43 insertions(+), 10 deletions(-)
 create mode 100644 railties/lib/rails/rack/content_length.rb
 delete mode 100644 railties/lib/rails/rack/static.rb

(limited to 'railties/lib')

diff --git a/railties/lib/rails/application.rb b/railties/lib/rails/application.rb
index 88ef95334f..816bff29b7 100644
--- a/railties/lib/rails/application.rb
+++ b/railties/lib/rails/application.rb
@@ -141,7 +141,7 @@ module Rails
 
     def default_middleware_stack
       ActionDispatch::MiddlewareStack.new.tap do |middleware|
-        middleware.use ::Rack::ContentLength, config.action_dispatch.x_sendfile_header
+        middleware.use ::Rails::Rack::ContentLength, config.action_dispatch.x_sendfile_header
 
         if rack_cache = config.action_controller.perform_caching && config.action_dispatch.rack_cache
           require "action_dispatch/http/rack_cache"
diff --git a/railties/lib/rails/rack.rb b/railties/lib/rails/rack.rb
index 1f20ceae44..d4a41b217e 100644
--- a/railties/lib/rails/rack.rb
+++ b/railties/lib/rails/rack.rb
@@ -1,8 +1,8 @@
 module Rails
   module Rack
-    autoload :Debugger,  "rails/rack/debugger"
-    autoload :Logger,    "rails/rack/logger"
-    autoload :LogTailer, "rails/rack/log_tailer"
-    autoload :Static,    "rails/rack/static"
+    autoload :ContentLength, "rails/rack/content_length"
+    autoload :Debugger,      "rails/rack/debugger"
+    autoload :Logger,        "rails/rack/logger"
+    autoload :LogTailer,     "rails/rack/log_tailer"
   end
 end
diff --git a/railties/lib/rails/rack/content_length.rb b/railties/lib/rails/rack/content_length.rb
new file mode 100644
index 0000000000..6839af4152
--- /dev/null
+++ b/railties/lib/rails/rack/content_length.rb
@@ -0,0 +1,38 @@
+require 'action_dispatch'
+require 'rack/utils'
+
+module Rails
+  module Rack
+    # Sets the Content-Length header on responses with fixed-length bodies.
+    class ContentLength
+      include ::Rack::Utils
+
+      def initialize(app, sendfile=nil)
+        @app = app
+        @sendfile = sendfile
+      end
+
+      def call(env)
+        status, headers, body = @app.call(env)
+        headers = HeaderHash.new(headers)
+
+        if !STATUS_WITH_NO_ENTITY_BODY.include?(status.to_i) &&
+           !headers['Content-Length'] &&
+           !headers['Transfer-Encoding'] &&
+           !(@sendfile && headers[@sendfile])
+
+          old_body = body
+          body, length = [], 0
+          old_body.each do |part|
+            body << part
+            length += bytesize(part)
+          end
+          old_body.close if old_body.respond_to?(:close)
+          headers['Content-Length'] = length.to_s
+        end
+
+        [status, headers, body]
+      end
+    end
+  end
+end
\ No newline at end of file
diff --git a/railties/lib/rails/rack/static.rb b/railties/lib/rails/rack/static.rb
deleted file mode 100644
index ebe8b9e103..0000000000
--- a/railties/lib/rails/rack/static.rb
+++ /dev/null
@@ -1,5 +0,0 @@
-require 'action_dispatch'
-
-module Rails::Rack
-  Static = ActiveSupport::Deprecation::DeprecatedConstantProxy.new('Rails::Rack::Static', ActionDispatch::Static)
-end
-- 
cgit v1.2.3