sambot 0.1.69 → 0.1.83

data/lib/sambot/domain/common/runtime.rb

@@ -5,17 +5,17 @@ module Sambot
     module Common
       module Runtime
 
-        def is_obsolete
+        def self.is_obsolete
           latest_version = Gems.new.versions('sambot')[0]["number"]
           Gem::Version.new(Sambot::VERSION) < Gem::Version.new(latest_version)
         end
 
-        def ensure_latest
+        def self.ensure_latest
           latest_version = Gems.new.versions('sambot')[0]["number"]
-          debug("Current Sambot version is #{Sambot::VERSION}.")
-          debug("Latest Sambot version is #{latest_version}.")
+          UI.debug("Current Sambot version is #{Sambot::VERSION}.")
+          UI.debug("Latest Sambot version is #{latest_version}.")
           if is_obsolete
-            say('A newer version of the sambot gem exists. Please update the gem before continuing.', :red)
+            UI.info('A newer version of the sambot gem exists. Please update the gem before continuing.', :red)
           end
         end
 

data/lib/sambot/domain/common/template_provider.rb

@@ -3,7 +3,7 @@ module Sambot
     module Common
       class TemplateProvider
 
-        def get_path(filename)
+        def self.get_path(filename)
           File.join(File.dirname(__FILE__), '..', '..', 'templates', filename)
         end
 

data/lib/sambot/domain/cookbook.rb

@@ -0,0 +1,103 @@
+require 'yaml'
+require 'git'
+
+module Sambot
+  module Domain
+    class Cookbook
+
+        def self.build(essential_files, generated_files)
+          config = Common::Config.new.read
+          validate_cookbook_structure(config['platforms'], essential_files, generated_files)
+          setup_test_kitchen(config)
+          build_metadata(config)
+          copy_git_hooks()
+          UI.info('The cookbook has been successfully built.')
+        end
+
+        def self.clean(generated_files)
+          UI.info('Removing all generated files from this cookbook.')
+          delete_file('metadata.rb')
+          delete_file('winrm_config')
+          delete_file('packer.json')
+          generated_files.each { |file| delete_file(file) }
+          Dir.glob('\.kitchen*\.yml').each do |file|
+            delete_file(file)
+          end
+          UI.info('The cookbook has been successfully cleaned.')
+        end
+
+        def self.generate(name, platforms, type, description, essential_files, generated_files)
+          Git.init(name)
+          Dir.chdir(name) do
+            FileUtils.mkdir('test')
+            FileUtils.mkdir('spec')
+            FileUtils.mkdir('recipes')
+            FileUtils.touch('README.md')
+            write_config(name, description, platforms, type)
+            build(essential_files, generated_files)
+          end
+          UI.info('The cookbook has been successfully generated.')
+        end
+
+        private
+
+        def self.write_config(name, description, platforms, type)
+          contents = {
+            'name' => name,
+            'version' => '0.0.1',
+            'platforms' => platforms,
+            'suites' => [{
+              'name' => 'default',
+              'run_list' => [
+                "recipe[#{name}]"
+              ],
+              'verifier' => {
+                'inspec_tests' => ['./test']
+              }
+            }],
+            'description' => description,
+          }.to_yaml
+          File.write('.config.yml', contents)
+          Domain::UI.debug("./.config.yml has been added to the cookbook.")
+        end
+
+        def self.copy_git_hooks
+          working_path = '.git/hooks/pre-push'
+          template_path = Common::TemplateProvider.get_path('pre-push')
+          File.delete(working_path) if File.exist?(working_path)
+          FileUtils.cp(template_path, working_path)
+          UI.debug("The pre-push Git hook has been added to the cookbook.")
+        end
+
+        def self.delete_file(filename)
+          filename = filename.gsub(/\.erb/, '') if filename.end_with?(".erb")
+          return unless File.exist?(filename)
+          File.delete(filename)
+          UI.debug("./#{filename} has been removed.")
+        end
+
+        def self.validate_cookbook_structure(platform, essential_files, generated_files)
+          essential_files.each { |path| Common::FileChecker.new.verify(path) }
+          if platform.include?('windows')
+            Common::FileChecker.new.update(['winrm_config'])
+          end
+          Common::FileChecker.new.update(generated_files)
+        end
+
+        def self.setup_test_kitchen(config)
+          files = Chef::Kitchen.generate_yml(config['name'], config['platforms'], config['suites'])
+          files.each do |filename, contents|
+            File.write(filename.to_s, contents)
+            UI.debug("#{filename.to_s} has been added to the cookbook.")
+          end
+        end
+
+        def self.build_metadata(config)
+          result = Chef::Metadata.generate(config['name'], config['platforms'], config['version'], config['description'], config['dependencies'], config['gems'])
+          File.write('metadata.rb', result)
+          UI.debug("A new metadata.rb file has been generated.")
+        end
+
+      end
+  end
+end

