module Demandbase
  RTID_KEY = ENV['DEMANDBASE_RTID_KEY']
  API_URLS = {
    :domain => "http://api.demandbase.com/api/v1/domain.json?key=#{RTID_KEY}",
    :ip     => "http://api.demandbase.com/api/v1/ip.json?key=#{RTID_KEY}" }

  class Record
    attr_accessor :company_name
    attr_accessor :demandbase_sid
    attr_accessor :marketing_alias
    attr_accessor :industry
    attr_accessor :sub_industry
    attr_accessor :employee_count
    attr_accessor :primary_sic
    attr_accessor :street_address
    attr_accessor :city
    attr_accessor :state
    attr_accessor :zip
    attr_accessor :country
    attr_accessor :country_name
    attr_accessor :phone
    attr_accessor :stock_ticker
    attr_accessor :web_site
    attr_accessor :annual_sales
    attr_accessor :revenue_range
    attr_accessor :employee_range
    attr_accessor :b2b
    attr_accessor :b2c
    attr_accessor :traffic
    attr_accessor :latitude
    attr_accessor :longitude
    attr_accessor :fortune_1000
    attr_accessor :forbes_2000

    # Instantiate a new Demandbase Record from a domain name.
    def initialize(domain)
      query = cleanse_domain(domain)
      url = Demandbase::API_URLS[:domain] + "&query=#{query}"

      begin
        response          = JSON.parse(RestClient.get(url))
        @company_name     = response["domain"]["company_name"]
        @demandbase_sid   = response["domain"]["demandbase_sid"]
        @marketing_alias  = response["domain"]["marketing_alias"]
        @industry         = response["domain"]["industry"]
        @sub_industry     = response["domain"]["sub_industry"]
        @employee_count   = response["domain"]["employee_count"]
        @primary_sic      = response["domain"]["primary_sic"]
        @street_address   = response["domain"]["street_address"]
        @city             = response["domain"]["city"]
        @state            = response["domain"]["state"]
        @zip              = response["domain"]["zip"]
        @country          = response["domain"]["country"]
        @country_name     = response["domain"]["country_name"]
        @phone            = response["domain"]["phone"]
        @stock_ticker     = response["domain"]["stock_ticker"]
        @web_site         = response["domain"]["web_site"]
        @annual_sales     = response["domain"]["annual_sales"]
        @revenue_range    = response["domain"]["revenue_range"]
        @employee_range   = response["domain"]["employee_range"]
        @b2b              = response["domain"]["b2b"]
        @b2c              = response["domain"]["b2c"]
        @traffic          = response["domain"]["traffic"]
        @latitude         = response["domain"]["latitude"]
        @longitude        = response["domain"]["longitude"]
        @fortune_1000     = response["domain"]["fortune_1000"]
        @forbes_2000      = response["domain"]["forbes_2000"]
      rescue => e
        puts "Problem querying the server: #{e.inspect}"
      end
    end

    # Clean the domain of things like 'http(s)://', 'www',
    # '?foo=bar', etc.
    #
    # Return the domain string.
    def cleanse_domain(domain)
      domain.downcase!
      domain = domain.sub(/^https?\:\/\//, '').sub(/^www./,'')
      domain = domain.split("/").first
      domain = domain.split("@").last

      domain = PublicSuffix.parse(domain)
      domain = "#{domain.sld}.#{domain.tld}"
      domain
    end
  end
end