lita-onewheel-finance 0.3.1 → 0.7.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 113679ad593f4f4ca6d87833981502ae6f85ec63291ea72fd22385cb66197558
-  data.tar.gz: '09f4b7139aa8f0c637391b7508101e2adec2ab044358289f41d0ea87555b2d0f'
+  metadata.gz: 15c662356a8c3af7ef1ddc3d9e2593e1f97407f14715342e9a4bb8b6e9eb1d6f
+  data.tar.gz: a06245525f9266e6d0860175a994399b8e9049c80ef7c5c90bec1244fd29b5e5
 SHA512:
-  metadata.gz: 945c1ccecab692ec3d96ac783344ace668aa2bc5d9e32f8deda7a110cf6b5236fe5928a7b1d5653bfe54d3097bf12adf595b036f334e91c2e01cb42ed1e67efb
-  data.tar.gz: 5cefd0c3fa137f66fda888281d6835f1189ba1d2ea771b6b7a1eb608baf1e4df5d49a2ec6c95cc017a30a020c7555e8f475ec61d847890ea4708341a2a9ff261
+  metadata.gz: 6c88a85f9a6b5dc86bc2a9683398c734debab9a5f7fdf9c8f0be8cd6e2bfd69aa9da16798318b7a7b05a73b5facd9a99b89c003363628629a3f92dfb2ab352c7
+  data.tar.gz: 91f5c5cc47f9690698faa731b7b7e93eaea9b16ec22badae7293dd075275861e072d5b331d2ef7b687c6c54160d6fe390cf120bb1cf673a743d22513ca1b1609

data/lib/lita/handlers/alphavantage_quote.rb

@@ -0,0 +1,68 @@
+class AlphaVantageQuote
+  attr_reader :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent, :exchange, :error, :name, :message, :is_index
+  attr_accessor :symbol
+
+  def initialize(symbol, apikey)
+    # Lita.logger.debug "parsing: #{json_blob}"
+    # hash = JSON.parse(json_blob)
+    @symbol = symbol
+    @apikey = apikey
+    @is_index = false
+
+    if @symbol[0] == '^'
+      @is_index = true
+    end
+
+    response = call_api
+    hash = JSON.parse response
+
+    quote = hash["Global Quote"]
+
+    quote.keys.each do |key|
+      case key
+      when "01. symbol"
+        @symbol = quote[key]
+      when "02. open"
+        @open = self.fix_number quote[key]
+      when "03. high"
+        @high = self.fix_number quote[key]
+      when "04. low"
+        @low = self.fix_number quote[key]
+      when "05. price"
+        @price = self.fix_number quote[key]
+      when "06. volume"
+        @volume = quote[key]
+      when "07. latest trading day"
+        @trading_day = quote[key]
+      when "08. previous close"
+        @prev_close = self.fix_number quote[key]
+      when "09. change"
+        @change = self.fix_number quote[key]
+      when "10. change percent"
+        @change_percent = self.fix_number quote[key]
+      end
+    end
+  end
+
+  def fix_number(price_str)
+    price_str.to_f.round(2)
+  end
+
+  def is_index?
+    @is_index
+  end
+
+  def call_api
+    url = "https://www.alphavantage.co/query"
+    params = {function: 'GLOBAL_QUOTE', symbol: @symbol, apikey: @apikey}
+    Lita.logger.debug "#{url} #{params.inspect}"
+    response = RestClient.get url, {params: params}
+
+    # if @exchange
+    #   params[:stock_exchange] = @exchange
+    # end
+
+    Lita.logger.debug "response: #{response}"
+    response
+  end
+end

data/lib/lita/handlers/irc_colors.rb

@@ -0,0 +1,43 @@
+module IrcColors
+  prefix = "\x03"
+  @white  = "#{prefix}00"
+  @black  = "#{prefix}01"
+  @blue   = "#{prefix}02"
+  @green  = "#{prefix}03"
+  @red    = "#{prefix}04"
+  @brown  = "#{prefix}05"
+  @purple = "#{prefix}06"
+  @orange = "#{prefix}07"
+  @yellow = "#{prefix}08"
+  @lime   = "#{prefix}09"
+  @teal   = "#{prefix}10"
+  @aqua   = "#{prefix}11"
+  @royal  = "#{prefix}12"
+  @pink   = "#{prefix}13"
+  @grey   = "#{prefix}14"
+  @silver = "#{prefix}15"
+  @reset  = prefix
+
+  class << self
+    attr_reader :white,
+                :black,
+                :blue,
+                :green,
+                :red,
+                :brown,
+                :purple,
+                :orange,
+                :yellow,
+                :lime,
+                :teal,
+                :aqua,
+                :royal,
+                :pink,
+                :grey,
+                :silver,
+                :reset
+  end
+
+  def initialize
+  end
+end

