tradier 0.1.0

data/lib/tradier/error/client_error.rb

@@ -0,0 +1,28 @@
+require 'tradier/error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns a 4xx HTTP status code or there's an error in Faraday
+    class ClientError < Tradier::Error
+      MESSAGE = "Client Error"
+
+      # Create a new error from an HTTP environment
+      #
+      # @param response [Hash]
+      # @return [Tradier::Error]
+      def self.from_response(response={})
+        new(response[:body], response[:response_headers])
+      end
+
+      # Initializes a new ServerError object
+      #
+      # @param message [String]
+      # @param response_headers [Hash]
+      # @return [Tradier::Error::ServerError]
+      def initialize(message=nil, response_headers={})
+        super((message || self.class.const_get(:MESSAGE)), response_headers)
+      end
+
+    end
+  end
+end

data/lib/tradier/error/configuration_error.rb

@@ -0,0 +1,6 @@
+module Tradier
+  class Error
+    class ConfigurationError < ::ArgumentError
+    end
+  end
+end

data/lib/tradier/error/decode_error.rb

@@ -0,0 +1,9 @@
+require 'tradier/error'
+
+module Tradier
+  class Error
+    # Raised when JSON parsing fails
+    class DecodeError < Tradier::Error
+    end
+  end
+end

data/lib/tradier/error/forbidden.rb

@@ -0,0 +1,10 @@
+require 'tradier/error/client_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 403
+    class Forbidden < Tradier::Error::ClientError
+      HTTP_STATUS_CODE = 403
+    end
+  end
+end

data/lib/tradier/error/gateway_timeout.rb

@@ -0,0 +1,11 @@
+require 'tradier/error/server_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 504
+    class GatewayTimeout < Tradier::Error::ServerError
+      HTTP_STATUS_CODE = 504
+      MESSAGE = "The Tradier servers are up, but the request couldn't be serviced due to some failure within our stack. Try again later."
+    end
+  end
+end

data/lib/tradier/error/internal_server_error.rb

@@ -0,0 +1,11 @@
+require 'tradier/error/server_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 500
+    class InternalServerError < Tradier::Error::ServerError
+      HTTP_STATUS_CODE = 500
+      MESSAGE = "Something is technically wrong."
+    end
+  end
+end

data/lib/tradier/error/not_acceptable.rb

@@ -0,0 +1,10 @@
+require 'tradier/error/client_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 406
+    class NotAcceptable < Tradier::Error::ClientError
+      HTTP_STATUS_CODE = 406
+    end
+  end
+end

data/lib/tradier/error/not_found.rb

@@ -0,0 +1,10 @@
+require 'tradier/error/client_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 404
+    class NotFound < Tradier::Error::ClientError
+      HTTP_STATUS_CODE = 404
+    end
+  end
+end

data/lib/tradier/error/raise_error.rb

@@ -0,0 +1,31 @@
+require 'faraday'
+require 'tradier/error/bad_gateway'
+require 'tradier/error/bad_request'
+require 'tradier/error/forbidden'
+require 'tradier/error/gateway_timeout'
+require 'tradier/error/internal_server_error'
+require 'tradier/error/not_acceptable'
+require 'tradier/error/not_found'
+require 'tradier/error/service_unavailable'
+require 'tradier/error/too_many_requests'
+require 'tradier/error/unauthorized'
+require 'tradier/error/unprocessable_entity'
+
+module Tradier
+  module Response
+    class RaiseError < Faraday::Response::Middleware
+
+      def on_complete(env)
+        status_code = env[:status].to_i
+        error_class = @klass.errors[status_code]
+        raise error_class.from_response(env) if error_class
+      end
+
+      def initialize(app, klass)
+        @klass = klass
+        super(app)
+      end
+
+    end
+  end
+end

data/lib/tradier/error/server_error.rb

@@ -0,0 +1,28 @@
+require 'tradier/error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns a 5xx HTTP status code
+    class ServerError < Tradier::Error
+      MESSAGE = "Server Error"
+
+      # Create a new error from an HTTP environment
+      #
+      # @param response [Hash]
+      # @return [Tradier::Error]
+      def self.from_response(response={})
+        new(response[:body], response[:response_headers])
+      end
+
+      # Initializes a new ServerError object
+      #
+      # @param message [String]
+      # @param response_headers [Hash]
+      # @return [Tradier::Error::ServerError]
+      def initialize(message=nil, response_headers={})
+        super((message || self.class.const_get(:MESSAGE)), response_headers)
+      end
+
+    end
+  end
+end

data/lib/tradier/error/service_unavailable.rb

@@ -0,0 +1,11 @@
+require 'tradier/error/server_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 503
+    class ServiceUnavailable < Tradier::Error::ServerError
+      HTTP_STATUS_CODE = 503
+      MESSAGE = "Tradier is over capacity."
+    end
+  end
+end

