Rename master to ctrl. Prettify output.

Author: Peter Benjamin

source/lib/auth.rb

@@ -5,26 +5,26 @@
 class Auth
 
   def self.get_mfa_code
-    print "Enter MFA: "
+    print Rainbow("Enter MFA: ").yellow
     STDOUT.flush
     STDIN.gets(7).chomp
   end
 
   def self.get_creds(username, defaults)
     region = defaults.fetch(:region)
-    master = defaults.fetch(:master)
-    master_account_number = master.fetch(:account_number)
-    master_role_prefix = master.fetch(:role_prefix)
-    master_role_name = master.fetch(:role_name)
+    ctrl = defaults.fetch(:ctrl)
+    ctrl_account_number = ctrl.fetch(:account_number)
+    ctrl_role_prefix = ctrl.fetch(:role_prefix)
+    ctrl_role_name = ctrl.fetch(:role_name)
 
     target = defaults.fetch(:target)
     target_account_number = target.fetch(:account_number)
     target_role_prefix = target.fetch(:role_prefix)
     target_role_name = target.fetch(:role_name)
-
-    serial_number = "arn:aws:iam::#{master_account_number}:mfa/#{username}"
-    puts "Logging into #{target.fetch(:label).upcase} using MFA: #{serial_number} (#{region})"
-    role_arn = "arn:aws:iam::#{master_account_number}:role#{master_role_prefix}#{master_role_name}"
+    target_label = target.fetch(:label)
+    serial_number = "arn:aws:iam::#{ctrl_account_number}:mfa/#{username}"
+    puts "Logging into #{Rainbow(target_label.upcase).yellow} using MFA: #{serial_number} (#{region})"
+    role_arn = "arn:aws:iam::#{ctrl_account_number}:role#{ctrl_role_prefix}#{ctrl_role_name}"
     session_name = username[0..31]
 
     sts_client = Aws::STS::Client.new(region: region)
@@ -53,7 +53,12 @@ def self.login(options, defaults, plane)
         Aws.config[:credentials] = Aws::SharedCredentials.new(profile_name: profile_name)
         creds = get_creds(options.fetch(:username), defaults)
       rescue KeyError => e
-        raise "Error parsing #{CONFIG_FILE}, (#{e.message}), please ensure it is properly formatted"
+        error = Rainbow("Error parsing #{CONFIG_FILE}, (#{e.message}), please ensure it is properly formatted").red
+        raise error
+      rescue => err
+        error = Rainbow(err.message).red
+        raise error
+        exit
       end
       # now save to yaml
       File.open(auth_file, 'w') do |f|