module ActionController::Live

def log_error(exception)
  logger = ActionController::Base.logger
  return unless logger
  logger.fatal do
    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  ")
    "#{message}\n\n"
  end
end

def new_controller_thread # :nodoc:
  Thread.new {
    t2 = Thread.current
    t2.abort_on_exception = true
    yield
  }
end

def new_controller_thread # :nodoc:
  yield
end

def process(name)
  t1 = Thread.current
  locals = t1.keys.map { |key| [key, t1[key]] }
  error = nil
  # 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
  new_controller_thread {
    ActiveSupport::Dependencies.interlock.running do
      t2 = Thread.current
      # 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
        if @_response.committed?
          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
        else
          error = e
        end
      ensure
        @_response.commit!
      end
    end
  }
  ActiveSupport::Dependencies.interlock.permit_concurrent_loads do
    @_response.await_commit
  end
  raise error if error
end

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