image_builder 0.0.1

data/lib/image_builder/backends/backend_base.rb

@@ -0,0 +1,10 @@
+require_relative '../../image_builder.rb'
+
+module ImageBuilder
+  module Backends
+    # Generic class doc comment
+    class Base
+      include ImageBuilder
+    end
+  end
+end

data/lib/image_builder/backends/packer.rb

@@ -0,0 +1,109 @@
+require_relative './backend_base.rb'
+require 'json'
+require 'mixlib/shellout'
+require 'tempfile'
+
+# rubocop:disable Style/ClassVars
+
+module ImageBuilder
+  module Backends
+    # Generic class doc comment
+    class Packer < ImageBuilder::Backends::Base
+      attr_accessor :description
+      attr_accessor :min_packer_version
+      attr_accessor :builders
+      attr_accessor :post_processors
+      attr_accessor :provisioners
+      attr_accessor :variables
+
+      @@default_path = ['/bin', '/usr/bin', '/sbin', '/usr/sbin', '/usr/local/bin']
+      @@default_path << ::File.join(ENV['GEM_HOME'], 'bin') if ENV.key? 'GEM_HOME'
+
+      @@packer_exe   = 'packer'
+
+      def initialize
+        @builders = []
+        @post_processors = []
+        @provisioners = []
+        @variables = {}
+      end
+
+      def self.packer_path(path)
+        @@packer_exe = ::File.expand_path(path)
+        @@default_path.unshift ::File.dirname(::File.expand_path(@@packer_exe))
+      end
+
+      def self.packer_path=(path)
+        packer_path(path)
+      end
+
+      # rubocop:disable Metrics/LineLength
+      def add_builder(builder_obj)
+        fail ArgumentError, 'Invalid builder object, does not respond to packer_hash()' unless builder_obj.respond_to?(:packer_hash)
+        @builders << builder_obj.packer_hash
+      end
+
+      def add_post_processor(post_obj)
+        fail ArgumentError, 'Invalid post-processor object, does not respond to packer_hash()' unless post_obj.respond_to?(:packer_hash)
+        @post_processors << post_obj.packer_hash
+      end
+
+      def add_provisioner(provisioner_obj)
+        fail ArgumentError, 'Invalid provisioner object, does not respond to packer_hash()' unless provisioner_obj.respond_to?(:packer_hash)
+        @provisioners << provisioner_obj.packer_hash
+      end
+      # rubocop:enable Metrics/LineLength
+
+      def add_variable(name, value = '')
+        @variables[name] = value
+      end
+
+      def build(args = [])
+        validate
+
+        cmd = [@@packer_exe, 'build'].concat(args)
+        cmd << '-'
+
+        # Some provisioners may take a long time to run without giving output (applying OS updates via Chef
+        # to CentOS6 comes to mind), so extend the timeout to an hour.
+        shellout = Mixlib::ShellOut.new(cmd, env: { 'PATH' => @@default_path.join(':') }, input: to_json, timeout: 3600)
+        shellout.live_stdout = STDOUT
+        shellout.live_stderr = STDERR
+        shellout.run_command
+        shellout.error!
+
+        true
+      end
+
+      def validate(args = [])
+        cmd = [@@packer_exe, 'validate'].concat(args)
+        cmd << '-'
+
+        shellout = Mixlib::ShellOut.new(cmd, env: { 'PATH' => @@default_path.join(':') }, input: to_json)
+        shellout.inspect
+        shellout.run_command
+        shellout.error!
+
+        shellout.stdout
+
+        true
+      end
+
+      def to_json
+        config = {}
+
+        attr_to_hash(config, :description)
+        attr_to_hash(config, :min_packer_version)
+        attr_to_hash(config, :variables)
+        attr_to_hash(config, :builders, true)
+        attr_to_hash(config, :provisioners)
+
+        # Special handling for post-processors, since the packer config name uses '-', and ruby doesn't
+        # like those (you know....subtraction operator, and all that)
+        config['post-processors'] = @post_processors unless @post_processors.nil? || @post_processors.empty?
+
+        config.to_json
+      end
+    end
+  end
+end

data/lib/image_builder/builders/aws_base.rb

