rack-traffic-logger 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f26f03dcdb7e324d22fe2d1eef5c171d2b0a2a7e
-  data.tar.gz: 4dc100276db42b05d2d57673ec9a633de4ecef8a
+  metadata.gz: 735bee50aea81ceaff58c19569eb8dbac38806a1
+  data.tar.gz: 2ae588f2b1990d712692d5bff1bf96687d581a71
 SHA512:
-  metadata.gz: b03a20d01b1c64ff07d84c450c21e4ecdba97626c0c892cc6c1a331a569c4ab10e64cb3a2e9e47f68c2e408b3c828fb5f01ed0e428670c69a49c1486fa721663
-  data.tar.gz: 6281b3c7a221db00a86e1d4eb6178b75fc53bac6e506922a742580cb823922f06dc7ab04f8a3a35593ff757091d05258f08b0150a10d42bb301c3a916606bc8c
+  metadata.gz: c9c2680d8fe14afd6f9b3afa9b744fa6debac3235dca0232095fef4da816aecb05958ce1d7c8c1738187ca1411884cc54b473d229e4325cb1c2a4738cff74441
+  data.tar.gz: adab23dfc915298bb226fee7079086f1e85a555c8a517882ad1c3495910f508d8ec6c163d314a1c1dc2c28764d95ca3c3cfabc5b45dedefa1e73ecc27ae48179

data/lib/rack/traffic_logger.rb

@@ -4,6 +4,7 @@ require_relative 'traffic_logger/header_hash'
 
 require 'forwardable'
 require 'rack/nulllogger'
+require 'json'
 
 module Rack
   class TrafficLogger
@@ -14,7 +15,9 @@ module Rack
         request_bodies:   {type: [TrueClass, FalseClass], default: true},
         response_headers: {type: [TrueClass, FalseClass], default: true},
         response_bodies:  {type: [TrueClass, FalseClass], default: true},
-        colors: {type: [TrueClass, FalseClass], default: false}
+        colors: {type: [TrueClass, FalseClass], default: false},
+        prevent_compression: {type: [TrueClass, FalseClass], default: false},
+        pretty_print: {type: [TrueClass, FalseClass], default: false}
     }
 
     PUBLIC_ATTRIBUTES.each do |attr, props|
@@ -42,6 +45,7 @@ module Rack
     end
 
     def call(env)
+      env.delete 'HTTP_ACCEPT_ENCODING' if prevent_compression
       log_request! env
       @app.call(env).tap { |response| log_response! env, response }
     end
@@ -105,7 +109,11 @@ module Rack
         if response_bodies
           body = response[2]
           body = ::File.open(body.path, 'rb') { |f| f.read } if body.respond_to? :path
-          body = body.tap(&:rewind).read if body.respond_to? :read
+          if body.respond_to? :read
+            stream = body
+            body = stream.tap(&:rewind).read
+            stream.rewind
+          end
           body = body.join if body.respond_to? :join
           log_body! body,
                     type: headers['Content-Type'],
@@ -115,7 +123,9 @@ module Rack
     end
 
     def log_body!(body, type: nil, encoding: nil)
-      body = "<#BINARY #{body.bytes.length} bytes>" if body =~ /[^[:print:]]/
+      body = Zlib::GzipReader.new(StringIO.new body).read if encoding == 'gzip'
+      body = JSON.pretty_generate(JSON.parse body) if type[/[^;]+/] == 'application/json' && pretty_print
+      body = "<#BINARY #{body.bytes.length} bytes>" if body =~ /[^[:print:]\r\n\t]/
       info body
     end
 

data/lib/rack/traffic_logger/echo.rb

@@ -6,7 +6,17 @@ module Rack
     class Echo
       def call(env)
         begin
-          [200, {'Content-Type' => 'application/json'}, [JSON.parse(env['rack.input'].tap(&:rewind).read)]]
+          body = JSON.parse(env['rack.input'].tap(&:rewind).read).to_json
+          headers = {'Content-Type' => 'application/json'}
+          if env['HTTP_ACCEPT_ENCODING'] =~ /\bgzip\b/
+            zipped = StringIO.new('w')
+            writer = Zlib::GzipWriter.new(zipped)
+            writer.write body
+            writer.close
+            body = zipped.string
+            headers['Content-Encoding'] = 'gzip'
+          end
+          [200, headers, [body]]
         rescue JSON::ParserError => error
           [
               500,

data/lib/rack/traffic_logger/version.rb

@@ -1,5 +1,5 @@
 module Rack
   class TrafficLogger
-    VERSION = '0.1.1'
+    VERSION = '0.1.2'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rack-traffic-logger
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Neil E. Pearson