formatron 0.1.14 → 0.1.15

data/lib/formatron/cloud_formation/scripts.rb

@@ -1,8 +1,9 @@
 class Formatron
   module CloudFormation
     # Generates scripts for setting up instances with CloudFormation init
+    # rubocop:disable Metrics/ModuleLength
     module Scripts
-      def self.hostname(sub_domain:, hosted_zone_name:)
+      def self.linux_common(sub_domain:, hosted_zone_name:)
         # rubocop:disable Metrics/LineLength
         <<-EOH.gsub(/^ {10}/, '')
           #/bin/bash -v
@@ -17,6 +18,75 @@ class Formatron
         # rubocop:enable Metrics/LineLength
       end
 
+      def self.windows_common(sub_domain:, hosted_zone_name:)
+        # rubocop:disable Metrics/LineLength
+        <<-EOH.gsub(/^ {10}/, '')
+          wmic computersystem where name="%COMPUTERNAME%" call rename name="#{sub_domain}"
+          REG ADD HKLM\\SYSTEM\\CurrentControlSet\\services\\Tcpip\\Parameters /v Domain /t REG_SZ /d #{hosted_zone_name} /f
+          shutdown.exe /r /t 00
+        EOH
+        # rubocop:enable Metrics/LineLength
+      end
+
+      # rubocop:disable Metrics/MethodLength
+      def self.windows_administrator(name:, password:)
+        # rubocop:disable Metrics/LineLength
+        <<-EOH.gsub(/^ {10}/, '')
+          $newAdminName = '#{name}'
+          $adminPassword = '#{password}'
+
+          # disable password policy
+          secedit /export /cfg c:\\secpol.cfg
+          (gc C:\\secpol.cfg).replace("PasswordComplexity = 1", "PasswordComplexity = 0") | Out-File C:\\secpol.cfg
+          secedit /configure /db c:\\windows\\security\\local.sdb /cfg c:\\secpol.cfg /areas SECURITYPOLICY
+          rm -force c:\\secpol.cfg -confirm:$false
+
+          # find the local administrator user
+          $computerName = $env:COMPUTERNAME
+          $computer = [ADSI] "WinNT://$computerName,Computer"
+          foreach ( $childObject in $computer.Children ) {
+            # Skip objects that are not users.
+            if ( $childObject.Class -ne "User" ) {
+              continue
+            }
+            $type = "System.Security.Principal.SecurityIdentifier"
+            $childObjectSID = new-object $type($childObject.objectSid[0],0)
+            if ( $childObjectSID.Value.EndsWith("-500") ) {
+              $adminName = $childObject.Name[0]
+
+              # set the new password
+              $adminUser = [ADSI] "WinNT://$computerName/$adminName,User"
+              $adminUser.SetPassword($adminPassword)
+
+              # set the new name
+              $user = Get-WMIObject Win32_UserAccount -Filter "Name='$adminName'"
+              $result = $user.Rename($newAdminName)
+
+              break
+            }
+          }
+        EOH
+        # rubocop:enable Metrics/LineLength
+      end
+      # rubocop:enable Metrics/MethodLength
+
+      # rubocop:disable Metrics/MethodLength
+      def self.windows_signal(wait_condition_handle:)
+        {
+          'Fn::Join' => [
+            '', [
+              'cfn-signal.exe -e 0 ',
+              {
+                'Fn::Base64' => {
+                  Ref: wait_condition_handle
+                }
+              }
+            ]
+          ]
+        }
+      end
+      # rubocop:enable Metrics/MethodLength
+
       # rubocop:disable Metrics/MethodLength
       def self.nat(cidr:)
         # rubocop:disable Metrics/LineLength
@@ -66,8 +136,9 @@ class Formatron
           set -e
 
           export HOME=/root
-
-          source /tmp/formatron/script-variables
+          export PATH=$PATH:/usr/local/sbin/
+          export PATH=$PATH:/usr/sbin/
+          export PATH=$PATH:/sbin
 
           apt-get -y update
           apt-get -y install wget ntp cron git libfreetype6 libpng3 python-pip
