sambot 0.1.43 → 0.1.44

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e9956d6ca9c4ab939d89580be328d56bc800ae79
-  data.tar.gz: e9863b2063c6b84127843ddf1d58082b3952fff7
+  metadata.gz: 1ff206043b3d053e2f72817f49310550e883eb52
+  data.tar.gz: c0571205dfe9c6d50e0cf3ffdece5f0425b7f71d
 SHA512:
-  metadata.gz: 556e2d28b7270fa2c943c67adfe540d2719ad228521afa6412e7653c9da823b639539f3702e778fec06c181eeb9759ec798e14185d00821b3068fb14aa33a759
-  data.tar.gz: fb213573790311cf5eee1f13892eea56353435e5bbfc9d1b7b86c71e6691e36c397a245cd3d5eae944948af992d4fc4b1db9daa3b578b36dd26efbe9b7c1de6d
+  metadata.gz: bc916312401b0e9ddc4ed9042c1980e5c75a6d87098d9018872fc1c299abbc18370ee3bac21e91600c1fc2fe993224f82d834d5f3d6ac9d13c2acab7c7d4e801
+  data.tar.gz: b1096e1a3eee52a6ba5d981e287dfbc5bce9d0ffda7caf92362d794ca2a880f4a0325a723c613452161a67ccacd2b564dba12c28debfe92316620c4bae06ca27

data/lib/sambot/cli.rb

@@ -2,6 +2,11 @@ require 'thor'
 require_relative 'domain/common/application_exception'
 require_relative 'domain/common/file_checker'
 require_relative 'domain/secrets/vault'
+require_relative 'domain/workstations/hosts'
+require_relative 'domain/workstations/env'
+require_relative 'domain/workstations/ssh_config_file'
+require_relative 'domain/workstations/ssh_config_section'
+require_relative 'domain/workstations/ssh_parser'
 require_relative 'domain/common/ui'
 require_relative 'domain/common/config'
 require_relative 'domain/common/runtime'
@@ -12,7 +17,7 @@ require_relative 'domain/cookbooks/metadata'
 require_relative 'commands/cookbook'
 require_relative 'commands/secret'
 require_relative 'commands/teamcity'
-require_relative 'commands/inventory'
+require_relative 'commands/workstation'
 
 module Sambot
   class CLI < Thor
@@ -26,8 +31,8 @@ module Sambot
     desc 'secret', 'Manage secrets inside Chef cookbooks'
     subcommand 'secret', Sambot::Commands::Secret
 
-    desc 'inventory', 'Manage instance inventories'
-    subcommand 'inventory', Sambot::Commands::Inventory
+    desc 'workstation', 'Manage engineer workstations'
+    subcommand 'workstation', Sambot::Commands::Workstation
 
   end
 end

data/lib/sambot/commands/secret.rb

@@ -22,6 +22,11 @@ module Sambot
         Sambot::Vault.new.write(path)
       end
 
+      desc 'list', 'List all secrets at a given path'
+      def list
+        Sambot::Vault.new.list(path)
+      end
+
     end
   end
 end

data/lib/sambot/commands/session.rb

@@ -0,0 +1,22 @@
+module Sambot
+  module Commands
+
+    class Session < Thor
+
+      namespace 'session'
+
+      desc "start", "Start a new DEV/QE session"
+      def start
+        username = ask('What is your DEV/QE username?')
+        password = ask('What is your DEV/QE password?')
+        puts "not implemented"
+      end
+
+      desc "stop", "Stop the DEV/QE session"
+      def start
+        puts "not implemented"
+      end
+
+    end
+  end
+end

data/lib/sambot/commands/workstation.rb

@@ -0,0 +1,23 @@
+module Sambot
+  module Commands
+
+    class Workstation < Thor
+
+      namespace 'workstation'
+
+      desc "cnfigure", "Sets up an engineering workstation"
+      def configure
+        username = ask("What is your DEV/QE Active Directory username i.e. john.smith?")
+        debug("Updating your SSH configuration.")
+        config = Domain::Workstations::SshParser.new.update('DEV\\' + username)
+        config.save
+        debug("Updating your environment variables.")
+        Domain::Workstations::Env.new.update
+        debug("Generating the datasource for Free Remote Desktop Manager, your RDP client.")
+        filename = Domain::Workstations::Hosts.new.generate
+        info("Your workstation is now ready for use. The RDM datasource is available here: #{filename}")
+      end
+
+    end
+  end
+end