data/lib/sambot/domain/dns.rb

@@ -0,0 +1,24 @@
+require 'hosts'
+
+module Sambot
+  module Domain
+
+    HOST_ENTRIES = {
+      'teamcity.brighter.io' => '127.0.0.1',
+      'chef.brighter.io' => '127.0.0.1',
+      'monitoring.brighter.io' => '127.0.0.1',
+      'jenkins.brighter.io' => '127.0.0.1',
+      'splunk.brighter.io' => '127.0.0.1',
+      'rundeck.brighter.io' => '127.0.0.1',
+      'grafana.brighter.io' => '127.0.0.1',
+      'prometheus.brighter.io' => '127.0.0.1'
+    }
+
+    class DNS
+
+      def self.update_hosts()
+      end
+
+    end
+  end
+end

data/lib/sambot/domain/packer.rb

@@ -0,0 +1,26 @@
+require 'yaml'
+
+module Sambot
+  module Domain
+      class Packer
+
+        def self.prepare(config_file = nil)
+          config = Common::Config.new.read(config_file)
+          cookbook_name = config['name']
+          target_platform = ENV['TARGET_IMAGE_PLATFORM']
+          unless target_platform == 'windows' || target_platform == 'linux'
+            raise ApplicationError.new("The target image platform must be either 'windows' or 'linux'.")
+          end
+          generate_config(cookbook_name, target_platform)
+        end
+
+        def self.generate_config(cookbook_name, target_platform)
+          filename = TemplateProvider.get_path("packer.#{target_platform}.json")
+          File.read(filename).gsub(/@@cookbook_name@@/, cookbook_name)
+          File.write("packer.json", contents)
+          UI.debug("The configuration file for Packer has been added.")
+        end
+
+    end
+  end
+end

data/lib/sambot/domain/session.rb

@@ -0,0 +1,25 @@
+module Sambot
+  module Domain
+    class Session
+
+      def self.start(username, password)
+        UI.debug("Opening a tunnel to the Rackspace DEV/QE environment...")
+        bastion = BastionHost.new.connect(username, password)
+        bastion.forwards.each do |forward|
+          UI.debug(" -- #{forward} will be forwarded to #{forward}")
+        end
+        UI.debug("Authenticating with Hashicorp Vault in Rackspace DEV/QE...")
+        token = Vault.authenticate(username, password)
+        UI.debug("Saving your Vault authentication token to ~/.vault-token...")
+        Vault.save_token(token)
+        UI.info("Your session has now started. Run `sambot session stop` to close it.")
+      end
+
+      def self.stop
+        BastionHost.disconnect
+        UI.info("Your session has been closed.")
+      end
+
+    end
+  end
+end

data/lib/sambot/domain/{workstations/ssh_config_file.rb → ssh/config_file.rb}

@@ -1,14 +1,14 @@
 module Sambot
   module Domain
-    module Workstations
-      class SshConfigFile
+    module Ssh
+      class ConfigFile
 
-        def initialize
+        def initialize(config_file = nil)
           @make_backups = true
           @header_lines = []
           @sections = []
           @sections_by_name = {}
-          read_config
+          read_config(config_file)
         end
 
         def sections
@@ -20,15 +20,15 @@ module Sambot
         end
 
         def add_section(name)
-          section = SshConfigSection.new(name)
+          section = ConfigSection.new(name)
           @sections << section
           @sections_by_name[section.name] = section
           section
         end
 
-        def read_config
+        def read_config(config_file = nil)
           current_section = nil
-          IO.readlines(File.expand_path("~/.ssh/config")).each_with_index do |line, i|
+          IO.readlines(config_file || File.expand_path("~/.ssh/config")).each_with_index do |line, i|
             line.rstrip!
             if line =~ /\bHost\s+(.+)/
               current_section = add_section($1)

data/lib/sambot/domain/{workstations/ssh_config_section.rb → ssh/config_section.rb}

@@ -1,7 +1,7 @@
 module Sambot
   module Domain
-    module Workstations
-      class SshConfigSection
+    module Ssh
+      class ConfigSection
 
         attr_accessor :name, :aliases, :lines, :settings
 

data/lib/sambot/domain/{workstations/ssh_parser.rb → ssh/parser.rb}

@@ -1,10 +1,10 @@
-require_relative 'ssh_config_file'
-require_relative 'ssh_config_section'
+require_relative 'config_file'
+require_relative 'config_section'
 
 module Sambot
   module Domain
-    module Workstations
-      class SshParser
+    module Ssh
+      class Parser
 
         ENTRIES = {
           'bastion': {
@@ -23,8 +23,8 @@ module Sambot
           }
         }
 
