batasrki / braintree.rb

Created April 04, 2011 — forked from dpickett/braintree.rb

if Rails.env.production?
  Braintree::Configuration.environment = Rails.env.staging? ? :sandbox : :production
  Braintree::Configuration.merchant_id = ENV["braintree_merchant_id"]
  Braintree::Configuration.public_key  = ENV["braintree_public_key"]
  Braintree::Configuration.private_key = ENV["braintree_private_key"]
else
  Braintree::Configuration.environment = :sandbox
  Braintree::Configuration.merchant_id = "<super secret>"
  Braintree::Configuration.public_key  = "<super secret>"
  Braintree::Configuration.private_key = "<super secret>"
end

if Rails.env.test?
  ::Rails.application.config.middleware.use BraintreeTestApp
end
 
class BraintreeTestApp
  def initialize(app)
    @app = app
  end
  
  def call(env)
    @env = env
    config = Braintree::Configuration.instantiate
    if request.path =~ /\/merchants\/#{config.merchant_id}\/transparent_redirect_requests$/ 
      #proxy post to braintree
      uri = URI.parse(config.protocol + "://" + config.server + ":" + 
        config.port.to_s + request.path)
      http = Net::HTTP.new(uri.host, uri.port)
      http.use_ssl = true
      res = http.post(uri.path, request.body.read)
 
      if res.code == "303"
        header_hash = res.header.to_hash
        header_hash["location"].first.gsub!("http://localhost:3000/", "http://www.example.com/")
        [303, {"location" => header_hash["location"].first}, ""]
      else
        raise "unexpected response from Braintree: expected a 303"
      end
    else
      @app.call(env)
    end
  end
 
  def request
    @request = Rack::Request.new(@env)
  end
end

module Rack
  module Utils
    def parse_nested_query(qs, d = nil)
      params = ActiveSupport::OrderedHash.new
 
      (qs || '').split(d ? /[#{d}] */n : DEFAULT_SEP).each do |p|
        k, v = unescape(p).split('=', 2)
        normalize_params(params, k, v)
      end
 
      return params
    end
  end
end
 
module Rack
  module Test
    class Session
      alias_method :old_env_for, :env_for
 
      def env_for(path, env)
        if env[:params] && env[:params].empty?
          env[:params] = ActiveSupport::OrderedHash.new
        end
 
        old_env_for(path, env)
      end
    end
  end
end

add_column :accounts, :payment_processor_subscription_id, :integer

VCR.config do |c|
  c.cassette_library_dir = 'features/cassettes'
  c.stub_with :webmock
  c.default_cassette_options = { :record => :new_episodes }
end
 
VCR.cucumber_tags do |t|
  t.tags '@braintree_valid', 
    '@braintree_invalid', 
    '@braintree_valid_bad_account'
end

VCR.config do |c|
  c.cassette_library_dir = 'features/cassettes'
  c.stub_with :webmock
  braintree_config = Braintree::Configuration.instantiate
  c.default_cassette_options = { :record => :new_episodes, :erb => {
    :braintree_merchant_id => braintree_config.merchant_id,
    :braintree_public_key  => braintree_config.public_key,
    :braintree_private_key => braintree_config.private_key
  } }
end
 
module VcrHelper
  def self.obscure_credentials
    braintree_config = Braintree::Configuration.instantiate
    [
      "spec/cassettes/**/*.yml",
      "features/cassettes/**/*.yml"
    ].each do |glob|
      Dir.glob(Rails.root.join(glob)).each do |f|
        contents = File.read(f)
        File.open(f, "w") do |j|
          j << contents.gsub(CGI.escape(braintree_config.merchant_id), "<%= braintree_merchant_id  %>").
            gsub(CGI.escape(braintree_config.public_key), "<%= braintree_public_key %>").
            gsub(CGI.escape(braintree_config.private_key), "<%= braintree_private_key %>")
        end
      end
    end
  end
end
 
 
at_exit do
  VcrHelper.obscure_credentials
end