data/lib/sambot/domain/secrets/vault.rb

@@ -1,6 +1,6 @@
 module Sambot
   module Domain
-    module Chef
+    module Secrets
       class Vault
 
         def initialize

data/lib/sambot/domain/workstations/install.sh

@@ -0,0 +1 @@
+brew install vault

data/lib/sambot/domain/workstations/ssh_config_file.rb

@@ -0,0 +1,81 @@
+module Sambot
+  module Domain
+    module Workstations
+      class SshConfigFile
+
+        def initialize
+          @make_backups = true
+          @header_lines = []
+          @sections = []
+          @sections_by_name = {}
+          read_config
+        end
+
+        def sections
+          @sections
+        end
+
+        def sections_by_name
+          @sections_by_name
+        end
+
+        def add_section(name)
+          section = SshConfigSection.new(name)
+          @sections << section
+          @sections_by_name[section.name] = section
+          section
+        end
+
+        def read_config
+          current_section = nil
+          IO.readlines(File.expand_path("~/.ssh/config")).each_with_index do |line, i|
+            line.rstrip!
+            if line =~ /\bHost\s+(.+)/
+              current_section = add_section($1)
+            else
+              if current_section
+                current_section.lines << line
+              else
+                @header_lines << line
+              end
+            end
+          end
+        end
+
+        def set(host_nick, key, value)
+          section = @sections_by_name[host_nick] || add_section(host_nick)
+          section[key] = value
+        end
+
+        def dump
+          to_text([@header_lines, @sections].flatten)
+        end
+
+        def rm(host_nick)
+          if @sections_by_name.key?(host_nick)
+            @sections_by_name.delete host_nick
+            @sections.delete_at(@sections.index{|s| s.name == host_nick})
+          end
+        end
+
+        def add_alias(host_nick, new_alias)
+          section = @sections_by_name[host_nick] || add_section(host_nick)
+          section.aliases.push(new_alias) unless section.aliases.member?(new_alias)
+        end
+
+        def save
+          File.open(File.expand_path("~/.ssh/config"), "w") do |file|
+            file.puts dump
+          end
+        end
+
+        private
+
+        def to_text(ray)
+          ray.map {|s| s.to_s } * "\n"
+        end
+      end
+
+    end
+  end
+end

data/lib/sambot/domain/workstations/ssh_config_section.rb

@@ -0,0 +1,84 @@
+module Sambot
+  module Domain
+    module Workstations
+      class SshConfigSection
+
+        attr_accessor :name, :aliases, :lines, :settings
+
+        def initialize(name)
+          all_names = name.split(/\s+/)
+          @name = all_names.shift
+          @aliases = all_names
+          @settings = {}
+          @lines = []
+        end
+
+        def [](setting)
+          unless @settings.key? setting
+            if line = lines[setting_index(setting)]
+              key,val = line.split(nil, 2)
+              @settings[key] = val
+            end
+          end
+          @settings[setting]
+        end
+
+        def unset(setting)
+          if line_num = setting_index(setting)
+            @settings.delete setting
+            @lines.delete_at line_num
+          end
+        end
+
+        def header
+          name_with_aliases = [@name, *aliases].join(" ")
+          "Host #{name_with_aliases}"
+        end
+
+        def to_s
+          [header, *lines] * "\n"
+        end
+
+        def []=(setting, value)
+          if value.is_a?(Array)
+            value.each do |val|
+              line_num = lines.length
+              lines[line_num] = format_line(setting, val)
+            end
+          elsif value != '-'
+            line_num = lines.length
+            lines[line_num] = format_line(setting, value)
+          else
+            @settings.delete(setting)
+            @lines.delete_at(setting_index(setting))
+          end
+        end
+
+        def matches?(text)
+          r = Regexp.new text
+          name =~ r || aliases.any? {|a| a =~ r} || lines.any? {|line| line =~ r}
+        end
+
+        def matches_exactly?(text)
+          name == text || has_alias?(text)
+        end
+
+        def has_alias?(text)
+          aliases.member?(text)
+        end
+
+        protected
+
+        def format_line(setting, value)
+          "    #{setting} #{value}"
+        end
+
+        def setting_index(setting)
+          r = Regexp.new(setting)
+          lines.index {|line| line =~ r}
+        end
+      end
+
+    end
+  end
+end

