stack_master 0.6.0 → 0.7.0

data/lib/stack_master/commands/init.rb

@@ -1,3 +1,5 @@
+require "erb"
+
 module StackMaster
   module Commands
     class Init

data/lib/stack_master/commands/list_stacks.rb

@@ -1,3 +1,5 @@
+require 'table_print'
+
 module StackMaster
   module Commands
     class ListStacks

data/lib/stack_master/commands/outputs.rb

@@ -1,3 +1,5 @@
+require 'table_print'
+
 module StackMaster
   module Commands
     class Outputs

data/lib/stack_master/commands/status.rb

@@ -1,3 +1,6 @@
+require 'table_print'
+require 'ruby-progressbar'
+
 module StackMaster
   module Commands
     class Status

data/lib/stack_master/parameter_resolver.rb

@@ -42,7 +42,7 @@ module StackMaster
     end
 
     def resolve_parameter_value(key, parameter_value)
-      return parameter_value.to_s if Numeric === parameter_value
+      return parameter_value.to_s if Numeric === parameter_value || parameter_value == true || parameter_value == false
       return parameter_value.join(',') if Array === parameter_value
       return parameter_value unless Hash === parameter_value
       validate_parameter_value!(key, parameter_value)

data/lib/stack_master/parameter_resolvers/secret.rb

@@ -1,3 +1,5 @@
+require 'dotgpg'
+
 module StackMaster
   module ParameterResolvers
     class Secret < Resolver

data/lib/stack_master/stack.rb

@@ -1,19 +1,21 @@
 module StackMaster
   class Stack
     MAX_TEMPLATE_SIZE = 51200
+    MAX_S3_TEMPLATE_SIZE = 460800
 
-    include Virtus.model
+    attr_reader :stack_name,
+                :region,
+                :stack_id,
+                :stack_status,
+                :parameters,
+                :template_body,
+                :notification_arns,
+                :outputs,
+                :stack_policy_body,
+                :tags,
+                :files
 
-    attribute :stack_name, String
-    attribute :region, String
-    attribute :stack_id, String
-    attribute :stack_status, String
-    attribute :parameters, Hash
-    attribute :template_body, String
-    attribute :notification_arns, Array[String]
-    attribute :outputs, Array
-    attribute :stack_policy_body, String
-    attribute :tags, Hash
+    include Utils::Initializable
 
     def template_hash
       if template_body
@@ -87,8 +89,13 @@ module StackMaster
           stack_policy_body: stack_policy_body)
     end
 
-    def too_big?
-      maybe_compressed_template_body.size > MAX_TEMPLATE_SIZE
+    def max_template_size(use_s3)
+      return MAX_S3_TEMPLATE_SIZE if use_s3
+      MAX_TEMPLATE_SIZE
+    end
+
+    def too_big?(use_s3 = false)
+      maybe_compressed_template_body.size > max_template_size(use_s3)
     end
 
     def aws_parameters

data/lib/stack_master/stack_definition.rb

@@ -1,34 +1,84 @@
 module StackMaster
   class StackDefinition
-    include Virtus.value_object(strict: true, required: false)
+    attr_accessor :region,
+                  :stack_name,
+                  :template,
+                  :tags,
+                  :notification_arns,
+                  :base_dir,
+                  :secret_file,
+                  :stack_policy_file,
+                  :additional_parameter_lookup_dirs,
+                  :s3,
+                  :files
 
-    values do
-      attribute :region, String
-      attribute :stack_name, String
-      attribute :template, String
-      attribute :tags, Hash
-      attribute :notification_arns, Array[String]
-      attribute :base_dir, String
-      attribute :secret_file, String
-      attribute :stack_policy_file, String
-      attribute :additional_parameter_lookup_dirs, Array[String]
+    include Utils::Initializable
+
+    def initialize(attributes = {})
+      @additional_parameter_lookup_dirs = []
+      @notification_arns = []
+      @s3 = {}
+      @files = []
+      super
+    end
+
+    def ==(other)
+      self.class === other &&
+        @region == other.region &&
+        @stack_name == other.stack_name &&
+        @template == other.template &&
+        @tags == other.tags &&
+        @notification_arns == other.notification_arns &&
+        @base_dir == other.base_dir &&
+        @secret_file == other.secret_file &&
+        @stack_policy_file == other.stack_policy_file &&
+        @additional_parameter_lookup_dirs == other.additional_parameter_lookup_dirs &&
+        @s3 == other.s3
+    end
+
+    def template_dir
+      File.join(base_dir, 'templates')
     end
 
     def template_file_path