@@ -0,0 +1,101 @@
+require_relative './builder_base.rb'
+
+module ImageBuilder
+  module Builders
+    module AWS
+      # Generic class doc comment
+      class Base < ImageBuilder::Builders::Base
+        attr_accessor :access_key
+        attr_accessor :ami_name
+        attr_accessor :instance_type
+        attr_accessor :region
+        attr_accessor :secret_key
+        attr_accessor :source_ami
+        attr_accessor :ssh_username
+        # - opt -
+        attr_accessor :ami_block_device_mappings
+        attr_accessor :ami_description
+        attr_accessor :ami_groups
+        attr_accessor :ami_product_codes
+        attr_accessor :ami_regions
+        attr_accessor :ami_users
+        attr_accessor :ami_virtualization_type
+        attr_accessor :associate_public_ip_address
+        attr_accessor :availability_zone
+        attr_accessor :enhanced_networking
+        attr_accessor :iam_instance_profile
+        attr_accessor :launch_block_device_mappings
+        attr_accessor :run_tags
+        attr_accessor :security_group_id
+        attr_accessor :security_group_ids
+        attr_accessor :spot_price
+        attr_accessor :spot_price_auto_product
+        attr_accessor :ssh_port
+        attr_accessor :ssh_private_key_file
+        attr_accessor :ssh_private_ip
+        attr_accessor :ssh_timeout
+        attr_accessor :subnet_id
+        attr_accessor :tags
+        attr_accessor :temporary_key_pair_name
+        attr_accessor :token
+        attr_accessor :user_data
+        attr_accessor :user_data_file
+        attr_accessor :vpc_id
+
+        attr_reader :type
+
+        def initialize
+          @access_key = ENV['AWS_ACCESS_KEY_ID'] || ENV['AWS_ACCESS_KEY']
+          @secret_key = ENV['AWS_SECRET_ACCESS_KEY'] || ENV['AWS_SECRET_KEY']
+          @region = 'us-east-1'
+        end
+
+        def self.default_block_device_mappings
+          [
+            { device_name: '/dev/sdf', virtual_name: 'ephemeral0' },
+            { device_name: '/dev/sdg', virtual_name: 'ephemeral1' },
+            { device_name: '/dev/sdh', virtual_name: 'ephemeral2' },
+            { device_name: '/dev/sdi', virtual_name: 'ephemeral3' }
+          ]
+        end
+
+        def self.default_launch_block_device_mappings
+          [
+            { device_name: '/dev/sdb', no_device: true },
+            { device_name: '/dev/sdc', no_device: true },
+            { device_name: '/dev/sdd', no_device: true }
+          ]
+        end
+
+        def packer_hash
+          hash = {}
+
+          req_attrs = [
+            :type, :name, :access_key, :ami_name, :instance_type,
+            :region, :secret_key, :source_ami, :ssh_username
+          ]
+
+          opt_attrs = [
+            :ami_description, :ami_groups, :ami_product_codes, :ami_regions, :ami_regions, :ami_users,
+            :ami_virtualization_type, :associate_public_ip_address, :availability_zone, :enhanced_networking,
+            :iam_instance_profile, :run_tags, :security_group_id, :security_group_ids, :spot_price,
+            :spot_price_auto_product, :ssh_port, :ssh_private_key_file, :ssh_private_ip, :ssh_timeout,
+            :subnet_id, :tags, :temporary_key_pair_name, :token, :user_data, :user_data_file, :vpc_id,
+            :launch_block_device_mappings, :ami_block_device_mappings
+          ]
+
+          # Required attrs
+          req_attrs.each do |a|
+            attr_to_hash(hash, a, true)
+          end
+
+          opt_attrs.each do |a|
+            attr_to_hash(hash, a)
+          end
+
+          hash
+        end
+      end
+    end
+  end
+end

data/lib/image_builder/builders/aws_chroot.rb

@@ -0,0 +1,33 @@
+require_relative './aws_base.rb'
+
+module ImageBuilder
+  module Builders
+    module AWS
+      # Generic class doc comment
+      class Chroot < ImageBuilder::Builders::AWS::Base
+        attr_accessor :chroot_mounts
+        attr_accessor :command_wrapper
+        attr_accessor :copy_files
+        attr_accessor :device_path
+        attr_accessor :mount_path
+
+        def initialize
+          super
+          @type = 'amazon-chroot'
+        end
+
+        def packer_hash
+          hash = super
+
+          attr_to_hash(hash, :chroot_mounts)
+          attr_to_hash(hash, :command_wrapper)
+          attr_to_hash(hash, :copy_files)
+          attr_to_hash(hash, :device_path)
+          attr_to_hash(hash, :mount_path)
+
+          hash
+        end
+      end
+    end
+  end
+end

data/lib/image_builder/builders/aws_ebs.rb

@@ -0,0 +1,15 @@
+require_relative './aws_base.rb'
+
+module ImageBuilder
+  module Builders
+    module AWS
+      # Generic class doc comment
+      class EBS < ImageBuilder::Builders::AWS::Base
+        def initialize
+          super
+          @type = 'amazon-ebs'
+        end
+      end
+    end
+  end
+end

data/lib/image_builder/builders/aws_instance.rb