@@ -134,5 +205,6 @@ class Formatron
       # rubocop:enable Metrics/ParameterLists
       # rubocop:enable Metrics/MethodLength
     end
+    # rubocop:enable Metrics/ModuleLength
   end
 end

data/lib/formatron/cloud_formation/template.rb

@@ -15,6 +15,8 @@ class Formatron
         external:,
         hosted_zone_name:,
         key_pair:,
+        administrator_name:,
+        administrator_password:,
         kms_key:,
         hosted_zone_id:,
         target:
@@ -25,6 +27,8 @@ class Formatron
         @external_outputs = external.outputs
         @hosted_zone_name = hosted_zone_name
         @key_pair = key_pair
+        @administrator_name = administrator_name
+        @administrator_password = administrator_password
         @kms_key = kms_key
         @hosted_zone_id = hosted_zone_id
         @bucket = formatron.bucket
@@ -45,6 +49,8 @@ class Formatron
             external: @external_formatron.vpc[key],
             hosted_zone_name: @hosted_zone_name,
             key_pair: @key_pair,
+            administrator_name: @administrator_name,
+            administrator_password: @administrator_password,
             kms_key: @kms_key,
             hosted_zone_id: @hosted_zone_id,
             bucket: @bucket,

data/lib/formatron/cloud_formation/template/vpc.rb

@@ -23,6 +23,8 @@ class Formatron
           external:,
           hosted_zone_name:,
           key_pair:,
+          administrator_name:,
+          administrator_password:,
           kms_key:,
           hosted_zone_id:,
           bucket:,
@@ -33,6 +35,8 @@ class Formatron
           @external = external
           @hosted_zone_name = hosted_zone_name
           @key_pair = key_pair
+          @administrator_name = administrator_name
+          @administrator_password = administrator_password
           @kms_key = kms_key
           @hosted_zone_id = hosted_zone_id
           @bucket = bucket
@@ -72,6 +76,8 @@ class Formatron
               vpc_guid: @guid,
               vpc_cidr: @cidr,
               key_pair: @key_pair,
+              administrator_name: @administrator_name,
+              administrator_password: @administrator_password,
               hosted_zone_name: @hosted_zone_name,
               kms_key: @kms_key,
               nats: Util::VPC.instances(:nat, @vpc),
@@ -104,6 +110,8 @@ class Formatron
               vpc_guid: @guid,
               vpc_cidr: @cidr,
               key_pair: @key_pair,
+              administrator_name: @administrator_name,
+              administrator_password: @administrator_password,
               hosted_zone_name: @hosted_zone_name,
               kms_key: @kms_key,
               nats: Util::VPC.instances(:nat, @external, @vpc),

data/lib/formatron/cloud_formation/template/vpc/subnet.rb

@@ -25,6 +25,8 @@ class Formatron
             vpc_guid:,
             vpc_cidr:,
             key_pair:,
+            administrator_name:,
+            administrator_password:,
             hosted_zone_name:,
             kms_key:,
             nats:,
@@ -41,6 +43,8 @@ class Formatron
             @cidr = @subnet.cidr
             @acl = @subnet.acl
             @key_pair = key_pair
+            @administrator_name = administrator_name
+            @administrator_password = administrator_password
             @hosted_zone_name = hosted_zone_name
             @kms_key = kms_key
             @nats = nats
