hstove/rbtc_arbitrage

module RbtcArbitrage
  module Client
    attr_accessor :options
    attr_writer :balance

    def initialize config={}
      @options = config
      @options = {}
      set_key config, :volume, 0.01
      set_key config, :cutoff, 2
      set_key config, :logger, Logger.new(STDOUT)
      set_key config, :verbose, true
      set_key config, :live, false
      self
    end

    def validate_keys *args
      args.each do |key|
        key = key.to_s.upcase
        if ENV[key].blank?
          raise ArgumentError, "Exiting because missing required ENV variable $#{key}."
        end
      end
    end

    def buy
      trade :buy
    end

    def sell
      trade :sell
    end

    def address
      ENV["#{exchange.to_s.upcase}_ADDRESS"]
    end

    def logger
      @options[:logger]
    end

    private

    def set_key config, key, default
      @options[key] = config.has_key?(key) ? config[key] : default
    end
  end
end