zillabyte-cli 0.0.24 → 0.1.0

data/lib/zillabyte/cli/auth.rb

@@ -10,22 +10,17 @@ class Zillabyte::Command::Auth < Zillabyte::Command::Base
 
   # auth:login
   #
-  # sets the Zillabyte auth token
+  # set the authentication token
   # 
-  # --auth_token TOKEN # provide the token
   def login
-    if options[:auth_token]
-      Zillabyte::Auth.login( options[:auth_token])
-    else
-      Zillabyte::Auth.ask_for_and_save_credentials
-    end
+    Zillabyte::Auth.ask_for_and_save_credentials
   end
   
   alias_command "login", "auth:login"
   
   # auth:logout
   #
-  # sets the Zillabyte auth token
+  # clear the authentication token
   #
   def logout
     Zillabyte::Auth.logout

data/lib/zillabyte/cli/base.rb

@@ -3,8 +3,8 @@ require "zillabyte/api"
 require "zillabyte/command"
 require "action_view"
 
-class Zillabyte::Command::Base
-  include Zillabyte::Helpers
+class Zillabyte::Command::Base 
+  include Zillabyte::Helpers 
   include ActionView::Helpers::DateHelper
 
   def self.namespace
@@ -19,8 +19,12 @@ class Zillabyte::Command::Base
     @options = options
   end
 
+  def api
+    @__api ||= Zillabyte::API.new(:api_key => Zillabyte::Auth.api_key, :session => self)
+  end  
+  
 
-protected
+  protected
 
   def self.inherited(klass)
     unless klass == Zillabyte::Command::Base
@@ -140,10 +144,27 @@ protected
     Zillabyte::Command.validate_arguments!
   end
   
-  def api
-    @__api ||= Zillabyte::API.new(:api_key => Zillabyte::Auth.api_key, :session => self)
-  end  
-  
+
+  def git_remotes(base_dir=Dir.pwd)
+    remotes = {}
+    original_dir = Dir.pwd
+    Dir.chdir(base_dir)
+
+    return unless File.exists?(".git")
+    git("remote -v").split("\n").each do |remote|
+      name, url, method = remote.split(/\s/)
+      if url =~ /^git@#{Zillabyte::Auth.git_host}(?:[\.\w]*):([\w\d-]+)\.git$/
+        remotes[name] = $1
+      end
+    end
+
+    Dir.chdir(original_dir)
+    if remotes.empty?
+      nil
+    else
+      remotes
+    end
+  end
   
   
   def session

data/lib/zillabyte/cli/components.rb

