bitstamp-rbtc-arbitrage 0.4.0

data/lib/bitstamp.rb

@@ -0,0 +1,77 @@
+require 'active_support/core_ext'
+require 'active_support/inflector'
+require 'active_model'
+require 'curb'
+require 'hmac-sha2'
+
+require 'bitstamp/net'
+require 'bitstamp/helper'
+require 'bitstamp/collection'
+require 'bitstamp/model'
+
+require 'bitstamp/orders'
+require 'bitstamp/transactions'
+require 'bitstamp/ticker'
+
+String.send(:include, ActiveSupport::Inflector)
+
+module Bitstamp
+  # API Key
+  mattr_accessor :key
+
+  # Bitstamp secret
+  mattr_accessor :secret
+  
+  # Bitstamp client ID
+  mattr_accessor :client_id
+
+  # Currency
+  mattr_accessor :currency
+  @@currency = :usd
+
+  def self.orders
+    self.sanity_check!
+
+    @@orders ||= Bitstamp::Orders.new
+  end
+
+  def self.user_transactions
+    self.sanity_check!
+
+    @@transactions ||= Bitstamp::UserTransactions.new
+  end
+
+  def self.transactions
+    return Bitstamp::Transactions.from_api
+  end
+
+  def self.balance
+    self.sanity_check!
+
+    JSON.parse Bitstamp::Net.post('/balance').body_str
+  end
+
+  def self.ticker
+    return Bitstamp::Ticker.from_api
+  end
+
+  def self.order_book
+    return JSON.parse Bitstamp::Net.get('/order_book/').body_str
+  end
+
+  def self.setup
+    yield self
+  end
+  
+  def self.configured?
+    self.key && self.secret && self.client_id
+  end
+
+  def self.sanity_check!
+    unless configured?
+      raise MissingConfigExeception.new("Bitstamp Gem not properly configured")
+    end
+  end
+
+  class MissingConfigExeception<Exception;end;
+end

data/lib/bitstamp/collection.rb

@@ -0,0 +1,30 @@
+module Bitstamp
+  class Collection
+    attr_accessor :access_token, :module, :name, :model, :path
+
+    def initialize(api_prefix="/api")
+      self.access_token = Bitstamp.key
+
+      self.module = self.class.to_s.singularize.underscore
+      self.name   = self.module.split('/').last
+      self.model  = self.module.camelize.constantize
+      self.path   = "#{api_prefix}/#{self.name.pluralize}"
+    end
+
+    def all(options = {})
+      Bitstamp::Helper.parse_objects! Bitstamp::Net::get(self.path).body_str, self.model
+    end
+
+    def create(options = {})
+      Bitstamp::Helper.parse_object! Bitstamp::Net::post(self.path, options).body_str, self.model
+    end
+
+    def find(id, options = {})
+      Bitstamp::Helper.parse_object! Bitstamp::Net::get("#{self.path}/#{id}").body_str, self.model
+    end
+
+    def update(id, options = {})
+      Bitstamp::Helper.parse_object! Bitstamp::Net::patch("#{self.path}/#{id}", options).body_str, self.model
+    end
+  end
+end

data/lib/bitstamp/helper.rb

@@ -0,0 +1,19 @@
+module Bitstamp
+  module Helper
+    def self.parse_objects!(string, klass)
+      # If Bitstamp returned nothing (which it does if the results yield empty) 'cast' it to an array
+      string = "[]" if string == ""
+
+      objects = JSON.parse(string)
+      objects.collect do |t_json|
+        parse_object!(t_json, klass)
+      end
+    end
+
+    def self.parse_object!(object, klass)
+      object = JSON.parse(object) if object.is_a? String
+
+      klass.new(object)
+    end
+  end
+end

data/lib/bitstamp/model.rb

@@ -0,0 +1,33 @@
+module Bitstamp
+  class Model
+    attr_accessor :error, :message
+
+    if ActiveModel::VERSION::MAJOR <= 3
+      include ActiveModel::Validations
+      include ActiveModel::Conversion
+      extend ActiveModel::Naming
+
+      def initialize(attributes = {})
+        self.attributes = attributes
+      end
+    else
+      include ActiveModel::Model
+    end
+
+    # Set the attributes based on the given hash
+    def attributes=(attributes = {})
+      attributes.each do |name, value|
+        begin
+          send("#{name}=", value)
+        rescue NoMethodError => e
+          puts "Unable to assign #{name}. No such method."
+        end
+      end
+    end
+
+    # Returns a hash with the current instance variables
+    def attributes
+      Hash[instance_variables.map { |name| [name, instance_variable_get(name)] }]
+    end
+  end
+end

data/lib/bitstamp/net.rb

