cfndk 0.1.1 → 0.1.3

data/lib/cfndk/key_pair.rb

@@ -1,17 +1,22 @@
 module CFnDK
   class KeyPair
-    attr_reader :key_file
+    attr_reader :key_file, :enabled, :pre_command, :post_command
     def initialize(name, data, option, global_config, credentials)
       @global_config = global_config
       @name = name
       data = {} unless data
       @key_file = data['key_file'] || nil
       @region = data['region'] || @global_config.region
+      @pre_command = data['pre_command'] || nil
+      @post_command = data['post_command'] || nil
+      @enabled = true
+      @enabled = false if data['enabled'] === false
       @option = option
       @client = Aws::EC2::Client.new(credentials: credentials, region: @region)
     end
 
     def create
+      return unless @enabled
       CFnDK.logger.info(('creating keypair: ' + name).color(:green))
       key_pair = @client.create_key_pair(
         key_name: name
@@ -22,6 +27,7 @@ module CFnDK
     end
 
     def destroy
+      return unless @enabled
       if exists?
         CFnDK.logger.info(('deleting keypair: ' + name).color(:green))
         @client.delete_key_pair(
@@ -51,6 +57,32 @@ module CFnDK
       @name
     end
 
+    def pre_command_execute
+      return unless @enabled
+      if @pre_command
+        CFnDK.logger.info(('execute pre command: ' + @pre_command).color(:green))
+        IO.popen(@pre_command, :err => [:child, :out]) do |io|
+          io.each_line do |line|
+            CFnDK.logger.info((line).color(:green))
+          end
+        end
+        raise 'pre command is error. status: ' + $?.exitstatus.to_s + ' command: ' + @pre_command if $?.exitstatus != 0
+      end
+    end
+
+    def post_command_execute
+      return unless @enabled
+      if @post_command
+        CFnDK.logger.info(('execute post command: ' + @post_command).color(:green))
+        IO.popen(@post_command, :err => [:child, :out]) do |io|
+          io.each_line do |line|
+            CFnDK.logger.info((line).color(:green))
+          end
+        end
+        raise 'post command is error. status: ' + $?.exitstatus.to_s + ' command: ' + @post_command if $?.exitstatus != 0
+      end
+    end
+
     private
 
     def create_key_file(key_pair)

data/lib/cfndk/key_pair_command.rb

@@ -2,6 +2,8 @@ module CFnDK
   class KeyPairCommand < Thor
     include SubcommandHelpReturnable
     include ConfigFileLoadable
+    include CredentialResolvable
+
     class_option :verbose, type: :boolean, aliases: 'v', desc: 'More verbose output.'
     class_option :color, type: :boolean, default: true, desc: 'Use colored output'
     class_option :config_path, type: :string, aliases: 'c', default: "#{Dir.getwd}/cfndk.yml", desc: 'The configuration file to use'
@@ -13,10 +15,15 @@ module CFnDK
     def create
       CFnDK.logger.info 'create...'.color(:green)
       data = load_config_data(options)
-
-      credentials = CFnDK::CredentialProviderChain.new.resolve
+      credentials = resolve_credential(data, options)
+      global_config = CFnDK::GlobalConfig.new(data, options)
       keypairs = CFnDK::KeyPairs.new(data, options, credentials)
+
+      global_config.pre_command_execute
+      keypairs.pre_command_execute
       keypairs.create
+      keypairs.post_command_execute
+      global_config.post_command_execute
       return 0
     rescue => e
       CFnDK.logger.error "#{e.class}: #{e.message}".color(:red)
@@ -31,8 +38,8 @@ module CFnDK
     def destroy
       CFnDK.logger.info 'destroy...'.color(:green)
       data = load_config_data(options)
+      credentials = resolve_credential(data, options)
 
-      credentials = CFnDK::CredentialProviderChain.new.resolve
       keypairs = CFnDK::KeyPairs.new(data, options, credentials)
 
       if options[:force] || yes?('Are you sure you want to destroy? (y/n)', :yellow)

data/lib/cfndk/key_pairs.rb

@@ -21,6 +21,18 @@ module CFnDK
       end
     end
 
+    def pre_command_execute
+      @keypairs.each_value do |keypair|
+        keypair.pre_command_execute
+      end
+    end
+
+    def post_command_execute
+      @keypairs.each_value do |keypair|
+        keypair.post_command_execute
+      end
+    end
+
     private
 
     def prepare_keypairs(data)

data/lib/cfndk/stack.rb

@@ -1,6 +1,6 @@
 module CFnDK
   class Stack
-    attr_reader :template_file, :parameter_input, :capabilities, :depends, :timeout_in_minutes, :region
+    attr_reader :template_file, :parameter_input, :capabilities, :depends, :timeout_in_minutes, :region, :role_arn, :package, :enabled, :pre_command, :post_command
     def initialize(name, data, option, global_config, credentials)
       @global_config = global_config
       @name = name
@@ -9,16 +9,24 @@ module CFnDK
       @capabilities = data['capabilities'] || []
       @depends = data['depends'] || []
       @region = data['region'] || @global_config.region
+      @role_arn = @global_config.role_arn
+      @package = data['package'] || @global_config.package
+      @pre_command = data['pre_command'] || nil
+      @post_command = data['post_command'] || nil
+      @enabled = true
+      @enabled = false if data['enabled'] === false 
       @timeout_in_minutes = data['timeout_in_minutes'] || @global_config.timeout_in_minutes
       @override_parameters = data['parameters'] || {}
       @option = option
       @client = Aws::CloudFormation::Client.new(credentials: credentials, region: @region)
       @s3_client = Aws::S3::Client.new(credentials: credentials, region: @region)
       @sts_client = Aws::STS::Client.new(credentials: credentials, region: @region)
+      @tp = CFnDK::TemplatePackager.new(@template_file, @region, @package, @global_config, @s3_client, @sts_client)
     end
 
     def create
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('creating stack: ' + name).color(:green))
       CFnDK.logger.debug('Name        :' + name)
       CFnDK.logger.debug('Parametres  :' + parameters.inspect)
@@ -42,10 +50,12 @@ module CFnDK
         timeout_in_minutes: timeout_in_minutes,
         tags: tags,
       }
-      if large_template?
-        hash[:template_url] = upload_template_file()
+      hash[:role_arn] = @role_arn if @role_arn
+
+      if @tp.large_template?
+        hash[:template_url] = @tp.upload_template_file()
       else
-        hash[:template_body] = template_body()
+        hash[:template_body] = @tp.template_body()
       end
       @client.create_stack(
         hash
@@ -54,6 +64,7 @@ module CFnDK
 
     def wait_until_create
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('waiting create stack: ' + name).color(:green))
       begin
         @client.wait_until(
@@ -74,6 +85,7 @@ module CFnDK
 
     def update
       return false if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('updating stack: ' + name).color(:green))
       CFnDK.logger.debug('Name        :' + name)
       CFnDK.logger.debug('Parametres  :' + parameters.inspect)
@@ -86,10 +98,11 @@ module CFnDK
           parameters: parameters,
           capabilities: capabilities,
         }
-        if large_template?
-          hash[:template_url] = upload_template_file()
+        hash[:role_arn] = @role_arn if @role_arn
+        if @tp.large_template?
+          hash[:template_url] = @tp.upload_template_file()
         else
-          hash[:template_body] = template_body()
+          hash[:template_body] = @tp.template_body()
         end
         @client.update_stack(
           hash
@@ -108,6 +121,7 @@ module CFnDK
 
     def wait_until_update
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('waiting update stack: ' + name).color(:green))
       @client.wait_until(
         :stack_update_complete,
@@ -121,12 +135,17 @@ module CFnDK
 
     def destroy
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       if exits?
         CFnDK.logger.info(('deleting stack: ' + name).color(:green))
         CFnDK.logger.debug('Name        :' + name)
         CFnDK.logger.debug('Region      :' + region)
+        hash = {
+          stack_name: name,
+        }
+        hash[:role_arn] = @role_arn if @role_arn
         @client.delete_stack(
-          stack_name: name
+          hash
         )
       else
         CFnDK.logger.info(('do not delete stack: ' + name).color(:red))
@@ -135,6 +154,7 @@ module CFnDK
 
     def wait_until_destroy
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       return unless exits?
       CFnDK.logger.info(('waiting delete stack: ' + name).color(:green))
       @client.wait_until(
@@ -149,6 +169,7 @@ module CFnDK
 
     def create_change_set
       return nil if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('creating change set: ' + change_set_name).color(:green))
       CFnDK.logger.debug('Parametres  :' + parameters.inspect)
       CFnDK.logger.debug('Capabilities:' + capabilities.inspect)
@@ -175,10 +196,11 @@ module CFnDK
         change_set_type: exits? ? 'UPDATE' : 'CREATE',
         tags: tags,
       }
-      if large_template?
-        hash[:template_url] = upload_template_file()
+      hash[:role_arn] = @role_arn if @role_arn
+      if @tp.large_template?
+        hash[:template_url] = @tp.upload_template_file()
       else
-        hash[:template_body] = template_body()
+        hash[:template_body] = @tp.template_body()
       end
       @client.create_change_set(
         hash
@@ -196,6 +218,7 @@ module CFnDK
 
     def wait_until_create_change_set
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       return unless exits?
       CFnDK.logger.info(('waiting create change set: ' + change_set_name).color(:green))
       @client.wait_until(
@@ -221,6 +244,7 @@ module CFnDK
 
     def execute_change_set
       return nil if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       if available_change_set?
         CFnDK.logger.info(('executing change set: ' + change_set_name).color(:green))
         @client.execute_change_set(
@@ -237,6 +261,7 @@ module CFnDK
 
     def delete_change_set
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('deleting change set: ' + change_set_name).color(:green))
       @client.delete_change_set(
         stack_name: name,
@@ -247,6 +272,7 @@ module CFnDK
 
     def report_change_set
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info('*****************************************************'.color(:green))
       CFnDK.logger.info(('change set: ' + change_set_name).color(:green))
       CFnDK.logger.info('*****************************************************'.color(:green))
@@ -310,13 +336,14 @@ module CFnDK
 
     def validate
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info(('validate stack: ' + name).color(:green))
       CFnDK.logger.debug('Name        :' + @name)
       hash = {}
-      if large_template?
-        hash[:template_url] = upload_template_file()
+      if @tp.large_template?
+        hash[:template_url] = @tp.upload_template_file()
       else
-        hash[:template_body] = template_body()
+        hash[:template_body] = @tp.template_body()
       end
       @client.validate_template(
         hash
@@ -365,6 +392,7 @@ module CFnDK
 
     def report
       return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
       CFnDK.logger.info('*****************************************************'.color(:green))
       CFnDK.logger.info(('stack: ' + name).color(:green))
       CFnDK.logger.info('*****************************************************'.color(:green))
@@ -476,14 +504,6 @@ module CFnDK
       [@name, @option[:change_set_uuid]].compact.join('-')
     end
 
-    def template_body
-      File.open(@template_file, 'r').read
-    end
-
-    def large_template?
-      File.size(@template_file) > 51200
-    end
-
     def parameters
       json = JSON.load(open(@parameter_input).read)
       json['Parameters'].map do |item|
@@ -495,49 +515,36 @@ module CFnDK
       end.compact
     end
 
-    private
-
-    def upload_template_file
-      begin
-        @s3_client.head_bucket(bucket: bucket_name)
-      rescue Aws::S3::Errors::NotFound, Aws::S3::Errors::Forbidden
-        @s3_client.create_bucket(bucket: bucket_name)
-        CFnDK.logger.info('Creatt S3 bucket: ' + bucket_name)
-        @s3_client.put_bucket_lifecycle_configuration(
-          bucket: bucket_name,
-          lifecycle_configuration: {
-            rules: [
-              {
-                expiration: {
-                  days: 1,
-                },
-                status: 'Enabled',
-                id: 'Delete Old Files',
-                prefix: '',
-                abort_incomplete_multipart_upload: {
-                  days_after_initiation: 1,
-                },
-              },
-            ],
-          }
-        )
+    def pre_command_execute
+      return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
+      if @pre_command
+        CFnDK.logger.info(('execute pre command: ' + @pre_command).color(:green))
+        IO.popen(@pre_command, :err => [:child, :out]) do |io|
+          io.each_line do |line|
+            CFnDK.logger.info((line).color(:green))
+          end
+        end
+        raise 'pre command is error. status: ' + $?.exitstatus.to_s + ' command: ' + @pre_command if $?.exitstatus != 0
       end
-      key = [@global_config.s3_template_hash, @template_file].compact.join('/')
-      @s3_client.put_object(
-        body: template_body,
-        bucket: bucket_name,
-        key: key
-      )
-      url = "https://s3.amazonaws.com/#{bucket_name}/#{key}"
-      CFnDK.logger.info('Put S3 object: ' + url)
-      url
     end
 
-    def bucket_name
-      resp = @sts_client.get_caller_identity({})
-      resp.account + '-' + @region + '-' + @global_config.s3_template_bucket
+    def post_command_execute
+      return if @option[:stack_names].instance_of?(Array) && !@option[:stack_names].include?(@name)
+      return unless @enabled
+      if @post_command
+        CFnDK.logger.info(('execute post command: ' + @post_command).color(:green))
+        IO.popen(@post_command, :err => [:child, :out]) do |io|
+          io.each_line do |line|
+            CFnDK.logger.info((line).color(:green))
+          end
+        end
+        raise 'post command is error. status: ' + $?.exitstatus.to_s + ' command: ' + @post_command if $?.exitstatus != 0
+      end
     end
 
+    private
+
     def colored_status(str)
       case str
       when 'CREATE_FAILED' then

data/lib/cfndk/stack_command.rb

@@ -2,11 +2,12 @@ module CFnDK
   class StackCommand < Thor
     include SubcommandHelpReturnable
     include ConfigFileLoadable
+    include CredentialResolvable
 
     class_option :verbose, type: :boolean, aliases: 'v', desc: 'More verbose output.'
     class_option :color, type: :boolean, default: true, desc: 'Use colored output'
     class_option :config_path, type: :string, aliases: 'c', default: "#{Dir.getwd}/cfndk.yml", desc: 'The configuration file to use'
-    class_option :stack_names, type: :array, desc: 'Target stack names'
+    class_option :stack_names, type: :array, aliases: 's', desc: 'Target stack names'
 
     desc 'create', 'Create stack'
     option :uuid, type: :string, aliases: 'u', default: ENV['CFNDK_UUID'] || nil, desc: 'Use UUID'
@@ -14,11 +15,16 @@ module CFnDK
     def create
       CFnDK.logger.info 'create...'.color(:green)
       data = load_config_data(options)
-
-      credentials = CFnDK::CredentialProviderChain.new.resolve
+      credentials = resolve_credential(data, options)
+      global_config = CFnDK::GlobalConfig.new(data, options)
       stacks = CFnDK::Stacks.new(data, options, credentials)
+      
+      global_config.pre_command_execute
+      stacks.pre_command_execute
       stacks.validate
       stacks.create
+      stacks.post_command_execute
+      global_config.post_command_execute
       return 0
     rescue => e
       CFnDK.logger.error "#{e.class}: #{e.message}".color(:red)
@@ -34,11 +40,16 @@ module CFnDK
     def update
       CFnDK.logger.info 'update...'.color(:green)
       data = load_config_data(options)
-
-      credentials = CFnDK::CredentialProviderChain.new.resolve
+      credentials = resolve_credential(data, options)
+      global_config = CFnDK::GlobalConfig.new(data, options)
       stacks = CFnDK::Stacks.new(data, options, credentials)
+
+      global_config.pre_command_execute
+      stacks.pre_command_execute
       stacks.validate
       stacks.update
+      stacks.post_command_execute
+      global_config.post_command_execute
       return 0
     rescue => e
       CFnDK.logger.error "#{e.class}: #{e.message}".color(:red)
@@ -54,8 +65,8 @@ module CFnDK
     def destroy
       CFnDK.logger.info 'destroy...'.color(:green)
       data = load_config_data(options)
+      credentials = resolve_credential(data, options)
 
-      credentials = CFnDK::CredentialProviderChain.new.resolve
       stacks = CFnDK::Stacks.new(data, options, credentials)
 
       if options[:force] || yes?('Are you sure you want to destroy? (y/n)', :yellow)
@@ -77,9 +88,15 @@ module CFnDK
     def validate
       CFnDK.logger.info 'validate...'.color(:green)
       data = load_config_data(options)
-      credentials = CFnDK::CredentialProviderChain.new.resolve
+      credentials = resolve_credential(data, options)
+      global_config = CFnDK::GlobalConfig.new(data, options)
       stacks = CFnDK::Stacks.new(data, options, credentials)
+
+      global_config.pre_command_execute
+      stacks.pre_command_execute
       stacks.validate
+      stacks.post_command_execute
+      global_config.post_command_execute
       return 0
     rescue => e
       CFnDK.logger.error "#{e.class}: #{e.message}".color(:red)
@@ -95,7 +112,8 @@ module CFnDK
     def report
       CFnDK.logger.info 'report...'.color(:green)
       data = load_config_data(options)
-      credentials = CFnDK::CredentialProviderChain.new.resolve
+      credentials = resolve_credential(data, options)
+
       stacks = CFnDK::Stacks.new(data, options, credentials)
       stacks.report
       return 0

data/lib/cfndk/stacks.rb

@@ -107,6 +107,22 @@ module CFnDK
       end
     end
 
+    def pre_command_execute
+      @sequence.each do |stacks|
+        stacks.each do |name|
+          @stacks[name].pre_command_execute
+        end
+      end
+    end
+
+    def post_command_execute
+      @sequence.each do |stacks|
+        stacks.each do |name|
+          @stacks[name].post_command_execute
+        end
+      end
+    end
+
     private
 
     def prepare_stack(data)