formatron 0.1.0

data/lib/formatron/generators/bootstrap/ec2.rb

@@ -0,0 +1,17 @@
+class Formatron
+  module Generators
+    module Bootstrap
+      # generates placeholder README.md
+      module EC2
+        def self.write(directory)
+          ec2_directory = File.join directory, 'config/_default/ec2'
+          FileUtils.mkdir_p ec2_directory
+          private_key = File.join ec2_directory, 'private_key'
+          File.write private_key, <<-EOH.gsub(/^ {12}/, '')
+            Remember to replace this file with the EC2 private key
+          EOH
+        end
+      end
+    end
+  end
+end

data/lib/formatron/generators/bootstrap/formatronfile.rb

@@ -0,0 +1,52 @@
+require 'erb'
+require 'curb'
+require 'formatron/generators/util'
+
+class Formatron
+  module Generators
+    module Bootstrap
+      # generates a bootstrap Formatronfile
+      module Formatronfile
+        # exports params to bootstrap ERB template
+        class Template
+          attr_reader :params, :ip
+
+          def initialize(params)
+            @params = params
+            @ip = Curl.get('http://whatismyip.akamai.com').body_str
+          end
+
+          def guid
+            Util.guid
+          end
+
+          def databag_secret
+            Util.databag_secret
+          end
+        end
+
+        def self.write(directory, params)
+          FileUtils.mkdir_p directory
+          formatronfile = File.join directory, 'Formatronfile'
+          File.write formatronfile, _content(params)
+        end
+
+        def self._content(params)
+          template = File.join(
+            File.dirname(File.expand_path(__FILE__)),
+            File.basename(__FILE__, '.rb'),
+            'Formatronfile.erb'
+          )
+          erb = ERB.new File.read(template)
+          erb.filename = template
+          erb_template = erb.def_class Template, 'render()'
+          erb_template.new(params).render
+        end
+
+        private_class_method(
+          :_content
+        )
+      end
+    end
+  end
+end

data/lib/formatron/generators/bootstrap/formatronfile/Formatronfile.erb

@@ -0,0 +1,79 @@
+formatron.name '<%= params[:name] %>'
+formatron.bucket '<%= params[:s3_bucket] %>'
+
+formatron.global do |global|
+  global.protect config['protected']
+  global.kms_key '<%= params[:kms_key] %>'
+  global.databag_secret '<%= databag_secret %>'
+  global.hosted_zone_id '<%= params[:hosted_zone_id] %>'
+
+  global.ec2 do |ec2|
+    ec2.key_pair '<%= params[:ec2_key_pair] %>'
+    ec2.private_key config['ec2']['private_key']
+  end
+end
+
+formatron.vpc 'vpc' do |vpc|
+  vpc.guid 'VPC<%= guid %>'
+  vpc.cidr '10.0.0.0/16'
+
+  vpc.subnet 'management' do |subnet|
+    subnet.guid 'SUBNET<%= guid %>'
+    subnet.availability_zone '<%= params[:availability_zone] %>'
+    subnet.cidr '10.0.0.0/24'
+    subnet.acl do |acl|
+      acl.source_cidr '<%= ip %>/32'
+    end
+
+    subnet.bastion 'bastion' do |bastion|
+      bastion.guid 'BASTION<%= guid %>'
+      bastion.sub_domain config['bastion']['sub_domain']
+      bastion.chef do |chef|
+        chef.cookbook 'cookbooks/bastion_instance'
+      end
+    end
+
+    subnet.nat 'nat' do |nat|
+      nat.guid 'NAT<%= guid %>'
+      nat.sub_domain config['nat']['sub_domain']
+      nat.chef do |chef|
+        chef.cookbook 'cookbooks/nat_instance'
+      end
+    end
+
+    subnet.chef_server 'chef_server' do |chef_server|
+      chef_server.guid 'CHEF<%= guid %>'
+      chef_server.version '12.2.0-1'
+      chef_server.sub_domain config['chef_server']['sub_domain']
+      chef_server.cookbooks_bucket config['chef_server']['cookbooks_bucket']
+      chef_server.organization do |organization|
+        organization.short_name '<%= params[:chef_server][:organization][:short_name] %>'
+        organization.full_name '<%= params[:chef_server][:organization][:full_name] %>'
+      end
+      chef_server.username '<%= params[:chef_server][:username] %>'
+      chef_server.email '<%= params[:chef_server][:email] %>'
+      chef_server.first_name '<%= params[:chef_server][:first_name] %>'
+      chef_server.last_name '<%= params[:chef_server][:last_name] %>'
+      chef_server.password '<%= params[:chef_server][:password] %>'
+      chef_server.ssl_key config['chef_server']['ssl']['key']
+      chef_server.ssl_cert config['chef_server']['ssl']['cert']
+      chef_server.ssl_verify config['chef_server']['ssl']['verify']
+      chef_server.chef do |chef|
+        chef.cookbook 'cookbooks/chef_server_instance'
+      end
+    end
+  end
+
+  vpc.subnet 'public' do |subnet|
+    subnet.guid 'SUBNET<%= guid %>'
+    subnet.availability_zone '<%= params[:availability_zone] %>'
+    subnet.cidr '10.0.1.0/24'
+  end
+
+  vpc.subnet 'private' do |subnet|
+    subnet.guid 'SUBNET<%= guid %>'
+    subnet.availability_zone '<%= params[:availability_zone] %>'
+    subnet.cidr '10.0.2.0/24'
+    subnet.gateway 'nat'
+  end
+end

data/lib/formatron/generators/bootstrap/ssl.rb

@@ -0,0 +1,35 @@
+class Formatron
+  module Generators
+    module Bootstrap
+      # generates placeholder SSL stuff
+      module SSL
+        def self.write_key(directory)
+          placeholder_key = File.join directory, 'key'
+          File.write placeholder_key, <<-EOH.gsub(/^ {12}/, '')
+            Remember to generate an SSL key
+          EOH
+        end
+
+        def self.write_cert(directory)
+          placeholder_cert = File.join directory, 'cert'
+          File.write placeholder_cert, <<-EOH.gsub(/^ {12}/, '')
+            Remember to generate an SSL certificate
+          EOH
+        end
+
+        def self.write(directory, target)
+          target_directory = File.join(
+            directory,
+            'config',
+            target.to_s,
+            'chef_server',
+            'ssl'
+          )
+          FileUtils.mkdir_p target_directory
+          write_key target_directory
+          write_cert target_directory
+        end
+      end
+    end
+  end
+end

data/lib/formatron/generators/credentials.rb

@@ -0,0 +1,17 @@
+class Formatron
+  module Generators
+    # generates a credentials JSON file
+    module Credentials
+      def self.generate(file, region, access_key_id, secret_access_key)
+        FileUtils.mkdir_p File.dirname(file)
+        File.write file, <<-EOH.gsub(/^ {10}/, '')
+          {
+            "region": "#{region}",
+            "access_key_id": "#{access_key_id}",
+            "secret_access_key": "#{secret_access_key}"
+          }
+        EOH
+      end
+    end
+  end
+end

data/lib/formatron/generators/instance.rb

@@ -0,0 +1,64 @@
+require_relative 'instance/formatronfile'
+require_relative 'instance/config'
+require_relative 'util/cookbook'
+require_relative 'util/readme'
+require_relative 'util/gitignore'
+
+class Formatron
+  module Generators
+    # generates an instance configuration
+    module Instance
+      def self.validate_hash_params(hash, params)
+        params.each do |param|
+          fail "params should contain #{param}" if hash[param].nil?
+        end
+      end
+
+      def self.validate_params(params)
+        validate_hash_params params, [
+          :name,
+          :instance_name,
+          :s3_bucket,
+          :bootstrap_configuration,
+          :vpc,
+          :subnet,
+          :targets
+        ]
+      end
+
+      def self.generate_targets(directory, targets, name)
+        targets.each do |target|
+          Config.write(
+            directory,
+            target,
+            name
+          )
+        end
+      end
+
+      def self.generate_cookbooks(directory, name)
+        Util::Cookbook.write(
+          directory,
+          "#{name}_instance",
+          "#{name} instance"
+        )
+      end
+
+      # rubocop:disable Metrics/MethodLength
+      def self.generate(directory, params)
+        validate_params params
+        Util::Readme.write directory, params[:name]
+        Util::Gitignore.write directory
+        Formatronfile.write directory, params
+        Config.write directory
+        generate_targets(
+          directory,
+          params[:targets],
+          params[:instance_name]
+        )
+        generate_cookbooks directory, params[:instance_name]
+      end
+      # rubocop:enable Metrics/MethodLength
+    end
+  end
+end

data/lib/formatron/generators/instance/config.rb

@@ -0,0 +1,47 @@
+class Formatron
+  module Generators
+    module Instance
+      # generates an empty config
+      module Config
+        def self.write_default(file)
+          File.write file, <<-EOH.gsub(/^ {12}/, '')
+            {
+            }
+          EOH
+        end
+
+        def self.write_target(file, target, name)
+          File.write file, <<-EOH.gsub(/^ {12}/, '')
+            {
+              "#{name}": {
+                "sub_domain": "#{name}-#{target}"
+              }
+            }
+          EOH
+        end
+
+        # rubocop:disable Metrics/MethodLength
+        def self.write(
+          directory,
+          target = nil,
+          name = nil
+        )
+          target_directory = File.join(
+            directory,
+            'config',
+            target.nil? ? '_default' : target.to_s
+          )
+          FileUtils.mkdir_p target_directory
+          file = File.join target_directory, '_default.json'
+          write_default(file) if target.nil?
+          write_target(
+            file,
+            target,
+            name
+          ) unless target.nil?
+        end
+        # rubocop:enable Metrics/MethodLength
+      end
+    end
+  end
+end

data/lib/formatron/generators/instance/formatronfile.rb

@@ -0,0 +1,47 @@
+require 'erb'
+require 'curb'
+require 'formatron/generators/util'
+
+class Formatron
+  module Generators
+    module Instance
+      # generates a bootstrap Formatronfile
+      module Formatronfile
+        # exports params to bootstrap ERB template
+        class Template
+          attr_reader :params
+
+          def initialize(params)
+            @params = params
+          end
+
+          def guid
+            Util.guid
+          end
+        end
+
+        def self.write(directory, params)
+          FileUtils.mkdir_p directory
+          formatronfile = File.join directory, 'Formatronfile'
+          File.write formatronfile, _content(params)
+        end
+
+        def self._content(params)
+          template = File.join(
+            File.dirname(File.expand_path(__FILE__)),
+            File.basename(__FILE__, '.rb'),
+            'Formatronfile.erb'
+          )
+          erb = ERB.new File.read(template)
+          erb.filename = template
+          erb_template = erb.def_class Template, 'render()'
+          erb_template.new(params).render
+        end
+
+        private_class_method(
+          :_content
+        )
+      end
+    end
+  end
+end

data/lib/formatron/generators/instance/formatronfile/Formatronfile.erb

@@ -0,0 +1,16 @@
+formatron.name '<%= params[:name] %>'
+formatron.bucket '<%= params[:s3_bucket] %>'
+
+formatron.depends '<%= params[:bootstrap_configuration] %>'
+
+formatron.vpc '<%= params[:vpc] %>' do |vpc|
+  vpc.subnet '<%= params[:subnet] %>' do |subnet|
+    subnet.instance '<%= params[:instance_name] %>' do |instance|
+      instance.guid 'INSTANCE<%= guid %>'
+      instance.sub_domain config['<%= params[:instance_name] %>']['sub_domain']
+      instance.chef do |chef|
+        chef.cookbook 'cookbooks/<%= params[:instance_name] %>_instance'
+      end
+    end
+  end
+end

data/lib/formatron/generators/util.rb

@@ -0,0 +1,14 @@
+class Formatron
+  module Generators
+    # Utility methods
+    module Util
+      def self.guid
+        Random.rand(36**8).to_s(36).upcase
+      end
+
+      def self.databag_secret
+        Random.rand(36**40).to_s(36).upcase
+      end
+    end
+  end
+end

data/lib/formatron/generators/util/cookbook.rb

@@ -0,0 +1,65 @@
+class Formatron
+  module Generators
+    module Util
+      # generates placeholder cookbooks
+      module Cookbook
+        def self.write(directory, name, description)
+          cookbook_directory = File.join(
+            directory,
+            'cookbooks',
+            name
+          )
+          _write_cookbook_metadata cookbook_directory, name
+          _write_cookbook_readme cookbook_directory, name, description
+          _write_cookbook_berksfile cookbook_directory
+          _write_cookbook_recipe cookbook_directory
+        end
+
+        def self._write_cookbook_metadata(directory, name)
+          FileUtils.mkdir_p directory
+          metadata = File.join directory, 'metadata.rb'
+          File.write metadata, <<-EOH.gsub(/^ {12}/, '')
+            name '#{name}'
+            version '0.1.0'
+            supports 'ubuntu'
+          EOH
+        end
+
+        def self._write_cookbook_readme(directory, name, description)
+          FileUtils.mkdir_p directory
+          readme = File.join directory, 'README.md'
+          File.write readme, <<-EOH.gsub(/^ {12}/, '')
+            # #{name}
+
+            Cookbook to perform additional configuration on the #{description}
+          EOH
+        end
+
+        def self._write_cookbook_berksfile(directory)
+          FileUtils.mkdir_p directory
+          berksfile = File.join directory, 'Berksfile'
+          File.write berksfile, <<-EOH.gsub(/^ {12}/, '')
+            source 'https://supermarket.chef.io'
+
+            metadata
+          EOH
+        end
+
+        def self._write_cookbook_recipe(directory)
+          recipes_directory = File.join directory, 'recipes'
+          FileUtils.mkdir_p recipes_directory
+          recipe = File.join recipes_directory, 'default.rb'
+          File.write recipe, <<-EOH.gsub(/^ {12}/, '')
+          EOH
+        end
+
+        private_class_method(
+          :_write_cookbook_metadata,
+          :_write_cookbook_readme,
+          :_write_cookbook_berksfile,
+          :_write_cookbook_recipe
+        )
+      end
+    end
+  end
+end