data/lib/lita/handlers/onewheel_finance.rb

@@ -1,144 +1,65 @@
+require_relative 'irc_colors'
+require_relative 'alphavantage_quote'
+require_relative 'worldtradedata_quote'
+require_relative 'yahoo_quote'
 require 'rest-client'
 
-class IrcColors
-  prefix = "\x03"
-  @white  = "#{prefix}00"
-  @black  = "#{prefix}01"
-  @blue   = "#{prefix}02"
-  @green  = "#{prefix}03"
-  @red    = "#{prefix}04"
-  @brown  = "#{prefix}05"
-  @purple = "#{prefix}06"
-  @orange = "#{prefix}07"
-  @yellow = "#{prefix}08"
-  @lime   = "#{prefix}09"
-  @teal   = "#{prefix}10"
-  @aqua   = "#{prefix}11"
-  @royal  = "#{prefix}12"
-  @pink   = "#{prefix}13"
-  @grey   = "#{prefix}14"
-  @silver = "#{prefix}15"
-  @reset  = prefix
-
-  class << self
-    attr_reader :white, :black, :blue, :green, :red, :brown, :purple, :orange, :yellow, :lime, :teal, :aqua, :royal, :pink, :grey, :silver, :reset
-  end
-
-  def initialize
-  end
-end
-
-class AlphaVantageQuote
-  attr_reader :symbol, :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent
-
-  def initialize(json_blob)
-    Lita.logger.debug "parsing: #{json_blob}"
-    hash = JSON.parse(json_blob)
-    quote = hash["Global Quote"]
-
-    quote.keys.each do |key|
-      case key
-      when "01. symbol"
-        @symbol = quote[key]
-      when "02. open"
-        @open = self.fix_number quote[key]
-      when "03. high"
-        @high = self.fix_number quote[key]
-      when "04. low"
-        @low = self.fix_number quote[key]
-      when "05. price"
-        @price = self.fix_number quote[key]
-      when "06. volume"
-        @volume = quote[key]
-      when "07. latest trading day"
-        @trading_day = quote[key]
-      when "08. previous close"
-        @prev_close = self.fix_number quote[key]
-      when "09. change"
-        @change = self.fix_number quote[key]
-      when "10. change percent"
-        @change_percent = self.fix_number quote[key]
-      end
-    end
-  end
-
-  def fix_number(price_str)
-    price_str.to_f.round(2)
-  end
-end
-
-class WorldTradeDataQuote
-  attr_reader :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent, :exchange, :error, :name
-  attr_accessor :symbol
-
-  def initialize(json_blob)
-    Lita.logger.debug "parsing: #{json_blob}"
-    hash = JSON.parse(json_blob)
-
-    if hash['Message'].to_s.include? 'Error'
-      @error = true
-      return
-    else
-      @error = false
-    end
-
-    quote = hash['data'][0]
-
-    quote.keys.each do |key|
-      case key
-      when "symbol"
-        @symbol = quote[key]
-      when "price_open"
-        @open = self.fix_number quote[key]
-      when "day_high"
-        @high = self.fix_number quote[key]
-      when "day_low"
-        @low = self.fix_number quote[key]
-      when "price"
-        @price = self.fix_number quote[key]
-      when "volume"
-        @volume = quote[key].to_i
-      when "last_trade_time"
-        @trading_day = quote[key]
-      when "08. previous close"
-        @prev_close = self.fix_number quote[key]
-      when "day_change"
-        @change = self.fix_number quote[key]
-      when "change_pct"
-        @change_percent = self.fix_number quote[key]
-      when 'stock_exchange_short'
-        @exchange = quote[key].sub /NYSEARCA/, 'NYSE'
-      when 'name'
-        @name = quote[key]
-      end
-    end
-  end
-
-  def fix_number(price_str)
-    price_str.to_f.round(2)
-  end
-end
-
 module Lita
   module Handlers
     class OnewheelFinance < Handler
       config :apikey, required: true
