mtncloud-cli 1.0.0

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

@@ -0,0 +1,352 @@
+require 'mtncloud/cli/cli_command'
+
+class MtnCloud::Cli::FileCopyRequestCommand
+  include MtnCloud::Cli::CliCommand
+
+  set_command_name :'file-copy-request'
+
+  register_subcommands :get, :execute, :download
+  #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
+    @file_copy_request_interface = @api_client.file_copy_request
+  end
+
+  def handle(args)
+    handle_subcommand(args)
+  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 a file copy request." + "\n" +
+                    "[uid] is required. This is the unique id of a file copy 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
+    file_copy_request_id = args[0]
+    begin
+      params.merge!(parse_list_options(options))
+      @file_copy_request_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @file_copy_request_interface.dry.get(file_copy_request_id, params)
+        return
+      end
+      json_response = @file_copy_request_interface.get(file_copy_request_id, params)
+      if options[:json]
+        puts as_json(json_response, options, "fileCopyRequest")
+        return 0
+      elsif options[:yaml]
+        puts as_yaml(json_response, options, "fileCopyRequest")
+        return 0
+      elsif options[:csv]
+        puts records_as_csv([json_response['fileCopyRequest']], options)
+        return 0
+      end
+
+      file_copy_request = json_response['fileCopyRequest']
+
+      # 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 ['complete','failed','expired'].include?(file_copy_request['status'])
+          # it is finished
+        else
+          print cyan
+          refresh_display_seconds = options[:refresh_interval] % 1.0 == 0 ? options[:refresh_interval].to_i : options[:refresh_interval]
+          print "File copy request has not yet finished. Refreshing every #{refresh_display_seconds} seconds"
+          while !['complete','failed','expired'].include?(file_copy_request['status']) do
+            sleep(options[:refresh_interval])
+            print cyan,".",reset
+            json_response = @file_copy_request_interface.get(file_copy_request_id, params)
+            file_copy_request = json_response['fileCopyRequest']
+          end
+          #sleep_with_dots(options[:refresh_interval])
+          print "\n", reset
+          # get(raw_args)
+        end
+      end
+
+      print_h1 "File Copy 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_file_copy_request_status(it) },
+        #"Created By" => lambda {|it| it['createdById'] },
+        #"Subdomain" => lambda {|it| it['subdomain'] },
+      }
+      print_description_list(description_cols, file_copy_request)      
+
+      if file_copy_request['stdErr']
+        print_h2 "Error"
+        puts file_copy_request['stdErr'].to_s.strip
+      end
+      if file_copy_request['stdOut']
+        print_h2 "Output"
+        puts file_copy_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
+    do_refresh = true
+    filename = nil
+    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, "File Copy Request ID") do |val|
+        params['requestId'] = val
+      end
+      opts.on('--file FILE', "Local file to be copied." ) do |val|
+        filename = val
+      end
+      opts.on('--target-path PATH', "Target path for file on destination host." ) do |val|
+        params['targetPath'] = val
+      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, :json, :dry_run, :quiet, :remote])
+      opts.footer = "Copy a file to a remote host(s)." + "\n" +
+                    "[server] or [instance] or [container] is required. This is the id of a server, instance or container." + "\n" +
+                    "[file] is required. This is the local filename that is to be copied." + "\n" +
+                    "[target-path] is required. This is the target path for the file on the destination host."
+    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
+    # if filename.nil?
+    #   puts_error "#{MtnCloud::Terminal.angry_prompt}missing required option: --file\n#{optparse}"
+    #   return 1
+    # end
+    # if params['targetPath'].nil?
+    #   puts_error "#{MtnCloud::Terminal.angry_prompt}missing required option: --target-path\n#{optparse}"
+    #   return 1
+    # end
+    begin
+      # construct payload
+      params.deep_merge!(options[:options].reject {|k,v| k.is_a?(Symbol) }) if options[:options]
+      full_filename = nil
+      if filename.nil?
+        v_prompt = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'file', 'type' => 'file', 'fieldLabel' => 'File', 'required' => true, 'description' => 'The local file to be copied'}], options[:options])
+        filename = v_prompt['file']
+      end
+      full_filename = File.expand_path(filename)        
+      if !File.exist?(full_filename)
+        print_red_alert "File not found: #{full_filename}"
+        return 1
+      end
+      local_file = File.new(full_filename, 'rb')
+      
+      if params['targetPath'].nil?
+        v_prompt = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'targetPath', 'type' => 'text', 'fieldLabel' => 'Target Path', 'required' => true, 'description' => 'The target path for the file on the destination host.'}], options[:options])
+        params['targetPath'] = v_prompt['targetPath']
+      end
+
+      @file_copy_request_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @file_copy_request_interface.dry.create(local_file, params)
+        return 0
+      end
+      # do it
+      json_response = @file_copy_request_interface.create(local_file, params)
+      # print and return result
+      if options[:quiet]
+        return 0
+      elsif options[:json]
+        puts as_json(json_response, options)
+        return 0
+      end
+      file_copy_request = json_response['fileCopyRequest']
+      print_green_success "Executing file copy request #{file_copy_request['uniqueId']}"
+      if do_refresh
+        get([file_copy_request['uniqueId'], "--refresh"])
+      else
+        get([file_copy_request['uniqueId']])
+      end
+      return 0
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def download(args)
+    options = {}
+    params = {}
+    script_content = nil
+    filename = nil
+    do_overwrite = false
+    do_mkdir = false
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = subcommand_usage("[uid] [file]")
+      opts.on('--file FILE', "Local file destination for the downloaded file." ) do |val|
+        filename = val
+      end
+      opts.on( '-f', '--force', "Overwrite existing [file] if it exists." ) do
+        do_overwrite = true
+        # do_mkdir = true
+      end
+      opts.on( '-p', '--mkdir', "Create missing directories for [file] if they do not exist." ) do
+        do_mkdir = true
+      end
+      #build_option_type_options(opts, options, add_user_source_option_types())
+      build_common_options(opts, options, [:options, :json, :dry_run, :quiet, :remote])
+      opts.footer = "Download a file associated with a file copy request." + "\n" +
+                    "[uid] is required. This is the unique id of a file copy request." + "\n" +
+                    "[file] is required. This is the full local filepath for the downloaded file."
+    end
+    optparse.parse!(args)
+    connect(options)
+    if args.count < 1 || args.count > 2
+      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
+    file_copy_request_id = args[0]
+    if args[1]
+      filename = args[1]
+    end
+    outfile = File.expand_path(filename)
+    if Dir.exist?(outfile)
+      print_red_alert "[file] is invalid. It is the name of an existing directory: #{outfile}"
+      return 1
+    end
+    destination_dir = File.dirname(outfile)
+    if !Dir.exist?(destination_dir)
+      if do_mkdir
+        print cyan,"Creating local directory #{destination_dir}",reset,"\n"
+        FileUtils.mkdir_p(destination_dir)
+      else
+        print_red_alert "[file] is invalid. Directory not found: #{destination_dir}"
+        return 1
+      end
+    end
+    if File.exist?(outfile)
+      if do_overwrite
+        # uhh need to be careful wih the passed filepath here..
+        # don't delete, just overwrite.
+        # File.delete(outfile)
+      else
+        print_error MtnCloud::Terminal.angry_prompt
+        puts_error "[file] is invalid. File already exists: #{outfile}", "Use -f to overwrite the existing file."
+        # puts_error optparse
+        return 1
+      end
+    end
+    begin
+      # construct payload
+      
+      @file_copy_request_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @file_copy_request_interface.dry.download_file_chunked(file_copy_request_id, outfile, params)
+        return 0
+      end
+      # do it
+      
+      if !options[:quiet]
+        print cyan + "Downloading file copy request #{file_copy_request_id} to #{outfile} ... "
+      end
+
+      http_response = @file_copy_request_interface.download_file_chunked(file_copy_request_id, outfile, params)
+
+      # FileUtils.chmod(0600, outfile)
+      success = http_response.code.to_i == 200
+      if success
+        if !options[:quiet]
+          print green + "SUCCESS" + reset + "\n"
+        end
+        return 0
+      else
+        if !options[:quiet]
+          print red + "ERROR" + reset + " HTTP #{http_response.code}" + "\n"
+        end
+        # F it, just remove a bad result
+        if File.exist?(outfile) && File.file?(outfile)
+          MtnCloud::Logging::DarkPrinter.puts "Deleting bad file download: #{outfile}" if MtnCloud::Logging.debug?
+          File.delete(outfile)
+        end
+        if options[:debug]
+          puts_error http_response.inspect
+        end
+        return 1
+      end
+
+    rescue RestClient::Exception => e
+      print_rest_exception(e, options)
+      exit 1
+    end
+  end
+
+  def format_file_copy_request_status(file_copy_request, return_color=cyan)
+    out = ""
+    status_str = file_copy_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
+
+end

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

