grape-middleware-logger 1.7.1 → 1.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c09e5c24a2fe69018c3193eae50e2ce2f9879600
-  data.tar.gz: 6e728c86c74a64debb9071909edcb7dc93e72273
+  metadata.gz: d7c3e7ac5229767e264e6037c0a1af92946af940
+  data.tar.gz: b471c789e7ad5107630deea636a10594f623da02
 SHA512:
-  metadata.gz: c9a08f38acf72a89882737bb91194a2b74c266d68c3b005bd24712aa42b0ade41c41cda9c88db027ca6653936a590e536b4f00bba7844d22fa61a1b5ac83836d
-  data.tar.gz: 2c0f0bb9dc3093a01adc33152de30235985541e6d59bfb32756b98286d922e204158500d4645546aebc279f463478e5244cc2b4dbef2b3e70f0e93c768a483b6
+  metadata.gz: 85a21c6f2dd2c0376eb2db6968683e2444a75dfe982206450c707fdade2a2327811eddaca6aba516c64f01c86b26b5ae1115b4cc575930105192ea768113d423
+  data.tar.gz: 3d4510679704050630b392f1246843a37885fac14cd56621b527bfb27490e5a6439e0cf198ed7b966ade5cf76a7a7639fa15718b83fefa134012f13c39cebd62

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+1.8.0 (4/22/2017)
+==================
+* [#17] Add a `:headers` option, which can be either `:all` or an array of strings. (Thanks [@yamamotok](https://github.com/yamamotok))
+
 1.7.1 (11/1/2016)
 ==================
 * Log the error class name (https://github.com/ridiculous/grape-middleware-logger/pull/13)

data/README.md

@@ -53,13 +53,17 @@ The middleware logger can be customized with the following options:
 
 * The `:logger` option can be any object that responds to `.info(String)`
 * The `:filter` option can be any object that responds to `.filter(Hash)` and returns a hash.
+* The `:headers` option can be either `:all` or array of strings.
+    + If `:all`, all request headers will be output.
+    + If array, output will be filtered by names in the array. (case-insensitive)
 
 For example:
 
 ```ruby
 insert_after Grape::Middleware::Formatter, Grape::Middleware::Logger, {
   logger: Logger.new(STDERR),
-  filter: Class.new { def filter(opts) opts.reject { |k, _| k.to_s == 'password' } end }.new
+  filter: Class.new { def filter(opts) opts.reject { |k, _| k.to_s == 'password' } end }.new,
+  headers: %w(version cache-control)
 }
 ```
 

data/grape-middleware-logger.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |spec|
   spec.name          = 'grape-middleware-logger'
-  spec.version       = '1.7.1'
+  spec.version       = '1.8.0'
   spec.platform      = Gem::Platform::RUBY
   spec.authors       = ['Ryan Buckley']
   spec.email         = ['arebuckley@gmail.com']

data/lib/grape/middleware/logger.rb

@@ -7,7 +7,7 @@ class Grape::Middleware::Logger < Grape::Middleware::Globals
   attr_reader :logger
 
   class << self
-    attr_accessor :logger, :filter
+    attr_accessor :logger, :filter, :headers
 
     def default_logger
       default = Logger.new(STDOUT)
@@ -19,6 +19,7 @@ class Grape::Middleware::Logger < Grape::Middleware::Globals
   def initialize(_, options = {})
     super
     @options[:filter] ||= self.class.filter
+    @options[:headers] ||= self.class.headers
     @logger = options[:logger] || self.class.logger || self.class.default_logger
   end
 
@@ -34,6 +35,7 @@ class Grape::Middleware::Logger < Grape::Middleware::Globals
     ]
     logger.info %Q(Processing by #{processed_by})
     logger.info %Q(  Parameters: #{parameters})
+    logger.info %Q(  Headers: #{headers}) if @options[:headers]
   end
 
   # @note Error and exception handling are required for the +after+ hooks
@@ -91,6 +93,18 @@ class Grape::Middleware::Logger < Grape::Middleware::Globals
     end
   end
 
+  def headers
+    request_headers = env[Grape::Env::GRAPE_REQUEST_HEADERS].to_hash
+    return Hash[request_headers.sort] if @options[:headers] == :all
+
+    headers_needed = Array(@options[:headers])
+    result = {}
+    headers_needed.each do |need|
+      result.merge!(request_headers.select { |key, value| need.to_s.casecmp(key).zero? })
+    end
+    Hash[result.sort]
+  end
+
   def start_time
     @start_time ||= Time.now
   end

data/spec/factories.rb

@@ -32,8 +32,10 @@ FactoryGirl.define do
     grape_request { build :grape_request }
     grape_endpoint { build(:grape_endpoint) }
     params { grape_request.params }
+    headers { grape_request.headers }
     post_params { { 'secret' => 'key', 'customer' => [] } }
     rails_post_params { { 'name' => 'foo', 'password' => 'access' } }
+    other_env_params { {} }
 
     initialize_with do
       new.merge(
@@ -42,10 +44,21 @@ FactoryGirl.define do
         'action_dispatch.request.request_parameters' => rails_post_params,
         Grape::Env::GRAPE_REQUEST => grape_request,
         Grape::Env::GRAPE_REQUEST_PARAMS => params,
+        Grape::Env::GRAPE_REQUEST_HEADERS => headers,
         Grape::Env::RACK_REQUEST_FORM_HASH => post_params,
         Grape::Env::API_ENDPOINT => grape_endpoint
-      )
+      ).merge(other_env_params)
     end
+
+    trait :prefixed_basic_headers do
+      other_env_params { {
+        'HTTP_VERSION' => 'HTTP/1.1',
+        'HTTP_CACHE_CONTROL' => 'max-age=0',
+        'HTTP_USER_AGENT' => 'Mozilla/5.0',
+        'HTTP_ACCEPT_LANGUAGE' => 'en-US'
+      } }
+    end
+
   end
 
   factory :grape_endpoint, class: Grape::Endpoint do
@@ -84,9 +97,20 @@ FactoryGirl.define do
   end
 
   factory :grape_request, class: OpenStruct do
+    headers { {} }
+
     initialize_with {
-      new(request_method: 'POST', path: '/api/1.0/users', headers: {}, params: { 'id' => '101001' })
+      new(request_method: 'POST', path: '/api/1.0/users', headers: headers, params: { 'id' => '101001' })
     }
+
+    trait :basic_headers do
+      headers { {
+        'Version' => 'HTTP/1.1',
+        'Cache-Control' => 'max-age=0',
+        'User-Agent' => 'Mozilla/5.0',
+        'Accept-Language' => 'en-US'
+      } }
+    end
   end
 
   factory :app do

data/spec/integration/lib/grape/middleware/headers_option_spec.rb

@@ -0,0 +1,65 @@
+require 'spec_helper'
+
+describe Grape::Middleware::Logger, type: :integration do
+  let(:app) { build :app }
+
+  subject { described_class.new(app, options) }
+
+  let(:grape_endpoint) { build(:grape_endpoint) }
+  let(:env) { build(:expected_env, :prefixed_basic_headers, grape_endpoint: grape_endpoint) }
+
+  context ':all option is set to option headers' do
+    let(:options) { {
+      filter: build(:param_filter),
+      headers: :all,
+      logger: Logger.new(Tempfile.new('logger'))
+    } }
+    it 'all headers will be shown, headers will be sorted by name' do
+      expect(subject.logger).to receive(:info).with ''
+      expect(subject.logger).to receive(:info).with %Q(Started POST "/api/1.0/users" at #{subject.start_time})
+      expect(subject.logger).to receive(:info).with %Q(Processing by TestAPI/users)
+      expect(subject.logger).to receive(:info).with %Q(  Parameters: {"id"=>"101001", "secret"=>"[FILTERED]", "customer"=>[], "name"=>"foo", "password"=>"[FILTERED]"})
+      expect(subject.logger).to receive(:info).with %Q(  Headers: {"Accept-Language"=>"en-US", "Cache-Control"=>"max-age=0", "User-Agent"=>"Mozilla/5.0", "Version"=>"HTTP/1.1"})
+      expect(subject.logger).to receive(:info).with /Completed 200 in \d+.\d+ms/
+      expect(subject.logger).to receive(:info).with ''
+      subject.call!(env)
+    end
+  end
+
+  context 'list of names ["User-Agent", "Cache-Control"] is set to option headers' do
+    let(:options) { {
+      filter: build(:param_filter),
+      headers: %w(User-Agent Cache-Control),
+      logger: Logger.new(Tempfile.new('logger'))
+    } }
+    it 'two headers will be shown' do
+      expect(subject.logger).to receive(:info).with ''
+      expect(subject.logger).to receive(:info).with %Q(Started POST "/api/1.0/users" at #{subject.start_time})
+      expect(subject.logger).to receive(:info).with %Q(Processing by TestAPI/users)
+      expect(subject.logger).to receive(:info).with %Q(  Parameters: {"id"=>"101001", "secret"=>"[FILTERED]", "customer"=>[], "name"=>"foo", "password"=>"[FILTERED]"})
+      expect(subject.logger).to receive(:info).with %Q(  Headers: {"Cache-Control"=>"max-age=0", "User-Agent"=>"Mozilla/5.0"})
+      expect(subject.logger).to receive(:info).with /Completed 200 in \d+.\d+ms/
+      expect(subject.logger).to receive(:info).with ''
+      subject.call!(env)
+    end
+  end
+
+  context 'a single string "Cache-Control" is set to option headers' do
+    let(:options) { {
+      filter: build(:param_filter),
+      headers: 'Cache-Control',
+      logger: Logger.new(Tempfile.new('logger'))
+    } }
+    it 'only Cache-Control header will be shown' do
+      expect(subject.logger).to receive(:info).with ''
+      expect(subject.logger).to receive(:info).with %Q(Started POST "/api/1.0/users" at #{subject.start_time})
+      expect(subject.logger).to receive(:info).with %Q(Processing by TestAPI/users)
+      expect(subject.logger).to receive(:info).with %Q(  Parameters: {"id"=>"101001", "secret"=>"[FILTERED]", "customer"=>[], "name"=>"foo", "password"=>"[FILTERED]"})
+      expect(subject.logger).to receive(:info).with %Q(  Headers: {"Cache-Control"=>"max-age=0"})
+      expect(subject.logger).to receive(:info).with /Completed 200 in \d+.\d+ms/
+      expect(subject.logger).to receive(:info).with ''
+      subject.call!(env)
+    end
+  end
+
+end

data/spec/lib/grape/middleware/headers_option_spec.rb

@@ -0,0 +1,57 @@
+require 'spec_helper'
+
+describe Grape::Middleware::Logger do
+  let(:app) { double('app') }
+
+  subject { described_class.new(app, options) }
+
+  describe '#headers' do
+    let(:grape_request) { build :grape_request, :basic_headers }
+    let(:env) { build :expected_env, grape_request: grape_request }
+
+    before { subject.instance_variable_set(:@env, env) }
+
+    context 'when @options[:headers] has a symbol :all' do
+      let(:options) { { headers: :all, logger: Object.new } }
+      it 'all request headers should be retrieved' do
+        expect(subject.headers.fetch('Accept-Language')).to eq('en-US')
+        expect(subject.headers.fetch('Cache-Control')).to eq('max-age=0')
+        expect(subject.headers.fetch('User-Agent')).to eq('Mozilla/5.0')
+        expect(subject.headers.fetch('Version')).to eq('HTTP/1.1')
+      end
+    end
+
+    context 'when @options[:headers] is a string "user-agent"' do
+      let(:options) { { headers: 'user-agent', logger: Object.new } }
+      it 'only "User-Agent" should be retrieved' do
+        expect(subject.headers.fetch('User-Agent')).to eq('Mozilla/5.0')
+        expect(subject.headers.length).to eq(1)
+      end
+    end
+
+    context 'when @options[:headers] is an array of ["user-agent", "Cache-Control", "Unknown"]' do
+      let(:options) { { headers: %w(user-agent Cache-Control Unknown), logger: Object.new } }
+      it '"User-Agent" and "Cache-Control" should be retrieved' do
+        expect(subject.headers.fetch('Cache-Control')).to eq('max-age=0')
+        expect(subject.headers.fetch('User-Agent')).to eq('Mozilla/5.0')
+      end
+      it '"Unknown" name does not make any effect' do
+        expect(subject.headers.length).to eq(2)
+      end
+    end
+  end
+
+  describe '#headers if no request header' do
+    let(:env) { build :expected_env }
+    before { subject.instance_variable_set(:@env, env) }
+
+    context 'when @options[:headers] is set, but no request header is there' do
+      let(:options) { { headers: %w(user-agent Cache-Control), logger: Object.new } }
+      it 'subject.headers should return empty hash' do
+        expect(subject.headers.length).to eq(0)
+      end
+    end
+  end
+
+end
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: grape-middleware-logger
 version: !ruby/object:Gem::Version
-  version: 1.7.1
+  version: 1.8.0
 platform: ruby
 authors:
 - Ryan Buckley
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-11-02 00:00:00.000000000 Z
+date: 2017-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: grape
@@ -112,8 +112,10 @@ files:
 - lib/grape/middleware/logger/railtie.rb
 - spec/factories.rb
 - spec/fixtures/rails_app.rb
+- spec/integration/lib/grape/middleware/headers_option_spec.rb
 - spec/integration/lib/grape/middleware/logger_spec.rb
 - spec/integration_rails/lib/grape/middleware/logger_spec.rb
+- spec/lib/grape/middleware/headers_option_spec.rb
 - spec/lib/grape/middleware/logger_spec.rb
 - spec/rails_helper.rb
 - spec/spec_helper.rb
@@ -146,8 +148,10 @@ summary: A logger for the Grape framework
 test_files:
 - spec/factories.rb
 - spec/fixtures/rails_app.rb
+- spec/integration/lib/grape/middleware/headers_option_spec.rb
 - spec/integration/lib/grape/middleware/logger_spec.rb
 - spec/integration_rails/lib/grape/middleware/logger_spec.rb
+- spec/lib/grape/middleware/headers_option_spec.rb
 - spec/lib/grape/middleware/logger_spec.rb
 - spec/rails_helper.rb
 - spec/spec_helper.rb