# Copyright: Savonix Corporation, 2009
# Author:    Albert Lash
# License:   MIT
module Rack
  class DocunextContentLength
    def initialize(app)
      @app = app
    end

    def call(env)
        status, headers, response = @app.call(env)
        headers.delete('Content-Length')
        if status == 304
          [status, headers, response]
        else
          response_body = ""
          response.each { |part| response_body += part }
          headers["Content-Length"] = response_body.length.to_s
          [status, headers, response_body]
        end
    end
  end
end