sambot 0.1.99 → 0.1.100

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6e779e1412fd356ebd1c2d95a1e2bba17f95f271
-  data.tar.gz: 34d07a2b89b11d01d7b38d8965df70a06fae25ef
+  metadata.gz: b5ec6ff097c32a95918f9ea05cdd036324ea8f54
+  data.tar.gz: 7d0a575a9b98dc0ae5db63e767e6256c793bf81a
 SHA512:
-  metadata.gz: 43e34ed4faad4c51a78006949492230236899c6461b8ac56813f099663bb41acc8fdfda5c37c4bd2c2d4783ac4849b6542a603dc26b4a7d8a2056d2dabf14b56
-  data.tar.gz: 1cf8f3347ab0464e23040bda5757e367f2e50440cc9c75fde66cf79688d45aabe8e8d9f2d22c5a6baf313fad0d209a598cd8f7c754b4f00b8563eaa1e8909147
+  metadata.gz: d6e81a25dd723813f9124bd882ebffa040fe7599b68ebc7d25b1ad0e2892713b9b25ba0095faaec0688ea9350e971219f07a11e23b62fd315414a4b7313980a7
+  data.tar.gz: 050c201c54cb984af3472048f05962a75e5b14753cf307d9f88bc8c2f97d4ea40687fd959f96174d34790266e74b77cc7fe9b92b4931a220152571556244574d

data/.gitignore

@@ -10,3 +10,4 @@
 
 # rspec failure tracking
 .rspec_status
+sambot-session.log

data/.rubocop.yml

@@ -7,5 +7,14 @@ Documentation:
 Metrics/AbcSize:
   Enabled: false
 
-Style/EmptyLinesAroundBody:
+Style/EmptyLinesAroundClassBody:
+  Enabled: false
+
+Style/EmptyLinesAroundBlockBody:
+  Enabled: false
+
+Metrics/ModuleLength:
+  Enabled: false
+
+Metrics/MethodLength:
   Enabled: false

data/bin/sambot

@@ -1,3 +1,6 @@
 #!/usr/bin/env ruby
+require 'hashie'
+$VERBOSE = nil
+Hashie.logger = Logger.new(nil)
 require_relative '../lib/sambot'
 Sambot::CLI.start

data/lib/sambot.rb

@@ -1,3 +1,5 @@
+require 'thor'
+
 require_relative 'sambot/application_error'
 require_relative 'sambot/config'
 require_relative 'sambot/ui'
@@ -17,11 +19,16 @@ require_relative 'sambot/rackspace/flavors'
 require_relative 'sambot/rackspace/images'
 require_relative 'sambot/rackspace/instances'
 
+require_relative 'sambot/dns/records'
+require_relative 'sambot/dns/repository'
+
 require_relative 'sambot/ssh/config_file'
 require_relative 'sambot/ssh/config_section'
 require_relative 'sambot/ssh/parser'
 
-require_relative 'sambot/developer_workflow/bastion_host'
+require_relative 'sambot/developer_workflow/tunnel'
+require_relative 'sambot/developer_workflow/brew'
+require_relative 'sambot/developer_workflow/networking'
 require_relative 'sambot/developer_workflow/dns'
 require_relative 'sambot/developer_workflow/session'
 require_relative 'sambot/developer_workflow/vault'
@@ -29,9 +36,12 @@ require_relative 'sambot/developer_workflow/workstation'
 
 require_relative 'sambot/commands/base_command'
 require_relative 'sambot/commands/cookbook'
+require_relative 'sambot/commands/rdp'
 require_relative 'sambot/commands/session'
 require_relative 'sambot/commands/workstation'
-require_relative 'sambot/commands/rundeck'
+require_relative 'sambot/commands/instance'
+require_relative 'sambot/commands/dns'
+require_relative 'sambot/commands/team'
 require_relative 'sambot/commands/report'
 
 require_relative 'sambot/cli'

data/lib/sambot/chef/cookbook.rb

@@ -1,3 +1,5 @@
+#frozen_string_literal: true
+
 require 'yaml'
 require 'git'
 
@@ -5,8 +7,7 @@ module Sambot
   module Chef
     class Cookbook
 
-        def self.build(essential_files, generated_files)
-          config = Config.new.read
+        def self.build(config, essential_files, generated_files)
           validate_cookbook_structure(config['platforms'], essential_files, generated_files)
           setup_test_kitchen(config)
           build_metadata(config)
