mtncloud-cli 1.0.0

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

@@ -0,0 +1,282 @@
+require 'mtncloud/cli/cli_command'
+
+class MtnCloud::Cli::DistributedWorkers
+  include MtnCloud::Cli::CliCommand
+
+  register_subcommands :list, :get, :add, :update, :remove
+
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @distributed_workers_interface = @api_client.distributed_workers
+  end
+
+  def handle(args)
+    handle_subcommand(args)
+  end
+
+  def list(args)
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage()
+      build_standard_list_options(opts, options)
+      opts.footer = "List distributed workers."
+    end
+    optparse.parse!(args)
+    connect(options)
+    verify_args!(args:args, optparse:optparse, count:0)
+    params = parse_list_options(options)
+    @distributed_workers_interface.setopts(options)
+    if options[:dry_run]
+      print_dry_run @distributed_workers_interface.dry.list(params)
+      return
+    end
+    json_response = @distributed_workers_interface.list(params)
+    render_response(json_response, options, 'distributedWorkers') do
+      distributed_workers = json_response['distributedWorkers']
+      print_h1 "Distributed Workers", parse_list_subtitles(options), options
+      if distributed_workers.empty?
+        print cyan,"No distributed workers found.",reset,"\n"
+      else
+        rows = distributed_workers.collect do |distributed_worker|
+          {
+            id: distributed_worker['id'],
+            name: distributed_worker['name'],
+            enabled: format_boolean(distributed_worker['enabled']),
+            proxyPort: distributed_worker['proxyPort'],
+            proxyHostList: distributed_worker['proxyHostList'],
+            applianceUrl: distributed_worker['applianceUrl']
+          }
+        end
+        columns = [:id, :name, :enabled, {:proxyPort => {:display_name => "Proxy Port"} }, {:proxyHostList => {:display_name => "Proxy Host List"} }, {:applianceUrl => {:display_name => "Appliance URL"} }]
+        print as_pretty_table(rows, columns, options)
+        print_results_pagination(json_response)
+      end
+      print reset,"\n"
+    end
+    return 0, nil
+  end
+
+  def get(args)
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[distributed-worker]")
+      build_standard_get_options(opts, options)
+      opts.footer = "Get details about a distributed worker.\n[distributed-worker] is required. This is the name or id of a distributed worker."
+    end
+    optparse.parse!(args)
+    verify_args!(args:args, optparse:optparse, count:1)
+    connect(options)
+    _get(args[0], options)
+  end
+
+  def _get(id, options)
+    @distributed_workers_interface.setopts(options)
+    if options[:dry_run]
+      if id.to_s =~ /\A\d{1,}\Z/
+        print_dry_run @distributed_workers_interface.dry.get(id.to_i)
+      else
+        print_dry_run @distributed_workers_interface.dry.list({name: id})
+      end
+      return
+    end
+    distributed_worker = find_distributed_worker_by_name_or_id(id)
+    return 1 if distributed_worker.nil?
+    json_response = {'distributedWorker' => distributed_worker}
+    render_response(json_response, options, 'distributedWorker') do
+      render_distributed_worker_details(distributed_worker)
+    end
+    return 0, nil
+  end
+
+  def add(args)
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[name] [options]")
+      build_option_type_options(opts, options, add_distributed_worker_option_types)
+      build_standard_add_options(opts, options)
+      opts.footer = <<-EOT
+Add a distributed worker.
+[name] is required. This is the name of the new distributed worker.
+The generated apiKey is only shown once, on creation. Be sure to save it.
+      EOT
+    end
+    optparse.parse!(args)
+    verify_args!(args:args, optparse:optparse, max:1)
+    connect(options)
+    options[:options] ||= {}
+    if args[0]
+      options[:options]['name'] ||= args[0]
+    end
+    payload = parse_payload(options, 'distributedWorker')
+    if payload.nil?
+      params = MtnCloud::Cli::OptionTypes.prompt(add_distributed_worker_option_types, options[:options], @api_client, options[:params])
+      payload = {'distributedWorker' => params}
+    end
+    @distributed_workers_interface.setopts(options)
+    if options[:dry_run]
+      print_dry_run @distributed_workers_interface.dry.create(payload)
+      return
+    end
+    json_response = @distributed_workers_interface.create(payload)
+    render_response(json_response, options, 'distributedWorker') do
+      distributed_worker = json_response['distributedWorker']
+      print_green_success "Added distributed worker #{distributed_worker['name']}"
+      api_key = distributed_worker['apiKey']
+      if api_key
+        print "\n"
+        print_h2 "API Key"
+        print yellow, "Save this API Key now. It will not be shown again.", reset, "\n"
+        print cyan, api_key, reset, "\n"
+      end
+      render_distributed_worker_details(distributed_worker)
+    end
+    return 0, nil
+  end
+
+  def update(args)
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[distributed-worker] [options]")
+      build_option_type_options(opts, options, update_distributed_worker_option_types)
+      build_standard_update_options(opts, options)
+      opts.footer = <<-EOT
+Update a distributed worker.
+[distributed-worker] is required. This is the name or id of a distributed worker.
+      EOT
+    end
+    optparse.parse!(args)
+    verify_args!(args:args, optparse:optparse, count:1)
+    connect(options)
+    distributed_worker = find_distributed_worker_by_name_or_id(args[0])
+    return 1 if distributed_worker.nil?
+    payload = parse_payload(options, 'distributedWorker')
+    if payload.nil?
+      params = options[:options] || {}
+      params = params.select {|k,v| params[k].to_s != "" }
+      if params.empty?
+        raise_command_error "Specify at least one option to update.\n#{optparse}"
+      end
+      payload = {'distributedWorker' => params}
+    end
+    payload['distributedWorker']['id'] = distributed_worker['id']
+    @distributed_workers_interface.setopts(options)
+    if options[:dry_run]
+      print_dry_run @distributed_workers_interface.dry.update(distributed_worker['id'], payload)
+      return
+    end
+    json_response = @distributed_workers_interface.update(distributed_worker['id'], payload)
+    render_response(json_response, options, 'distributedWorker') do
+      print_green_success "Updated distributed worker #{distributed_worker['name']}"
+      _get(distributed_worker['id'], options)
+    end
+    return 0, nil
+  end
+
+  def remove(args)
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = subcommand_usage("[distributed-worker]")
+      build_standard_remove_options(opts, options)
+      opts.footer = <<-EOT
+Delete a distributed worker.
+[distributed-worker] is required. This is the name or id of a distributed worker.
+      EOT
+    end
+    optparse.parse!(args)
+    verify_args!(args:args, optparse:optparse, count:1)
+    connect(options)
+    distributed_worker = find_distributed_worker_by_name_or_id(args[0])
+    return 1 if distributed_worker.nil?
+    unless options[:yes] || MtnCloud::Cli::OptionTypes.confirm("Are you sure you want to delete the distributed worker #{distributed_worker['name']}?", options)
+      return 9, "aborted command"
+    end
+    @distributed_workers_interface.setopts(options)
+    if options[:dry_run]
+      print_dry_run @distributed_workers_interface.dry.destroy(distributed_worker['id'])
+      return
+    end
+    json_response = @distributed_workers_interface.destroy(distributed_worker['id'])
+    render_response(json_response, options, 'distributedWorker') do
+      print_green_success "Removed distributed worker #{distributed_worker['name']}"
+    end
+    return 0, nil
+  end
+
+  private
+
+  def render_distributed_worker_details(distributed_worker)
+    print_h1 "Distributed Worker Details"
+    print cyan
+    description_cols = {
+      "ID" => 'id',
+      "Name" => 'name',
+      "Description" => 'description',
+      "Enabled" => lambda {|it| format_boolean(it['enabled']) },
+      "Active" => lambda {|it| format_boolean(it['active']) },
+      "Proxy Port" => 'proxyPort',
+      "Proxy Host List" => 'proxyHostList',
+      "Appliance URL" => 'applianceUrl',
+      "Created" => lambda {|it| format_local_dt(it['dateCreated']) },
+      "Updated" => lambda {|it| format_local_dt(it['lastUpdated']) }
+    }
+    print_description_list(description_cols, distributed_worker)
+    print reset,"\n"
+  end
+
+  def find_distributed_worker_by_name_or_id(val)
+    if val.to_s =~ /\A\d{1,}\Z/
+      return find_distributed_worker_by_id(val)
+    else
+      return find_distributed_worker_by_name(val)
+    end
+  end
+
+  def find_distributed_worker_by_id(id)
+    begin
+      json_response = @distributed_workers_interface.get(id.to_i)
+      return json_response['distributedWorker']
+    rescue RestClient::Exception => e
+      if e.response && e.response.code == 404
+        print_red_alert "Distributed Worker not found by id #{id}"
+        return nil
+      else
+        raise e
+      end
+    end
+  end
+
+  def find_distributed_worker_by_name(name)
+    distributed_workers = @distributed_workers_interface.list({name: name.to_s})['distributedWorkers']
+    if distributed_workers.empty?
+      print_red_alert "Distributed Worker not found by name #{name}"
+      return nil
+    elsif distributed_workers.size > 1
+      print_red_alert "#{distributed_workers.size} distributed workers found by name #{name}"
+      rows = distributed_workers.collect do |it|
+        {id: it['id'], name: it['name']}
+      end
+      print "\n"
+      puts as_pretty_table(rows, [:id, :name], {color: red})
+      return nil
+    else
+      return distributed_workers[0]
+    end
+  end
+
+  def add_distributed_worker_option_types
+    [
+      {'fieldName' => 'name', 'fieldLabel' => 'Name', 'type' => 'text', 'required' => true, 'displayOrder' => 1, 'description' => 'A unique name for the distributed worker.'},
+      {'fieldName' => 'description', 'fieldLabel' => 'Description', 'type' => 'text', 'required' => false, 'displayOrder' => 2},
+      {'fieldName' => 'enabled', 'fieldLabel' => 'Enabled', 'type' => 'checkbox', 'required' => false, 'defaultValue' => true, 'displayOrder' => 3},
+      {'fieldName' => 'proxyHostList', 'fieldLabel' => 'Proxy Host List', 'type' => 'text', 'required' => false, 'displayOrder' => 4, 'description' => 'A comma or space separated list of proxy hosts.'},
+      {'fieldName' => 'applianceUrl', 'fieldLabel' => 'Appliance URL', 'type' => 'text', 'required' => false, 'displayOrder' => 5},
+    ]
+  end
+
+  def update_distributed_worker_option_types
+    list = add_distributed_worker_option_types
+    list.each {|it| it['required'] = false; it.delete('defaultValue') }
+    list
+  end
+
+end

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

