module ActionController::Live

def log_error(exception)
  logger = ActionController::Base.logger
  return unless logger
  message = "\n#{exception.class} (#{exception.message}):\n"
  message << exception.annoted_source_code.to_s if exception.respond_to?(:annoted_source_code)
  message << "  " << exception.backtrace.join("\n  ")
  logger.fatal("#{message}\n\n")
end

def process(name)
  t1 = Thread.current
  locals = t1.keys.map { |key| [key, t1[key]] }
  # This processes the action in a child thread. It lets us return the
  # response code and headers back up the rack stack, and still process
  # the body in parallel with sending data to the client
  Thread.new {
    t2 = Thread.current
    t2.abort_on_exception = true
    # Since we're processing the view in a different thread, copy the
    # thread locals from the main thread to the child thread. :'(
    locals.each { |k,v| t2[k] = v }
    begin
      super(name)
    rescue => e
      begin
        @_response.stream.write(ActionView::Base.streaming_completion_on_exception) if request.format == :html
        @_response.stream.call_on_error
      rescue => exception
        log_error(exception)
      ensure
        log_error(e)
        @_response.stream.close
      end
    ensure
      @_response.commit!
    end
  }
  @_response.await_commit
end

def response_body=(body)
  super
  response.stream.close if response
end

def set_response!(request)
  if request.env["HTTP_VERSION"] == "HTTP/1.0"
    super
  else
    @_response         = Live::Response.new
    @_response.request = request
  end
end