@@ -78,6 +82,8 @@ class Formatron
                 args = {
                   symbol => instance,
                   key_pair: @key_pair,
+                  administrator_name: @administrator_name,
+                  administrator_password: @administrator_password,
                   availability_zone: @availability_zone,
                   subnet_guid: @guid,
                   hosted_zone_name: @hosted_zone_name,
@@ -116,6 +122,8 @@ class Formatron
                 args = {
                   symbol => instance,
                   key_pair: @key_pair,
+                  administrator_name: @administrator_name,
+                  administrator_password: @administrator_password,
                   availability_zone: @availability_zone,
                   subnet_guid: @guid,
                   hosted_zone_name: @hosted_zone_name,

data/lib/formatron/cloud_formation/template/vpc/subnet/bastion.rb

@@ -12,6 +12,8 @@ class Formatron
             def initialize(
               bastion:,
               key_pair:,
+              administrator_name:,
+              administrator_password:,
               availability_zone:,
               subnet_guid:,
               hosted_zone_name:,
@@ -25,10 +27,13 @@ class Formatron
               target:
             )
               @bastion = bastion
+              _set_os
               _add_open_ports
               @instance = Instance.new(
                 instance: bastion,
                 key_pair: key_pair,
+                administrator_name: administrator_name,
+                administrator_password: administrator_password,
                 availability_zone: availability_zone,
                 subnet_guid: subnet_guid,
                 hosted_zone_name: hosted_zone_name,
@@ -45,6 +50,12 @@ class Formatron
             # rubocop:enable Metrics/ParameterLists
             # rubocop:enable Metrics/MethodLength
 
+            def _set_os
+              @bastion.os(
+                'ubuntu'
+              )
+            end
+
             def _add_open_ports
               @bastion.security_group do |security_group|
                 security_group.open_tcp_port 22
@@ -56,6 +67,7 @@ class Formatron
             end
 
             private(
+              :_set_os,
               :_add_open_ports
             )
           end

data/lib/formatron/cloud_formation/template/vpc/subnet/chef_server.rb

@@ -20,6 +20,8 @@ class Formatron
             def initialize(
               chef_server:,
               key_pair:,
+              administrator_name:,
+              administrator_password:,
               availability_zone:,
               subnet_guid:,
               hosted_zone_name:,
@@ -70,6 +72,7 @@ class Formatron
               @organization_short_name = organization.short_name
               @organization_full_name = organization.full_name
               _set_default_instance_type
+              _set_os
               _add_ssl_cert_policy
               _add_keys_policy
               _add_open_ports
@@ -77,6 +80,8 @@ class Formatron
               @instance = Instance.new(
                 instance: @chef_server,
                 key_pair: key_pair,
+                administrator_name: administrator_name,
+                administrator_password: administrator_password,
                 availability_zone: availability_zone,
                 subnet_guid: subnet_guid,
                 hosted_zone_name: hosted_zone_name,
@@ -100,6 +105,12 @@ class Formatron
               ) if @chef_server.instance_type.nil?
             end
 
+            def _set_os
+              @chef_server.os(
+                'ubuntu'
+              )
+            end
+
             def _add_ssl_cert_policy
               @chef_server.policy do |policy|
                 policy.statement do |statement|
@@ -190,6 +201,7 @@ class Formatron
 
             private(
               :_set_default_instance_type,
+              :_set_os,
               :_add_ssl_cert_policy,
               :_add_keys_policy,
               :_add_open_ports,

data/lib/formatron/cloud_formation/template/vpc/subnet/instance.rb

@@ -33,6 +33,8 @@ class Formatron
             def initialize(
               instance:,
               key_pair:,
+              administrator_name:,
+              administrator_password:,
               availability_zone:,
               subnet_guid:,
               hosted_zone_name:,
@@ -62,7 +64,10 @@ class Formatron
                 "#{SecurityGroup::SECURITY_GROUP_PREFIX}#{@guid}"
               @availability_zone = availability_zone
               @instance_type = @instance.instance_type || 't2.micro'
+              @os = @instance.os || 'ubuntu'
               @key_pair = key_pair
+              @administrator_name = administrator_name
+              @administrator_password = administrator_password
               @subnet_guid = subnet_guid
               @subnet_id = "#{Subnet::SUBNET_PREFIX}#{@subnet_guid}"
               @sub_domain = @instance.sub_domain
@@ -108,6 +113,7 @@ class Formatron
               )
               policy.merge resources: resources
               security_group = SecurityGroup.new(
+                os: @os,
                 security_group: @security_group,
                 instance_guid: @guid,
                 vpc_guid: @vpc_guid,
@@ -121,17 +127,22 @@ class Formatron
                 availability_zone: @availability_zone,
                 instance_type: @instance_type,
                 key_name: @key_pair,
+                administrator_name: @administrator_name,
+                administrator_password: @administrator_password,
                 subnet: @subnet_id,
                 name: "#{@sub_domain}.#{@hosted_zone_name}",
                 wait_condition_handle: @wait_condition_handle_id,
                 security_group: @security_group_id,
                 logical_id: @instance_id,
-                source_dest_check: @source_dest_check
+                source_dest_check: @source_dest_check,
+                os: @os
               )
               setup = Setup.new(
                 setup: @setup,
                 sub_domain: @sub_domain,
-                hosted_zone_name: @hosted_zone_name
+                hosted_zone_name: @hosted_zone_name,
+                os: @os,
+                wait_condition_handle: @wait_condition_handle_id
               )
               setup.merge instance: instance
               block_devices = BlockDevices.new(

data/lib/formatron/cloud_formation/template/vpc/subnet/instance/security_group.rb

@@ -7,16 +7,19 @@ class Formatron
         class Subnet
           class Instance
             # generates CloudFormation security group resource
+            # rubocop:disable Metrics/ClassLength
             class SecurityGroup
               SECURITY_GROUP_PREFIX = 'securityGroup'
 
               # rubocop:disable Metrics/MethodLength
               def initialize(
+                os:,
                 security_group:,
                 instance_guid:,
                 vpc_guid:,
                 vpc_cidr:
               )
+                @os = os
                 @security_group = security_group
                 @vpc_guid = vpc_guid
                 @cidr = vpc_cidr
@@ -32,7 +35,11 @@ class Formatron
 
               # rubocop:disable Metrics/MethodLength
               def merge(resources:)
-                ingress_rules = _base_ingress_rules
+                if @os.eql? 'windows'
+                  ingress_rules = _base_windows_ingress_rules
+                else
+                  ingress_rules = _base_ingress_rules
+                end
                 ingress_rules.concat(
                   @open_tcp_ports.collect do |port|
                     {
@@ -104,11 +111,49 @@ class Formatron
               end
               # rubocop:enable Metrics/MethodLength
 
+              # rubocop:disable Metrics/MethodLength
+              def _base_windows_ingress_rules
+                [{
+                  cidr: @cidr,
+                  protocol: 'tcp',
+                  from_port: '0',
+                  to_port: '65535'
+                }, {
+                  cidr: @cidr,
+                  protocol: 'udp',
+                  from_port: '0',
+                  to_port: '65535'
+                }, {
+                  cidr: @cidr,
+                  protocol: 'icmp',
+                  from_port: '-1',
+                  to_port: '-1'
+                }, {
+                  cidr: '0.0.0.0/0',
+                  protocol: 'tcp',
+                  from_port: '3389',
+                  to_port: '3389'
+                }, {
+                  cidr: '0.0.0.0/0',
+                  protocol: 'tcp',
+                  from_port: '5985',
+                  to_port: '5985'
+                }, {
+                  cidr: '0.0.0.0/0',
+                  protocol: 'tcp',
+                  from_port: '5986',
+                  to_port: '5986'
+                }]
+              end
+              # rubocop:enable Metrics/MethodLength
+
               private(
                 :_base_egress_rules,
-                :_base_ingress_rules
+                :_base_ingress_rules,
+                :_base_windows_ingress_rules
               )
             end
+            # rubocop:enable Metrics/ClassLength
           end
         end
       end

data/lib/formatron/cloud_formation/template/vpc/subnet/instance/setup.rb

@@ -7,52 +7,116 @@ class Formatron
         class Subnet
           class Instance
             # Adds setup scripts to an instance
+            # rubocop:disable Metrics/ClassLength
             class Setup
-              def initialize(setup:, sub_domain:, hosted_zone_name:)
+              # rubocop:disable Metrics/MethodLength
+              def initialize(
+                setup:,
+                sub_domain:,
+                hosted_zone_name:,
+                os:,
+                wait_condition_handle:
+              )
                 @setup = setup
+                @wait_condition_handle = wait_condition_handle
                 @sub_domain = sub_domain
                 @hosted_zone_name = hosted_zone_name
                 @scripts = @setup.script unless @setup.nil?
                 @variables = @setup.variable unless @setup.nil?
+                @os = os
               end
+              # rubocop:enable Metrics/MethodLength
 
               # rubocop:disable Metrics/MethodLength
               # rubocop:disable Metrics/AbcSize
               def merge(instance:)
-                files = {
-                  '/tmp/formatron/script-0.sh' => {
-                    content: Scripts.hostname(
+                env = {}
+                @variables.each do |key, value|
+                  env[key] = value.value
+                end unless @variables.nil?
+                if @os.eql? 'windows'
+                  script_key = 'script-0'
+                  script = "C:\\formatron\\#{script_key}.bat"
+                  files = {}
+                  commands = {}
+                  @scripts.each_index do |index|
+                    script_key = "script-#{index}"
+                    script = "C:\\formatron\\#{script_key}.bat"
+                    files[script] = {
+                      content: @scripts[index]
+                    }
+                    commands[script_key] = {
+                      command: script,
+                      env: env
+                    }
+                  end unless @scripts.nil?
+                  setup_script_index = @scripts.nil? ? 0 : @scripts.length
+                  signal_script_index = setup_script_index + 1
+                  script_key = "script-#{setup_script_index}"
+                  script = "C:\\formatron\\#{script_key}.bat"
+                  files[script] = {
+                    content: Scripts.windows_common(
                       sub_domain: @sub_domain,
                       hosted_zone_name: @hosted_zone_name
-                    ),
-                    mode: '000755',
-                    owner: 'root',
-                    group: 'root'
+                    )
                   }
-                }
-                @scripts.each_index do |index|
-                  files["/tmp/formatron/script-#{index + 1}.sh"] = {
-                    content: @scripts[index],
-                    mode: '000755',
-                    owner: 'root',
-                    group: 'root'
+                  commands[script_key] = {
+                    command: script,
+                    env: env,
+                    waitAfterCompletion: 'forever'
                   }
-                end unless @scripts.nil?
-                variables = []
-                @variables.each do |key, value|
-                  variables.concat(["#{key}=", value.value, "\n"])
-                end unless @variables.nil?
-                files['/tmp/formatron/script-variables'] = {
-                  content: Template.join(*variables),
-                  mode: '000644',
-                  owner: 'root',
-                  group: 'root'
-                } unless variables.length == 0
+                  script_key = "script-#{signal_script_index}"
+                  script = "C:\\formatron\\#{script_key}.bat"
+                  files[script] = {
+                    content: Scripts.windows_signal(
+                      wait_condition_handle: @wait_condition_handle
+                    )
+                  }
+                  commands[script_key] = {
+                    command: script,
+                    env: env
+                  }
+                else
+                  script_key = 'script-0'
+                  script = "/tmp/formatron/#{script_key}.sh"
+                  files = {
+                    "#{script}" => {
+                      content: Scripts.linux_common(
+                        sub_domain: @sub_domain,
+                        hosted_zone_name: @hosted_zone_name
+                      ),
+                      mode: '000755',
+                      owner: 'root',
+                      group: 'root'
+                    }
+                  }
+                  commands = {
+                    "#{script_key}" => {
+                      command: script,
+                      env: env
+                    }
+                  }
+                  @scripts.each_index do |index|
+                    script_key = "script-#{index + 1}"
+                    script = "/tmp/formatron/#{script_key}.sh"
+                    files[script] = {
+                      content: @scripts[index],
+                      mode: '000755',
+                      owner: 'root',
+                      group: 'root'
+                    }
+                    commands[script_key] = {
+                      command: script,
+                      env: env
+                    }
+                  end unless @scripts.nil?
+                end
                 instance[:Metadata] = {
                   Comment1: 'Create setup scripts',
                   'AWS::CloudFormation::Init' => {
                     config: {
-                      files: files
+                      files: files,
+                      commands: commands
                     }
                   }
                 }
@@ -60,6 +124,7 @@ class Formatron
               # rubocop:enable Metrics/AbcSize
               # rubocop:enable Metrics/MethodLength
             end
+            # rubocop:enable Metrics/ClassLength
           end
         end
       end