-        def update(username)
-          config = SshConfigFile.new
+        def update(username, config_file = nil)
+          config = ConfigFile.new(config_file)
           sections = config.sections_by_name
           ENTRIES.each_pair do |key, val|
             config.rm(key.to_s)
@@ -34,12 +34,13 @@ module Sambot
                   config.add_alias(key.to_s, new_alias.to_s)
                 end
               elsif x == :username
-                config.set(key.to_s, 'Username', username)
+                config.set(key.to_s, 'User', username)
               else
                 config.set(key.to_s, x.to_s, y)
               end
             end
           end
+          config
         end
 
       end

data/lib/sambot/domain/ui.rb

@@ -0,0 +1,19 @@
+module Sambot
+  module Domain
+      module UI
+
+        def self.debug(msg)
+          Thor.new.say("debug: #{msg}", :yellow)
+        end
+
+        def self.info(msg)
+          Thor.new.say(" info: #{msg}", :green)
+        end
+
+        def self.error(msg)
+          Thor.new.say("error: #{msg}", :red)
+        end
+
+      end
+  end
+end

data/lib/sambot/domain/vault.rb

@@ -0,0 +1,32 @@
+require 'vault'
+
+module Sambot
+  module Domain
+    class Vault
+
+      ENVIRONMENT_VARIABLES = {
+        VAULT_ADDR: 'https://localhost:8200',
+        VAULT_SKIP_VERIFY: true
+      }
+
+      ENV_ROOT = '/etc/profile.d/'
+
+      def self.authenticate(username, password)
+        secret = ::Vault.auth.ldap(username, password)
+        secret.auth.client_token
+      end
+
+      def self.setup_environment(root = ENV_ROOT)
+        ENVIRONMENT_VARIABLES.each_pair do |key, value|
+          path = File.join(root, "#{key}.sh")
+          File.write(path, "export #{key}=#{value}")
+        end
+      end
+
+      def self.save_token(token)
+        File.write("~/.vault-token", token)
+      end
+
+    end
+  end
+end

data/lib/sambot/domain/workstation.rb

@@ -0,0 +1,25 @@
+module Sambot
+  module Domain
+    class Workstation
+
+      def self.configure(username)
+        UI.debug("Updating your SSH configuration.")
+        update_ssh_configuration(username)
+        UI.debug("Updating your environment variables.")
+        update_environment_variables()
+        UI.debug("Updating your hosts file.")
+        DNS.update_hosts()
+        UI.info("Your workstation is now ready for use. Please close this shell and open up a new one to start making use of your new environment.")
+      end
+
+      def self.update_ssh_configuration(username)
+        config = Domain::Ssh::Parser.new.update('DEV\\' + username)
+        config.save
+      end
+
+      def self.update_environment_variables
+      end
+
+    end
+  end
+end

data/lib/sambot/templates/{.kitchen.gcp.windows.yml → .kitchen.gcp.yml.erb}

@@ -3,11 +3,43 @@ provisioner:
   name: chef_zero
   log_level: <%= ENV['TEST_KITCHEN_LOG_LEVEL'] || 'info' %>
   deprecations_as_errors: true
+  require_chef_omnibus: 12
   cookbooks_path:
     - .
 
 platforms:
+<% if @platforms.include?('centos') %>
+  - name: centos
+    transport:
+      username: chefuser
+      ssh_key:
+        - <%= ENV['GCP_SSH_KEY'] || "" %>
+    driver:
+      name: sfmc_google
+      region: <%= ENV['GCP_REGION'] %>
+      project: <%= ENV['GCP_PROJECT'] %>
+      image_project: <%= ENV['GCP_CENTOS_IMAGE_PROJECT'] %>
+      image_family: <%= ENV['GCP_CENTOS_IMAGE_FAMILY'] %>
+      network: <%= ENV['GCP_NETWORK'] %>
+      subnet: <%= ENV['GCP_SUBNETWORK'] %>
+      use_private_ip: false
+      preemptible: true
+      service_account_name: <%= ENV['GCP_SERVICE_ACCOUNT_NAME'] %>
+      service_account_scopes:
+        - userinfo-email
+        - logging-write
+        - monitoring-write
+      tags:
+        - "test-kitchen"
+        - "consul-agent"
+        - "vault-client"
+<% end %>
+<% if @platforms.include?('windows') %>
   - name: windows
+    transport:
+      name: winrm
+      username: chefuser
+      elevated: true
     driver:
       name: sfmc_google
       region: <%= ENV['GCP_REGION'] %>
@@ -32,11 +64,7 @@ platforms:
         - "test-kitchen"
         - "consul-agent"
         - "vault-client"
-
-transport:
-  name: winrm
-  elevated: true
-  username: chefuser
+<% end %>
 
 verifier:
   name: inspec