stack_master 1.6.0-x64-mingw32

data/lib/stack_master/command.rb

@@ -0,0 +1,57 @@
+module StackMaster
+  module Command
+    def self.included(base)
+      base.extend ClassMethods
+      base.prepend Perform
+    end
+
+    module ClassMethods
+      def perform(*args)
+        new(*args).tap do |command|
+          command.perform
+        end
+      end
+
+      def command_name
+        name.split('::').last.underscore
+      end
+    end
+
+    module Perform
+      def perform
+        catch(:halt) do
+          super
+        end
+      rescue Aws::CloudFormation::Errors::ServiceError, TemplateCompiler::TemplateCompilationFailed => e
+        failed error_message(e)
+      end
+    end
+
+    def success?
+      @failed != true
+    end
+
+    private
+
+    def error_message(e)
+      msg = "#{e.class} #{e.message}"
+      msg << "\n Caused by: #{e.cause.class} #{e.cause.message}" if e.cause
+      msg
+    end
+
+    def failed(message = nil)
+      StackMaster.stderr.puts(message) if message
+      @failed = true
+    end
+
+    def failed!(message = nil)
+      failed(message)
+      halt!
+    end
+
+    def halt!(message = nil)
+      StackMaster.stdout.puts(message) if message
+      throw :halt
+    end
+  end
+end

data/lib/stack_master/commands/apply.rb

@@ -0,0 +1,221 @@
+module StackMaster
+  module Commands
+    class Apply
+      include Command
+      include Commander::UI
+      include StackMaster::Prompter
+      TEMPLATE_TOO_LARGE_ERROR_MESSAGE = 'The (space compressed) stack is larger than the limit set by AWS. See http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/cloudformation-limits.html'.freeze
+
+      def initialize(config, stack_definition, options = Commander::Command::Options.new)
+        @config = config
+        @s3_config = stack_definition.s3
+        @stack_definition = stack_definition
+        @from_time = Time.now
+        @options = options
+        @options.on_failure ||= nil
+      end
+
+      def perform
+        diff_stacks
+        ensure_valid_parameters!
+        ensure_valid_template_body_size!
+        create_or_update_stack
+        tail_stack_events
+        set_stack_policy
+      end
+
+      private
+
+      def cf
+        @cf ||= StackMaster.cloud_formation_driver
+      end
+
+      def s3
+        @s3 ||= StackMaster.s3_driver
+      end
+
+      def stack
+        @stack ||= Stack.find(region, stack_name)
+      end
+
+      def proposed_stack
+        @proposed_stack ||= Stack.generate(@stack_definition, @config)
+      end
+
+      def stack_exists?
+        !stack.nil?
+      end
+
+      def use_s3?
+        !@s3_config.empty?
+      end
+
+      def diff_stacks
+        StackDiffer.new(proposed_stack, stack).output_diff
+      end
+
+      def create_or_update_stack
+        if stack_exists?
+          update_stack
+        else
+          create_stack
+        end
+      end
+
+      def create_stack
+        upload_files
+        if use_change_set?
+          create_stack_by_change_set
+        else
+          create_stack_directly
+        end
+      end
+
+      def use_change_set?
+        @options.on_failure.nil?
+      end
+
+      def create_stack_by_change_set
+        begin
+          @change_set = ChangeSet.create(stack_options.merge(change_set_type: 'CREATE'))
+          if @change_set.failed?
+            ChangeSet.delete(@change_set.id)
+            halt!(@change_set.status_reason)
+          end
+
+          @change_set.display(StackMaster.stdout)
+          unless ask?('Create stack (y/n)? ')
+            cf.delete_stack(stack_name: stack_name)
+            halt!('Stack creation aborted')
+          end
+        rescue StackMaster::CtrlC
+          cf.delete_stack(stack_name: stack_name)
+          raise
+        end
+
+        execute_change_set
+      end
+
+      def create_stack_directly
+        failed!('Stack creation aborted') unless ask?('Create stack (y/n)? ')
+        cf.create_stack(stack_options.merge(on_failure: @options.on_failure))
+      end
+
+      def ask_to_cancel_stack_update
+        if ask?("Cancel stack update?")
+          StackMaster.stdout.puts "Attempting to cancel stack update"
+          cf.cancel_update_stack(stack_name: stack_name)
+          tail_stack_events
+        end
+      end
+
+      def update_stack
+        upload_files
+        @change_set = ChangeSet.create(stack_options)
+        if @change_set.failed?
+          ChangeSet.delete(@change_set.id)
+          halt!(@change_set.status_reason)
+        end
+
+        @change_set.display(StackMaster.stdout)
+        unless ask?("Apply change set (y/n)? ")
+          ChangeSet.delete(@change_set.id)
+          halt! "Stack update aborted"
+        end
+        execute_change_set
+      end
+
+      def upload_files
+        return unless use_s3?
+        s3.upload_files(s3_options)
+      end
+
+      def template_method
+        use_s3? ? :template_url : :template_body
+      end
+
+      def template_value
+        if use_s3?
+          s3.url(bucket: @s3_config['bucket'], prefix: @s3_config['prefix'], region: @s3_config['region'], template: @stack_definition.s3_template_file_name)
+        else
+          proposed_stack.template
+        end
+      end
+
+      def files_to_upload
+        return {} unless use_s3?
+        @stack_definition.s3_files.tap do |files|
+          files[@stack_definition.s3_template_file_name] = {
+            path: @stack_definition.template_file_path,
+            body: proposed_stack.template
+          }
+        end
+      end
+
+      def stack_options
+        {
+          stack_name: stack_name,
+          parameters: proposed_stack.aws_parameters,
+          tags: proposed_stack.aws_tags,
+          capabilities: ['CAPABILITY_IAM', 'CAPABILITY_NAMED_IAM'],
+          role_arn: proposed_stack.role_arn,
+          notification_arns: proposed_stack.notification_arns,
+          template_method => template_value
+        }
+      end
+
+      def s3_options
+        {
+          bucket: @s3_config['bucket'],
+          prefix: @s3_config['prefix'],
+          region: @s3_config['region'],
+          files: files_to_upload
+        }
+      end
+
+      def tail_stack_events
+        StackEvents::Streamer.stream(stack_name, region, io: StackMaster.stdout, from: @from_time)
+      rescue StackMaster::CtrlC
+        ask_to_cancel_stack_update
+      end
+
+      def execute_change_set
+        ChangeSet.execute(@change_set.id, stack_name)
+      rescue StackMaster::CtrlC
+        ask_to_cancel_stack_update
+      end
+
+      def ensure_valid_parameters!
+        if @proposed_stack.missing_parameters?
+          StackMaster.stderr.puts "Empty/blank parameters detected, ensure values exist for those parameters. Parameters will be read from the following locations:"
+          @stack_definition.parameter_files.each do |parameter_file|
+            StackMaster.stderr.puts " - #{parameter_file}"
+          end
+          halt!
+        end
+      end
+
+      def ensure_valid_template_body_size!
+        if proposed_stack.too_big?(use_s3?)
+          failed! TEMPLATE_TOO_LARGE_ERROR_MESSAGE
+        end
+      end
+
+      def set_stack_policy
+        current_policy = stack && stack.stack_policy_body
+        proposed_policy = proposed_stack.stack_policy_body
+        # No need to reset a stack policy if it's nil or not changed
+        return if proposed_policy.nil? || proposed_policy == current_policy
+        StackMaster.stdout.print 'Setting a stack policy...'
+        cf.set_stack_policy(
+          stack_name: stack_name,
+          stack_policy_body: proposed_policy
+        )
+        StackMaster.stdout.puts 'done.'
+      end
+
+      extend Forwardable
+      def_delegators :@stack_definition, :stack_name, :region
+    end
+  end
+end

data/lib/stack_master/commands/delete.rb

@@ -0,0 +1,53 @@
+module StackMaster
+  module Commands
+    class Delete
+      include Command
+      include StackMaster::Prompter
+
+      def initialize(region, stack_name)
+        @region = region
+        @stack_name = stack_name
+        @from_time = Time.now
+      end
+
+      def perform
+
+        return unless check_exists
+
+        unless ask?("Really delete stack #{@stack_name} (y/n)? ")
+          StackMaster.stdout.puts "Stack update aborted"
+          return
+        end
+
+        delete_stack
+        tail_stack_events
+      end
+
+      private
+
+      def delete_stack
+        cf.delete_stack({stack_name: @stack_name})
+      end
+
+      def check_exists
+        cf.describe_stacks({stack_name: @stack_name})
+        true
+      rescue Aws::CloudFormation::Errors::ValidationError
+        StackMaster.stdout.puts "Stack does not exist"
+        false
+      end
+
+      def cf
+        StackMaster.cloud_formation_driver
+      end
+
+      def tail_stack_events
+        StackEvents::Streamer.stream(@stack_name, @region, io: StackMaster.stdout, from: @from_time)
+        StackMaster.stdout.puts "Stack deleted"
+      rescue Aws::CloudFormation::Errors::ValidationError
+        # Unfortunately the stack as a tendency of going away before we get the final delete event.
+        StackMaster.stdout.puts "Stack deleted"
+      end
+    end
+  end
+end

data/lib/stack_master/commands/diff.rb

@@ -0,0 +1,31 @@
+module StackMaster
+  module Commands
+    class Diff
+      include Command
+      include Commander::UI
+
+      def initialize(config, stack_definition, options = {})
+        @config = config
+        @stack_definition = stack_definition
+      end
+
+      def perform
+        StackMaster::StackDiffer.new(proposed_stack, stack).output_diff
+      end
+
+      private
+
+      def stack_definition
+        @stack_definition ||= @config.find_stack(@region, @stack_name)
+      end
+
+      def stack
+        @stack ||= Stack.find(@stack_definition.region, @stack_definition.stack_name)
+      end
+
+      def proposed_stack
+        @proposed_stack ||= Stack.generate(stack_definition, @config)
+      end
+    end
+  end
+end

data/lib/stack_master/commands/events.rb

@@ -0,0 +1,39 @@
+module StackMaster
+  module Commands
+    class Events
+      include Command
+      include Commander::UI
+
+      def initialize(config, stack_definition, options = {})
+        @config = config
+        @stack_definition = stack_definition
+        @options = options
+      end
+
+      def perform
+        events = StackEvents::Fetcher.fetch(@stack_definition.stack_name, @stack_definition.region)
+        filter_events(events).each do |event|
+          StackEvents::Presenter.print_event(StackMaster.stdout, event)
+        end
+        if @options.tail
+          StackEvents::Streamer.stream(@stack_definition.stack_name, @stack_definition.region, io: StackMaster.stdout)
+        end
+      end
+
+      private
+
+      def filter_events(events)
+        if @options.all
+          events
+        else
+          n = @options.number || 25
+          from = events.count - n
+          if from < 0
+            from = 0
+          end
+          events[from..-1]
+        end
+      end
+    end
+  end
+end

data/lib/stack_master/commands/init.rb

@@ -0,0 +1,111 @@
+require "erb"
+
+module StackMaster
+  module Commands
+    class Init
+      include Command
+
+      def initialize(overwrite, region, stack_name)
+        @overwrite = overwrite
+        @region = region
+        @stack_name = stack_name
+      end
+
+      def perform
+        if check_files
+          create_stack_master_yml
+          create_stack_json_yml
+          create_parameters_yml
+        end
+      end
+
+      private
+
+      def check_files
+        @stack_master_filename = "stack_master.yml"
+        @stack_json_filename = "templates/#{@stack_name}.json"
+        @parameters_filename = File.join("parameters", "#{underscored_stack_name}.yml")
+        @region_parameters_filename = File.join("parameters", @region, "#{underscored_stack_name}.yml")
+
+        if !@overwrite
+          [@stack_master_filename, @stack_json_filename, @parameters_filename, @region_parameters_filename].each do |filename|
+            if File.exists?(filename)
+              StackMaster.stderr.puts("Aborting: #{filename} already exists. Use --overwrite to force overwriting file.")
+              return false
+            end
+          end
+        end
+        true
+      end
+
+      def create_stack_json_yml
+        StackMaster.stdout.puts "Writing #{@stack_json_filename}"
+        FileUtils.mkdir_p(File.dirname(@stack_json_filename))
+        IO.write(@stack_json_filename, stack_json_output)
+      end
+
+      def stack_json_output
+        render ERB.new(File.read(stack_json_template))
+      end
+
+      def stack_json_template
+        File.join(StackMaster.base_dir, "stacktemplates", "stack.json.erb")
+      end
+
+      def create_stack_master_yml
+        StackMaster.stdout.puts "Writing #{@stack_master_filename}"
+        IO.write("#{@stack_master_filename}", stack_master_yml_output)
+      end
+
+      def stack_master_yml_output
+        render ERB.new(File.read(stack_master_template))
+      end
+
+      def stack_master_template
+        File.join(StackMaster.base_dir, "stacktemplates", "stack_master.yml.erb")
+      end
+
+      def create_parameters_yml
+        StackMaster.stdout.puts "Writing #{@parameters_filename}"
+        StackMaster.stdout.puts "Writing #{@region_parameters_filename}"
+        FileUtils.mkdir_p("parameters/#{@region}")
+        IO.write(@parameters_filename, parameter_stack_name_yml_output)
+        IO.write(@region_parameters_filename, parameter_region_yml_output)
+      end
+
+      def parameter_stack_name_yml_output
+        File.read(parameter_stack_name_template)
+      end
+
+      def parameter_region_yml_output
+        File.read(parameter_region_template)
+      end
+
+      def parameter_stack_name_template
+        File.join(StackMaster.base_dir, "stacktemplates", "parameter_stack_name.yml")
+      end
+
+      def parameter_region_template
+        File.join(StackMaster.base_dir, "stacktemplates", "parameter_region.yml")
+      end
+
+      def underscored_stack_name
+        @stack_name.gsub('-', '_')
+      end
+
+      def render(renderer)
+        binding = InitBinding.new(region: @region, stack_name: @stack_name).get_binding
+        renderer.result(binding)
+      end
+
+      class InitBinding
+        def initialize(region:, stack_name:)
+          @region = region
+          @stack_name = stack_name
+        end
+
+        attr_reader :region, :stack_name
+      end
+    end
+  end
+end