lita-onewheel-finance 0.3.1 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 113679ad593f4f4ca6d87833981502ae6f85ec63291ea72fd22385cb66197558
-  data.tar.gz: '09f4b7139aa8f0c637391b7508101e2adec2ab044358289f41d0ea87555b2d0f'
+  metadata.gz: 316a69ee9c4dd4c0e88009906080be2d326bb4cac5723526956e29c569466607
+  data.tar.gz: e9bee0f2b4e414dcc58391707f61db410d39062ecda51ee94f185e5f613160c6
 SHA512:
-  metadata.gz: 945c1ccecab692ec3d96ac783344ace668aa2bc5d9e32f8deda7a110cf6b5236fe5928a7b1d5653bfe54d3097bf12adf595b036f334e91c2e01cb42ed1e67efb
-  data.tar.gz: 5cefd0c3fa137f66fda888281d6835f1189ba1d2ea771b6b7a1eb608baf1e4df5d49a2ec6c95cc017a30a020c7555e8f475ec61d847890ea4708341a2a9ff261
+  metadata.gz: c45a1b428948933a9680df0b80cdd1686939484e4819f0cc9a2c22046029c4a61d1337eea10586918a0721cc06d5151c6f6a3788310dfd7fe4cc33a318bbd726
+  data.tar.gz: 1051b4205d68f27c59875ef6ee986b78aecd9cbeb58acda85c8d420cc257562c8a3955f9d76671c104074b64e1766bc8751690f7e4f3e6869aa5cd9a2ef00df6

data/lib/lita/handlers/onewheel_finance.rb

@@ -71,6 +71,100 @@ 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)
@@ -146,15 +240,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

data/lita-onewheel-finance.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |spec|
   spec.name          = "lita-onewheel-finance"
-  spec.version       = "0.3.1"
+  spec.version       = "0.4.0"
   spec.authors       = ["Andrew Kreps"]
   spec.email         = ["andrew.kreps@gmail.com"]
   spec.description   = "Wee li'l stock quote bot"

data/spec/fixtures/worldtradedata-search-1.json

@@ -0,0 +1,17 @@
+{
+  "total_returned": 1,
+  "total_results": 1,
+  "total_pages": 1,
+  "limit": 5,
+  "page": 1,
+  "data": [
+    {
+      "symbol": "NKE",
+      "name": "NIKE, Inc.",
+      "currency": "USD",
+      "price": "102.20",
+      "stock_exchange_long": "New York Stock Exchange",
+      "stock_exchange_short": "NYSE"
+    }
+  ]
+}

data/spec/lita/handlers/onewheel_finance_spec.rb

@@ -30,4 +30,10 @@ describe Lita::Handlers::OnewheelFinance, lita_handler: true do
     send_command 'quote in'
     expect(replies.last).to eq('`in` not found on any stock exchange.')
   end
+
+  #it 'searches with 1 result' do
+  #  mock_up 'worldtradedata-search-1'
+  #  send_command 'quote nike'
+  #  expect(replies.last).to eq('NKE')
+  #end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lita-onewheel-finance
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
 platform: ruby
 authors:
 - Andrew Kreps
@@ -144,6 +144,7 @@ files:
 - spec/fixtures/worldtradedata-error.json
 - spec/fixtures/worldtradedata-quote-down.json
 - spec/fixtures/worldtradedata-quote-up.json
+- spec/fixtures/worldtradedata-search-1.json
 - spec/lita/handlers/onewheel_finance_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/onewheelskyward/lita-onewheel-finance
@@ -177,5 +178,6 @@ test_files:
 - spec/fixtures/worldtradedata-error.json
 - spec/fixtures/worldtradedata-quote-down.json
 - spec/fixtures/worldtradedata-quote-up.json
+- spec/fixtures/worldtradedata-search-1.json
 - spec/lita/handlers/onewheel_finance_spec.rb
 - spec/spec_helper.rb