lita-onewheel-finance 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0650867d30743080da877fdbad93bd7ddbd9f525a7921fa566c7da0a64d9850c'
-  data.tar.gz: 0ad45310e357e2e5ed9b2e10bb1721cd0ba91da863507ff5a4dc728b10f14bd9
+  metadata.gz: ef2f754f9b1fe08aa1fe7749a6cc7b338e83b0ecb5d2745decb441f8773f278e
+  data.tar.gz: 95d1163691c52c32eeae155112c0d86fad7bec95226ec56156e28ef94eceb616
 SHA512:
-  metadata.gz: 495d556d163bf461fa1d521ed7176054636a591c1fb2772fcad41634740bc54d5bc4c8d8a464dc936a58d9eb2d6066336fed012bed60e292949bdc5344b3e677
-  data.tar.gz: 9b7bc63735e850c56bfe52dd83d39cce2979be507b400237dea7b5f30b50ba35fad55dd5f24fe220b9f927da241295ac65dd73a41426f23bc93a2cc99b933be6
+  metadata.gz: 33982576f1623d697254375a928dab14974cac4173d8f6079d30c381d8ce56b62d13013ec799f7ca07c1b6e6c8a5f585cdce4d2ae0aaa329072483890b155e71
+  data.tar.gz: cd8eb5c4a74651005cf8579511af016206fbceca03337c4b323b892e6c4204057c8d7df8bc00b05c7df5788648aa7ce0940197c2533ed920d0488c5e4d5d40cb

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,226 +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, :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
-
 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
@@ -233,14 +39,27 @@ module Lita
             dollar_sign = ''
           end
 
-          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}"
+          # 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} ↯#{dollar_sign}#{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
 
@@ -256,13 +75,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