@@ -16,26 +17,20 @@ module Sambot
 
         def self.clean(generated_files)
           UI.info('Removing all generated files from this cookbook.')
-          delete_file('metadata.rb')
-          delete_file('winrm_config')
-          delete_file('Berksfile.lock')
-          delete_file('packer.json')
-          generated_files.each { |file| delete_file(file) }
-          Dir.glob('\.kitchen*\.yml').each do |file|
+          targets = ['metadata.rb', 'winrm_config', 'Berksfile.lock'] + generated_files + Dir.glob('\.kitchen*\.yml')
+          targets.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)
+        def self.generate(config, name, platforms, type, description, essential_files, generated_files)
           Git.init(name)
           Dir.chdir(name) do
-            FileUtils.mkdir('test')
-            FileUtils.mkdir('spec')
-            FileUtils.mkdir('recipes')
+            ['test', 'spec', 'recipes'].each {|target|  FileUtils.mkdir(target) }
             FileUtils.touch('README.md')
             write_config(name, description, platforms, type)
-            build(essential_files, generated_files)
+            build(config, essential_files, generated_files)
           end
           UI.info('The cookbook has been successfully generated.')
         end

data/lib/sambot/chef/kitchen.rb

@@ -1,11 +1,12 @@
+#frozen_string_literal: true
+
 module Sambot
     module Chef
       class Kitchen
 
-        KITCHEN_LOCAL_YML = ".kitchen.yml"
-        KITCHEN_GCP_YML = ".kitchen.gcp.yml"
-        KITCHEN_RACKSPACE_YML = ".kitchen.rackspace.yml"
-
+        KITCHEN_LOCAL_YML = '.kitchen.yml'
+        KITCHEN_GCP_YML = '.kitchen.gcp.yml'
+        KITCHEN_RACKSPACE_YML = '.kitchen.rackspace.yml'
 
         def self.generate_yml(cookbook_name, platforms, suites = nil)
           raise ApplicationError, 'Missing platforms when trying to generate Test-Kitchen YAML.' unless platforms
@@ -28,7 +29,7 @@ module Sambot
           filename = File.join(FileManagement::TemplateProvider.get_path("#{template}.erb"))
           contents = File.read(filename).gsub(/@@cookbook_name@@/, cookbook_name)
           eruby = Erubis::Eruby.new(contents)
-          yaml = eruby.evaluate({platforms: platforms}).gsub(/\_@/, "<%=").gsub(/@\_/, "%>")
+          yaml = eruby.evaluate({platforms: platforms}).gsub(/\_@/, '<%=').gsub(/@\_/, '%>')
           YAML.load(yaml)
         end
 

data/lib/sambot/chef/metadata.rb

@@ -1,3 +1,5 @@
+#frozen_string_literal: true
+
 require 'erubis'
 
 module Sambot

data/lib/sambot/chef/server.rb

@@ -1,3 +1,5 @@
+#frozen_string_literal: true
+
 require 'ridley'
 require 'json'
 
@@ -6,23 +8,23 @@ module Sambot
       class Server
 
         SHORT_NAMES = {
-          "wsus" => "WSUS",
-          "nginx-proxy" => "PRXY",
-          "jetstream" => "JTS",
-          "task-scheduler" => "TASKS",
-          "queue-service" => "QUE",
-          "skylight" => "SKY",
-          "prometheus" => "PRT",
-          "octopus" => "OCTO",
-          "vault" => "VAULT",
-          "consul" => "CONSUL",
-          "rabbitmq" => "RMQ",
-          "etcd" => "ETCD",
-          "teamcity-agent" => "TCA",
-          "teamcity-server" => "TCS",
-          "bastion" =>   "BST",
-          "base" => "BASE",
-          "grafana" => "MON"
+          'wsus' => 'WSUS',
+          'nginx-proxy' => 'PRXY',
+          'jetstream' => 'JTS',
+          'task-scheduler' => 'TASKS',
+          'queue-service' => 'QUE',
+          'skylight' => 'SKY',
+          'prometheus' => 'PRT',
+          'octopus' => 'OCTO',
+          'vault' => 'VAULT',
+          'consul' => 'CONSUL',
+          'rabbitmq' => 'RMQ',
+          'etcd' => 'ETCD',
+          'teamcity-agent' => 'TCA',
+          'teamcity-server' => 'TCS',
+          'bastion' =>   'BST',
+          'base' => 'BASE',
+          'grafana' => 'MON'
         }
 
         def initialize(ridley = nil)