data/lib/sambot/domain/workstations/ssh_parser.rb

@@ -0,0 +1,48 @@
+require_relative 'ssh_config_file'
+require_relative 'ssh_config_section'
+
+module Sambot
+  module Domain
+    module Workstations
+      class SshParser
+
+        ENTRIES = {
+          'bastion': {
+            username: '',
+            'HostName': '146.177.10.174',
+            'LocalForward': [
+              '8200 vault.brighter.io:8200',
+              '9997 splunk.brighter.io:9997'
+            ]
+          },
+          'AVG-*': {
+            aliases: ['avg-*', 'GD-*', 'gd-*', 'NW-*', 'nw-*','PHX-* phx-* RS-* rs-* THD-* thd-*'],
+            username: '',
+            'ProxyJump': 'bastion',
+            'StrictHostKeyChecking': 'no'
+          }
+        }
+
+        def update(username)
+          config = SshConfigFile.new
+          sections = config.sections_by_name
+          ENTRIES.each_pair do |key, val|
+            config.rm(key.to_s)
+            val.each_pair do |x, y|
+              if x == :aliases
+                val[x].each do |new_alias|
+                  config.add_alias(key.to_s, new_alias.to_s)
+                end
+              elsif x == :username
+                config.set(key.to_s, 'Username', username)
+              else
+                config.set(key.to_s, x.to_s, y)
+              end
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/sambot/version.rb

@@ -1,3 +1,3 @@
 module Sambot
-  VERSION = '0.1.43'.freeze
+  VERSION = '0.1.44'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sambot
 version: !ruby/object:Gem::Version
-  version: 0.1.43
+  version: 0.1.44
 platform: ruby
 authors:
 - Olivier Kouame
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-05-29 00:00:00.000000000 Z
+date: 2017-05-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: git
@@ -190,9 +190,10 @@ files:
 - lib/sambot.rb
 - lib/sambot/cli.rb
 - lib/sambot/commands/cookbook.rb
-- lib/sambot/commands/inventory.rb
 - lib/sambot/commands/secret.rb
+- lib/sambot/commands/session.rb
 - lib/sambot/commands/teamcity.rb
+- lib/sambot/commands/workstation.rb
 - lib/sambot/domain/common/application_exception.rb
 - lib/sambot/domain/common/config.rb
 - lib/sambot/domain/common/file_checker.rb
@@ -203,6 +204,12 @@ files:
 - lib/sambot/domain/cookbooks/kitchen.rb
 - lib/sambot/domain/cookbooks/metadata.rb
 - lib/sambot/domain/secrets/vault.rb
+- lib/sambot/domain/workstations/env.rb
+- lib/sambot/domain/workstations/hosts.rb
+- lib/sambot/domain/workstations/install.sh
+- lib/sambot/domain/workstations/ssh_config_file.rb
+- lib/sambot/domain/workstations/ssh_config_section.rb
+- lib/sambot/domain/workstations/ssh_parser.rb
 - lib/sambot/templates/.gitignore
 - lib/sambot/templates/.kitchen.centos.yml
 - lib/sambot/templates/.kitchen.gcp.centos.yml

data/lib/sambot/commands/inventory.rb

@@ -1,15 +0,0 @@
-module Sambot
-  module Commands
-
-    class Inventory < Thor
-
-      namespace 'inventory'
-
-      desc "generate", "Gets the Rackspace inventory list for use with Remote Desktop Manager"
-      def generate
-        say("Not implemented yet")
-      end
-
-    end
-  end
-end