ssp 0.1 → 0.2

data/bin/ssp

@@ -1,4 +1,4 @@
-#!/usr/bin/env ruby
+#!/usr/bin/env ruby -rubygems
 
 $:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
 require 'ssp/app'

data/lib/ssp/app.rb

@@ -2,7 +2,15 @@ require 'thor'
 require 'thor/group'
 
 module SSP
+  module ThorExtensions
+    def banner(task, namespace = nil, subcommand = false)
+      "#{File.basename($0)} #{task.formatted_usage(self, $thor_runner, subcommand)}"
+    end
+  end
+
   class App < Thor
+    extend ThorExtensions
+
     namespace :app
     map "-T" => :list
 
@@ -61,6 +69,9 @@ module SSP
         unless @_commands_initialized
           Dir[File.join(File.dirname(__FILE__), "application", "*.rb")].each do |command|
             require(command)
+            klass_name = Thor::Util.camel_case(File.basename(command).sub(/\.rb$/, ''))
+            klass = SSP::App.const_get(klass_name)
+            klass.extend(ThorExtensions)
           end
           @_commands_initialized = true
         end

data/lib/ssp/application/node.rb

@@ -24,8 +24,8 @@ class SSP::App::Node < Thor
                 :desc => "Sets the flavor of the server (1: 256MB, 2: 512MB, 3: 1024MB, ...)",
                 :default => 2
   method_option :image, :aliases => "-i", :type => :numeric,
-                :desc => "Sets the image to be used (49: Ubuntu 10.04, 4151016: chef-node)",
-                :default => 4151016
+                :desc => "Sets the image to be used (49: Ubuntu 10.04, 4256704: chef-node)",
+                :default => 4256704
   def create(fqdn, *roles)
     server = connection.servers.new
 
@@ -110,6 +110,19 @@ validation_client_name "#{chef_config[:validation_client_name]}"
     ssh.run
   end
 