@@ -0,0 +1,49 @@
+module Bitstamp
+  module Net
+    def self.to_uri(path)
+      return "https://www.bitstamp.net/api#{path}/"
+    end
+
+    def self.curl(verb, path, options={})
+      verb = verb.upcase.to_sym
+
+      c = Curl::Easy.new(self.to_uri(path))
+
+      if Bitstamp.configured?
+        options[:key] = Bitstamp.key
+        options[:nonce] = (Time.now.to_f*10000).to_i.to_s
+        options[:signature] = HMAC::SHA256.hexdigest(Bitstamp.secret, options[:nonce]+Bitstamp.client_id.to_s+options[:key]).upcase
+      end
+
+      c.post_body = options.to_query
+
+      c.http(verb)
+
+      return c
+    end
+
+    def self.get(path, options={})
+      request = self.curl(:GET, path, options)
+
+      return request
+    end
+
+    def self.post(path, options={})
+      request = self.curl(:POST, path, options)
+
+      return request
+    end
+
+    def self.patch(path, options={})
+      request = self.curl(:PATCH, path, options)
+
+      return request
+    end
+
+    def self.delete(path, options={})
+      request = self.curl(:DELETE, path, options)
+
+      return request
+    end
+  end
+end

data/lib/bitstamp/orders.rb

@@ -0,0 +1,41 @@
+module Bitstamp
+  class Orders < Bitstamp::Collection
+    def all(options = {})
+      Bitstamp::Helper.parse_objects! Bitstamp::Net::post('/open_orders').body_str, self.model
+    end
+
+    def create(options = {})
+      path = (options[:type] == Bitstamp::Order::SELL ? "/sell" : "/buy")
+      Bitstamp::Helper.parse_object! Bitstamp::Net::post(path, options).body_str, self.model
+    end
+
+    def sell(options = {})
+      options.merge!({type: Bitstamp::Order::SELL})
+      self.create options
+    end
+
+    def buy(options = {})
+      options.merge!({type: Bitstamp::Order::BUY})
+      self.create options
+    end
+
+    def find(order_id)
+      all = self.all
+      index = all.index {|order| order.id.to_i == order_id}
+
+      return all[index] if index
+    end
+  end
+
+  class Order < Bitstamp::Model
+    BUY  = 0
+    SELL = 1
+
+    attr_accessor :type, :amount, :price, :id, :datetime
+    attr_accessor :error, :message
+
+    def cancel!
+      Bitstamp::Net::post('/cancel_order', {id: self.id}).body_str
+    end
+  end
+end

data/lib/bitstamp/ticker.rb

@@ -0,0 +1,16 @@
+module Bitstamp
+  class Ticker < Bitstamp::Model
+    attr_accessor :last, :high, :low, :volume, :bid, :ask, :timestamp, :vwap
+
+    def self.from_api
+      Bitstamp::Helper.parse_object!(Bitstamp::Net.get('/ticker').body_str, self)
+    end
+
+    def self.method_missing method, *args
+      ticker = self.from_api
+      return ticker.send(method) if ticker.respond_to? method
+
+      super
+    end
+  end
+end

data/lib/bitstamp/transactions.rb

@@ -0,0 +1,37 @@
+module Bitstamp
+  class UserTransactions < Bitstamp::Collection
+    def all(options = {})
+      # Default time delta to an hour
+      options[:timedelta] = "3600" unless options[:timedelta]
+
+      Bitstamp::Helper.parse_objects! Bitstamp::Net::post("/user_transactions", options).body_str, self.model
+    end
+
+    def find(order_id)
+      all = self.all
+      index = all.index {|order| order.id.to_i == order_id}
+
+      return all[index] if index
+    end
+
+    def create(options = {})
+    end
+
+    def update(options={})
+    end
+  end
+
+  class UserTransaction < Bitstamp::Model
+    attr_accessor :datetime, :id, :type, :usd, :btc, :fee, :order_id, :btc_usd, :nonce
+  end
+
+  # adding in methods to pull the last public trades list
+  class Transactions < Bitstamp::Model
+    attr_accessor :date, :price, :tid, :amount
+
+    def self.from_api
+      Bitstamp::Helper.parse_objects! Bitstamp::Net::get("/transactions").body_str, self
+    end
+
+  end
+end

data/spec/bitstamp_spec.rb

@@ -0,0 +1,57 @@
+require 'spec_helper'
+
+describe Bitstamp do
+
+  describe :sanity_check! do
+    context 'not properly configured' do
+      it { -> { Bitstamp.sanity_check! }.should raise_error }
+    end
+    context 'properly configured' do
+      before {
+        Bitstamp.setup do |config|
+          config.key = 'test'
+          config.secret = 'test'
+          config.client_id = 'test'
+        end
+      }
+      it { -> { Bitstamp.sanity_check! }.should_not raise_error }
+    end
+  end
+
+  describe :orders do
+    it { should respond_to :orders }
+  end
+
+  describe :ticket, vcr: {cassette_name: 'bitstamp/ticker'} do
+    subject { Bitstamp.ticker }
+    it { should be_kind_of Bitstamp::Ticker }
+    its(:high) { should == "124.90" }
+    its(:last) { should == "124.55" }
+    its(:timestamp) { should == "1380237724" }
+    its(:bid) { should == "124.55" }
+    its(:volume) { should == "7766.46908740" }
+    its(:low) { should == "123.00" }
+    its(:ask) { should == "124.56" }
+  end
+
+  describe :balance, vcr: {cassette_name: 'bitstamp/balance'} do
+    context "configured" do
+      subject { Bitstamp.balance }
+      before { setup_bitstamp }
+      it { should == {"btc_reserved"=>"0", "fee"=>"0.4000", "btc_available"=>"0", "usd_reserved"=>"1.02", "btc_balance"=>"0", "usd_balance"=>"6953.07", "usd_available"=>"6952.05"} }
+    end
+    context "not configured" do
+      it { expect { Bitstamp.balance }.to raise_exception(Bitstamp::MissingConfigExeception, "Bitstamp Gem not properly configured") }
+    end
+  end
+
+  describe :order_book, vcr: {cassette_name: 'bitstamp/order_book'} do
+    let(:order_book) { Bitstamp.order_book }
+    subject { order_book }
+    it { should be_kind_of Hash }
+    it { should have_key("asks") }
+    it { should have_key("bids") }
+    it { order_book["asks"].should be_kind_of Array }
+    it { order_book["bids"].should be_kind_of Array }
+  end
+end

