Merge pull request #3 from jercherng/feature-configure

Configure feature

Author: slietz

README.md

@@ -30,6 +30,7 @@ ACTIONS
     deploy	Deploys a new stack
     migrate	Migrates from green to blue stack
     remove	Removes the given deployment
+    configure	Configure the target account in the restacker.yml file
     dump	Dumps the default configuration for a given template or module
     console	Opens the AWS Console
 

source/bin/restacker

@@ -12,6 +12,7 @@ ACTIONS = {
     deploy: 'Deploys a new stack',
     migrate: 'Migrates from green to blue stack',
     remove: 'Removes the given deployment',
+    configure: 'Configure the target account in the restacker.yml file',
     dump: 'Dumps the default configuration for a given template or module',
     console: 'Opens the AWS Console'
 }
@@ -116,6 +117,17 @@ begin
   puts(VERSION) || exit(0) if options[:version]
   usage("Please specify an ACTION") && exit(0) if action.nil?
 
+  if action == 'configure'
+    plane = options[:location] ? options[:location] : DEFAULT_PLANE
+    if !plane
+      puts "Location not specified, plane is set to default: [#{DEFAULT_PLANE}]"
+    end
+    puts "Configuring target plane [#{plane}]: "
+    RestackerConfig.load_config(plane.to_sym)
+    Restacker.configure(plane.to_sym)
+    exit(0)
+  end
+
   if ACTIONS.keys.push(:aws).include?(action.to_sym)
     restacker = Restacker.new(options.to_h) unless ['dump'].include?(action)
     case action
@@ -158,6 +170,16 @@ begin
       else
         usage "Please specify a stack name (-n) to remove"
       end
+    # when 'configure'
+    #   plane = ""
+    #   if !options[:location]
+    #     puts "Location not specified, default plane #{DEFAULT_PLANE} used."
+    #     plane = DEFAULT_PLANE
+    #   else
+    #     plane = options[:location]
+    #   end
+    #   puts "Configuring target account for plane [#{plane}]:\n"
+    #   RestackerConfig.rc_configure(options[:location])
     when 'dump'
       if options[:template]
         # puts "Dumping stack parameters"

source/bin/rstkr

@@ -21,7 +21,7 @@ class BaseStacker
   def initialize(options)
     @plane = options[:location] ? options[:location].to_sym : DEFAULT_PLANE
     config = RestackerConfig.load_config(@plane)
-    # use default regio if not passed in from cli
+    # use default region if not passed in from cli
     config[:region] = options[:region] if options[:region]
     options[:region] = config[:region] unless options[:region]
     @cf, @creds = Auth.login(options, config, @plane)

source/lib/base_stacker.rb

@@ -290,4 +290,51 @@ def self.dump_stack_params(options)
       end
     end
   end
+
+  def self.configure(location)
+    puts "config file location: #{CONFIG_FILE}"
+    config = YAML.load_file(CONFIG_FILE)
+    if !config
+      puts "Make sure your ~/.restacker/restacker.yml file is configured. \n
+          See source/restacker-sample.yml for example."
+      exit(1)
+    end
+    target = config.fetch(location, {}).fetch(:target)
+
+    new_account_number, new_role_name, new_role_prefix = ""
+
+    old_account_number = target[:account_number].to_s
+    # old_account_number[0...7] = "********"
+
+    print "Label [\"#{target[:label]}\"]: "
+    new_label = gets.chomp
+
+    loop do
+      print "Account Number [#{old_account_number}]: "
+
+      new_account_number = gets.chomp
+      break if (new_account_number =~ /\d{12,}/ || new_account_number.empty?)
+    end
+
+    loop do
+      print "Role Name [#{target[:role_name]}]: "
+      new_role_name = gets.chomp
+      break if (new_role_name =~ /[\w&&\S\-]/ || new_role_name.empty?)
+    end
+
+    loop do
+      print "Role Prefix [#{target[:role_prefix]}]: "
+      new_role_prefix = gets.chomp
+      break if (new_role_prefix =~ /[\w&&\S\-\/]/ || new_role_prefix.empty?)
+    end
+
+    target[:label] = new_label.empty? ? target[:label] : new_label
+    target[:account_number] = new_account_number.empty? ? target[:account_number] : new_account_number
+    target[:role_name] = new_role_name.empty? ? target[:role_name] : new_role_name
+    target[:role_prefix] = new_role_prefix.empty? ? target[:role_prefix] : new_role_prefix
+
+    File.open(CONFIG_FILE, 'w') do |f|
+      f.write config.to_yaml
+    end
+  end
 end

source/lib/restacker.rb

@@ -1,5 +1,3 @@
-
-
 class RestackerConfig
   def self.load_config(plane)
     begin
@@ -22,5 +20,4 @@ def self.load_config(plane)
     end
     config[plane]
   end
-
 end