rbtc_arbitrage 1.3.0 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 073f034a32dc76e95d0e5a425a0716e4c97d2208
-  data.tar.gz: fc051d210da396065843debde1122612f09bf331
+  metadata.gz: 35a3e079318e368f33db37bb070cef7acd9dc3d5
+  data.tar.gz: 23327807ba84e9d20b99e170d25b4088c64b7587
 SHA512:
-  metadata.gz: 09aa1b9843ff89cf3254d97801e1d1b2547cc4ae9be908d9abac68f5989250e447cc7b0f129c2540952937d58bb79ba4328acabc35767033f77cb24f53b2f1df
-  data.tar.gz: c90035506f8efeefc460761e1e2d4e61bc2313d4e69f251cfdb1c9d51df5facb1ba1335773d657a3c6b85ef04e315b048a403ccd74bc652f44f85c76a8bc5408
+  metadata.gz: 8dbab32dba09e0ae422ca3ecc5bfe20dba8f5e76de1af1efaa0c8f8a537151d78fd6345472bd737288f623f088834915acffdd7e44f4872cf4dc6aaf54d09fbd
+  data.tar.gz: cb577476f076d7f1285cff313bd80dac03f077b8ec53a973df02aa0605ff071d406c8f8c5988d7bf0dbd817264a8d26105ebe535a2a52f4cfaf43c9b958d9839

data/.travis.yml

@@ -19,3 +19,6 @@ env:
   - secure: IFyJX6lAmBpef0I50UPCy6L5IRxqRKoPhv74s5y0FPB4chrir7iMZ7Sd1IczxV7ly4Dt7hdALHP1qMTmkcGmTV6YOzwR46G4JAwWtX2xlf4OYk4jey7qjNsS6iijD6dvdYjqSKUuGaX3Til3kUOLcA3hhINvHxj5unv2pHB2W70=
   - secure: YiYtCBJNfgWuzQ059RMmZxM00bs4c/GuqaEaKufLPPByJHHt3z4jCNT5lurox/RIqSYTv+1Nvgor1xJZF4VPDIhtoCpVRe3g5Pvubrp4gWQry11ta0f1JhPSrm5ORlF1hpHPfzHHSh+/YVmKcO8agQAaIgfKcXDzsE6j5QqiGWQ=
   - secure: TCrSkWTG2A7ZHxU/J82R1oRn3+ISYHOZmukvBT/OkOh35yjhtig3KhroYki67guCClVxov+ZPvCRUKgmsEoVng1ljk/ZXyyNf4QNVccEJxHCwIRiqs36jlC2ebd8g+JJUPrnnWGO/U5vBm+WlrOb4lFlIsBuCYI+FBAFYTPmKEs=
+  - secure: cePvBAt6j7UwybsP6Cs5a7mwaXMHDteXC1ENsNMy69qV5mae7oRKqsfavZ8ar33apNXqALXDWiDnHH7wPccMl3GXIquHE7I73lAOcEu4V+f7nHqqFuJezA12IL0swOQhDNK23C5ibkFftlpG2jTxHpaRLaDNCuv404cREalwwgA=
+  - secure: PLTV2N3BYlqXmDDuOuZGk83YApX3oEIpskkWHUGj8f3wBpjOSPxuLBdpSGMXRqT2kIMJqle65gqlYpwh4NcTgqL93iNAT9fh7WDyNx/SUL+Mz04kphAC6N0VPyg7nvTi33bDbkmAAFNc+1yoWKAlB2gjKKXFtdxTwMp4SjvNsfs=
+  - secure: nyFUNdswDOM0UgW3T/jeEvIyZPSGq+ZEpyYw6KGpwb7a5F1DbAfoSy6zJGWK6S3vKf6814e43HmawhJlhcD9KYOV783O7KYu54FTmhZ/SP9QhNV+w1hiKKRQcTlJdjbPIuO5piuMDMCPacCJa+C9YO0lhhxfZ+/6F4SNR1fteN0=

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rbtc_arbitrage (1.3.0)
+    rbtc_arbitrage (1.4.0)
       activemodel (>= 3.1)
       activesupport (>= 3.1)
       bitstamp
@@ -9,6 +9,7 @@ PATH
       btce (= 0.2.4)
       coinbase (= 1.2.4)
       mtgox
+      pony
       thor
 
 GEM
@@ -78,6 +79,10 @@ GEM
       rb-fsevent (>= 0.9.3)
       rb-inotify (>= 0.9)
     lumberjack (1.0.4)
+    mail (2.2.7)
+      activesupport (>= 2.3.6)
+      mime-types
+      treetop (>= 1.4.5)
     method_source (0.8.2)
     mime-types (2.0)
     minitest (4.7.5)
