module YogiBerra
  class ExceptionMiddleware
    def initialize(app)
      @app = app
      YogiBerra::Catcher.quick_connection
    end

    def call(env)
      begin
        response = dup._call(env)
        environment = {
          :session => env['rack.session'],
          :params => env['action_controller.request.path_parameters'].merge(env['rack.request.query_hash']),
          :user_agent => env['HTTP_USER_AGENT'],
          :server_name => env['SERVER_NAME'],
          :server_port => env['SERVER_PORT'],
          :server_address => env['SERVER_ADDR'],
          :remote_address => env['REMOTE_ADDR']
        }
      rescue Exception => raised
        YogiBerra.exceptionize(raised, environment, YogiBerra::Catcher.connection)
        raise raised
      end

      if env['rack.exception']
        YogiBerra.exceptionize(raised, environment, YogiBerra::Catcher.connection)
      end
      response
    end

    def _call(env)
      @status, @headers, @response = @app.call(env)
      [@status, @headers, self]
    end

    def each(&block)
      @response.each(&block)
    end

  end
end