+  desc "list", "Lists available servers on Rackspace"
+  def list
+    server_list = [ ["ID", "Name", "Public IP", "Private IP", "Flavor ID"].map {|t| shell.set_color(t, :bold)} ]
+    server_list += connection.servers.all.map do |server|
+      [
+        server.id.to_s, server.name,
+        server.addresses["public"][0], server.addresses["private"][0],
+        server.flavor_id.to_s
+      ]
+    end
+    print_table(server_list)
+  end
+
   protected
     def connection
       @connection ||= Fog::Rackspace::Servers.new(

data/lib/ssp/application/office_node.rb

@@ -0,0 +1,180 @@
+require 'chef/config'
+require 'chef/knife/ssh'
+require 'ssp/config'
+
+class SSP::App::OfficeNode < Thor
+  namespace :officenode
+
+  class_option :ssh_user,
+               :desc => "The ssh username",
+               :default => "root"
+
+  class_option :ssh_password,
+               :desc => "The ssh password",
+               :default => SSP::Config[:vanilla][:root_password]
+
+  class_option :chef_config, :aliases => "-c",
+               :desc => "Location of the chef configuration file",
+               :default => File.join(ENV['HOME'], '.chef', 'knife.rb')
+
+
+  desc "create FQDN [ROLES...]", "Creates a new chef client on Vanilla"
+  method_option :memory, :aliases => "-m", :type => :numeric,
+                :desc => "Sets the memory size for the node in MB",
+                :default => 512
+  method_option :iplastpart, :aliases => "-i", :type => :numeric,
+                :desc => "Sets the last part of the node's IP",
+                :required => true
+  def create(fqdn, *roles)
+    ip = "192.168.1.#{options[:iplastpart]}"
+    disk_size = options[:memory] * 20 / 1024
+    hostname = fqdn.split(".").first
+
+    # Convert roles array to proper run list
+    run_list = roles.map { |r| r =~ /recipe\[(.*)\]/ ? $1 : "role[#{r}]" }
+
+    # Always include the chef-client role in the run list
+    unless run_list.include? "role[chef-client]"
+      run_list.unshift "role[chef-client]"
+    end
+
+    vncpass = pwgen
+    rootpass = pwgen
+
+    say_status "Name: ", fqdn, :cyan
+    say_status "Memory: ", "#{options[:memory]}MB", :cyan
+    say_status "Disk: ", "#{disk_size}GB", :cyan
+    say_status "IP: ", ip, :cyan
+    say_status "VNC password: ", vncpass, :cyan
+    say_status "root password: ", rootpass, :cyan
+
+    say "\nRequesting server", :magenta
+
+    command = <<EOH
+bash -c '
+/root/bin/mkdomu.sh -m #{options[:memory]} -d #{disk_size} -v #{vncpass} -p #{rootpass} #{hostname} #{options[:iplastpart]}
+xm create /etc/xen/#{hostname}.cfg
+'
+EOH
+
+    ssh_run "vanilla", command
+
+    say "\nServer ready, waiting 15 seconds to bootstrap."
+    sleep 15
+
+    say "\nBootstrapping #{shell.set_color(fqdn, :bold)}..."
+
+    command = <<EOH
+bash -c '
+mkdir -p /etc/chef
+
+(
+cat <<'EOP'
+127.0.0.1	localhost localhost.localdomain
+#{ip}	#{fqdn} #{fqdn.split(".").first}
+
+192.168.1.82	basil basil.secretsaucepartners.com chef.secretsaucepartners.com
+EOP
+) > /etc/hosts
+
+(
+cat <<'EOP'
+#{IO.read(chef_config[:validation_key])}
+EOP
+) > /tmp/validation.pem
+awk NF /tmp/validation.pem > /etc/chef/validation.pem
+rm /tmp/validation.pem
+
+(
+cat <<'EOP'
+log_level        :info
+log_location     STDOUT
+chef_server_url  "#{chef_config[:chef_server_url]}"
+validation_client_name "#{chef_config[:validation_client_name]}"
+EOP
+) > /etc/chef/client.rb
+
+(
+cat <<'EOP'
+#{{ "run_list" => run_list }.to_json}
+EOP
+) > /etc/chef/first-boot.json
+
+apt-get -y update
+apt-get -y upgrade
+
+/usr/local/bin/chef-client -j /etc/chef/first-boot.json'
+EOH
+
+    begin
+      ssh_run ip, command, "root", rootpass
+    rescue Net::SSH::HostKeyMismatch => key_ex
+      key_ex.remember_host!
+      retry
+    end
+  end
+
+  desc "list", "Lists available servers on Vanilla"
+  def list
+    ssh_run "vanilla", "xm list"
+  end
+
+
+  desc "destroy FQDN", "Removes a chef client and also destorys the Xen guest on Vanilla"
+  def destroy(fqdn)
+    hostname = fqdn.split(".").first
+
+    if ask("Are you sure you want to destroy the `#{fqdn}' node? [yes/no]").downcase != "yes"
+      raise Thor::Error, "Aborting."
+    end
+
+    command = <<EOH
+bash -c '
+xm shutdown #{hostname}
+sleep 3
+rm /etc/xen/#{hostname}.cfg
+lvremove -f vg/#{hostname}-{root,swap}'
+EOH
+
+    ssh_run "vanilla", command
+
+    say %x{knife node delete #{fqdn} -y -c #{options[:chef_config]} 2>/dev/null}
+    say %x{knife client delete #{fqdn} -y -c #{options[:chef_config]} 2>/dev/null}
+  end
+
+  protected
+    def chef_config
+      unless defined?(@_chef_config_loaded)
+        Chef::Config.from_file(options[:chef_config])
+        @_chef_config_loaded = true
+      end
+      Chef::Config
+    end
+
+    def ssh_run(host, command, user = options[:ssh_user], pass = options[:ssh_password])
+      ssh = Chef::Knife::Ssh.new
+      ssh.name_args = [ host, command ]
+      ssh.config[:ssh_user] = user
+      ssh.config[:password] = pass
+      ssh.config[:manual] = true
+
+      begin
+        ssh.run
+      rescue Net::SSH::AuthenticationFailed
+        unless config[:ssh_password]
+          puts "Failed to authenticate #{config[:ssh_user]} - trying password auth"
+          ssh = Chef::Knife::Ssh.new
+          ssh.name_args = [ host, command ]
+          ssh.config[:ssh_user] = user
+          ssh.config[:manual] = true
+          ssh.config[:password] = ssh.get_password
+          ssh.run
+        end
+      end
+    end
+
+    def pwgen(length = 16)
+      chars = (('a'..'z').to_a + ('A'..'Z').to_a + ('0'..'9').to_a) - %w(i o 0 1 l I O)
+      (1..length).collect {|a| chars[rand(chars.size)]}.join
+    end
+end

data/lib/ssp/version.rb

@@ -1,3 +1,3 @@
 module SSP
-  VERSION = "0.1"
+  VERSION = "0.2"
 end

metadata

@@ -1,12 +1,12 @@
 --- !ruby/object:Gem::Specification 
 name: ssp
 version: !ruby/object:Gem::Version 
-  hash: 9
+  hash: 15
   prerelease: false
   segments: 
   - 0
-  - 1
-  version: "0.1"
+  - 2
+  version: "0.2"
 platform: ruby
 authors: 
 - !binary |
@@ -16,7 +16,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-06-30 00:00:00 +02:00
+date: 2010-07-05 00:00:00 +02:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -64,6 +64,7 @@ files:
 - lib/ssp/app.rb
 - lib/ssp/application/bags.rb
 - lib/ssp/application/node.rb
+- lib/ssp/application/office_node.rb
 - lib/ssp/application/pair.rb
 - lib/ssp/config.rb
 - lib/ssp/version.rb