request_visualizer 0.0.2

data/Gemfile

@@ -0,0 +1,10 @@
+source 'http://rubygems.org'
+
+gemspec
+
+group :test do
+  gem 'ey_api_hmac', '>= 0.4.0'
+  gem "sinatra"
+  gem 'rspec'
+  gem 'capybara'
+end

data/Gemfile.lock

@@ -0,0 +1,68 @@
+PATH
+  remote: .
+  specs:
+    request_visualizer (0.0.1)
+      colored
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    capybara (1.1.2)
+      mime-types (>= 1.16)
+      nokogiri (>= 1.3.3)
+      rack (>= 1.0.0)
+      rack-test (>= 0.5.4)
+      selenium-webdriver (~> 2.0)
+      xpath (~> 0.1.4)
+    childprocess (0.3.1)
+      ffi (~> 1.0.6)
+    colored (1.2)
+    diff-lcs (1.1.3)
+    ey_api_hmac (0.4.4)
+      json
+      rack-client
+      rack-idempotent (>= 0.0.3)
+    ffi (1.0.11)
+    json (1.6.5)
+    mime-types (1.17.2)
+    multi_json (1.1.0)
+    nokogiri (1.5.2)
+    rack (1.4.1)
+    rack-client (0.4.0)
+      rack (>= 1.0.0)
+    rack-idempotent (0.0.3)
+    rack-protection (1.2.0)
+      rack
+    rack-test (0.6.1)
+      rack (>= 1.0)
+    rspec (2.8.0)
+      rspec-core (~> 2.8.0)
+      rspec-expectations (~> 2.8.0)
+      rspec-mocks (~> 2.8.0)
+    rspec-core (2.8.0)
+    rspec-expectations (2.8.0)
+      diff-lcs (~> 1.1.2)
+    rspec-mocks (2.8.0)
+    rubyzip (0.9.6.1)
+    selenium-webdriver (2.20.0)
+      childprocess (>= 0.2.5)
+      ffi (~> 1.0)
+      multi_json (~> 1.0)
+      rubyzip
+    sinatra (1.3.2)
+      rack (~> 1.3, >= 1.3.6)
+      rack-protection (~> 1.2)
+      tilt (~> 1.3, >= 1.3.3)
+    tilt (1.3.3)
+    xpath (0.1.4)
+      nokogiri (~> 1.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  capybara
+  ey_api_hmac (>= 0.4.0)
+  request_visualizer!
+  rspec
+  sinatra

data/lib/request_visualizer.rb

@@ -0,0 +1,70 @@
+require 'colored'
+class RequestVisualizer
+  def initialize(app, &lookup)
+    @app = app
+    @lookup = lookup
+  end
+
+  def parse(string)
+    colorize((@lookup && @lookup.call(string.to_s)) || string.to_s)
+  end
+
+  def colorize(string)
+    @@colors ||= [:black_on_cyan, :black_on_magenta, :black_on_yellow, :black_on_green]
+    @@colorized_before ||= {}
+    @@colorized_before[string] ||= string.send(@@colors.pop || :black_on_white)
+    @@colorized_before[string]
+  end
+
+  def indent
+    " "*@@indent
+  end
+
+  def log_request(request)
+    from = parse(request.user_agent.to_s)
+    to = parse(request.url.to_s)
+    puts "#{self.indent}#{from} -> #{request.request_method.upcase.bold} (#{request.url.underline}) -> #{to}"
+    request.body.rewind
+    req_body = request.body.read
+    begin
+      json = JSON.parse(req_body)
+      indent_prefix = self.indent
+      puts indent_prefix + json.pretty_inspect.gsub("\n","\n#{indent_prefix}")
+    rescue => e
+      # puts "WARN: JSON request with non-json body! (#{req_body})"
+    end
+    request.body.rewind
+    [from, to]
+  end
+
+  def log_response(from, to, headers, body, status)
+    location = headers["Location"]
+    if location
+      puts "#{self.indent}#{from} <--#{status}-- #{location.underline} <- #{to}"
+    else
+      puts "#{self.indent}#{from} <--#{status}-- #{to}"
+    end
+    do_inspect = true
+    if headers["Content-Type"].to_s.match(/json/) && do_inspect
+      body.each do |bod|
+        begin
+          json = JSON.parse(bod)
+          indent_prefix = self.indent
+          puts indent_prefix + json.pretty_inspect.gsub("\n","\n#{indent_prefix}")
+        rescue => e
+          puts "WARN: JSON response with non-json body! (#{bod})"
+        end
+      end
+    end
+  end
+
+  def call(env)
+    @@indent ||= 0
+    @@indent += 5
+    from, to = log_request(Rack::Request.new(env))
+    status, headers, body = @app.call(env)
+    log_response(from, to, headers, body, status)
+    @@indent -= 5
+    [status, headers, body]
+  end
+end

data/lib/request_visualizer/version.rb

@@ -0,0 +1,3 @@
+class RequestVisualizer
+  VERSION = "0.0.2"
+end

data/request_visualizer.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "request_visualizer/version"
+
+Gem::Specification.new do |s|
+  s.name        = "request_visualizer"
+  s.version     = RequestVisualizer::VERSION
+  s.authors     = ["Jacob Burkhart"]
+  s.email       = ["jacob@engineyard.com"]
+  s.homepage    = ""
+  s.summary     = %q{Rack middleware for visualizing HTTP/JSON requests}
+  s.description = %q{Rack middleware for visualizing HTTP/JSON requests, and stuff}
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency 'colored'
+end

data/spec/sample.rb

@@ -0,0 +1,93 @@
+require 'ey_api_hmac'
+class Connection < EY::ApiHMAC::BaseConnection
+
+  def initialize(user_agent, app)
+    super(user_agent)
+    self.backend = app
+  end
+
+  def get(url)
+    super(url) do |json_body, location|
+      json_body
+    end
+  end
+
+  def post(url, params)
+    super(url, params) do |json_body, location|
+      json_body
+    end
+  end
+
+end
+
+require 'sinatra/base'
+class Service < Sinatra::Base
+  enable :raise_errors
+  disable :dump_errors
+  disable :show_exceptions
+
+  def conn
+    @conn ||= Connection.new(self.class.to_s, Capybara.app)
+  end
+end
+
+class Lookup
+  def self.services
+    @services ||= {}
+  end
+
+  def self.lookup(service_url)
+    services.each do |base_url, service|
+      if service_url.match base_url
+        return service.to_s
+      end
+    end
+    service_url
+  end
+
+end
+
+class SandwhichMaker < Service
+  BASE_URL = "https://sandwhichmaker.example.com/"
+  Lookup.services[BASE_URL] = self
+
+  get "/" do
+    bread = conn.get("#{Baker::BASE_URL}get_loaf")
+    conn.post("#{BreadSlicer::BASE_URL}slice", {:bread_url => bread["url"]})
+    content_type "application/json"
+    {:sandwhich => true}.to_json
+  end
+
+end
+
+class BreadSlicer < Service
+  BASE_URL = "https://breadslicer.example.com/"
+  Lookup.services[BASE_URL] = self
+
+  post "/slice" do
+    post_body = JSON.parse(request.body.read)
+    bread = conn.get(post_body["bread_url"])
+    content_type "application/json"
+    {:slices => ["heel","slice","slice","slice","heel"]}.to_json
+  end
+
+end
+
+class Baker < Service
+  BASE_URL = "https://bakery.example.com/"
+  Lookup.services[BASE_URL] = self
+
+  get "/loaves/1" do
+    content_type "application/json"
+    {:type => "sourdough"}.to_json
+  end
+
+  get "/get_loaf" do
+    content_type "application/json"
+    {:url => "#{BASE_URL}loaves/1"}.to_json
+  end
+
+end
+
+
+

data/spec/spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path("../sample", __FILE__)
+require 'request_visualizer'
+require 'capybara/rspec'
+
+RSpec.configure do |config|
+  config.include(Capybara::RSpecMatchers)
+  config.include(Capybara::DSL)
+end
+
+describe "things" do
+  before do
+    Capybara.app = Rack::Builder.new do
+      use RequestVisualizer do |string|
+        Lookup.lookup(string.to_s)
+      end
+      Lookup.services.each do |url, service|
+        map url do
+          run service
+        end
+      end
+    end
+    page.driver.header 'User-Agent', "Capybara"
+  end
+
+  it "makes a sandwhich" do
+    visit SandwhichMaker::BASE_URL
+  end
+
+end

metadata

@@ -0,0 +1,87 @@
+--- !ruby/object:Gem::Specification 
+name: request_visualizer
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 2
+  version: 0.0.2
+platform: ruby
+authors: 
+- Jacob Burkhart
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-03-16 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: colored
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Rack middleware for visualizing HTTP/JSON requests, and stuff
+email: 
+- jacob@engineyard.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- Gemfile
+- Gemfile.lock
+- README
+- lib/request_visualizer.rb
+- lib/request_visualizer/version.rb
+- request_visualizer.gemspec
+- spec/sample.rb
+- spec/spec.rb
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Rack middleware for visualizing HTTP/JSON requests
+test_files: 
+- spec/sample.rb
+- spec/spec.rb