@@ -0,0 +1,61 @@
+require 'mtncloud/cli/cli_command'
+
+# This is for use in dotfile scripts for printing
+# It is also responsible for maintaining a map of variables
+# that are also used in custom shell prompts.
+class MtnCloud::Cli::Echo
+  include MtnCloud::Cli::CliCommand
+  set_command_name :echo
+  set_command_hidden
+
+  unless defined?(COLOR_VARIABLE_MAP)
+    COLOR_VARIABLE_MAP = {'%cyan' => Term::ANSIColor.cyan, '%magenta' => Term::ANSIColor.magenta, '%red' => Term::ANSIColor.red, '%green' => Term::ANSIColor.green, '%yellow' => Term::ANSIColor.yellow, '%white' => Term::ANSIColor.white, '%dark' => Term::ANSIColor.dark, '%reset' => Term::ANSIColor.reset}
+  end
+
+  def self.variable_map
+    @output_variable_map ||= recalculate_variable_map()
+  end
+
+  def self.recalculate_variable_map()
+    var_map = {}
+    if Term::ANSIColor.coloring?
+      var_map.merge!(COLOR_VARIABLE_MAP)
+    else
+      COLOR_VARIABLE_MAP.each {|k,v| var_map[k] = "" }
+    end
+
+    appliance = ::MtnCloud::Cli::Remote.load_active_remote()
+    if appliance
+      var_map.merge!({'%remote' => appliance[:name], '%remote_url' => (appliance[:host].to_s || appliance[:url].to_s), '%username' => appliance[:username].to_s})
+    else
+      var_map.merge!({'%remote' => '', '%remote_url' => '', '%username' => ''})
+    end
+    @output_variable_map = var_map
+  end
+
+  def handle(args)
+    append_newline = true
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: #{prog_name} #{command_name} [<message>]"
+      opts.on( '-n', '--nonewline', "do not append a newline to your words" ) do
+        append_newline = false
+      end
+      build_common_options(opts, options, [])
+    end
+    optparse.parse!(args)
+    out = ""
+    out << args.join(' ')
+
+    self.class.variable_map.each do |k, v|
+      out.gsub!(k.to_s, v.to_s)
+    end
+    if append_newline
+      out << "\n"
+    end
+    # print out 
+    print cyan + out + reset
+    return 0
+  end
+
+end

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

@@ -0,0 +1,57 @@
+require 'mtncloud/cli/cli_command'
+
+# Command for editing the .mtncloud_profile initialization script
+class MtnCloud::Cli::EditProfileCommand
+  include MtnCloud::Cli::CliCommand
+  set_command_name :'edit-profile'
+  #set_command_hidden
+
+  def handle(args)
+    options = {}
+    editor = ENV['EDITOR'] || 'nano'
+    filename = MtnCloud::Cli::DotFile.mtncloud_profile_filename
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: #{prog_name} #{command_name}"
+      opts.on( '-e', '--editor PROGRAM', "Editor program to use. The default is $EDITOR." ) do |val|
+        editor = val
+      end
+      build_common_options(opts, options, [])
+      opts.footer = <<-EOT
+Edit your .mtncloud_profile script located in the mtncloud home directory.
+This script runs whenever the #{prog_name} terminal command is executed.
+It provides a way to initialize your cli environment for all #{prog_name} commands.
+
+Example:
+
+# disable coloring to exclude ansi characters in output
+coloring off -q
+
+# Enable debugging to print extra output for troubleshooting
+debug on
+
+EOT
+    end
+    optparse.parse!(args)
+
+    if !editor
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error "You have not defined an EDITOR."
+      puts_error "Try export EDITOR=emacs"
+      #puts "Trying nano..."
+      #editor = "nano"
+      return 1
+    end
+    puts "opening #{filename} for editing"
+    system(editor, filename)
+    if !$?.success?
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error "edit command failed with #{$?.exitstatus}: #{editor} #{filename}"
+      return $?.exitstatus
+    end
+    if MtnCloud::Cli::Shell.has_instance?
+      puts "use 'reload' to re-execute your startup script"
+    end
+    return 0 # $?
+  end
+
+end

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

@@ -0,0 +1,77 @@
+require 'mtncloud/cli/cli_command'
+
+# Command for editing the .mtncloud_profile initalization script
+class MtnCloud::Cli::EditRcCommand
+  include MtnCloud::Cli::CliCommand
+  set_command_name :'edit-rc'
+  #set_command_hidden
+
+  def handle(args)
+    options = {}
+    editor = nil
+    filename = MtnCloud::Cli::DotFile.mtncloudrc_filename
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: #{prog_name} #{command_name}"
+      opts.on( '-e', '--editor PROGRAM', "Editor program to use. The default is $EDITOR." ) do |val|
+        editor = val.gsub("'",'')
+      end
+      build_common_options(opts, options, [])
+      opts.footer = <<-EOT
+Edit your .mtncloudrc script located in the mtncloud home directory.
+This script runs at the beginning of the interactive #{prog_name} shell command.
+It provides a way to initialize your interactive #{prog_name} shell environment.
+
+Example:
+
+# customize the mtncloud shell prompt
+set-prompt "%cyan%username%reset@%magenta%remote %cyanmtncloud> %reset"
+
+# print the mtncloud cli version
+version
+
+# print the current remote appliance version
+remote get current
+
+# greetings
+echo "Welcome back %username"
+EOT
+    end
+    optparse.parse!(args)
+    
+
+    if !editor
+      editor = ENV['EDITOR']
+    end
+
+    # try something...
+    if !editor
+      if MtnCloud::Cli.windows?
+        editor = "notepad"
+      else
+        editor = "nano"
+      end
+    end
+
+    if !editor
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error "You have not defined an EDITOR."
+      puts_error "Try export EDITOR=emacs"
+      #puts "Trying nano..."
+      #editor = "nano"
+      return 1
+    end
+    has_editor = true
+    puts "opening #{filename} for editing"
+    system(editor, filename)
+    if !$?.success?
+      print_error MtnCloud::Terminal.angry_prompt
+      puts_error "edit command failed with #{$?.exitstatus}: #{editor} #{filename}"
+      return $?.exitstatus
+    end
+    if MtnCloud::Cli::Shell.has_instance?
+      puts "use 'reload' to re-execute your startup script"
+    end
+    return 0 # $?
+  end
+
+end