@@ -91,6 +96,9 @@ GEM
     multi_json (1.8.2)
     multi_xml (0.5.5)
     multipart-post (1.2.0)
+    polyglot (0.3.3)
+    pony (1.6.2)
+      mail (>= 2.0)
     pry (0.9.12.4)
       coderay (~> 1.0)
       method_source (~> 0.8)
@@ -137,6 +145,9 @@ GEM
     tilt (1.4.1)
     timers (1.1.0)
     tins (0.13.1)
+    treetop (1.4.15)
+      polyglot
+      polyglot (>= 0.3.1)
     tzinfo (0.3.38)
     vcr (2.8.0)
     webmock (1.16.1)

data/bin/rbtc

@@ -1,4 +1,5 @@
 #!/usr/bin/env ruby
 
+$LOAD_PATH.unshift File.join(File.dirname(__FILE__), '..', 'lib')
 require 'rbtc_arbitrage'
 RbtcArbitrage::CLI.start ARGV

data/lib/rbtc_arbitrage/cli.rb

@@ -9,6 +9,7 @@ module RbtcArbitrage
     option :buyer, type: :string, default: "bitstamp"
     option :seller, type: :string, default: "mtgox"
     option :repeat, type: :numeric, default: nil
+    option :notify, type: :boolean, default: false
     def trade
       RbtcArbitrage::Trader.new(options).trade
     end

data/lib/rbtc_arbitrage/trader.rb

@@ -17,6 +17,7 @@ module RbtcArbitrage
       set_key opts, :verbose, true
       set_key opts, :live, false
       set_key opts, :repeat, nil
+      set_key opts, :notify, false
       exchange = opts[:buyer] || :bitstamp
       @buy_client = client_for_exchange(exchange)
       exchange = opts[:seller] || :mtgox
@@ -38,9 +39,12 @@ module RbtcArbitrage
 
       execute_trade if options[:live]
 
+      notify
+
       if @options[:repeat]
         trade_again
       end
+
       self
     end
 
@@ -104,6 +108,15 @@ module RbtcArbitrage
       [@sell_client, @buy_client].each do |client|
         client.validate_env
       end
+      if options[:notify]
+        ["PASSWORD","USERNAME","EMAIL"].each do |key|
+          key = "SENDGRID_#{key}"
+          unless ENV[key]
+            raise ArgumentError, "Exiting because missing required ENV variable $#{key}."
+          end
+        end
+        setup_pony
+      end
     end
 
     def client_for_exchange market
@@ -119,5 +132,49 @@ module RbtcArbitrage
         raise ArgumentError, "Invalid exchange - '#{market}'"
       end
     end
+
+    def notify
+      return false unless options[:notify]
+      return false unless @percent > options[:cutoff]
+      setup_pony
+
+      options[:logger].info "Sending email to #{ENV['SENDGRID_EMAIL']}"
+      Pony.mail({
+        body: notification,
+        to: ENV['SENDGRID_EMAIL'],
+      })
+    end
+
+    def setup_pony
+      Pony.options = {
+        from: "info@uludum.org",
+        subject: "rbtc_arbitrage notification",
+        via: :smtp,
+        via_options: {
+          address: 'smtp.sendgrid.net',
+          port: '587',
+          domain: 'heroku.com',
+          user_name: ENV['SENDGRID_USERNAME'],
+          password: ENV['SENDGRID_PASSWORD'],
+          authentication: :plain,
+          enable_starttls_auto: true
+        }
+      }
+    end
+
+    def notification
+      lower_ex = @buy_client.exchange.to_s.capitalize
+      higher_ex = @sell_client.exchange.to_s.capitalize
+      <<-eos
+      Update from your friendly rbtc_arbitrage trader:
+      #{lower_ex}: $#{buyer[:price].round(2)}
+      #{higher_ex}: $#{seller[:price].round(2)}
+      buying #{@options[:volume]} btc from #{lower_ex} for $#{@paid.round(2)}
+      selling #{@options[:volume]} btc on #{higher_ex} for $#{@received.round(2)}
+      profit: $#{(@received - @paid).round(2)} (#{@percent.round(2)}%)
+
+      options
+      eos
+    end
   end
 end

data/lib/rbtc_arbitrage/version.rb

@@ -1,3 +1,3 @@
 module RbtcArbitrage
-  VERSION = "1.3.0"
+  VERSION = "1.4.0"
 end

data/lib/rbtc_arbitrage.rb

@@ -4,6 +4,7 @@ require_relative 'rbtc_arbitrage/campbx.rb'
 require 'bitstamp'
 require 'btce'
 require 'coinbase'
