stockfighter 0.1.4 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9303ecd757e438d43ccd60406c9f9dd435754db7
-  data.tar.gz: c5c827c6ae4651d831aa01a18849acebb9956610
+  metadata.gz: 154ddcc262a23ff8b0405fc5e0ba6d786acbbfd3
+  data.tar.gz: 269848186df9487af20330d68c6c2a44c0fbcb05
 SHA512:
-  metadata.gz: 8d44a34db3496aa5b8958e4471eb0f4bb3cd7276fc513b95028e9b6e1dbfa175b22f3efa6f95505a1832afbea3965029f6f0eff00e573eb996ee3eaedc8c76b9
-  data.tar.gz: ba974cc66552de0c09ffb6f49bd8d275a72e56635189ef6f2a9d6b471dacc440147e5ec4dd484b66a97f2a6d5eb649db9d82a5fe3e731768e5fb4bfbc07c2369
+  metadata.gz: c013b1a77db9436e7a2d8524a3f92ee8fd10871d8af16e7128511958b6b1cce27a08cc64397a3e15adbadc8d6571cbcca6720c9a9d8047db0226fd39c9d16915
+  data.tar.gz: 2d6540bfd7b0e1c4d144745772326249e44e94c559df1ddee27c7c070f2e75b009f6ad06b2b3fb94dcb440605dc5869df51f20259fd88db091324302cb279582

data/README.md

@@ -26,7 +26,20 @@ require 'stockfighter'
 
 # Use the GM to fetch the info automatically
 
-gm = Stockfighter::GM.new(key: "supersecretapikey1234567", level: "first_steps")
+gm = Stockfighter::GM.new(key: "supersecretapikey1234567", level: "first_steps", polling:true)
+
+# Register message callbacks - GM needs to be constructed with polling:true for these callbacks to work
+gm.add_message_callback('success') { |message|
+	puts "\e[#32m#{message}\e[0m"
+}
+gm.add_message_callback('info') { |message|
+	puts "\e[#34m#{message}\e[0m"
+}
+gm.add_state_change_callback { |previous_state, new_state|
+	if new_state == 'won'
+		puts "You've won!"
+	end
+}
 
 api = Stockfighter::Api.new(gm.config)
 

data/lib/stockfighter/api.rb

@@ -15,6 +15,18 @@ module Stockfighter
       HTTParty.get("#{BASE_URL}/venues/#{@venue}/stocks/#{@symbol}/quote", {"X-Starfighter-Authorization" => @api_key}).parsed_response
     end
 
+    def block_until_first_trade()
+      puts "Waiting until first trade of #{@symbol}.#{@venue}"
+
+      quote = nil
+      loop do
+        quote = get_quote
+        puts quote
+        break if quote['last'] != nil
+      end
+      quote
+    end
+
     def place_order(price:, quantity:, direction:, order_type:)
       order = {
         "account" => @account,
@@ -30,6 +42,17 @@ module Stockfighter
       headers: {"X-Starfighter-Authorization" => @api_key}).parsed_response
     end
 
+    def block_until_order_filled(order_id)
+      puts "Blocking until order #{order_id} is filled"
+
+      loop do 
+        quote = get_quote
+        order_status = order_status(order_id)
+        puts "Order Status: #{order_status['direction']} #{order_status['originalQty']} #{order_status['symbol']}.#{order_status['venue']} @ #{order_status['price']} FilledQuantity:#{order_status['totalFilled']} Last Price: #{quote['last']}"
+        break if not order_status["open"]
+      end       
+    end
+
     def cancel_order(order_id)
       HTTParty.delete("#{BASE_URL}/venues/#{@venue}/stocks/#{@symbol}/orders/#{order_id}", headers: {"X-Starfighter-Authorization" => @api_key})
     end

data/lib/stockfighter/gm.rb

@@ -1,4 +1,5 @@
 require 'httparty'
+require 'rufus-scheduler'
 
 module Stockfighter
   class GM
@@ -6,13 +7,41 @@ module Stockfighter
 
     attr_accessor :instance_id
 
-    def initialize(key:, level:)
+    def initialize(key:, level:, polling:false)
       @api_key = key
       @level = level
 
-      resp = HTTParty.post("#{GM_URL}/levels/#{level}", :headers => {"X-Starfighter-Authorization" => @api_key})
+      @callback_types = ['success', 'info']
+      @message_callbacks = Hash.new { |h,k| h[k] = [] }
+      @state_change_callback = []
 