@@ -43,18 +45,14 @@ module Sambot
         end
 
         def find_role_name(cookbook)
-          naive_name = cookbook.gsub(/as-role-/, "").gsub(/as-app-role-/, "")
+          naive_name = cookbook.gsub(/as-role-/, '').gsub(/as-app-role-/, '')
           if SHORT_NAMES.key?(naive_name)
             SHORT_NAMES[naive_name]
           else
-            raise "Could not generate an instance name"
+            raise 'Could not generate an instance name'
           end
         end
 
       end
     end
   end
-
-
-#Find out what's happening on the security front
-#Summarize conversation with Daven & take

data/lib/sambot/cli.rb

@@ -1,4 +1,4 @@
-require 'thor'
+# frozen_string_literal: true
 
 module Sambot
   class CLI < Thor
@@ -12,5 +12,8 @@ module Sambot
     desc 'workstation', 'Manage engineer workstations'
     subcommand 'workstation', Sambot::Commands::Workstation
 
+    desc 'dns', 'Manage DEV/QE DNS'
+    subcommand 'dns', Sambot::Commands::DNS
+
   end
 end

data/lib/sambot/commands/base_command.rb

@@ -1,5 +1,56 @@
+# frozen_string_literal: true
+
 require 'thor/hollaback'
 
+class Thor
+  module Shell
+    class Basic
+      def print_wrapped(message, options = {})
+        indent = options[:indent] || 0
+        width = 100 - indent
+        paras = message.split("\n\n")
+
+        paras.map! do |unwrapped|
+          unwrapped.strip.tr("\n", " ").squeeze(" ").gsub(/.{1,#{width}}(?:\s|\Z)/) { ($& + 5.chr).gsub(/\n\005/, "\n").gsub(/\005/, "\n") }
+        end
+
+        paras.each do |para|
+          para.split("\n").each do |line|
+            stdout.puts line.insert(0, " " * indent)
+          end
+          stdout.puts unless para == paras.last
+        end
+      end
+    end
+  end
+end
+
+
+class Thor
+  class << self
+    def command_help(shell, command_name)
+      meth = normalize_command_name(command_name)
+      command = all_commands[meth]
+      handle_no_command_error(meth) unless command
+      shell.say
+      shell.say("Usage:", :green)
+      shell.say
+      shell.say "  sambot #{namespace} #{command_name} "
+      shell.say
+      shell.say
+      class_options_help(shell, nil => command.options.values)
+      if command.long_description
+        shell.say "Description:", :green
+        shell.say
+        shell.print_wrapped(docs("#{namespace}/#{command_name}"), :indent => 2)
+      else
+        shell.say command.description
+      end
+      shell.say
+    end
+  end
+end
+
 module Sambot
   module Commands
 
@@ -8,9 +59,22 @@ module Sambot
       before :check_version
 
       no_commands do
-        def check_version
+
+        def execute
           Runtime.ensure_latest
+          yield
+        rescue ApplicationError => e
+          UI.error(e.message)
         end
+
+        def config
+           Config.new.read
+        end
+
+        def self.docs(path)
+          File.read(File.join(File.dirname(__FILE__), '../docs', path + '.txt'))
+        end
+
       end
 
     end

data/lib/sambot/commands/cookbook.rb

@@ -1,4 +1,4 @@
-require 'thor'
+# frozen_string_literal: true
 
 module Sambot
   module Commands
@@ -8,75 +8,33 @@ module Sambot
 
     class Cookbook < BaseCommand
 
-      GUIDE =
-      {
-        clean: {
-          SHORT_DESC: 'Remove all generated build files from a cookbook',
-          LONG_DESC:  <<-LONGDESC
-          `sambot cookbook clean` will remove all files generated by `sambot cookbook build`.
-          This is required to ensure the generated files are not stored in source control.
-          LONGDESC
-        },
-        build: {
-          SHORT_DESC: 'Builds a cookbook from its configuration file',
-          LONG_DESC:  <<-LONGDESC
-          `sambot cookbook build` will generate all the files required for the functioning
-          of a Chef cookbook from a configuration file. The motivation behind this setup is to
-          standardize and simplify the Chef cookbook creation workflow. The configuration file should be
-          called .config.yml and stored in the root of the Chef cookbook.
-          LONGDESC
-        },
-        generate: {
-          SHORT_DESC: 'Creates a new cookbook',
-          LONG_DESC: <<-LONGDESC
-          `sambot cookbook generate` will create a new cookbook. This can be either a
-          wrapper cookbook or an instance role cookbook.
-          LONGDESC
-        },
-        version: {
-          SHORT_DESC: 'Gives the cookbook version as a TeamCity service message',
-          LONG_DESC: ''
-        }
-      }
-
       namespace 'cookbook'
 
-      desc 'clean', GUIDE[:clean][:SHORT_DESC]
-      long_desc GUIDE[:clean][:LONG_DESC]
+      desc 'clean', 'Remove all generated build files from a cookbook'
       def clean
-        Chef::Cookbook.clean(GENERATED_FILES - ['.gitignore'])
-      rescue ApplicationError => e
-        error(e.message)
+        execute { Chef::Cookbook.clean(GENERATED_FILES - ['.gitignore']) }
       end
 
-      desc 'build', GUIDE[:clean][:SHORT_DESC]
-      long_desc GUIDE[:clean][:LONG_DESC]
+      desc 'build', 'Builds a cookbook from its configuration file'
       def build
-        Chef::Cookbook.build(ESSENTIAL_FILES, GENERATED_FILES)
-      rescue ApplicationError => e
-        error(e.message)
+        execute { Chef::Cookbook.build(config, ESSENTIAL_FILES, GENERATED_FILES) }
       end
 
-      desc 'generate', GUIDE[:generate][:SHORT_DESC]
-      long_desc GUIDE[:generate][:LONG_DESC]
-      def generate()
-        name = ask(' What is the name of this cookbook?')
-        description = ask(' What does this cookbook do?')
-        platforms = ask(' What operating system will this cookbook run on?', :limited_to => ['windows', 'centos', 'both'])
-        type = ask(' What type of cookbook will this be?', :limited_to => ['wrapper', 'role'])
-        platforms = platforms == 'both' ? ['centos', 'windows'] : [platforms]
-        Chef::Cookbook.generate(name, platforms, type, description, ESSENTIAL_FILES, GENERATED_FILES)
-      rescue ApplicationError => e
-        error(e.message)
+      desc 'generate', 'Creates a new cookbook'
+      def generate
+        execute do
+          name = ask(' What is the name of this cookbook?')
+          description = ask(' What does this cookbook do?')
+          platforms = ask(' What operating system will this cookbook run on?', :limited_to => ['windows', 'centos', 'both'])
+          type = ask(' What type of cookbook will this be?', :limited_to => ['wrapper', 'role'])
+          platforms = platforms == 'both' ? ['centos', 'windows'] : [platforms]
+          Chef::Cookbook.generate(config, name, platforms, type, description, ESSENTIAL_FILES, GENERATED_FILES)
+        end
       end
 
-      desc 'version', GUIDE[:version][:SHORT_DESC]
-      long_desc GUIDE[:version][:LONG_DESC]
-      def version()
-        result = Config.new.read
-        puts "##teamcity[buildNumber '#{result['version'].to_s}']"
-      rescue ApplicationError => e
-        error(e.message)
+      desc 'version', 'Gives the cookbook version as a TeamCity service message'
+      def version
+        execute { puts "##teamcity[buildNumber '#{config['version'].to_s}']" }
       end
 
     end

data/lib/sambot/commands/dns.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'git'
+
+module Sambot
+  module Commands
+    class DNS < BaseCommand
+
+      namespace 'dns'
+
+      desc 'remove', 'Removes the given DNS entry'
+      option :hostname, :required => true, :desc => 'The DNS name for the A record'
+      def remove
+        execute do
+          Sambot::DNS::Repository.new.records.remove(options[:hostname])
+        end
+      end
+
+      desc 'list', 'Shows all DNS entries'
+      def list
+        execute do
+          records = Sambot::DNS::Repository.new.list_records
+          say
+          print_table(records)
+        end
+      end
+
+      desc 'show', 'Shows a given DNS entry'
+      def show
+      rescue ApplicationError => e
+        UI.error(e.message)
+      end
+
+      desc 'add', 'Adds or modify a DNS entry'
+      option :address, :required => true, :desc => 'The internal IP address for the A record'
+      option :hostname, :required => true, :desc => 'The DNS name for the A record'
+      option :team, :desc => 'The team owning the instance pointed to. Leave blank if the hostname contains stable, patch or unstable i.e. stable-api'
+      long_desc docs('cookbook/clean')
+      def add
+          execute { Sambot::DNS::Repository.new.records.add(options[:hostname], options[:address], options[:team]) }
+      end
+
+    end
+  end
+end