+require 'pony'
 require_relative 'rbtc_arbitrage/client.rb'
 Dir["#{File.dirname(__FILE__)}/rbtc_arbitrage/**/*.rb"].each { |f| require(f) }
 

data/rbtc_arbitrage.gemspec

@@ -29,4 +29,5 @@ Gem::Specification.new do |spec|
   spec.add_dependency "btce", '0.2.4'
   spec.add_dependency "bitstamp"
   spec.add_dependency "coinbase", '1.2.4'
+  spec.add_dependency "pony"
 end

data/spec/trader_spec.rb

@@ -37,6 +37,20 @@ describe RbtcArbitrage::Trader do
         end
       end
     end
+
+    ["PASSWORD","USERNAME","EMAIL"].each do |key|
+      key = "SENDGRID_#{key}"
+      it "should raise if --notify and not ENV[#{key}]" do
+        trader.sell_client.stub(:validate_env)
+        trader.buy_client.stub(:validate_env)
+        old_val = ENV[key]
+        ENV[key] = nil
+        trader.options[:notify] = true
+        expect { trader.validate_env }.to raise_error(ArgumentError)
+        trader.options[:notify] = false
+        expect { trader.validate_env }.not_to raise_error
+      end
+    end
   end
 
   describe "#fetch_prices" do
@@ -71,6 +85,7 @@ describe RbtcArbitrage::Trader do
         live: true,
         seller: :bitstamp,
         buyer: :mtgox,
+        notify: true,
       }
     }
 
@@ -142,6 +157,7 @@ describe RbtcArbitrage::Trader do
   describe "#trade" do
     it "calls the right methods" do
       trader.should_receive(:fetch_prices)
+      trader.should_receive(:notify)
       trader.should_not_receive(:log_info)
       trader.should_not_receive(:execute_trade)
 
@@ -225,4 +241,79 @@ describe RbtcArbitrage::Trader do
     end
   end
 
+  describe "#notify" do
+    it "returns false when notify == false" do
+      trader.notify.should == false
+    end
+
+    it "returns false unless cutoff < percent" do
+      trader.options[:notify] = true
+      trader.instance_variable_set :@percent, 1
+      trader.notify.should == false
+    end
+
+    it "calls Sendgrid when notify == true" do
+      trader.options[:notify] = true
+      trader.instance_variable_set :@percent, 3
+      trader.instance_variable_set :@paid, 5
+      trader.instance_variable_set :@received, 6
+      trader.buyer[:price] = 1
+      trader.seller[:price] = 1
+
+      trader.options[:logger].should_receive(:info)
+      Pony.should_receive(:mail).with({
+        to: ENV['SENDGRID_EMAIL'],
+        body: trader.notification,
+      })
+
+      trader.notify
+    end
+  end
+
+  describe "#setup_pony" do
+    it "gets called on validation when --notify" do
+      ["PASSWORD","USERNAME","EMAIL"].each do |key|
+        key = "SENDGRID_#{key}"
+        ENV[key] ||= "something"
+      end
+      trader.options[:notify] = true
+      trader.should_receive(:setup_pony).once
+      trader.validate_env
+    end
+    it "sets up pony correctly" do
+      opts = {
+        from: "info@uludum.org",
+        subject: "rbtc_arbitrage notification",
+        :via => :smtp,
+        :via_options => {
+          :address => 'smtp.sendgrid.net',
+          :port => '587',
+          :domain => 'heroku.com',
+          :user_name => ENV['SENDGRID_USERNAME'],
+          :password => ENV['SENDGRID_PASSWORD'],
+          :authentication => :plain,
+          :enable_starttls_auto => true
+        }
+      }
+      Pony.should_receive(:options=).with(opts)
+      trader.setup_pony
+    end
+  end
+
+  describe "#notification" do
+    it "includes the right values" do
+      trader.instance_variable_set :@percent, 3
+      trader.instance_variable_set :@paid, 5
+      trader.instance_variable_set :@received, 8
+      trader.buyer[:price] = 1.00453
+      trader.seller[:price] = 3.50453
+
+      value = trader.notification
+      value.should include("$1.0")
+      value.should include("$3.5")
+      value.should include("$3.0")
+      value.should include("3.0%")
+    end
+  end
+
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbtc_arbitrage
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: ruby
 authors:
 - Hank Stoever
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-12-23 00:00:00.000000000 Z
+date: 2014-01-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -150,6 +150,20 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 1.2.4
+- !ruby/object:Gem::Dependency
+  name: pony
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A gem for conducting arbitrage with Bitcoin.
 email:
 - hstove@gmail.com