lita-onewheel-finance 0.5.0 → 0.7.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 86924c83a1b54b7034f09552a2ff66c46723de2baa4c3b061efbdfc9bcb39d70
-  data.tar.gz: 4b2908349083d6dc6c740fd5c2ddc4b18e7cd90d6180fc672644c09be07530cc
+  metadata.gz: af486197ad5808a1a74b9c18f7a87badf1ffa46349bb80796be2d26ee6a0f240
+  data.tar.gz: 979ed93c618a32f47a925e28676bf599c108f85e4c219a8daeebcc54b8a54d1f
 SHA512:
-  metadata.gz: 750a2b4e836adfaf7bea9913051037a6277c8f26787861fe14ae38389c5479c5e97e8412a9a1cd3251d4c6ad451914cf42776c400d1e97babf88adcdcdbfb4d9
-  data.tar.gz: d67f291c283fc66939cb3a3c59b2954247f7d807c4ba208c8beda96bc4ac6c5c504823dae63f31d944f1f43a6a3cbb51ba40f718a31f39eeb15bc725dd5aed6f
+  metadata.gz: 6efe6ae097efeb85946835686c30d140754f140d003267a5aa91f27959668ad9b93c4e7a4e743cdd5bf5b2447d89c444ee9c6f0259c05e1a5e7de2160156bfe3
+  data.tar.gz: a04f6034575fb5fea5fe6dae265c4ace8dc765546b93f13e3dca32f47d7844505a20382b736f2a374ba414657f7b93bca948ad6940ccdc0a1099614e0f579acf

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,200 +1,32 @@
+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, :message
-  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
-
-    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
-  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
-
 module Lita
   module Handlers
     class OnewheelFinance < Handler
       config :apikey, required: 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
           if stock.message
             str = stock.message
@@ -202,14 +34,40 @@ module Lita
             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
 
@@ -225,13 +83,14 @@ 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
-      #end
+      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)
     end

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