-      File.join(base_dir, 'templates', template)
+      File.join(template_dir, template)
+    end
+
+    def files_dir
+      File.join(base_dir, 'files')
+    end
+
+    def s3_files
+      files.inject({}) do |hash, file|
+        path = File.join(files_dir, file)
+        hash[file] = {
+          path: path,
+          body: File.read(path)
+        }
+        hash
+      end
+    end
+
+    def s3_template_file_name
+      Utils.change_extension(template, 'json')
     end
 
     def parameter_files
-      [ default_parameter_file_path, region_parameter_file_path ] + additional_parameter_lookup_file_paths
+      [ default_parameter_file_path, region_parameter_file_path, additional_parameter_lookup_file_paths ].flatten.compact
     end
 
     def stack_policy_file_path
       File.join(base_dir, 'policies', stack_policy_file) if stack_policy_file
     end
 
+    def s3_configured?
+      !s3.nil?
+    end
+
     private
 
     def additional_parameter_lookup_file_paths
+      return unless additional_parameter_lookup_dirs
       additional_parameter_lookup_dirs.map do |a|
         File.join(base_dir, 'parameters', a, "#{underscored_stack_name}.yml")
       end

data/lib/stack_master/stack_differ.rb

@@ -1,3 +1,5 @@
+require "diffy"
+
 module StackMaster
   class StackDiffer
     def initialize(proposed_stack, current_stack)

data/lib/stack_master/stack_events/fetcher.rb

@@ -1,5 +1,5 @@
 module StackMaster
-  class StackEvents
+  module StackEvents
     class Fetcher
       def self.fetch(*args)
         new(*args).fetch

data/lib/stack_master/stack_events/presenter.rb

@@ -1,5 +1,5 @@
 module StackMaster
-  class StackEvents
+  module StackEvents
     class Presenter
       def self.print_event(io, event)
         new(io).print_event(event)

data/lib/stack_master/stack_events/streamer.rb

@@ -1,5 +1,5 @@
 module StackMaster
-  class StackEvents
+  module StackEvents
     class Streamer
       def self.stream(*args, &block)
         new(*args, &block).stream

data/lib/stack_master/template_compiler.rb

@@ -3,7 +3,9 @@ module StackMaster
     TemplateCompilationFailed = Class.new(RuntimeError)
 
     def self.compile(config, template_file_path)
-      template_compiler_for_file(template_file_path, config).compile(template_file_path)
+      compiler = template_compiler_for_file(template_file_path, config)
+      compiler.require_dependencies
+      compiler.compile(template_file_path)
     rescue
       raise TemplateCompilationFailed.new("Failed to compile #{template_file_path}.")
     end

data/lib/stack_master/template_compilers/cfndsl.rb

@@ -1,9 +1,11 @@
 module StackMaster::TemplateCompilers
   class Cfndsl
+    def self.require_dependencies
+      require 'cfndsl'
+    end
 
     def self.compile(template_file_path)
-      require 'cfndsl'
-      CfnDsl.eval_file_with_extras(template_file_path).to_json
+      ::CfnDsl.eval_file_with_extras(template_file_path).to_json
     end
 
     StackMaster::TemplateCompiler.register(:cfndsl, self)

data/lib/stack_master/template_compilers/json.rb

@@ -3,6 +3,10 @@ module StackMaster::TemplateCompilers
     MAX_TEMPLATE_SIZE = 51200
     private_constant :MAX_TEMPLATE_SIZE
 
+    def self.require_dependencies
+      require 'json'
+    end
+
     def self.compile(template_file_path)
       template_body = File.read(template_file_path)
       if template_body.size > MAX_TEMPLATE_SIZE

data/lib/stack_master/template_compilers/sparkle_formation.rb

@@ -1,5 +1,10 @@
 module StackMaster::TemplateCompilers
   class SparkleFormation
+    def self.require_dependencies
+      require 'sparkle_formation'
+      require 'stack_master/sparkle_formation/user_data_file'
+    end
+
     def self.compile(template_file_path)
       ::SparkleFormation.sparkle_path = File.dirname(template_file_path)
       JSON.pretty_generate(::SparkleFormation.compile(template_file_path))

data/lib/stack_master/template_compilers/yaml.rb

@@ -1,5 +1,9 @@
 module StackMaster::TemplateCompilers
   class Yaml
+    def self.require_dependencies
+      require 'yaml'
+      require 'json'
+    end
 
     def self.compile(template_file_path)
       template_body = File.read(template_file_path)

data/lib/stack_master/test_driver/cloud_formation.rb

@@ -1,48 +1,62 @@
 module StackMaster
   module TestDriver
     class Stack