data/spec/collection_spec.rb

@@ -0,0 +1,12 @@
+require 'spec_helper'
+
+class Bitstamp::Coin < Bitstamp::Model;end
+class Bitstamp::Coins < Bitstamp::Collection;end
+
+describe Bitstamp::Coins do
+  subject { Bitstamp::Coins.new }
+  its(:name) { should eq 'coin' }
+  its(:module) { should eq "bitstamp/coin" }
+  its(:model) { should be Bitstamp::Coin }
+  its(:path) { should eq "/api/coins" }
+end

data/spec/fixtures/vcr_cassettes/bitstamp/balance.yml

@@ -0,0 +1,63 @@
+---
+http_interactions:
+- request:
+    method: post
+    uri: https://www.bitstamp.net/api/balance/
+    body:
+      encoding: US-ASCII
+      string: key=BITSTAMP_KEY&nonce=1380237671&signature=67768DD1BBDF45203840CC8688847A2DF5D224CA15B3ABDDC804EA251AC12211
+    headers: {}
+  response:
+    status:
+      code: 200
+      message: !binary |-
+        T0s=
+    headers:
+      !binary "U2VydmVy":
+      - !binary |-
+        Y2xvdWRmbGFyZS1uZ2lueA==
+      !binary "RGF0ZQ==":
+      - !binary |-
+        VGh1LCAyNiBTZXAgMjAxMyAyMzoyMToxOCBHTVQ=
+      !binary "Q29udGVudC1UeXBl":
+      - !binary |-
+        YXBwbGljYXRpb24vanNvbg==
+      !binary "VHJhbnNmZXItRW5jb2Rpbmc=":
+      - !binary |-
+        Y2h1bmtlZA==
+      !binary "Q29ubmVjdGlvbg==":
+      - !binary |-
+        a2VlcC1hbGl2ZQ==
+      !binary "U2V0LUNvb2tpZQ==":
+      - !binary |-
+        X19jZmR1aWQ9ZDg2ZTkzYTIxNzY5Y2UxZDQzODc2YzQ1N2ZjNTU3ZmNjMTM4
+        MDIzNzY3Nzk1ODsgZXhwaXJlcz1Nb24sIDIzLURlYy0yMDE5IDIzOjUwOjAw
+        IEdNVDsgcGF0aD0vOyBkb21haW49LmJpdHN0YW1wLm5ldA==
+      !binary "Q29udGVudC1MYW5ndWFnZQ==":
+      - !binary |-
+        ZW4=
+      !binary "RXhwaXJlcw==":
+      - !binary |-
+        VGh1LCAyNiBTZXAgMjAxMyAyMzoyMToxOCBHTVQ=
+      !binary "VmFyeQ==":
+      - !binary |-
+        QWNjZXB0LUxhbmd1YWdl
+      !binary "Q2FjaGUtQ29udHJvbA==":
+      - !binary |-
+        bWF4LWFnZT0w
+      !binary "TGFzdC1Nb2RpZmllZA==":
+      - !binary |-
+        VGh1LCAyNiBTZXAgMjAxMyAyMzoyMToxOCBHTVQ=
+      !binary "Q2YtUmF5":
+      - !binary |-
+        YjQzYjA4ZjM5ZDgwMDk3
+    body:
+      encoding: ASCII-8BIT
+      string: !binary |-
+        eyJidGNfcmVzZXJ2ZWQiOiAiMCIsICJmZWUiOiAiMC40MDAwIiwgImJ0Y19h
+        dmFpbGFibGUiOiAiMCIsICJ1c2RfcmVzZXJ2ZWQiOiAiMS4wMiIsICJidGNf
+        YmFsYW5jZSI6ICIwIiwgInVzZF9iYWxhbmNlIjogIjY5NTMuMDciLCAidXNk
+        X2F2YWlsYWJsZSI6ICI2OTUyLjA1In0=
+    http_version: 
+  recorded_at: Thu, 26 Sep 2013 23:21:18 GMT
+recorded_with: VCR 2.6.0