-      route /quote\s+(.+)/i, :handle_quote, command: true
+      config :mode, default: 'irc'
+      config :handler, default: 'alphavantage'
+      route /qu*o*t*e*\s+(.+)/i, :handle_quote, command: true
+      # route /q2\s+(.+)/i, :handle_alphavantage, command: true
 
       def handle_quote(response)
-        stock = handle_world_trade_data response.matches[0][0]
+        stock = nil
+        if config.handler == 'worldtradedata'
+          stock = handle_world_trade_data response.matches[0][0]
+        elsif config.handler == 'alphavantage'
+          stock = handle_alphavantage response.matches[0][0]
+        elsif config.handler == 'yahoo'
+          stock = handle_yahoo response.matches[0][0]
+        else
+          Lita.logger.error "Unknown/missing config.handler #{config.handler}.  Try 'worldtradedata' or 'alphavantage'"
+          return
+        end
 
+        # Continue!
         if stock.error
-          str = "`#{stock.symbol}` not found on any stock exchange."
+          if stock.message
+            str = stock.message
+          else
+            str = "`#{stock.symbol}` not found on any stock exchange."
+          end
         else
-          str = "#{IrcColors::grey}#{stock.exchange} - #{IrcColors::reset}#{stock.symbol}: #{IrcColors::blue}$#{stock.price}#{IrcColors::reset} "
-          if stock.change >= 0
-            # if irc
-            str += "#{IrcColors::green} ⬆$#{stock.change}#{IrcColors::reset}, #{IrcColors::green}#{stock.change_percent}%#{IrcColors::reset} "
-            str += "#{IrcColors::grey}(#{stock.name})#{IrcColors::reset}"
+          dollar_sign = '$'
+          if stock.is_index?
+            dollar_sign = ''
+          end
+
+          # IRC mode
+          if config.mode == 'irc'
+            str = "#{IrcColors::grey}#{stock.exchange} - #{IrcColors::reset}#{stock.symbol}: #{IrcColors::blue}#{dollar_sign}#{stock.price}#{IrcColors::reset} "
+            if stock.change >= 0
+              # if irc
+              str += "#{IrcColors::green} ⬆#{dollar_sign}#{stock.change}#{IrcColors::reset}, #{IrcColors::green}#{stock.change_percent}%#{IrcColors::reset} "
+              str += "#{IrcColors::grey}(#{stock.name})#{IrcColors::reset}"
+            else
+              str += "#{IrcColors::red} ↯#{dollar_sign}#{stock.change}#{IrcColors::reset}, #{IrcColors::red}#{stock.change_percent}%#{IrcColors::reset} "
+              str += "#{IrcColors::grey}(#{stock.name})#{IrcColors::reset}"
+            end
           else
-            str += "#{IrcColors::red} ↯$#{stock.change}#{IrcColors::reset}, #{IrcColors::red}#{stock.change_percent}%#{IrcColors::reset} "
-            str += "#{IrcColors::grey}(#{stock.name})#{IrcColors::reset}"
+            str = "#{stock.exchange} - #{stock.symbol}: #{dollar_sign}#{stock.price} "
+            if stock.change >= 0
+              # if irc
+              str += " :arrow_up:#{dollar_sign}#{stock.change}, :heavy_plus_sign:#{stock.change_percent}% "
+              str += "(#{stock.name})"
+            else
+              str += " :chart_with_downwards_trend:#{dollar_sign}#{stock.change}, :heavy_minus_sign:#{stock.change_percent}% "
+              str += "(#{stock.name})"
+            end
           end
         end
 
@@ -146,15 +67,7 @@ module Lita
       end
 
       def handle_world_trade_data(symbol)
-        url = "https://api.worldtradingdata.com/api/v1/stock"
-        params = {symbol: symbol, api_token: config.apikey}
-
-        Lita.logger.debug "#{url} #{params.inspect}"
-
-        resp = RestClient.get url, {params: params}
-        Lita.logger.debug "response: #{resp}"
-
-        stock = WorldTradeDataQuote.new resp
+        stock = WorldTradeDataQuote.new symbol, config.apikey
         if stock.error
           stock.symbol = symbol
         end
