morpheus-cli 2.10.1 → 2.10.2

data/lib/morpheus/cli/dot_file.rb

@@ -0,0 +1,207 @@
+require 'fileutils'
+require 'time'
+require 'morpheus/cli/cli_registry'
+require 'morpheus/logging'
+require 'term/ansicolor'
+
+class Morpheus::Cli::DotFile
+  include Term::ANSIColor
+
+  DEFAULT_EXEC_PROC = lambda {|args|
+
+  }
+  EXPORTED_ALIASES_HEADER = "# exported aliases"
+
+  # the path of the profile source file
+  # this is executed when `morpheus` is run
+  def self.morpheus_profile_filename
+    File.join(Morpheus::Cli.home_directory, ".morpheus_profile")
+  end
+
+  # the path of the shell source file
+  # this is executed when `morpheus shell` is run
+  def self.morpheusrc_filename
+    File.join(Morpheus::Cli.home_directory, ".morpheusrc")
+  end
+
+  attr_reader :filename
+  attr_reader :file_contents
+  attr_reader :commands
+  attr_reader :cmd_results
+
+  def initialize(fn)
+    @filename = fn
+    #load_file()
+  end
+
+  # execute this file as a morpheus shell script
+  # @param stop_on_failure [true, false] the will halt execution if a command returns false. 
+  #   Default is false, keep going...
+  # @block [Proc] if a block is given, each command in the file will be yielded to it
+  #   The default is executes the command with the CliRegistry.exec(cmd, args)
+  # @return [Array] exit codes of all the commands that were run.
+  def execute(stop_on_failure=false, &block)
+    if !File.exists?(@filename)
+      print "#{Term::ANSIColor.red}source file not found: #{@filename}#{Term::ANSIColor.reset}\n" # if Morpheus::Logging.debug?
+    else
+      print "#{dark} #=> executing source file #{@filename}#{reset}\n" if Morpheus::Logging.debug?
+    end
+    file_contents ||= File.read(@filename)
+    lines = file_contents.split("\n")
+    cmd_results = []
+    line_num = 0
+    lines.each_with_index do |line, line_index|
+      line_num = line_index + 1
+      line = line.strip
+      next if line.empty?
+      next if line =~ /^\#/ # skip comments
+      
+      # print "#{dark} #=> executing source file line #{line_num}: #{line}#{reset}\n" if Morpheus::Logging.debug?
+
+      # todo: allow semicolons inside arguments too..
+      command_list = line.strip.split(';').compact
+      command_list.each do |input|
+        input = input.strip
+        if input.empty?
+          next
+        end
+        argv = Shellwords.shellsplit(input)
+
+        if Morpheus::Cli::CliRegistry.has_command?(argv[0]) || Morpheus::Cli::CliRegistry.has_alias?(argv[0])
+          #log_history_command(input)
+          cmd_result = nil
+          begin
+            cmd_result = Morpheus::Cli::CliRegistry.exec(argv[0], argv[1..-1])
+          rescue SystemExit => err
+            if err.success?
+              cmd_result = true
+            else
+              puts "#{red} source file: #{@filename} line: #{line_num} command: #{argv[0]}#{reset} error: exited non zero - #{err}"
+              cmd_result = false
+            end
+          rescue => err
+            # raise err
+            puts "#{red} source file: #{@filename} line: #{line_num} command: #{argv[0]}#{reset} error: unexpected error - #{err}"
+            cmd_result = false
+          end
+          cmd_results << cmd_result
+          # next command please!
+        else
+          puts "#{red}Unrecognized source command file: #{@filename} line: #{line_num} command: #{argv[0]}#{reset}"
+          cmd_result = false
+        end
+        if cmd_result == false
+          if stop_on_failure
+            return cmd_results
+          end
+        end
+      end
+
+    end
+    return cmd_results
+  end
+
+  # this saves the source file, upserting alias definitions at the bottom
+  # todo: smarter logic to allow the user to put stuff AFTER this section too
+  # under the section titled '# exported aliases'
+  # @param alias_definitions [Map] Map of alias_name => command_string
+  # @return nil
+  def export_aliases(alias_definitions)
+    if !@filename
+      print "#{Term::ANSIColor.dark}Skipping source file save because filename has not been set#{Term::ANSIColor.reset}\n" if Morpheus::Logging.debug?
+      return false
+    end
+    if !Dir.exists?(File.dirname(@filename))
+      FileUtils.mkdir_p(File.dirname(@filename))
+    end
+    if !File.exists?(@filename)
+      print "#{Term::ANSIColor.dark}Initializing source file #{@filename}#{Term::ANSIColor.reset}\n" if Morpheus::Logging.debug?
+      FileUtils.touch(@filename)
+    else
+      print "#{dark} #=> Saving source file #{@filename}#{reset}\n" if Morpheus::Logging.debug?
+    end
+
+
+    config_text = File.read(@filename)
+    config_lines = config_text.split(/\n/)
+    new_config_lines = []
+    existing_alias_definitions = {}
+    header_line_index = config_lines.index {|line| line.strip.include?(EXPORTED_ALIASES_HEADER) }
+    if header_line_index
+      # keep everything before the exported alias section
+      new_config_lines = config_lines[0..header_line_index-1]
+      existing_alias_lines = config_lines[header_line_index..config_lines.size-1]
+      # parse out the existing alias definitions
+      existing_alias_lines.each do |line|
+        if line =~ /^alias\s+/
+          alias_name, command_string = Morpheus::Cli::CliRegistry.parse_alias_definition(line)
+          if alias_name.empty? || command_string.empty?
+            print "#{dark} #=> removing bad config line #{line_num} invalid alias definition: #{line}\n" if Morpheus::Logging.debug?
+          else
+            existing_alias_definitions[alias_name] = command_string
+          end
+        end
+      end
+    else
+      new_config_lines = config_lines
+      new_config_lines << "" # blank line before alias header
+    end
+
+    # append header line
+    new_config_lines << EXPORTED_ALIASES_HEADER
+    new_config_lines << "# Do not put anything below here, or it will be lost when aliases are exported"
+    #new_config_lines << ""
+
+    # update aliases, sort them, and append the lines
+    new_alias_definitions = existing_alias_definitions.merge(alias_definitions)
+    new_alias_definitions.keys.sort.each do |alias_name|
+      new_config_lines << "alias #{alias_name}='#{new_alias_definitions[alias_name]}'"
+    end
+    
+    # include a blank line after this section
+    new_config_lines << ""
+    new_config_lines << ""
+
+    new_config_text = new_config_lines.join("\n")
+    
+    File.open(@filename, 'w') {|f| f.write(new_config_text) }
+    return true
+  end
+
+  # this saves the source file, removing alias definitions at the bottom
+  # todo: smarter logic to allow the user to put stuff AFTER this section too
+  # under the section titled '# exported aliases'
+  # @param alias_definitions [Map] Map of alias_name => command_string
+  # @return nil
+  def remove_aliases(alias_names)
+    if !@filename
+      print "#{Term::ANSIColor.dark}Skipping source file save because filename has not been set#{Term::ANSIColor.reset}\n" if Morpheus::Logging.debug?
+      return false
+    end
+    if !Dir.exists?(File.dirname(@filename))
+      FileUtils.mkdir_p(File.dirname(@filename))
+    end
+    if !File.exists?(@filename)
+      print "#{Term::ANSIColor.dark}Initializing source file #{@filename}#{Term::ANSIColor.reset}\n" if Morpheus::Logging.debug?
+      FileUtils.touch(@filename)
+    else
+      print "#{dark} #=> Saving source file #{@filename}#{reset}\n" if Morpheus::Logging.debug?
+    end
+
+
+    config_text = File.read(@filename)
+    config_lines = config_text.split(/\n/)
+    
+    new_config_lines = config_lines.reject {|line|
+      alias_names.find {|alias_name| /^alias\s+#{Regexp.escape(alias_name)}\s?\=/ }
+    }
+    
+    new_config_text = new_config_lines.join("\n")
+    
+    File.open(@filename, 'w') {|f| f.write(new_config_text) }
+    return true
+  end
+
+
+
+end

data/lib/morpheus/cli/echo_command.rb

@@ -0,0 +1,32 @@
+require 'optparse'
+require 'morpheus/cli/cli_command'
+require 'json'
+
+# This is for use in dotfile scripts
+class Morpheus::Cli::EchoCommand
+  include Morpheus::Cli::CliCommand
+  set_command_name :echo
+  set_command_hidden
+
+  def handle(args)
+    append_newline = true
+    options = {}
+    optparse = Morpheus::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: morpheus #{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(' ')
+    if append_newline
+      out << "\n"
+    end
+    # print out 
+    print cyan + out + reset
+    return true
+  end
+
+end

data/lib/morpheus/cli/error_handler.rb

@@ -12,12 +12,14 @@ class Morpheus::Cli::ErrorHandler
     end
     case (err)
     when OptionParser::InvalidOption, OptionParser::AmbiguousOption, OptionParser::MissingArgument, OptionParser::InvalidArgument
-      # raise e
+      # raise err
       print_red_alert "#{err.message}"
       puts "Try -h for help with this command."
     when Errno::ECONNREFUSED
       print_red_alert "#{err.message}"
       # more special errors?
+    when OpenSSL::SSL::SSLError
+      print_red_alert "Error Communicating with the Appliance. #{err.message}"
     when RestClient::Exception
       print_rest_exception(err, options)
     else

data/lib/morpheus/cli/groups.rb

@@ -128,12 +128,14 @@ class Morpheus::Cli::Groups
     use_it = false
     optparse = Morpheus::Cli::OptionParser.new do|opts|
       opts.banner = subcommand_usage("[name]")
-      opts.on( '-l', '--location LOCATION', "Location" ) do |val|
-        params[:location] = val
-      end
+      build_option_type_options(opts, options, add_group_option_types())
+      # opts.on( '-l', '--location LOCATION', "Location" ) do |val|
+      #   params[:location] = val
+      # end
       opts.on( '--use', '--use', "Make this the current active group" ) do
         use_it = true
       end
+
       build_common_options(opts, options, [:options, :json, :dry_run, :remote])
       opts.footer = "Create a new group."
     end
@@ -187,9 +189,10 @@ class Morpheus::Cli::Groups
     params = {}
     optparse = Morpheus::Cli::OptionParser.new do|opts|
       opts.banner = subcommand_usage("[name] [options]")
-      opts.on( '-l', '--location LOCATION', "Location" ) do |val|
-        params[:location] = val
-      end
+      build_option_type_options(opts, options, update_group_option_types())
+      # opts.on( '-l', '--location LOCATION', "Location" ) do |val|
+      #   params[:location] = val
+      # end
       build_common_options(opts, options, [:options, :json, :dry_run, :remote])
       opts.footer = "Update an existing group."
     end
@@ -203,13 +206,11 @@ class Morpheus::Cli::Groups
       group = find_group_by_name_or_id(args[0])
       group_payload = {id: group['id']}
 
-      all_option_types = update_group_option_types()
-      #params = Morpheus::Cli::OptionTypes.prompt(all_option_types, options[:options], @api_client, {})
+      #params = Morpheus::Cli::OptionTypes.prompt(update_group_option_types, options[:options], @api_client, {})
       params = options[:options] || {}
 
       if params.empty?
-        puts optparse.banner
-        print_available_options(all_option_types)
+        puts optparse
         exit 1
       end
 
@@ -364,8 +365,8 @@ class Morpheus::Cli::Groups
     optparse = Morpheus::Cli::OptionParser.new do|opts|
       opts.banner = subcommand_usage("[name]")
       opts.footer = "" +
-        "This sets the current active group.\n" +
-        "It will be auto-selected for use during provisioning.\n" +
+        "This sets the active group.\n" +
+        "The active group will be auto-selected for use during provisioning.\n" +
         "You can still use the --group option to override this."
       build_common_options(opts, options, [])
     end
@@ -401,9 +402,10 @@ class Morpheus::Cli::Groups
   def unuse(args)
     options = {}
     optparse = Morpheus::Cli::OptionParser.new do|opts|
-      opts.banner = subcommand_usage() + "\n\n" +
-        "This will clear the active group." + "\n" +
-        "You will be prompted for a Group during provisioning." + "\n\n"
+      opts.banner = subcommand_usage()
+      opts.footer = "" +
+        "This will clear the current active group.\n" +
+        "You will be prompted for a Group during provisioning."
       build_common_options(opts, options, [])
     end
     optparse.parse!(args)
@@ -429,10 +431,7 @@ class Morpheus::Cli::Groups
     optparse = Morpheus::Cli::OptionParser.new do|opts|
       opts.banner = subcommand_usage()
       build_common_options(opts, options, [])
-      opts.footer = "" +
-        "This will set the current remote appliance.\n" +
-        "It will be used for all subsequent commands.\n" +
-        "You may still use the --remote option to override this."
+      opts.footer = "Prints the name of the current active group"
     end
     optparse.parse!(args)
     connect(options)
@@ -509,7 +508,6 @@ public
     end
 
     # Provides the current active group information (just the ID right now)
-    # appliance_name should probably be required.. or just use this instead: Groups.active_groups[appliance_name]
     def active_group(appliance_name=nil)
       if appliance_name == nil
         appliance_name, appliance_url = Morpheus::Cli::Remote.active_appliance

data/lib/morpheus/cli/hosts.rb

@@ -10,7 +10,7 @@ require 'json'
 class Morpheus::Cli::Hosts
   include Morpheus::Cli::CliCommand
   include Morpheus::Cli::ProvisioningHelper
-  register_subcommands :list, :get, :stats, :add, :remove, :logs, :start, :stop, :resize, :run_workflow, :make_managed, :upgrade_agent, :server_types
+  register_subcommands :list, :get, :stats, :add, :remove, :logs, :start, :stop, :resize, :run_workflow, {:'make-managed' => :install_agent}, :upgrade_agent, :server_types
   alias_subcommand :details, :get
   set_default_subcommand :list
 
@@ -147,6 +147,11 @@ class Morpheus::Cli::Hosts
       puts "Type: #{server['computeServerType'] ? server['computeServerType']['name'] : 'unmanaged'}"
       puts "Platform: #{server['serverOs'] ? server['serverOs']['name'].upcase : 'N/A'}"
       puts "Plan: #{server['plan'] ? server['plan']['name'] : ''}"
+      if server['agentInstalled']
+        puts "Agent: #{server['agentVersion'] || ''} updated at #{format_local_dt(server['lastAgentUpdate'])}"
+      else
+        puts "Agent: (not installed)"
+      end
       puts "Status: #{format_host_status(server)}"
       puts "Power: #{format_server_power_state(server)}"
       if ((stats['maxMemory'].to_i != 0) || (stats['maxStorage'].to_i != 0))
@@ -325,7 +330,7 @@ class Morpheus::Cli::Hosts
       opts.on( '-t', '--type TYPE', "Server Type Code" ) do |val|
         options[:server_type_code] = val
       end
-      build_common_options(opts, options, [:options, :json, :dry_run, :remote])
+      build_common_options(opts, options, [:options, :json, :dry_run, :quiet, :remote])
     end
     optparse.parse!(args)
     connect(options)
@@ -457,8 +462,8 @@ class Morpheus::Cli::Hosts
       if options[:json]
         print JSON.pretty_generate(json_response)
         print "\n"
-      else
-        print_green_success "Provisioning Server..."
+      elsif !options[:quiet]
+        print_green_success "Provisioning Server..." 
         list([])
       end
     rescue RestClient::Exception => e
@@ -478,7 +483,7 @@ class Morpheus::Cli::Hosts
       opts.on( '-S', '--skip-remove-infrastructure', "Skip removal of underlying cloud infrastructure" ) do
         query_params[:removeResources] = 'off'
       end
-      build_common_options(opts, options, [:auto_confirm, :json, :dry_run, :remote])
+      build_common_options(opts, options, [:auto_confirm, :json, :dry_run, :quiet, :remote])
     end
     optparse.parse!(args)
     if args.count < 1
@@ -500,9 +505,9 @@ class Morpheus::Cli::Hosts
       if options[:json]
         print JSON.pretty_generate(json_response)
         print "\n"
-      else
-        print_green_success "Removing Server..."
-        list([])
+      elsif !options[:quiet]
+        print_green_success "Host #{server['name']} is being removed..."
+        #list([])
       end
     rescue RestClient::Exception => e
       print_rest_exception(e, options)
@@ -514,7 +519,7 @@ class Morpheus::Cli::Hosts
     options = {}
     optparse = OptionParser.new do|opts|
       opts.banner = subcommand_usage("[name]")
-      build_common_options(opts, options, [:json, :remote])
+      build_common_options(opts, options, [:json, :dry_run, :quiet, :remote])
     end
     optparse.parse!(args)
     if args.count < 1
@@ -524,12 +529,16 @@ class Morpheus::Cli::Hosts
     connect(options)
     begin
       host = find_host_by_name_or_id(args[0])
+      if options[:dry_run]
+        print_dry_run @servers_interface.dry.start(host['id'])
+        return
+      end
       json_response = @servers_interface.start(host['id'])
       if options[:json]
         print JSON.pretty_generate(json_response)
         print "\n"
-      else
-        puts "Host #{host['name']} started."
+      elsif !options[:quiet]
+        print_green_success "Host #{host['name']} started."
       end
       return
     rescue RestClient::Exception => e
@@ -542,7 +551,7 @@ class Morpheus::Cli::Hosts
     options = {}
     optparse = OptionParser.new do|opts|
       opts.banner = subcommand_usage("[name]")
-      build_common_options(opts, options, [:json, :remote])
+      build_common_options(opts, options, [:json, :dry_run, :quiet, :remote])
     end
     optparse.parse!(args)
     if args.count < 1
@@ -560,8 +569,8 @@ class Morpheus::Cli::Hosts
       if options[:json]
         print JSON.pretty_generate(json_response)
         print "\n"
-      else
-        puts "Host #{host['name']} stopped." unless options[:quiet]
+      elsif !options[:quiet]
+        puts "Host #{host['name']} stopped."
       end
       return
     rescue RestClient::Exception => e
@@ -654,7 +663,7 @@ class Morpheus::Cli::Hosts
         print "\n"
       else
         unless options[:quiet]
-          print_green_success "Resizing server #{server['name']}"
+          puts "Host #{server['name']} resizing..."
           list([])
         end
       end
@@ -664,11 +673,63 @@ class Morpheus::Cli::Hosts
     end
   end
 
-  def upgrade(args)
+  def install_agent(args)
+    options = {}
+    optparse = OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[name]")
+      build_option_type_options(opts, options, install_agent_option_types(false))
+      build_common_options(opts, options, [:json, :dry_run, :quiet, :remote])
+    end
+    optparse.parse!(args)
+    if args.count < 1
+      puts optparse
+      exit 1
+    end
+    connect(options)
+    begin
+      host = find_host_by_name_or_id(args[0])
+      if host['agentInstalled']
+        print_red_alert "Agent already installed on host '#{host['name']}'"
+        return false
+      end
+      payload = {
+        'server' => {}
+      }
+      params = Morpheus::Cli::OptionTypes.prompt(install_agent_option_types, options[:options], @api_client, options[:params])
+      server_os = params.delete('serverOs')
+      if server_os
+        payload['server']['serverOs'] = {id: server_os}
+      end
+      account_id = params.delete('account') # not yet implemented
+      if account_id
+        payload['server']['account'] = {id: account}
+      end
+      payload['server'].merge!(params)
+
+      if options[:dry_run]
+        print_dry_run @servers_interface.dry.install_agent(host['id'], payload)
+        return
+      end
+      json_response = @servers_interface.install_agent(host['id'], payload)
+      if options[:json]
+        print JSON.pretty_generate(json_response)
+        print "\n"
+      elsif !options[:quiet]
+        print_green_success "Host #{host['name']} is being converted to managed."
+        puts "Public Key:\n#{json_response['publicKey']}\n(copy to your authorized_keys file)"
+      end
+      return true
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def upgrade_agent(args)
     options = {}
     optparse = OptionParser.new do|opts|
       opts.banner = subcommand_usage("[name]")
-      build_common_options(opts, options, [:json, :quiet, :remote])
+      build_common_options(opts, options, [:json, :dry_run, :quiet, :remote])
     end
     optparse.parse!(args)
     if args.count < 1
@@ -678,6 +739,10 @@ class Morpheus::Cli::Hosts
     connect(options)
     begin
       host = find_host_by_name_or_id(args[0])
+      if options[:dry_run]
+        print_dry_run @servers_interface.dry.upgrade(host['id'])
+        return
+      end
       json_response = @servers_interface.upgrade(host['id'])
       if options[:json]
         print JSON.pretty_generate(json_response)
@@ -696,13 +761,13 @@ class Morpheus::Cli::Hosts
     options = {}
     optparse = OptionParser.new do|opts|
       opts.banner = subcommand_usage("run-workflow", "[name]", "[workflow]")
-      build_common_options(opts, options, [:json, :dry_run, :remote])
+      build_common_options(opts, options, [:json, :dry_run, :quiet, :remote])
     end
+    optparse.parse!(args)
     if args.count < 2
-      puts "\n#{optparse}\n\n"
+      puts optparse
       exit 1
     end
-    optparse.parse!(args)
     connect(options)
     host = find_host_by_name_or_id(args[0])
     workflow = find_workflow_by_name(args[1])
@@ -738,8 +803,8 @@ class Morpheus::Cli::Hosts
       if options[:json]
         print JSON.pretty_generate(json_response)
         print "\n"
-      else
-        puts "Running workflow..."
+      elsif !options[:quiet]
+        puts "Workflow #{workflow['name']} is running..."
       end
     rescue RestClient::Exception => e
       print_rest_exception(e, options)
@@ -873,4 +938,13 @@ class Morpheus::Cli::Hosts
     out
   end
 
+   def install_agent_option_types(connected=true)
+    [
+      #{'fieldName' => 'account', 'fieldLabel' => 'Account', 'type' => 'select', 'optionSource' => 'accounts', 'required' => true},
+      {'fieldName' => 'sshUsername', 'fieldLabel' => 'SSH Username', 'type' => 'text', 'required' => true},
+      {'fieldName' => 'sshPassword', 'fieldLabel' => 'SSH Password', 'type' => 'password', 'required' => false},
+      {'fieldName' => 'serverOs', 'fieldLabel' => 'OS Type', 'type' => 'select', 'optionSource' => 'osTypes', 'required' => false},
+    ]
+  end
+
 end