require 'rest-core/middleware'

class RestCore::Oauth2Header
  def self.members; [:access_token]; end
  include RestCore::Middleware

  def call env
    start_time = Time.now
    headers = {'Authorization' => "OAuth #{access_token(env)}"}.
                merge(env[REQUEST_HEADERS] || {}) if access_token(env)

    event = Event::WithHeader.new(Time.now - start_time,
              "Authorization: #{headers['Authorization']}") if headers

    app.call(log(cache_key(
      env.merge(REQUEST_HEADERS => headers || env[REQUEST_HEADERS])), event))
  end

  def cache_key env
    env.merge('cache.key' => "#{request_uri(env)}&#{access_token(env)}")
  end
end