mtncloud-cli 1.0.0

data/lib/mtncloud/cli/commands/execution_request_command.rb

@@ -0,0 +1,333 @@
+require 'mtncloud/cli/cli_command'
+
+class MtnCloud::Cli::ExecutionRequestCommand
+  include MtnCloud::Cli::CliCommand
+  # include MtnCloud::Cli::InfrastructureHelper
+  # include MtnCloud::Cli::ProvisioningHelper
+
+  set_command_name :'execution-request'
+
+  register_subcommands :get, :execute
+  #register_subcommands :'execute-against-lease' => :execute_against_lease
+  
+  # set_default_subcommand :list
+
+  
+  def initialize()
+    # @appliance_name, @appliance_url = MtnCloud::Cli::Remote.active_appliance
+  end
+
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    # @instances_interface = @api_client.instances
+    # @containers_interface = @api_client.containers
+    # @servers_interface = @api_client.servers
+    @execution_request_interface = @api_client.execution_request
+  end
+
+  def handle(args)
+    handle_subcommand(args)
+  end
+
+  def default_refresh_interval
+    5
+  end
+
+  def get(args)
+    raw_args = args
+    options = {}
+    params = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[uid]")
+      build_common_options(opts, options, [:query, :json, :yaml, :csv, :fields, :dry_run, :remote])
+      opts.on('--refresh [SECONDS]', String, "Refresh until execution is finished. Default interval is #{default_refresh_interval} seconds.") do |val|
+        options[:refresh_until_finished] = true
+        if !val.to_s.empty?
+          options[:refresh_interval] = val.to_f
+        end
+      end
+      opts.footer = "Get details about an execution request." + "\n" +
+                    "[uid] is required. This is the unique id of an execution request."
+    end
+    optparse.parse!(args)
+    connect(options)
+    if args.count != 1
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error  "wrong number of arguments, expected 1 and got (#{args.count}) #{args.inspect}\n#{optparse}"
+      return 1
+    end
+    execution_request_id = args[0]
+    begin
+      params.merge!(parse_list_options(options))
+      @execution_request_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @execution_request_interface.dry.get(execution_request_id, params)
+        return
+      end
+      json_response = @execution_request_interface.get(execution_request_id, params)
+      if options[:json]
+        puts as_json(json_response, options, "executionRequest")
+        return 0
+      elsif options[:yaml]
+        puts as_yaml(json_response, options, "executionRequest")
+        return 0
+      elsif options[:csv]
+        puts records_as_csv([json_response['executionRequest']], options)
+        return 0
+      end
+
+      execution_request = json_response['executionRequest']
+
+      # refresh until a status is reached
+      if options[:refresh_until_finished]
+        if options[:refresh_interval].nil? || options[:refresh_interval].to_f < 0
+          options[:refresh_interval] = default_refresh_interval
+        end
+        if is_finished(execution_request)
+          # it is finished
+        else
+          print cyan
+          refresh_display_seconds = options[:refresh_interval] % 1.0 == 0 ? options[:refresh_interval].to_i : options[:refresh_interval]
+          print "Execution request has not yet finished. Refreshing every #{refresh_display_seconds} seconds"
+          while !is_finished(execution_request) do
+            sleep(options[:refresh_interval])
+            print cyan,".",reset
+            json_response = @execution_request_interface.get(execution_request_id, params)
+            execution_request = json_response['executionRequest']
+          end
+          #sleep_with_dots(options[:refresh_interval])
+          print "\n", reset
+          # get(raw_args)
+        end
+      end
+
+      print_h1 "Execution Request Details"
+      print cyan
+      description_cols = {
+        #"ID" => lambda {|it| it['id'] },
+        "Unique ID" => lambda {|it| it['uniqueId'] },
+        "Server ID" => lambda {|it| it['serverId'] },
+        "Instance ID" => lambda {|it| it['instanceId'] },
+        "Container ID" => lambda {|it| it['containerId'] },
+        "Expires At" => lambda {|it| format_local_dt it['expiresAt'] },
+        "Exit Code" => lambda {|it| it['exitCode'] },
+        "Status" => lambda {|it| format_execution_request_status(it) },
+        #"Created By" => lambda {|it| it['createdById'] },
+        #"Subdomain" => lambda {|it| it['subdomain'] },
+      }
+      description_cols.delete("Server ID") if execution_request['serverId'].nil?
+      description_cols.delete("Instance ID") if execution_request['instanceId'].nil?
+      description_cols.delete("Container ID") if execution_request['containerId'].nil?
+      description_cols.delete("Exit Code") if execution_request['exitCode'].nil?
+      print_description_list(description_cols, execution_request)      
+
+      if execution_request['stdErr'].to_s.strip != '' && execution_request['stdErr'] != "stdin: is not a tty\n"
+        print_h2 "Error"
+        puts execution_request['stdErr'].to_s.strip
+      end
+      if execution_request['stdOut']
+        print_h2 "Output"
+        puts execution_request['stdOut'].to_s.strip
+      end
+      print reset, "\n"
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      return 1
+    end
+  end
+
+  def execute(args)
+    options = {}
+    params = {}
+    script_content = nil
+    send_keys = nil
+    options[:refresh_until_finished] = true
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[options]")
+      opts.on('--server ID', String, "Server ID") do |val|
+        params['serverId'] = val
+      end
+      opts.on('--instance ID', String, "Instance ID") do |val|
+        params['instanceId'] = val
+      end
+      opts.on('--container ID', String, "Container ID") do |val|
+        params['containerId'] = val
+      end
+      opts.on('--request ID', String, "Execution Request ID") do |val|
+        params['requestId'] = val
+      end
+      opts.on('--script SCRIPT', "Script to be executed" ) do |val|
+        script_content = val
+      end
+      opts.on('--optionTypes [true|false]', String, "Include optionTypes in the response. Default is false.") do |val|
+        params['optionTypes'] = (val.to_s == '' || val.to_s == 'on' || val.to_s == 'true')
+      end
+      opts.on('--send-keys [true|false]', "Send key mappings to the hypervisor console so commands such as <LEFT>, <RIGHT> and <WAIT> can be used." ) do |val|
+        send_keys = val.to_s == 'on' || val.to_s == 'true' || val.to_s.empty?
+      end
+      opts.on('--refresh [SECONDS]', String, "Refresh until execution is finished. Default interval is #{default_refresh_interval} seconds.") do |val|
+        options[:refresh_until_finished] = true
+        if !val.to_s.empty?
+          options[:refresh_interval] = val.to_f
+        end
+      end
+      opts.on(nil, '--no-refresh', "Do not refresh. The default behavior is to refresh until finished." ) do
+        options[:refresh_until_finished] = false
+      end
+      #build_option_type_options(opts, options, add_user_source_option_types())
+      build_common_options(opts, options, [:options, :payload, :json, :dry_run, :quiet, :remote])
+      opts.footer = "Execute an arbitrary script." + "\n" +
+                    "[server] or [instance] or [container] is required. This is the id of a server, instance or container." + "\n" +
+                    "[script] is required. This is the script that is to be executed."
+    end
+    optparse.parse!(args)
+    connect(options)
+    if args.count != 0
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error  "wrong number of arguments, expected 0 and got (#{args.count}) #{args.inspect}\n#{optparse}"
+      return 1
+    end
+    if params['serverId'].nil? && params['instanceId'].nil? && params['containerId'].nil? && params['requestId'].nil?
+      puts_error "#{MtnCloud::Terminal.angry_prompt}missing required option: --server or --instance or --container\n#{optparse}"
+      return 1
+    end
+    begin
+      # construct payload
+      payload = {}
+      if options[:payload]
+        payload = options[:payload]
+      else
+        payload.deep_merge!(options[:options].reject {|k,v| k.is_a?(Symbol) }) if options[:options]
+        # could prompt for Server or Container or Instance
+        # prompt for Script
+        if script_content.nil?
+          v_prompt = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'script', 'type' => 'code-editor', 'fieldLabel' => 'Script', 'required' => true, 'description' => 'The script content'}], options[:options])
+          script_content = v_prompt['script']
+        end
+        payload['script'] = script_content
+        if !send_keys.nil?
+          payload['sendKeys'] = send_keys
+        end
+      end
+      @execution_request_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @execution_request_interface.dry.create(params, payload)
+        return 0
+      end
+      # do it
+      json_response = @execution_request_interface.create(params, payload)
+      # print and return result
+      if options[:quiet]
+        return 0
+      elsif options[:json]
+        puts as_json(json_response, options)
+        return 0
+      end
+      execution_request = json_response['executionRequest']
+      print_green_success "Executing request #{execution_request['uniqueId']}"
+      if options[:refresh_until_finished]
+        get([execution_request['uniqueId'], "--refresh", options[:refresh_interval] ? options[:refresh_interval].to_s : nil].compact + (options[:remote] ? ["-r",options[:remote]] : []))
+      else
+        get([execution_request['uniqueId']] + (options[:remote] ? ["-r",options[:remote]] : []))
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def execute_against_lease(args)
+    options = {}
+    params = {}
+    do_refresh = true
+    script_content = nil
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[uid] [options]")
+      opts.on('--script SCRIPT', "Script to be executed" ) do |val|
+        script_content = val
+      end
+      opts.on('--file FILE', "File containing the script. This can be used instead of --script" ) do |filename|
+        full_filename = File.expand_path(filename)
+        if File.exist?(full_filename)
+          script_content = File.read(full_filename)
+        else
+          print_red_alert "File not found: #{full_filename}"
+          exit 1
+        end
+      end
+      opts.on(nil, '--no-refresh', "Do not refresh until finished" ) do
+        do_refresh = false
+      end
+      #build_option_type_options(opts, options, add_user_source_option_types())
+      build_common_options(opts, options, [:options, :payload, :json, :dry_run, :quiet, :remote])
+      opts.footer = "Execute request against lease.\n" +
+                    "[uid] is required. This is the unique id of the execution request.\n" +
+                    "[script] is required. This is the script that is to be executed."
+    end
+    optparse.parse!(args)
+    connect(options)
+    if args.count != 1
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error  "wrong number of arguments, expected 1 and got (#{args.count}) #{args.inspect}\n#{optparse}"
+      return 1
+    end
+    execution_request_id = args[0]
+    begin
+      # construct payload
+      payload = {}
+      if options[:payload]
+        payload = options[:payload]
+      else
+        payload.deep_merge!(options[:options].reject {|k,v| k.is_a?(Symbol) }) if options[:options]
+        if script_content
+          payload['script'] = script_content
+        end
+      end
+      @execution_request_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @execution_request_interface.dry.execute_against_lease(execution_request_id, params, payload)
+        return 0
+      end
+      # do it
+      json_response = @execution_request_interface.execute_against_lease(execution_request_id, params, payload)
+      # print and return result
+      if options[:quiet]
+        return 0
+      elsif options[:json]
+        puts as_json(json_response, options)
+        return 0
+      end
+      execution_request = json_response['executionRequest']
+      print_green_success "Executing request #{execution_request['uniqueId']} against lease"
+      if do_refresh
+        get([execution_request['uniqueId'], "--refresh"] + (options[:remote] ? ["-r",options[:remote]] : []))
+      else
+        get([execution_request['uniqueId']] + (options[:remote] ? ["-r",options[:remote]] : []))
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def format_execution_request_status(execution_request, return_color=cyan)
+    out = ""
+    status_str = execution_request['status']
+    if status_str == 'complete'
+      out << "#{green}#{status_str.upcase}#{return_color}"
+    elsif status_str == 'failed' || status_str == 'expired'
+      out << "#{red}#{status_str.upcase}#{return_color}"
+    else
+      out << "#{cyan}#{status_str.upcase}#{return_color}"
+    end
+    out
+  end
+
+  def is_finished(execution_request)
+    ['executing', 'pending', 'new'].include?(execution_request['status']) ? false : true
+  end
+
+end

data/lib/mtncloud/cli/commands/exit_command.rb

@@ -0,0 +1,34 @@
+require 'mtncloud/cli/cli_command'
+
+# Utility command for exiting a shell
+class MtnCloud::Cli::ExitCommand
+  include MtnCloud::Cli::CliCommand
+  
+  set_command_name :'exit'
+  
+  set_command_hidden
+
+  def handle(args)
+    exit_code = 0
+    options = {}
+    filename = MtnCloud::Cli::DotFile.mtncloud_profile_filename
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: exit [code]"
+      build_common_options(opts, options, [])
+      opts.footer = "Exit #{prog_name}. Default is 0"
+    end
+    optparse.parse!(args)
+    
+    if args[0]
+      exit_code = args[0].to_i
+    end
+
+    if options[:debug]
+      puts "#{dark}exit #{exit_code}#{reset}"
+    end
+
+    exit exit_code
+
+  end
+
+end