@@ -0,0 +1,45 @@
+require_relative './aws_base.rb'
+
+module ImageBuilder
+  module Builders
+    module AWS
+      # Generic class doc comment
+      class Instance < ImageBuilder::Builders::AWS::Base
+        attr_accessor :account_id
+        attr_accessor :s3_bucket
+        attr_accessor :x509_cert_path
+        attr_accessor :x509_key_path
+        # - opt -
+        attr_accessor :bundle_destination
+        attr_accessor :bundle_prefix
+        attr_accessor :bundle_upload_command
+        attr_accessor :bundle_vol_command
+        attr_accessor :x509_upload_path
+
+        def initialize
+          super
+          @type = 'amazon-instance'
+        end
+
+        def packer_hash
+          hash = super
+
+          # Required
+          attr_to_hash(hash, :account_id, true)
+          attr_to_hash(hash, :s3_bucket, true)
+          attr_to_hash(hash, :x509_cert_path, true)
+          attr_to_hash(hash, :x509_key_path, true)
+
+          # Optional
+          attr_to_hash(hash, :bundle_destination)
+          attr_to_hash(hash, :bundle_prefix)
+          attr_to_hash(hash, :bundle_upload_command)
+          attr_to_hash(hash, :bundle_vol_command)
+          attr_to_hash(hash, :x509_upload_path)
+
+          hash
+        end
+      end
+    end
+  end
+end

data/lib/image_builder/builders/builder_base.rb

@@ -0,0 +1,12 @@
+require_relative '../../image_builder.rb'
+
+module ImageBuilder
+  module Builders
+    # Generic class doc comment
+    class Base
+      include ImageBuilder
+
+      attr_accessor :name
+    end
+  end
+end

data/lib/image_builder/builders/null.rb

@@ -0,0 +1,39 @@
+require_relative './builder_base.rb'
+require 'etc'
+
+module ImageBuilder
+  module Builders
+    # Generic class doc comment
+    class Null < ImageBuilder::Builders::Base
+      attr_accessor :host
+      attr_accessor :ssh_password
+      attr_accessor :ssh_username
+      attr_accessor :ssh_private_key_file
+      attr_accessor :port
+
+      attr_reader :type
+
+      def initialize
+        @type = 'null'
+        @host = '127.0.0.1'
+        @port = 22
+        @ssh_username = Etc.getlogin
+        @ssh_private_key_file = ::File.join(::Dir.home, '.ssh', 'id_rsa')
+      end
+
+      def packer_hash
+        hash = {}
+
+        attr_to_hash(hash, :type, true)
+        attr_to_hash(hash, :name, true)
+        attr_to_hash(hash, :host, true)
+        attr_to_hash(hash, :ssh_username, true)
+        attr_to_hash(hash, :ssh_password)
+        attr_to_hash(hash, :ssh_private_key_file)
+        attr_to_hash(hash, :port)
+
+        hash
+      end
+    end
+  end
+end

data/lib/image_builder/post_processors/compress.rb

@@ -0,0 +1,24 @@
+require_relative './post_processors_base.rb'
+
+module ImageBuilder
+  module PostProcessors
+    # Generic class doc comment
+    class Compress < ImageBuilder::PostProcessors::Base
+      attr_accessor :output
+      attr_reader :type
+
+      def initialize
+        @type = 'compress'
+      end
+
+      def packer_hash
+        hash = {}
+
+        attr_to_hash(hash, :type, true)
+        attr_to_hash(hash, :output, true)
+
+        hash
+      end
+    end
+  end
+end

data/lib/image_builder/post_processors/post_processors_base.rb

@@ -0,0 +1,10 @@
+require_relative '../../image_builder.rb'
+
+module ImageBuilder
+  module PostProcessors
+    # Generic class doc comment
+    class Base
+      include ImageBuilder
+    end
+  end
+end

data/lib/image_builder/post_processors/vagrant.rb

@@ -0,0 +1,41 @@
+require_relative './post_processors_base.rb'
+
+module ImageBuilder
+  module PostProcessors
+    # Generic class doc comment
+    class Vagrant < ImageBuilder::PostProcessors::Base
+      attr_accessor :compression_level
+      attr_accessor :include
+      attr_accessor :keep_input_artifact
+      attr_accessor :output
+      attr_accessor :vagrantfile_template
+      attr_accessor :override
+
+      attr_reader :type
+
+      def initialize
+        @type = 'vagrant'
+        @compression_level = 6
+        @keep_input_artifact = false
+      end
+
+      def add_override(provider, override_hash)
+        @override[provider] = override_hash
+      end
+
+      def packer_hash
+        hash = {}
+
+        attr_to_hash(hash, :type, true)
+        attr_to_hash(hash, :compression_level)
+        attr_to_hash(hash, :include)
+        attr_to_hash(hash, :keep_input_artifact)
+        attr_to_hash(hash, :output)
+        attr_to_hash(hash, :vagrantfile_template)
+        attr_to_hash(hash, :override)
+
+        hash
+      end
+    end
+  end
+end