data/lib/tradier/error/too_many_requests.rb

@@ -0,0 +1,12 @@
+require 'tradier/error/client_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 429
+    class TooManyRequests < Tradier::Error::ClientError
+      HTTP_STATUS_CODE = 429
+    end
+    EnhanceYourCalm = TooManyRequests
+    RateLimited = TooManyRequests
+  end
+end

data/lib/tradier/error/unauthorized.rb

@@ -0,0 +1,10 @@
+require 'tradier/error/client_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 401
+    class Unauthorized < Tradier::Error::ClientError
+      HTTP_STATUS_CODE = 401
+    end
+  end
+end

data/lib/tradier/error/unprocessable_entity.rb

@@ -0,0 +1,10 @@
+require 'tradier/error/client_error'
+
+module Tradier
+  class Error
+    # Raised when Tradier returns the HTTP status code 404
+    class UnprocessableEntity < Tradier::Error::ClientError
+      HTTP_STATUS_CODE = 422
+    end
+  end
+end

data/lib/tradier/event.rb

@@ -0,0 +1,8 @@
+require 'tradier/base'
+
+module Tradier
+  class Event < Tradier::Base
+
+    attr_reader :symbol, :meta, :description, :event_type, :amount
+  end
+end

data/lib/tradier/history.rb

@@ -0,0 +1,20 @@
+require 'tradier/base'
+require 'date'
+
+module Tradier
+  class History < Tradier::Base
+
+    attr_reader :open , :close, :high, :low, :volume
+
+    def date
+      return unless @date || @attrs[:date]
+
+      @date ||= Date.parse(@attrs[:date])
+    end
+
+    def self.from_response(body={})
+      new(body)
+    end
+
+  end
+end

data/lib/tradier/option_quote.rb

@@ -0,0 +1,37 @@
+require 'tradier/base'
+require 'tradier/symbol'
+
+module Tradier
+  class OptionQuote < Tradier::Base
+
+    attr_reader :symbol, :strike, :last, :bid, :ask, :change, :open_interest, :bid_size, :ask_size, :volume
+
+    def put?
+      parsed_symbol.put?
+    end
+
+    def call?
+      parsed_symbol.call?
+    end
+
+    def underlier
+      parsed_symbol.underlier
+    end
+
+    def expiration_date
+      return unless @expiration_date || @attrs[:expiration_date]
+
+      @expiration_date ||= Date.parse(@attrs[:expiration_date])
+    end
+
+    def ==(option_quote)
+      self.symbol == option_quote.symbol
+    end
+
+    private
+
+    def parsed_symbol
+      @parsed ||= Tradier::Symbol.parse(@attrs[:symbol])
+    end
+  end
+end

data/lib/tradier/order.rb

@@ -0,0 +1,14 @@
+require 'tradier/base'
+
+module Tradier
+  class Order < Tradier::Base
+    attr_reader :id, :order_id, :type, :price, :symbol, :side, :quantity, \
+    :status, :time_in_force, :option_type, :exec_inst, :exec_status, :extended_hours, \
+    :trailing_limit_type, :trailing_stop_type, :request_date, :response_date, :num_legs, \
+    :errors, :result
+
+    def self.from_response(body={})
+      new(body[:order] || body)
+    end
+  end
+end

data/lib/tradier/position.rb

@@ -0,0 +1,7 @@
+require 'tradier/base'
+
+module Tradier
+  class Position < Tradier::Base
+    attr_reader :costbasis, :date_acquired, :quantity, :symbol
+  end
+end

data/lib/tradier/profile.rb

@@ -0,0 +1,14 @@
+require 'tradier/base'
+
+module Tradier
+  class Profile < Tradier::Base
+
+    def name
+      @name ||= @attrs[:profile] && @attrs[:profile][:name]
+    end
+
+    def accounts
+      @accounts ||= @attrs[:profile] && [@attrs[:profile][:account]].flatten.map { |a| Tradier::Account.new(a); }
+    end
+  end
+end

data/lib/tradier/quote.rb

@@ -0,0 +1,12 @@
+require 'tradier/base'
+
+module Tradier
+  class Quote < Tradier::Base
+    attr_reader :symbol
+
+    def ==(quote)
+      self.symbol == quote.symbol
+    end
+
+  end
+end

data/lib/tradier/response/parse_json.rb

@@ -0,0 +1,25 @@
+require 'faraday'
+require 'multi_json'
+
+module Tradier
+  module Response
+    class ParseJson < Faraday::Response::Middleware
+
+      def parse(body)
+        case body
+        when /\A^\s*$\z/, nil
+          nil
+        else
+          MultiJson.decode(body, :symbolize_keys => true)
+        end
+      end
+
+      def on_complete(env)
+        if respond_to?(:parse)
+          env[:body] = parse(env[:body]) unless [204, 301, 302, 304].include?(env[:status])
+        end
+      end
+
+    end
+  end
+end

