zillabyte-cli 0.1.0 → 0.1.1

data/lib/zillabyte/cli/executes.rb

@@ -1,180 +0,0 @@
-require "zillabyte/cli/base"
-require "zillabyte/cli/config"
-require "zillabyte/common"
-require "pty"
-require 'indentation'
-require 'open3'
-require 'securerandom'
-require 'colorize'
-require 'time_difference'
-
-
-# executes arbitrary input against apps
-#
-# HIDDEN
-class Zillabyte::Command::Executes < Zillabyte::Command::Base
-
-  POLL_DELAY = 3
-
-
-  # executes:exec ID INPUT_PARAMS [INPUT_PARAMS]
-  #
-  # executes an app against with the given parameters. INPUT_PARAMS
-  # is expected to be in JSON format. 
-  # 
-  # --json            # return the results as JSON
-  # --async           # returns immediately.  Use executes:results to fetch results
-  # --select FIELDS   # returns only the fields in this list. useful for scripting
-  #
-  # Examples: 
-  #
-  # $ zillabyte exec "web_screenshot", "{url: 'google.com'}"
-  # $ zillabyte exec "web_screenshot", "{url: 'google.com'}", "{url: 'facebook.com'}" 
-  def exec
-    
-    # INIT 
-    app_id = options[:id] || shift_argument
-    async = options[:async] || false
-    select_field = options[:select]
-    
-    # Get the inputs
-    input_params = []
-    while ((ip = shift_argument()) != nil)
-      input_params << JSON.parse(ip)
-    end
-    
-    # Sanity
-    error("At least one input parameter is required") if input_params.size == 0
-    error("Unable to execute multiple requests synchronously. Add the --async flag or call with only one input_param") if input_params.size > 1 and !async
-    error("The --select option and --async cannot be used at the same time") if async and select_field
-    
-    # Work with concrete ids or names
-    if !(app_id =~ /^\d*$/)
-      options[:is_name] = true
-    end
-    
-    # Build the hash
-    options[:input] = input_params
-    
-    # Make the request
-    res = api.request(
-      :expects => 200, 
-      :method => :post,
-      :body    => options.to_json,
-      :path => "/flows/#{CGI.escape(app_id)}/executes"
-    ).body
-    
-    
-    if async    
-      
-      # Async. Return right away... 
-      if res['status'] == 'success'
-        if options[:display_type] == :json
-          display res.to_s
-        else
-          # TODO: 
-          display res.to_s
-        end
-      else
-        error res['error_message']
-      end
-      
-    else
-      
-      # Synchronous... poll until done.. 
-      execute_ids = res['execute_ids']
-      while (true)
-        
-        res = get_results(app_id, execute_ids)
-        if res['status'] == 'success'
-          
-          execute_ids.each do |execute_id|
-            h = res['results'][execute_id]
-            if h['status'] == 'complete'
-              if select_field
-                display(h['data'][select_field])
-              else
-                display(h.to_json)
-              end
-              execute_ids.delete(execute_id)
-            end
-          end
-          
-        else
-          error(res.to_json)
-        end
-        
-        if execute_ids.size == 0
-          # success. done
-          return
-        else
-          # still more. keep polling
-          sleep(POLL_DELAY)
-        end
-        
-      end
-      
-    end
-    
-  end
-  alias_command "exec", "executes:exec"
-  alias_command "execute", "executes:exec"
-  
-  
-  
-  
-
-  # executes:results ID EXECUTE_ID [EXECUTE_ID]
-  #
-  # fetches the results associated with the given execute_id.
-  # 
-  # --json      # return the results as JSON
-  #
-  def results
-    
-    # INIT 
-    app_id = options[:id] || shift_argument
-    execute_ids = []
-    while ((eid = shift_argument()) != nil)
-      execute_ids << eid
-    end
-    
-    # Sanity
-    error("at least one execute id parameter is required") if execute_ids.size == 0
-    
-    # Work with concrete ids or names
-    if !(app_id =~ /^\d*$/)
-      options[:is_name] = true
-    end
-    
-    # Fetch the results... 
-    display( get_results(app_id, execute_ids).to_json )
-    
-    
-    
-  end
-  alias_command "results", "executes:results"
-  
-  
-  
-  private 
-  
-  def get_results(app_id, execute_ids)
-    
-    hash = {
-      :execute_ids => execute_ids
-    }
-    
-    res = api.request(
-      :expects => 200, 
-      :method => :get,
-      :body    => hash.to_json,
-      :path => "/flows/#{CGI.escape(app_id)}/executes"
-    ).body
-    
-    return res
-    
-  end
-  
-  
-end