-      include Virtus.model
-      attribute :stack_id, String
-      attribute :stack_name, String
-      attribute :description, String
-      attribute :parameters, Array[OpenStruct]
-      attribute :creation_time, String
-      attribute :last_update_time, String
-      attribute :stack_status, String
-      attribute :stack_status_reason, String
-      attribute :disable_rollback, String
-      attribute :notification_arns, Array
-      attribute :timeout_in_minutes, Integer
-      attribute :capabilities, Array
-      attribute :outputs, Array[OpenStruct]
-      attribute :tags, Array[OpenStruct]
+      attr_reader :stack_id,
+                  :stack_name,
+                  :description,
+                  :parameters,
+                  :creation_time,
+                  :last_update_time,
+                  :stack_status,
+                  :stack_status_reason,
+                  :disable_rollback,
+                  :notification_arns,
+                  :timeout_in_minutes,
+                  :capabilities,
+                  :outputs,
+                  :tags
+
+      include Utils::Initializable
+
+      def parameters
+        @parameters.map do |hash|
+          OpenStruct.new(parameter_key: hash[:parameter_key],
+                         parameter_value: hash[:parameter_value])
+        end
+      end
     end
 
     class StackEvent
-      include Virtus.model
-      attribute :stack_id, String
-      attribute :event_id, String
-      attribute :stack_name, String
-      attribute :logical_resource_id, String
-      attribute :physical_resource_id, String
-      attribute :resource_type, String
-      attribute :timestamp, Time
-      attribute :resource_status, String
-      attribute :resource_status_reason, String
-      attribute :resource_properties, String
+      attr_reader :stack_id,
+                  :event_id,
+                  :stack_name,
+                  :logical_resource_id,
+                  :physical_resource_id,
+                  :resource_type,
+                  :timestamp,
+                  :resource_status,
+                  :resource_status_reason,
+                  :resource_properties
+
+      include Utils::Initializable
+
+      def timestamp
+        Time.parse(@timestamp) if @timestamp
+      end
     end
 
     class StackResource
-      include Virtus.model
-      attribute :stack_name, String
-      attribute :stack_id, String
-      attribute :logical_resource_id, String
-      attribute :physical_resource_id, String
-      attribute :resource_type, String
-      attribute :timestamp, Time
-      attribute :resource_status, String
-      attribute :resource_status_reason, String
-      attribute :description, String
+      attr_reader :stack_name,
+                  :stack_id,
+                  :logical_resource_id,
+                  :physical_resource_id,
+                  :resource_type,
+                  :timestamp,
+                  :resource_status,
+                  :resource_status_reason,
+                  :description
+
+      include Utils::Initializable
     end
 
     class CloudFormation

data/lib/stack_master/test_driver/s3.rb

@@ -0,0 +1,34 @@
+module StackMaster
+  module TestDriver
+    class S3
+      def initialize
+        reset
+      end
+
+      def set_region(_)
+      end
+
+      def reset
+        @files = Hash.new { |hash, key| hash[key] = Hash.new }
+      end
+
+      def upload_files(bucket: nil, prefix: nil, region: nil, files: {})
+        return if files.empty?
+
+        files.each do |template, file|
+          object_key = [prefix, template].compact.join('/')
+          @files[bucket][object_key] = file[:body]
+        end
+      end
+
+      def url(bucket:, prefix:, region:, template:)
+        ["https://s3-#{region}.amazonaws.com", bucket, prefix, template].compact.join('/')
+      end
+
+      # test only method
+      def find_file(bucket:, object_key:)
+        @files[bucket][object_key]
+      end
+    end
+  end
+end

data/lib/stack_master/testing.rb

@@ -5,4 +5,5 @@ Aws.config[:stub_responses] = true
 require 'stack_master/test_driver/cloud_formation'
 
 StackMaster.cloud_formation_driver = StackMaster::TestDriver::CloudFormation.new
+StackMaster.s3_driver = StackMaster::TestDriver::S3.new
 StackMaster.non_interactive!

data/lib/stack_master/utils.rb

@@ -1,7 +1,26 @@
 module StackMaster
   module Utils
+    module Initializable
+      def initialize(attributes = {})
+        self.attributes = attributes
+      end
+
+      def attributes=(attributes)
+        attributes.each do |k, v|
+          instance_variable_set("@#{k}", v)
+        end
+      end
+    end
+
     extend self
 
+    def change_extension(file_name, extension)
+      [
+        File.basename(file_name, '.*'),
+        extension
+      ].join('.')
+    end
+
     def hash_to_aws_parameters(params)
       params.inject([]) do |params, (key, value)|
         params << { parameter_key: key, parameter_value: value }