@@ -162,12 +75,13 @@ module Lita
       end
 
       # deprecated for now
-      def handle_alphavantage
-        url = "https://www.alphavantage.co/query"
-        params = {function: 'GLOBAL_QUOTE', symbol: response.matches[0][0], apikey: config.apikey}
-        Lita.logger.debug "#{url} #{params.inspect}"
-        resp = RestClient.get url, {params: params}
-        stock = GlobalQuote.new resp
+      def handle_alphavantage(symbol)
+       stock = AlphaVantageQuote.new symbol, config.apikey
+      end
+
+      # welp
+      def handle_yahoo(symbol)
+       stock = YahooQuote.new symbol, config.apikey
       end
 
       Lita.register_handler(self)

data/lib/lita/handlers/worldtradedata_quote.rb

@@ -0,0 +1,128 @@
+class WorldTradeDataQuote
+  attr_reader :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent, :exchange, :error, :name, :message, :is_index
+  attr_accessor :symbol
+
+  def initialize(symbol, api_key)
+    @base_uri = 'https://api.worldtradingdata.com/api/v1'
+    @symbol = symbol
+    @api_key = api_key
+
+    if @symbol[':']
+      (@exchange, @symbol) = @symbol.split /:/
+    end
+
+    if @symbol['^']
+      @is_index = true
+    else
+      @is_index = false
+    end
+
+    self.call_api
+
+    hash = JSON.parse(@response)
+
+    # We couldn't find the stock.  Let's look for it real quick.
+    if hash['Message'].to_s.include? 'Error'
+      @error = true
+      self.run_search
+
+      if @message
+        @error = true
+        return
+      else
+        self.call_api
+        hash = JSON.parse(@response)
+        @error = false
+      end
+    else
+      @error = false
+    end
+
+    unless hash['data']
+      @message = "Error getting data for #{@symbol}"
+      @error = true
+      return
+    end
+
+    quote = hash['data'][0]
+
+    quote.keys.each do |key|
+      case key
+      when "symbol"
+        @symbol = quote[key]
+      when "price_open"
+        @open = self.fix_number quote[key]
+      when "day_high"
+        @high = self.fix_number quote[key]
+      when "day_low"
+        @low = self.fix_number quote[key]
+      when "price"
+        @price = self.fix_number quote[key]
+      when "volume"
+        @volume = quote[key].to_i
+      when "last_trade_time"
+        @trading_day = quote[key]
+      when "08. previous close"
+        @prev_close = self.fix_number quote[key]
+      when "day_change"
+        @change = self.fix_number quote[key]
+      when "change_pct"
+        @change_percent = self.fix_number quote[key]
+      when 'stock_exchange_short'
+        @exchange = quote[key].sub /NYSEARCA/, 'NYSE'
+      when 'name'
+        @name = quote[key]
+      end
+    end
+
+    def is_index?
+      is_index
+    end
+  end
+
+  # Let's see what we can get from the api.
+  def call_api
+    url = "#{@base_uri}/stock"
+    params = {symbol: @symbol, api_token: @api_key}
+
+    if @exchange
+      params[:stock_exchange] = @exchange
+    end
+
+    Lita.logger.debug "call_api: #{url} #{params.inspect}"
+
+    @response = RestClient.get url, {params: params}
+
+    Lita.logger.debug "response: #{@response}"
+  end
+
+  def run_search
+    url = "#{@base_uri}/stock_search"
+    params = {search_term: @symbol,
+              search_by: 'symbol,name',
+              stock_exchange: 'NASDAQ,NYSE',
+              limit: 5,
+              page: 1,
+              api_token: @api_key
+    }
+
+    Lita.logger.debug "run_search: #{url} #{params.inspect}"
+
+    response = RestClient.get url, {params: params}
+
+    Lita.logger.debug "response: #{response}"
+    result = JSON.parse(response)
+
+    if result['total_returned'] == 1
+      @symbol = result['data'][0]['symbol']
+    elsif result['total_returned'].to_i > 1
+      Lita.logger.debug "many search results: #{result.inspect}"
+      x = result['data'].map { |k| k.values[0] }
+      @message = "`#{symbol}` not found, did you mean one of #{x.join(', ')}?"
+    end
+  end
+
+  def fix_number(price_str)
+    price_str.to_f.round(2)
+  end
+end