lita-onewheel-finance 0.4.0 → 0.7.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 316a69ee9c4dd4c0e88009906080be2d326bb4cac5723526956e29c569466607
-  data.tar.gz: e9bee0f2b4e414dcc58391707f61db410d39062ecda51ee94f185e5f613160c6
+  metadata.gz: 705648b5037494e24e403eff8e89f00041e4df73fe32a1948b2a4483a07360ee
+  data.tar.gz: cd5a3b83ae4a98f140d824061c2433146481cec08ce0879c0cd668055cb3d043
 SHA512:
-  metadata.gz: c45a1b428948933a9680df0b80cdd1686939484e4819f0cc9a2c22046029c4a61d1337eea10586918a0721cc06d5151c6f6a3788310dfd7fe4cc33a318bbd726
-  data.tar.gz: 1051b4205d68f27c59875ef6ee986b78aecd9cbeb58acda85c8d420cc257562c8a3955f9d76671c104074b64e1766bc8751690f7e4f3e6869aa5cd9a2ef00df6
+  metadata.gz: 12fa54d511e7bc069bd475d9de53bdae53aa0990fd4790c794b50d36aeaec2f93b1594c609bd5d58de9ab1c5d4b1e99b33814bbb066537b961937196e4f3ccab
+  data.tar.gz: 965db07646a717a3adf927a7c94897ad2c21c93d3e5bc691cb173ca3db1d204eccbc5e4958fcfbc7a73277115e849873fffd90897420321f7dfb337b4f9ef148

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,238 +1,73 @@
+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(symbol, api_key)
-    @base_uri = 'https://api.worldtradingdata.com/api/v1'
-    @symbol = symbol
-    @api_key = api_key
-
-    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
-      self.call_api
-      hash = JSON.parse(@response)
-      @error = false
-    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
-
-  # Let's see what we can get from the api.
-  def call_api
-    url = "#{@base_uri}/stock"
-    params = {symbol: @symbol, api_token: @api_key}
-
-    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']
-    end
-  end
-
-  def fix_number(price_str)
-    price_str.to_f.round(2)
-  end
-end
-
-class WorldTradeDataSearch
-  attr_reader :open, :high, :low, :price, :volume, :trading_day, :prev_close, :change, :change_percent, :exchange, :error, :name
-
-  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}#{"%.2f" % stock.price}#{IrcColors::reset} "
+            if stock.change >= 0
+              # if irc
+              str += "#{IrcColors::green} ⬆#{dollar_sign}#{"%.2f" % stock.change}#{IrcColors::reset}, #{IrcColors::green}#{stock.change_percent}%#{IrcColors::reset} "
+              if stock.name
+                str += "#{IrcColors::grey}(#{stock.name})#{IrcColors::reset}"
+              end
+            else
+              str += "#{IrcColors::red} ↯#{dollar_sign}#{"%.2f" % stock.change}#{IrcColors::reset}, #{IrcColors::red}#{stock.change_percent}%#{IrcColors::reset} "
+              if stock.name
+                str += "#{IrcColors::grey}(#{stock.name})#{IrcColors::reset}"
+              end
+            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}#{"%.2f" % stock.price} "
+            if stock.change >= 0
+              # if irc
+              str += " :arrow_up:#{dollar_sign}#{"%.2f" % stock.change}, :heavy_plus_sign:#{stock.change_percent}% "
+              if stock.name
+                str += "(#{stock.name})"
+              end
+            else
+              str += " :chart_with_downwards_trend:#{dollar_sign}#{"%.2f" % stock.change}, :heavy_minus_sign:#{stock.change_percent}% "
+              if stock.name
+                str += "(#{stock.name})"
+              end
+            end
           end
         end
 
@@ -248,12 +83,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