crypto_arbitrer 0.0.3 → 0.0.4

data/lib/crypto_arbitrer/version.rb

@@ -1,3 +1,3 @@
 module CryptoArbitrer
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

data/lib/crypto_arbitrer.rb

@@ -5,6 +5,14 @@ require "open-uri"
 
 module CryptoArbitrer
   class Base
+    def self.btc_usd_strategy
+      @btc_usd_strategy || :mtgox
+    end
+
+    def self.btc_usd_strategy=(strategy)
+      @btc_usd_strategy = strategy
+    end
+
     # Quasi constant, all supported fiat currencies.
     def self.supported_fiat
       %w(usd ars uyu brl clp sgd eur vef)
@@ -68,9 +76,13 @@ module CryptoArbitrer
     # The returned hash has 'sell' and 'buy' keys for the different prices,
     # the prices mean at which price you could buy and sell from them, respectively.
     def fetch_btc_usd
-      response = open('http://data.mtgox.com/api/2/BTCUSD/money/ticker_fast').read
-      json = JSON.parse(response)['data']
-      {'sell' => json['sell']['value'].to_f, 'buy' => json['buy']['value'].to_f}
+      if self.class.btc_usd_strategy == :mtgox
+        response = open('http://data.mtgox.com/api/2/BTCUSD/money/ticker_fast').read
+        json = JSON.parse(response)['data']
+        {'sell' => json['sell']['value'].to_f, 'buy' => json['buy']['value'].to_f}
+      else
+        parse_btce_prices open("https://btc-e.com/exchange/btc_usd").read
+      end
     end
     derive_reversal(:btc, :usd)
     
@@ -93,15 +105,18 @@ module CryptoArbitrer
     non_btc_cryptos = supported_cryptos - ['btc']
     non_btc_cryptos.each do |currency|
       define_method("fetch_#{currency}_btc") do
-        response = open("https://btc-e.com/exchange/#{currency}_btc").read
-        response =~ /<span id=.max_price..(.*?)..span./
-        sell = $1.to_f
-        response =~ /<span id=.min_price..(.*?)..span./
-        buy = $1.to_f
-        {'sell' => sell, 'buy' => buy}
+        parse_btce_prices open("https://btc-e.com/exchange/#{currency}_btc").read
       end
       derive_reversal(currency, :btc)
     end
+    
+    def parse_btce_prices(response)
+      response =~ /<span id=.max_price..(.*?)..span./
+      sell = $1.to_f
+      response =~ /<span id=.min_price..(.*?)..span./
+      buy = $1.to_f
+      {'sell' => sell, 'buy' => buy}
+    end
 
     # Fiat prices are fetch from rate-exchange.appspot.com but there are no buy/sell prices.
     # We use USD as the common denominator for all fiat currencies.

data/spec/base_spec.rb

@@ -1,6 +1,7 @@
 require_relative '../lib/crypto_arbitrer.rb'
 require 'webmock/rspec'
 require 'csv'
+require 'ruby-debug'
 RSpec.configure do |config|
   config.include CryptoArbitrer::SpecHelper
 end
@@ -51,6 +52,15 @@ describe CryptoArbitrer::Base do
     subject.fetch_btc_usd
     stub.should have_been_requested
   end
+  
+  it 'hits btce for the btc to usd rate when configured that way' do
+    CryptoArbitrer::Base.btc_usd_strategy = :btce
+    stub = stub_crypto_arbitrer_get("https://btc-e.com/exchange/btc_usd", "btce_btc_usd.html")
+    rate = subject.fetch_btc_usd
+    rate['buy'].should == 99.999
+    rate['sell'].should == 99.502
+    stub.should have_been_requested
+  end
 
   it 'hits dolarparalelo for venezuelan bolivar' do
     stub = stub_crypto_arbitrer_get('http://www.dolarparalelo.org/', 'dolarparalelo.html')