@@ -0,0 +1,245 @@
+require 'yaml'
+require "zillabyte/cli/flows"
+require "zillabyte/cli/config"
+require "zillabyte/common"
+
+# manage custom components
+#
+class Zillabyte::Command::Components < Zillabyte::Command::Flows
+
+  # components
+  #
+  # list custom components
+  # --output_type OUTPUT_TYPE     # specify an output type i.e. json
+  #
+  def index
+    self.list
+  end
+  
+
+
+  # components
+  #
+  # list custom components
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  def list
+    type = options[:output_type]
+    
+    headings = ["id", "name", "state", "inputs", "outputs"]
+    rows = api.component.list.map do |row|
+      if headings.size == 0
+        headings = row.keys 
+        headings.delete("rel_dir")
+      end
+      
+      v = row.values_at *headings
+      v
+    end
+
+    display "components:\n" if type.nil?
+    display TableOutputBuilder.build_table(headings, rows, type)
+    display "Total number of components: " + rows.length.to_s if type.nil?
+  end
+
+  
+  # components:pull ID DIR
+  #
+  # pulls a component source to a directory.
+  #
+  # --force # pulls even if the directory exists
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  # --directory DIR # Directory of the component
+  #
+  # Examples:
+  #
+  # $ zillabyte components:pull . 
+  #
+  def pull
+    super
+  end
+
+  # components:status [DIR]
+  # 
+  # fetches detailed status of the component
+  # 
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  # --directory DIR # Directory of the component
+  #
+  def status
+    super
+  end
+
+
+
+  # components:delete ID
+  #
+  # deletes a component.
+  #
+  # -f, --force # don't ask for confirmation
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  #
+  def delete
+    super
+  end
+
+  # components:kill ID
+  #
+  # kills the given component rpc
+  # 
+  # --config CONFIG_FILE # use the given config file
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  #
+  def kill
+    id = options[:id] || shift_argument
+    type = options[:output_type]
+
+    if id.nil?
+      id = read_name_from_conf(options)
+      options[:is_name] = true
+    elsif !(id =~ /^\d*$/)
+      options[:is_name] = true
+    end
+
+    display "Killing component ##{id}...please wait..." if type.nil?
+    api.components.kill(id, options)
+
+    if type == "json"
+      display "{}"
+    else
+      display "Component ##{id} killed"
+    end
+  end
+
+  # components:info [DIR]
+  # 
+  # outputs the info for the component in the dir.
+  # 
+  # --pretty # Pretty prints the info output
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  # --directory DIR # Directory of the component
+  #
+  def info
+    super
+  end
+
+
+  # components:test
+  #
+  # tests a local component with sample data
+  # 
+  # --config CONFIG_FILE            # use the given config file
+  # --input INPUT_FILE              # uses a CSV for component input
+  # --output OUTPUT_FILE            # write output to a CSV
+  # --directory DIR                 # app directory
+  #
+  def test
+    super
+  end  
+
+  # components:prep [DIR]
+  # 
+  # performs any necessary initialization for the component
+  # 
+  # --directory DIR       # component directory
+  # --output_type OUTPUT_TYPE           # specify an output type i.e. json
+  #  
+  def prep
+    super
+  end
+
+  # components:errors ID
+  #
+  # show recent errors generated by the componeny rpc
+  # --output_type OUTPUT_TYPE           # specify an output type i.e. json
+  #
+  def errors
+    super
+  end
+
+  # components:live_run [OPERATION_NAME] [PIPE_NAME] [DIR]
+  #
+  # runs a local component with live data
+  # 
+  # --config CONFIG_FILE # use the given config file
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  # --directory DIR # Directory of the component
+  #
+  # HIDDEN:
+  def live_run
+    super
+  end
+ 
+  # components:rpc [ID] [QUERY_1_INPUT_1,QUERY_1_INPUT_2,...] [QUERY_2_INPUT_1,QUERY_2_INPUT_2,...], ...
+  #
+  # runs a component as an rpc
+  #
+  # --output_type OUTPUT_TYPE # specify an output type i.e. json
+  def rpc
+    component_id = options[:id] || shift_argument
+
+    if component_id.nil?
+      component_id = read_name_from_conf(options)
+      options[:is_name] = true
+    elsif !(component_id =~ /^\d*$/)
+      options[:is_name] = true
+    end
+
+    type = options[:output_type]
+
+    component_args = []
+    while(true) do
+      next_arg = shift_argument
+      break if next_arg.nil?
+      component_args << next_arg.split(",")
+    end
+
+    res = api.components.rpc(component_id, {:rpc_inputs => component_args})
+    if res['error']
+      error("error: #{res['error_message']}", type)
+    else
+      if type.nil?
+        display "Request submitted to component ##{res['id']}. The run ids are (query: run_id):"
+        res["execute_ids"].each do |pars, id|
+          display "\t #{pars}: #{id}"
+        end
+        display "Please use \"zillabyte components:results [run id]\" to check on the status of your query and to retrieve your results.\nYou may also wish to check \"zillabyte logs\" for errors."
+      end
+    end
+  end
+
+  # components:results [ID] [RUN_ID]
+  #
+  # gets the result of an rpc request if it is done running, otherwise gets current run status
+  #
+  def results
+    component_id = options[:id] || shift_argument
+
+    if component_id.nil?
+      component_id = read_name_from_conf(options)
+      options[:is_name] = true
+    elsif !(component_id =~ /^\d*$/)
+      options[:is_name] = true
+    end
+
+    type = options[:output_type]
+
+    run_ids = []
+    while(true) do
+      next_arg = shift_argument
+      break if next_arg.nil?
+      run_ids << next_arg
+    end
+
+    res = api.components.get_rpc_results(component_id, {:execute_ids => run_ids})
+    if res['error']
+      error("error: #{res['error_message']}", type)
+    else
+      if type.nil?
+        res["results"].each do |id, hash|
+          display "#{id}: #{hash["status"]}"
+          display "\t #{hash["data"]}" if hash["status"] == "complete"
+        end
+      end
+    end
+  end
+end