-      update_config(resp)
+      new_level_response = perform_request("post", "#{GM_URL}/levels/#{level}")
+      previous_state = new_level_response['state']
+
+      if polling
+        # websocket API functionality instead of polling would be great here
+        scheduler = Rufus::Scheduler.new(:overlap => false)
+        scheduler.every '10s' do
+          response = get_instance()
+
+          current_state = response['state']
+          if previous_state != current_state
+            @state_change_callback.each { |callback|
+              callback.call(previous_state, current_state)
+            }
+            previous_state = current_state
+          end
+        end
+      end
+    end
+
+    def add_message_callback(type, &block)
+      raise "Unknown message callback type #{type}" unless @callback_types.include? type
+      @message_callbacks[type] << block
+    end
+
+    def add_state_change_callback(&block)
+      @state_change_callback << block
     end
 
     def config
@@ -24,49 +53,71 @@ module Stockfighter
     end
 
     def restart
-      if @instance_id
-        resp = HTTParty.post("#{GM_URL}/instances/#{@instance_id}/restart", :headers => {"X-Starfighter-Authorization" => @api_key})
-        update_config(resp)
-      end
+      perform_request("post", "#{GM_URL}/instances/#{@instance_id}/restart")
     end
 
     def stop
-      if @instance_id
-        resp = HTTParty.post("#{GM_URL}/instances/#{@instance_id}/stop", :headers => {"X-Starfighter-Authorization" => @api_key})
-        update_config(resp)
-      end
+      perform_request("post", "#{GM_URL}/instances/#{@instance_id}/stop")
     end
 
     def resume
-      if @instance_id
-        resp = HTTParty.post("#{GM_URL}/instances/#{@instance_id}/resume", :headers => {"X-Starfighter-Authorization" => @api_key})
-        update_config(resp)
-      end
+      perform_request("post", "#{GM_URL}/instances/#{@instance_id}/resume")
     end
 
     def active?
-      response = HTTParty.get("#{GM_URL}/instances/#{@instance_id}", :headers => {"X-Starfighter-Authorization" => @api_key})
-      response["done"] && response["ok"]
+      response = get_instance()
+      response["done"]
     end
 
-    def update_config(resp)
+    def get_instance
+      perform_request("get", "#{GM_URL}/instances/#{@instance_id}")
+    end
 
-      if resp.code == 200
-        if resp["ok"]
-          @config = {}
-          @config[:key] = @api_key
-          @config[:account] = resp["account"]
-          @config[:venue] = resp["venues"][0]
-          @config[:symbol] = resp["tickers"][0]
+    def perform_request(action, url)
+      response = HTTParty.method(action).call(url, :headers => {"X-Starfighter-Authorization" => @api_key})
+      if response.code != 200
+        raise "HTTP error response received from #{url}: #{response.code}"
+      end
+      if not response["ok"]
+        raise "Error response received from #{url}: #{response['error']}"
+      end
 
-          @instance_id = resp["instanceId"]
+      if response.key?('flash')
+        flash = response['flash']
+        if flash.key?('success')
+          @message_callbacks['success'].each { |callback|
+            callback.call(flash['success'])
+          }
+        elsif flash.key?('info')
+          @message_callbacks['info'].each { |callback|
+            callback.call(flash['info'])
+          }
         else
-          raise "Error response received from #{GM_URL}: #{resp['error']}"
+          raise "TODO: Unhandled flash scenario: #{response}"
         end
-      else
-        raise "HTTP error response received from #{GM_URL}: #{resp.code}"
       end
+
+      if response.key?('instructions')
+        if response['instructions'].key?('Instructions')
+          puts "\e[#34m#{response['instructions']['Instructions']}\e[0m"
+        else
+          raise "TODO: Unhandled instructions scenario: #{response}"
+        end
+      end
+
+      if action == 'post'
+        @config = {}
+        @config[:key] = @api_key
+        @config[:account] = response["account"]
+        @config[:venue] = response["venues"][0]
+        @config[:symbol] = response["tickers"][0]
+
+        @instance_id = response["instanceId"]        
+      end
+
+      response
     end
-    private :update_config
+ 
+    private :perform_request
   end
 end

data/lib/stockfighter/version.rb

@@ -1,3 +1,3 @@
 module Stockfighter
-  VERSION = "0.1.4"
+  VERSION = "0.2.0"
 end

data/stockfighter.gemspec

@@ -19,6 +19,7 @@ Gem::Specification.new do |spec|
   spec.require_paths = ["lib"]
 
   spec.add_runtime_dependency "httparty", "~> 0.13.7"
+  spec.add_runtime_dependency "rufus-scheduler", "~> 3.1.10"
   spec.add_development_dependency "bundler", "~> 1.7"
   spec.add_development_dependency "rake", "~> 10.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stockfighter
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.2.0
 platform: ruby
 authors:
 - Robert J Samson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-12-16 00:00:00.000000000 Z
+date: 2015-12-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -24,6 +24,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.13.7
+- !ruby/object:Gem::Dependency
+  name: rufus-scheduler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.1.10
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.1.10
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement