amigrind 0.1.1

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/amigrind.gemspec

@@ -0,0 +1,36 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'amigrind/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "amigrind"
+  spec.version       = Amigrind::VERSION
+  spec.authors       = ["Ed Ropple"]
+  spec.email         = ["ed@edropple.com"]
+
+  spec.summary       = "An easy, convention-over-configuration builder for Packer images."
+  spec.homepage      = "https://github.com/eropple/amigrind"
+  spec.license       = "Apache 2.0"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "bin"
+  spec.executables   = [ 'amigrind' ]
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.11"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+
+  spec.add_runtime_dependency 'cri', '~> 2.7.0'
+  spec.add_runtime_dependency 'os', '~> 0.9.6'
+  spec.add_runtime_dependency 'racker', '~> 0.2.0'
+  spec.add_runtime_dependency 'virtus', '~> 1.0.5'
+  spec.add_runtime_dependency 'activesupport', '~> 4.2.6'
+  spec.add_runtime_dependency 'activemodel', '~> 4.2.6'
+  spec.add_runtime_dependency 'erubis', '~> 2.7.0'
+  spec.add_runtime_dependency 'ptools', '~> 1.3', '>= 1.3.3'
+  spec.add_runtime_dependency 'settingslogic', '~> 2.0.9'
+
+  spec.add_runtime_dependency 'amigrind-core', '~> 0.1.0'
+end

data/bin/amigrind

@@ -0,0 +1,8 @@
+#! /usr/bin/env ruby
+
+lib = File.expand_path(File.dirname(__FILE__) + '/../lib')
+$LOAD_PATH.unshift(lib) if File.directory?(lib) && !$LOAD_PATH.include?(lib)
+
+require 'amigrind/cli'
+
+Amigrind::CLI.run(ARGV)

data/lib/amigrind.rb

@@ -0,0 +1,29 @@
+require 'amigrind/version'
+require 'amigrind/core'
+require 'amigrind/config'
+require 'amigrind/repo'
+
+require 'ptools'
+require 'open3'
+require 'fileutils'
+require 'aws-sdk'
+require 'racker'
+require 'json'
+require 'yaml'
+require 'ice_nine'
+require 'set'
+require 'erubis'
+require 'virtus'
+require 'active_support/core_ext/string/inflections'
+require 'active_support/core_ext/hash'
+require 'active_support/core_ext/numeric/time'
+
+# Pre-includes
+module Amigrind
+end
+
+Dir["#{__dir__}/**/*.rb"].reject { |f| f.include?('/cli') }.each { |f| require_relative f }
+
+# Post-includes
+module Amigrind
+end

data/lib/amigrind/blueprints/aws_config.rb

@@ -0,0 +1,56 @@
+module Amigrind
+  module Blueprints
+    class AWSConfig
+      include Virtus.model(constructor: false, mass_assignment: false)
+
+      class BlockDeviceMapping
+      end
+
+      attribute :instance_type, String
+      # TODO: ssh_username is currently required for all builds, but should
+      #       instead inherit from the parent blueprint if one exists. This
+      #       is annoying and recursive right now, and a small pain point,
+      #       so I've punted it.
+      attribute :ssh_username, String
+
+      attribute :region, String
+      attribute :copy_regions, Array[String]
+      attribute :associate_public_ip_address, Boolean
+      attribute :ebs_optimized, Boolean
+      attribute :enhanced_networking, Boolean
+      attribute :iam_instance_profile, String
+      attribute :ssh_keypair_name, String
+      attribute :ssh_private_ip, Boolean
+      attribute :user_data, String
+      attribute :windows_password_timeout, ActiveSupport::Duration
+
+      attribute :run_tags, Hash[String => String]
+      attribute :run_volume_tags, Hash[String => String]
+      attribute :security_group_ids, Array[String]
+
+      attribute :vpc_id, String
+      attribute :subnet_ids, Array[String]
+
+      # TODO: object-ize the ami block device mappings
+      attribute :ami_block_device_mappings, Array[BlockDeviceMapping]
+      attribute :launch_block_device_mappings, Array[BlockDeviceMapping]
+
+      attr_reader :custom
+
+      def initialize
+        @copy_regions = []
+
+        @run_tags = {}
+        @run_volume_tags = {}
+        @security_group_ids = []
+
+        @subnet_ids = []
+
+        @ami_block_device_mappings = []
+        @launch_block_device_mappings = []
+
+        @custom = {}
+      end
+    end
+  end
+end

data/lib/amigrind/blueprints/base_ami_source.rb