data/lib/tradier/response/raise_error.rb

@@ -0,0 +1,31 @@
+require 'faraday'
+require 'tradier/error/bad_gateway'
+require 'tradier/error/bad_request'
+require 'tradier/error/forbidden'
+require 'tradier/error/gateway_timeout'
+require 'tradier/error/internal_server_error'
+require 'tradier/error/not_acceptable'
+require 'tradier/error/not_found'
+require 'tradier/error/service_unavailable'
+require 'tradier/error/too_many_requests'
+require 'tradier/error/unauthorized'
+require 'tradier/error/unprocessable_entity'
+
+module Tradier
+  module Response
+    class RaiseError < Faraday::Response::Middleware
+
+      def on_complete(env)
+        status_code = env[:status].to_i
+        error_class = @klass.errors[status_code]
+        raise error_class.from_response(env) if error_class
+      end
+
+      def initialize(app, klass)
+        @klass = klass
+        super(app)
+      end
+
+    end
+  end
+end

data/lib/tradier/symbol.rb

@@ -0,0 +1,147 @@
+module Tradier
+  class Symbol
+
+    OPTION_SYMBOL_REGEXP = /([a-z]+)(\d?)(\d{2})(\d{2})(\d{2})(C|P)(\d{5})(\d{3})/ix
+
+    MONTHS = {
+      1 => 'Jan',
+      2 => 'Feb',
+      3 => 'Mar',
+      4 => 'Apr',
+      5 => 'May',
+      6 => 'Jun',
+      7 => 'Jul',
+      8 => 'Aug',
+      9 => 'Sep',
+      10 => 'Oct',
+      11 => 'Nov',
+      12 => 'Dec'
+    }
+
+    attr_accessor :symbol, :adjustment, :month, :day, :year, :option_type, :dollars, :cents
+
+    def initialize(symbol)
+      @symbol = symbol
+    end
+
+    def self.parse(symbol)
+      symbol.strip!
+
+      sym = if match = symbol.match(OPTION_SYMBOL_REGEXP)
+        match = match.to_a
+        match.shift
+
+        s = Tradier::Symbol.new(match[0])
+        s.adjustment  = match[1]
+        s.year        = match[2]
+        s.month       = match[3]
+        s.day         = match[4]
+        s.option_type = match[5]
+        s.dollars     = match[6]
+        s.cents       = match[7]
+        s
+      elsif symbol =~ /\d/
+        nil
+      elsif symbol =~ /[a-zA-Z]+/
+        Tradier::Symbol.new(symbol)
+      else
+        nil
+      end
+
+      raise ArgumentError.new("Invalid symbol: #{symbol}") unless sym && sym.valid?
+      sym
+    end
+
+    def adjustment?
+      adjustment && adjustment.to_i >= 1
+    end
+
+    def put?
+      return unless option_type
+      option_type.upcase == 'P'
+    end
+
+    def call?
+      return unless option_type
+      option_type.upcase == 'C'
+    end
+
+    def option?
+      put? || call?
+    end
+
+    def equity?
+      !option?
+    end
+
+    def valid?
+      equity? || (option? && valid_month?)
+    end
+
+    def to_s
+      s = equity? ? underlier : "#{underlier}#{@adjustment if adjustment?}#{@year}#{occ_month}#{occ_day}#{@option_type}#{occ_dollars}#{occ_cents}"
+      s.upcase
+    end
+
+    def description
+      equity? ? underlier : "#{underlier} #{expiration_description} $#{strike_description} #{option_type_description}"
+    end
+
+    def underlier
+      @symbol
+    end
+
+    def occ_month
+      @month.to_i >= 10 ? @month : "0#{@month.to_i}"
+    end
+
+    def occ_day
+      @day.to_i >= 10 ? @day : "0#{@day.to_i}"
+    end
+
+    def occ_cents
+      case @cents.size
+      when 3 then @cents
+      when 2 then "#{@cents}0"
+      when 1 then "#{@cents}00"
+      when 0 then "000"
+      end
+    end
+
+    def occ_dollars
+      return @dollars if @dollars.size == 5
+      ('0' * (5 - @dollars.size)) + @dollars.to_s
+    end
+
+    def expiration_description
+      "#{MONTHS[@month.to_i]} #{@day} #{year_description}"
+    end
+
+    def year_description
+      "20#{@year}"
+    end
+
+    def strike_description
+      "#{@dollars.to_i}.#{occ_cents[0,2]}"
+    end
+
+    def option_type_description
+      put? ? 'PUT' : 'CALL'
+    end
+
+    def expiration
+      Date.new(year_description.to_i, @month.to_i, @day.to_i)
+    end
+
+    def strike_price
+      "#{@dollars}.#{@cents}".to_f
+    end
+
+    private
+
+    def valid_month?
+      @month.to_i <= 12 && @month.to_i >= 1
+    end
+
+  end
+end