rack-request_replication 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 7c53f6dc67d9b644ff865a0e0e5db200e19756ad
-  data.tar.gz: 77e9cd22aeab694556c8d2195e766d56c9df25f9
+  metadata.gz: 9f86b1abca6aee060bc978f2d9d03cc0241ff51e
+  data.tar.gz: 8f5e982046204fc40f2c4369805b5b073795c679
 SHA512:
-  metadata.gz: 6c9e718ca2a6a0688088766e7d7a96e550f32a521e6991fcf5d37937ca905a281f3035c1e6f1532d26eebfd523ba55315ea0b464711f356a64e2e7af1fa46a68
-  data.tar.gz: afd0d4ffa1b81cf3cd5b2cdeb39af9a3e7ee70009f2335c4686c5bc0c33474266442b2d1b5a5bdc5d709d36128a99ae1c7cf46d3a682b8ea2b35b8e98eb1d052
+  metadata.gz: b67f6d7ecbf05611b54da1c862ce67fdd6e3ea58e53c01867bc0eb6f62d71aa845b2d23b762ac81aa5ad17a8033778f232eef6b264a4ad4854935c173f373a75
+  data.tar.gz: cfe08f90f63f11b7232330c81c2ba43faea3f5bc9c13ea42cb9a5ba5c8fc8f0fab060b4d4eb37f8ac7126528edd0d0390530e8ebf68c540058c8089d5bde189e

data/lib/rack/request_replication/forwarder.rb

@@ -2,6 +2,7 @@ require 'logger'
 require 'json'
 require 'net/http'
 require 'uri'
+require 'redis'
 
 module Rack
   module RequestReplication
@@ -18,12 +19,16 @@ module Rack
       # @param  [Hash{Symbol => Object}] options
       # @option options [String]  :host  ('localhost')
       # @option options [Integer] :port  (8080)
+      # @option options [String]  :session_key ('rack.session')
       #
       def initialize( app, options = {} )
         @app = app
         @options = {
           host: 'localhost',
-          port: 8080
+          port: 8080,
+          session_key: 'rack.session',
+          root_url: '/',
+          redis: {}
         }.merge options
       end
 
@@ -54,16 +59,38 @@ module Rack
         forward_request.add_field("Accept", opts[:accept])
         forward_request.add_field("Accept-Encoding", opts[:accept_encoding])
         forward_request.add_field("Host", request.host)
-        forward_request.add_field("Cookie", opts[:cookies]) # TODO: we need to link the source session to the session on the destination app. Maybe we can use Redis to store this.
+
         Thread.new do
           begin
-            http.request(forward_request)
-          rescue
-            logger.debug "Request to Forward App failed."
+            forward_request.add_field("Cookie", cookies( request ))
+            update_cookies( request, http.request(forward_request) )
+          rescue => e
+            logger.debug "Replicating request failed with: #{e.message}"
           end
         end
       end
 
+      def update_cookies( request, response )
+        return unless cookies_id( request )
+        cookie = response.to_hash['set-cookie'].collect{|ea|ea[/^.*?;/]}.join rescue {}
+        cookie = Hash[cookie.split(";").map{|d|d.split('=')}] rescue {}
+        redis.set( cookies_id( request), cookie )
+      end
+
+      def cookies( request )
+        return ( request.cookies || "" ) unless cookies_id( request )
+        cs = redis.get( cookies_id( request )) ||
+          request.cookies ||
+          {}
+      end
+
+      def cookies_id( request )
+        cs = request.cookies
+        sess = cs && cs[options[:session_key]]
+        sess_id = sess && sess.split("\n--").last
+        sess_id
+      end
+
       def create_get_request( uri, opts = {} )
         Net::HTTP::Get.new uri.request_uri
       end
@@ -108,7 +135,6 @@ module Rack
           body
           request_method
           content_charset
-          cookies
           media_type
           media_type_params
           params
@@ -131,12 +157,28 @@ module Rack
       # @returns [URI]
       #
       def forward_uri( request )
-        url = "#{request.scheme}://#{options[:host]}"
-        url << ":#{options[:port]}" unless port_matches_scheme? request
+        url = "#{request.scheme}://#{forward_host_with_port( request )}"
         url << request.fullpath
         URI(url)
       end
 
+      def forward_host_with_port( request )
+        host = options[:host].to_s
+        host = "#{host}:#{options[:port]}" unless port_matches_scheme? request
+        host
+      end
+
+      ##
+      # Persistent Redis connection that is used
+      # to store cookies.
+      #
+      def redis
+        @redis ||= Redis.new({
+          host: 'localhost',
+          port: 6379,
+          db: 'rack-request-replication'
+        }.merge(options[:redis]))
+      end
       ##
       # Checks if the request scheme matches the destination port.
       #

data/lib/rack/request_replication/version.rb

@@ -1,5 +1,5 @@
 module Rack
   module RequestReplication
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/rack-request_replication.gemspec

@@ -24,6 +24,8 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency 'rspec',     '~> 3.0'
   spec.add_development_dependency 'yard' ,     '>= 0.5.5'
   spec.add_development_dependency 'sinatra'
+  spec.add_development_dependency 'sinatra-contrib'
 
   spec.add_runtime_dependency 'rack', '>= 1.0.0'
+  spec.add_runtime_dependency 'redis', '>= 1.0.0'
 end

data/spec/app/test_app.rb

@@ -1,4 +1,5 @@
 require 'sinatra/base'
+require 'sinatra/cookies'
 require 'rack/request_replication'
 
 $destination_responses ||= []
@@ -36,10 +37,16 @@ class TestApp < Sinatra::Base
 end
 
 class DestApp < Sinatra::Base
+  helpers Sinatra::Cookies
+
   set :port, 4568
 
   enable :logging
 
+  before do
+    cookies.merge! 'boo' => 'far', 'zar' => 'bab'
+  end
+
   get '/' do
     $destination_responses << 'GET OK'
     'Hello, World!'

data/spec/spec_helper.rb

@@ -15,7 +15,6 @@ RSpec.configure do |config|
 
   config.after :suite do
     puts "Quitting example applications..."
-    #@spid.kill
     @dpid.kill
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rack-request_replication
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Wouter de Vos
@@ -94,6 +94,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: sinatra-contrib
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rack
   requirement: !ruby/object:Gem::Requirement
@@ -108,6 +122,20 @@ dependencies:
     - - '>='
       - !ruby/object:Gem::Version
         version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.0
 description: Replicate or record HTTP requests to your Rack application and replay
   them elsewhere or at another time.
 email: