stack_master 0.0.1

data/lib/stack_master/command.rb

@@ -0,0 +1,13 @@
+module StackMaster
+  module Command
+    def self.included(base)
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      def perform(*args)
+        new(*args).perform
+      end
+    end
+  end
+end

data/lib/stack_master/commands/apply.rb

@@ -0,0 +1,104 @@
+module StackMaster
+  module Commands
+    class Apply
+      include Command
+      include Commander::UI
+      include StackMaster::Prompter
+
+      def initialize(config, stack_definition, options = {})
+        @config = config
+        @stack_definition = stack_definition
+        @from_time = Time.now
+        @updating = false
+      end
+
+      def perform
+        diff_stacks
+        unless ask?("Continue and apply the stack (y/n)? ")
+          StackMaster.stdout.puts "Stack update aborted"
+          return
+        end
+        begin
+          return if stack_too_big
+          create_or_update_stack
+          tail_stack_events
+        rescue StackMaster::CtrlC
+          cancel
+        end
+      end
+
+      private
+
+      def cf
+        @cf ||= StackMaster.cloud_formation_driver
+      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
+
+      def stack_exists?
+        !stack.nil?
+      end
+
+      def diff_stacks
+        StackDiffer.new(proposed_stack, stack).output_diff
+      end
+
+      def cancel
+        if @updating
+          if ask?("Cancel stack update?")
+            StackMaster.stdout.puts "Attempting to cancel stack update"
+            cf.cancel_update_stack({stack_name: @stack_definition.stack_name})
+            tail_stack_events
+          end
+        end
+      end
+
+      def create_or_update_stack
+        if stack_exists?
+          update_stack
+        else
+          create_stack
+        end
+      end
+
+      def stack_too_big
+        if proposed_stack.too_big?
+          StackMaster.stdout.puts 'The (space compressed) stack is larger than the limit set by AWS. See http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/cloudformation-limits.html'
+          true
+        else
+          false
+        end
+      end
+
+      def update_stack
+        @updating = true
+        cf.update_stack(stack_options)
+      end
+
+      def create_stack
+        cf.create_stack(stack_options.merge(tags: proposed_stack.aws_tags))
+      end
+
+      def stack_options
+        {
+          stack_name: @stack_definition.stack_name,
+          template_body: proposed_stack.maybe_compressed_template_body,
+          parameters: proposed_stack.aws_parameters,
+          capabilities: ['CAPABILITY_IAM'],
+          notification_arns: proposed_stack.notification_arns,
+          stack_policy_body: proposed_stack.stack_policy_body
+        }
+      end
+
+      def tail_stack_events
+        StackEvents::Streamer.stream(@stack_definition.stack_name, @stack_definition.region, io: StackMaster.stdout, from: @from_time)
+      end
+    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 (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,109 @@
+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

data/lib/stack_master/commands/list_stacks.rb

@@ -0,0 +1,16 @@
+module StackMaster
+  module Commands
+    class ListStacks
+      include Command
+      include Commander::UI
+
+      def initialize(config)
+        @config = config
+      end
+
+      def perform
+        tp @config.stacks, :region, :stack_name
+      end
+    end
+  end
+end

data/lib/stack_master/commands/outputs.rb

@@ -0,0 +1,27 @@
+module StackMaster
+  module Commands
+    class Outputs
+      include Command
+      include Commander::UI
+
+      def initialize(config, stack_definition, options = {})
+        @config = config
+        @stack_definition = stack_definition
+      end
+
+      def perform
+        if stack
+          tp stack.outputs, :output_key, :output_value, :description
+        else
+          StackMaster.stdout.puts "Stack doesn't exist"
+        end
+      end
+
+      private
+
+      def stack
+        @stack ||= Stack.find(@stack_definition.region, @stack_definition.stack_name)
+      end
+    end
+  end
+end

data/lib/stack_master/commands/resources.rb

@@ -0,0 +1,33 @@
+module StackMaster
+  module Commands
+    class Resources
+      include Command
+      include Commander::UI
+
+      def initialize(config, stack_definition, options = {})
+        @config = config
+        @stack_definition = stack_definition
+      end
+
+      def perform
+        if stack_resources
+          tp stack_resources, :logical_resource_id, :resource_type, :timestamp, :resource_status, :resource_status_reason, :description
+        else
+          StackMaster.stdout.puts "Stack doesn't exist"
+        end
+      end
+
+      private
+
+      def stack_resources
+        @stack_resources = cf.describe_stack_resources(stack_name: @stack_definition.stack_name).stack_resources
+      rescue Aws::CloudFormation::Errors::ValidationError
+        nil
+      end
+
+      def cf
+        @cf ||= StackMaster.cloud_formation_driver
+      end
+    end
+  end
+end

data/lib/stack_master/commands/status.rb

@@ -0,0 +1,47 @@
+module StackMaster
+  module Commands
+    class Status
+      include Command
+
+      def initialize(config)
+        @config = config
+      end
+
+      def perform
+        tp.set :io, StackMaster.stdout
+        tp @config.stacks.map { |stack_definition| get_status(stack_definition) }
+      end
+
+      private
+
+      def sort_params(hash)
+        hash.sort.to_h
+      end
+
+      def get_status(stack_definition)
+        region = stack_definition.region
+        stack_name = stack_definition.stack_name
+        begin
+          driver = StackMaster.cloud_formation_driver
+          driver.set_region(region)
+          stack = Stack.find(region, stack_name)
+          if stack
+            proposed_stack = Stack.generate(stack_definition, @config)
+            differ = StackMaster::StackDiffer.new(proposed_stack, stack)
+            different = differ.body_different? || differ.params_different?
+            stack_status = stack.stack_status
+          else
+            different = true
+            stack_status = nil
+          end
+        rescue Aws::CloudFormation::Errors::ValidationError
+          stack_status = "missing"
+          different = true
+        end
+
+        { region: region, stack_name: stack_name, stack_status: stack_status, different: different ? "Yes" : "No" }
+      end
+
+    end
+  end
+end