@@ -0,0 +1,133 @@
+require 'mtncloud/cli/cli_command'
+
+class MtnCloud::Cli::ForgotPassword
+  include MtnCloud::Cli::CliCommand
+  
+  set_command_name :'forgot'
+  set_command_description "Send a forgot password email and reset your password."
+
+  def connect(opts)
+    @api_client = establish_remote_appliance_connection(opts)
+    @forgot_interface = @api_client.forgot
+  end
+
+  def handle(args)
+  #   handle_subcommand(args)
+  # end
+
+  # def email(args)
+    options = {}
+    params = {}
+    optparse = MtnCloud::Cli::OptionParser.new do |opts|
+      opts.banner = "Usage: #{prog_name} #{command_name} [username]"
+      opts.on( '-U', '--username USERNAME', "Username of the user to be emailed." ) do |val|
+        options[:options]['username'] = val
+      end
+      opts.on("--email [USERNAME]", String, "Email only. Only send an email, skip the reset password.") do |val|
+        options[:email_only] = true
+        if !val.to_s.empty?
+          options[:options]['username'] = val
+        end
+      end
+      opts.on("--reset [TOKEN]", "Reset only. Only reset password, skip sending an email.") do |val|
+        options[:reset_only] = true
+        if !val.to_s.empty?
+          options[:options]['token'] = val
+        end
+      end
+      opts.on( '-T', '--token TOKEN', "Token, the secret token that was emailed to the user. Only reset password, skip sending an email." ) do |val|
+        options[:reset_only] = true
+        options[:options]['token'] = val
+      end
+      opts.on( '-P', '--password PASSWORD', "New Password, the new password for the user." ) do |val|
+        options[:options]['password'] = val
+      end
+
+      build_standard_post_options(opts, options, [], [:remote_username,:remote_password,:remote_token])
+      opts.footer = <<-EOT
+Send a forgot password email and reset your password.
+[username] is required. This is the username to be notified.
+By default this command prompts to perform two actions. 
+First it sends a forgot password email to the specified user.
+Then it attempts to reset the password with the secret token and a new password.
+Use the --email and --token options to only perform one of these actions, instead of prompting to do both.
+That is, only send the email or only reset the password.
+
+EOT
+    end
+    optparse.parse!(args)
+    connect(options)
+    verify_args!(args:args, optparse:optparse, max:1)
+
+    if options[:email_only] && options[:options]['token']
+      raise_command_error "Invalid usage. --email cannot be used with --token or --reset, use one or the other", args, optparse
+    end
+
+    if args[0]
+      options[:options]['username'] = args[0]
+    end
+    
+    params.merge!(parse_query_options(options))
+
+    # Step 1. Send Forgot Password Email
+    if options[:reset_only] != true
+      print_h1 "Forgot Password", [], options unless options[:quiet] || options[:json] || options[:yaml]
+      payload = {}
+      payload['username'] = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'username', 'fieldLabel' => 'Username', 'type' => 'text', 'description' => "Enter the username of your MTN Cloud user.", 'required' => true, :fmt => :natural}], options[:options],@api_client)['username']
+      @forgot_interface.setopts(options)
+      if options[:dry_run]
+        print_dry_run @forgot_interface.dry.send_email(payload, params)
+        return
+      end
+      json_response = @forgot_interface.send_email(payload, params)
+      if options[:email_only]
+        render_response(json_response, options) do
+          print_green_success(json_response["msg"] || "Email has been sent") unless options[:quiet]
+        end
+        return 0, nil
+      else
+        print_green_success(json_response["msg"] || "Email has been sent") unless options[:quiet]
+      end
+    end
+
+    # Step 2. Reset Password
+    print_h1 "Reset Password", [], options unless options[:quiet] || options[:json] || options[:yaml]
+    payload = {}
+    if options[:payload]
+      payload = options[:payload]
+      payload.deep_merge!(parse_passed_options(options))
+    else
+      payload.deep_merge!(parse_passed_options(options))
+      # prompt for Token
+      payload['token'] = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'token', 'fieldLabel' => 'Token', 'type' => 'text', 'description' => "Enter the token that you obtained from the forgot password email.", 'required' => true, :fmt => :natural}], options[:options],@api_client)['token']
+      # New Password
+      # todo: prompt_password_with_confirmation()
+      password_value = options[:options]['password']
+      confirm_password_value = password_value
+      while password_value.to_s.empty?
+        password_value = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'password', 'fieldLabel' => 'New Password', 'type' => 'password', 'description' => "Enter your new password.", 'required' => true, :fmt => :natural}], options[:options],@api_client)['password']
+        confirm_password_value = MtnCloud::Cli::OptionTypes.prompt([{'fieldName' => 'password', 'fieldLabel' => 'Confirm New Password', 'type' => 'password', 'description' => "Enter your new password again to confirm it is what you intend.", 'required' => true, :fmt => :natural}], options[:options],@api_client)['password']
+        if password_value != confirm_password_value
+          print_red_alert("Passwords did not match. Please try again.")
+          password_value = nil
+        end
+      end
+      payload['password'] = password_value
+    end
+    
+    @forgot_interface.setopts(options)
+    if options[:dry_run]
+      print_dry_run @forgot_interface.dry.reset_password(payload, params)
+      return
+    end
+    json_response = @forgot_interface.reset_password(payload, params)
+    render_response(json_response, options) do
+      print_green_success(json_response["msg"] || "Password has been updated") unless options[:quiet]
+    end
+    return 0, nil
+  end
+
+  protected
+
+end
+

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

@@ -0,0 +1,36 @@
+require 'mtncloud/cli/cli_command'
+
+class MtnCloud::Cli::GetPromptCommand
+  include MtnCloud::Cli::CliCommand
+  set_command_name :'get-prompt'
+  set_command_hidden
+
+  def handle(args)
+    use_echo = false
+    options = {}
+    optparse = MtnCloud::Cli::OptionParser.new do|opts|
+      opts.banner = "Usage: #{prog_name} #{command_name}"
+      build_common_options(opts, options, [])
+      opts.on('-e', '--echo', "Use echo to display the prompt, displaying ansi colors and variables." ) do
+        use_echo = true
+      end
+      opts.footer = <<-EOT
+Display the current #{prog_name} shell prompt value.
+This value can be set using `set-prompt [shell-prompt]`.
+EOT
+    end
+    optparse.parse!(args)
+    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 use_echo
+      my_terminal.execute("echo #{my_terminal.prompt}")
+    else
+      puts self.my_terminal.prompt
+    end
+    return 0
+  end
+
+end