@@ -0,0 +1,15 @@
+module Amigrind
+  module Blueprints
+    class BaseAMISource
+      include Virtus.model(constructor: false, mass_assignment: false)
+
+      attribute :family, Symbol
+      attribute :version, String
+      attribute :ids, Hash[String => String]
+
+      def initialize
+        @ids = {}
+      end
+    end
+  end
+end

data/lib/amigrind/blueprints/blueprint.rb

@@ -0,0 +1,22 @@
+require 'amigrind/blueprints/provisioner'
+
+module Amigrind
+  module Blueprints
+    class Blueprint
+      include Virtus.model(constructor: false, mass_assignment: false)
+
+      attribute :name, String
+      attribute :build_channel, Symbol
+      attribute :description, String, default: ''
+      attr_accessor :source
+      attribute :aws, Amigrind::Blueprints::AWSConfig
+
+      attribute :provisioners, Array[Amigrind::Blueprints::Provisioner], default: []
+
+      def initialize
+        @aws = AWSConfig.new
+        @provisioners = []
+      end
+    end
+  end
+end

data/lib/amigrind/blueprints/evaluator.rb

@@ -0,0 +1,269 @@
+module Amigrind
+  module Blueprints
+    class Evaluator
+      include Amigrind::Core::Logging::Mixin
+      include Amigrind::Blueprints::Provisioners
+
+      attr_reader :blueprint
+
+      def initialize(filename, environment = nil)
+        @blueprint = Blueprint.new
+        @blueprint.name = File.basename(filename, ".rb")
+
+        regex = Amigrind::Core::BLUEPRINT_NAME_REGEX
+        raise "blueprint name (#{@blueprint.name}) must match #{regex.source}" \
+          unless regex.match(@blueprint.name)
+
+        @properties =
+          if environment.nil?
+            debug_log "no environment found to use with blueprint"
+            {}
+          else
+            debug_log "using environment '#{environment.name}' with blueprint"
+            environment.properties.merge(environment_name: environment.name)
+          end
+
+        unless environment.nil?
+          @blueprint.aws.vpc_id = environment.aws.vpc
+          @blueprint.aws.subnet_ids = environment.aws.subnets
+
+          @blueprint.aws.region = environment.aws.region
+          @blueprint.aws.copy_regions = environment.aws.copy_regions
+          @blueprint.aws.ssh_keypair_name = environment.aws.ssh_keypair_name
+        end
+
+        instance_eval(IO.read(filename), File.expand_path(filename), 0)
+      end
+
+      def self.evaluate(filename, environment = nil)
+        Evaluator.new(filename, environment).blueprint
+      end
+
+      def properties
+        @properties
+      end
+
+      private
+
+      def build_channel(b)
+        raise "'build_channel' must be a String or Symbol." \
+          unless b.is_a?(String) || b.is_a?(Symbol)
+
+        @blueprint.build_channel = b
+      end
+
+      def description(d)
+        raise "'description' must be a String." unless d.is_a?(String)
+        @blueprint.description = d
+      end
+
+      def parent_blueprint(p)
+        raise "'parent_blueprint' must be a String." unless p.is_a?(String)
+        @blueprint.source = p
+      end
+
+      def source(type, &block)
+        case type
+        when :ami
+          BaseAMIEvaluator.new(@blueprint, self, &block)
+        when :parent
+          ParentBlueprintEvaluator.new(@blueprint, self, &block)
+        else
+          raise "Invalid source type: #{type} (must be :ami, :parent)"
+        end
+      end
+
+      def aws(&block)
+        AWSConfigEvaluator.new(@blueprint, self, &block)
+      end
+
+      def provisioner(name, provisioner_class, weight: nil, &block)
+        weight ||= (@blueprint.provisioners.max_by(&:weight) || 0) + 1
+
+        raise "'name' must be a String or Symbol." \
+          unless name.is_a?(String) || name.is_a?(Symbol)
+        raise "'provisioner_class' must inherit from Amigrind::Blueprints::Provisioner" \
+          unless provisioner_class.ancestors.include?(Amigrind::Blueprints::Provisioner)
+        raise "'weight' must be a Fixnum." unless weight.is_a?(Fixnum)
+
+        @blueprint.provisioners <<
+          ProvisionerEvaluator.new(name, self, weight, provisioner_class, &block).provisioner
+      end
+
+      class BaseAMIEvaluator
+        def initialize(bp, evaluator, &block)
+          @bp = bp
+          @bp.source = Amigrind::Blueprints::BaseAMISource.new
+
+          @evaluator = evaluator
+
+          instance_eval(&block)
+        end
+
+        private
+
+        def properties
+          @evaluator.properties
+        end
+
+        def family(f)
+          raise "'family' must implement #to_sym." unless f.respond_to?(:to_sym)
+
+          @bp.source.family = f.to_sym.to_s.strip.to_sym
+        end
+
+        def version(v)
+          raise "'version' must be a String." unless v.is_a?(String)
+
+          @bp.source.version = v.strip
+        end
+
+        def id(region, image_id)
+          regex = Amigrind::Core::AMI_REGEX
+
+          raise "'region' must be stringable." unless region.respond_to?(:to_s)
+          raise "'image_id' must be in AMI format (/#{regex.source}/)" \
+            unless regex.match(image_id)
+
+          @bp.source.ids[region.to_s.strip] = image_id
+        end
+
+        def ids(ami_ids)
+          raise "'ami_ids' must be a Hash-alike." unless ami_ids.respond_to?(:each_pair)
+          ami_ids.each_pair { |region, image_id| id(region, image_id) }
+        end
+      end
+
+      class ParentBlueprintEvaluator
+        def initialize(bp, evaluator, &block)
+          @bp = bp
+          @bp.source = Amigrind::Blueprints::ParentBlueprintSource.new
+
+          @evaluator = evaluator
+
+          instance_eval(&block)
+        end
+
+        private
+
+        def properties
+          @evaluator.properties
+        end
+
+        def name(n)
+          raise "'name' must be a String." unless n.is_a?(String)
+
+          @bp.source.name = n
+        end
+
+        def channel(c)
+          raise "'channel' must implement #to_sym." unless c.respond_to?(:to_sym)
+
+          @bp.source.channel = c
+        end
+      end
+
+      class AWSConfigEvaluator
+        def initialize(bp, evaluator, &block)
+          @bp = bp
+
+          @evaluator = evaluator
+
+          instance_eval(&block)
+        end
+
+        private
+
+        def properties
+          @evaluator.properties
+        end
+
+        def run_tag(key, value)
+          raise "'key' must be a String." unless key.is_a?(String)
+          raise "'value' must be stringable." unless value.respond_to?(:to_s)
+
+          @bp.aws.run_tags[key] = value.to_s
+        end
+
+        def run_tags(tags)
+          raise "'tags' must be a Hash." unless tags.respond_to?(:each_pair)
+          tags.each_pair { |key, value| run_tag(key, value) }
+        end
+
+        def run_volume_tag(key, value)
+          raise "'key' must be a String." unless key.is_a?(String)
+          raise "'value' must be stringable." unless value.respond_to?(:to_s)
+
+          @bp.aws.run_volume_tags[key] = value.to_s
+        end
+
+        def run_volume_tags(tags)
+          raise "'tags' must be a Hash." unless tags.respond_to?(:each_pair)
+          tags.each_pair { |key, value| run_volume_tag(key, value) }
+        end
+
+        def vpc(vpc_id)
+          regex = Amigrind::Core::VPC_REGEX
+          raise "'vpc_id' must be a resource (#{regex.source})" unless regex.match(vpc_id)
+
+          @bp.aws.vpc_id = vpc_id
+        end
+
+        def subnet(subnet_id)
+          regex = Amigrind::Core::SUBNET_REGEX
+          raise "'subnet_id' must be a resource (#{regex.source})" unless regex.match(subnet_id)
+
+          @bp.aws.subnet_ids << subnet_id
+        end
+
+        def security_group(sg_id)
+          regex = Amigrind::Core::SG_REGEX
+          raise "'sg_id' must be a resource (#{regex.source})." unless regex.match(sg_id)
+
+          @bp.aws.security_group_ids << sg_id
+        end
+
+        def custom(key, value)
+          raise "custom 'key' must be a Symbol." unless key.is_a?(Symbol)
+          raise "custom 'value' must be non-nil." if value.nil?
+
+          @bp.aws.custom[key] = value
+        end
+
+        def method_missing(m, *args)
+          @bp.aws.send(:"#{m}=", args[0])
+        end
+      end
+    end
+
+    class ProvisionerEvaluator
+      attr_reader :provisioner
+
+      def initialize(name, evaluator, weight, provisioner_class, &block)
+        @provisioner = provisioner_class.new
+        @provisioner.name = name.to_s
+        @provisioner.weight = weight
+
+        @evaluator = evaluator
+
+        instance_eval(&block)
+      end
+
+      def method_missing(m, *args)
+        eq_msg = :"#{m}="
+
+        if @provisioner.respond_to?(eq_msg)
+          @provisioner.send(eq_msg, args[0])
+        else
+          @provisioner.send(m, *args)
+        end
+      end
+
+      private
+
+      def properties
+        @evaluator.properties
+      end
+    end
+  end
+end