Skip to content

Commit

Permalink
Merge branch 'pm_add_log_scrubbing' of https://github.com/petems/faraday
Browse files Browse the repository at this point in the history
 into petems-pm_add_log_scrubbing

# Conflicts:
#	lib/faraday/response/logger.rb
  • Loading branch information
iMacTia committed Dec 29, 2016
2 parents e66210b + a557e87 commit dadc002
Show file tree
Hide file tree
Showing 3 changed files with 43 additions and 5 deletions.
8 changes: 8 additions & 0 deletions README.md
Expand Up @@ -30,6 +30,14 @@ conn = Faraday.new(:url => 'http://sushi.com') do |faraday|
faraday.adapter Faraday.default_adapter # make requests with Net::HTTP
end

# filter sensitive information from logs with a regex matcher

conn = Faraday.new(:url => 'http://sushi.com/api_key=s3cr3t') do |faraday|
faraday.response :logger do | logger |
logger.filter(/(api_key=)(\w+)/,'\1[REMOVED]')
end
end

## GET ##

response = conn.get '/nigiri/sake.json' # GET http://sushi.com/nigiri/sake.json
Expand Down
19 changes: 14 additions & 5 deletions lib/faraday/response/logger.rb
Expand Up @@ -12,22 +12,24 @@ def initialize(app, logger = nil, options = {})
require 'logger'
::Logger.new(STDOUT)
end
@filter = []
@options = DEFAULT_OPTIONS.merge(options)
yield self if block_given?
end

def_delegators :@logger, :debug, :info, :warn, :error, :fatal

def call(env)
info "#{env.method} #{env.url.to_s}"
debug('request') { dump_headers env.request_headers } if log_headers?(:request)
debug('request') { dump_body(env[:body]) } if env[:body] && log_body?(:request)
info "#{env.method} #{apply_filters(env.url.to_s)}"
debug('request') { apply_filters(dump_headers env.request_headers) } if log_headers?(:request)
debug('request') { apply_filters(dump_body(env[:body])) } if env[:body] && log_body?(:request)
super
end

def on_complete(env)
info('Status') { env.status.to_s }
debug('response') { dump_headers env.response_headers } if log_headers?(:response)
debug('response') { dump_body env[:body] } if env[:body] && log_body?(:response)
debug('response') { apply_filters(dump_headers env.response_headers) } if log_headers?(:response)
debug('response') { apply_filters(dump_body env[:body]) } if env[:body] && log_body?(:response)
end

private
Expand Down Expand Up @@ -62,5 +64,12 @@ def log_body?(type)
else @options[:bodies]
end
end

def apply_filters(output)
@filter.each do |pattern, replacement|
output = output.to_s.gsub(pattern, replacement)
end
output
end
end
end
21 changes: 21 additions & 0 deletions test/adapters/logger_test.rb
Expand Up @@ -9,6 +9,10 @@ def conn(logger, logger_options={})

Faraday.new do |b|
b.response :logger, logger, logger_options
b.response :logger, @logger do | logger |
logger.filter(/foo/,'[REDACTED]')
logger.filter(/(api_key:).*"(.+)."/,'\1[API_KEY]')
end
b.adapter :test do |stubs|
stubs.get('/hello') { [200, {'Content-Type' => 'text/html'}, 'hello'] }
stubs.post('/ohai') { [200, {'Content-Type' => 'text/html'}, 'fred'] }
Expand Down Expand Up @@ -94,5 +98,22 @@ def test_log_response_body_object
app.get '/rubbles', nil, :accept => 'text/html'
assert_match %([\"Barney\", \"Betty\", \"Bam Bam\"]\n), @io.string
end

def test_logs_filter_url
app = conn(@logger)
app.get '/foo', nil, :accept => 'text/html'
assert_match %([REDACTED]), @io.string
refute_match %(foo), @io.string
end

def test_logs_filter_header
app = conn(@logger)
app.headers = {'api_key' => 'ABC123'}
app.get '/bar', nil, :accept => 'text/html'
assert_match %(api_key:), @io.string
assert_match %([API_KEY]), @io.string
refute_match %(ABC123), @io.string
end

end
end

0 comments on commit dadc002

Please sign in to comment.