ssvm_cli 0.0.2 → 0.0.3

data/bin/ssvm

@@ -17,7 +17,7 @@ class Ssvm < Thor
   method_option :owners, :aliases => '-o', :desc => "Owners of the VM", :type => :array
   method_option :env, :aliases => '-e', :required => true, :desc => "Environment for the VM (e.g. prod or np)"
   method_option :datacenter, :aliases => '-d', :required => true, :desc => "Datacenter for the VM (e.g. dc1 or dc2)"
-  method_option :build_spec, :required => true, :desc => "Default or customized VM request"
+  method_option :build_spec, :aliases => '-b', :required => true, :desc => "Default or customized VM request"
   method_option :os, :desc => "What OS to use for the VM (e.g. Scientific Linux (Carbon) 6.0 x86_64)"
   method_option :instances, :desc => "How many instances to make"
   method_option :start_at, :desc => "What number to start at for multi instances request", :default => 1
@@ -26,14 +26,33 @@ class Ssvm < Thor
   method_option :memory, :aliases => '-m', :desc => "How much memory for the VM"
   method_option :cpu, :aliases => '-c', :desc => "How CPU for the VM"
   method_option :disk, :desc => "How much disk space for the VM"
-  method_option :no_exp, :type => :boolean, :desc => "No expiration for the VM"
+  method_option :no_expiration, :type => :boolean, :desc => "No expiration for the VM"
   method_option :justifications, :aliases => '-j', :desc => "Justification for any special request"
+  method_option :location_specs, :aliases => '-l', 
+                :desc => "Specificiations for where to create the VM (format [include|exclude]:[rack|chassis|node]:location1,location2)",
+                :type => :array
   def create_vm
     conf = load_conf(options[:conf_file] || CONFIG_FILE)
     rhelper = RequestHelper.new(conf)
     rhelper.create_vm(options)
   end
 
+  desc 'perform_action', 'perform the action on the specified VMs'
+  method_option :hosts, :aliases => '-h', :required => true, :desc => "List of VMs to perform the action on", :type => :array
+  method_option :action, :aliases => '-a', :required => true, :desc => "What action to perform. Available actions are rebuild, reboot and delete"
+  method_option :silent, :type => :boolean, :desc => "Don't prompt"
+  def perform_action
+    conf = load_conf(options[:conf_file] || CONFIG_FILE)
+    rhelper = RequestHelper.new(conf)
+    puts "You're about to #{options['action']} the following VM(s): #{options['hosts'].inspect}"
+    if options['silent']
+      continue = true
+    else
+      continue = ask("Are you sure you want to do that? (y/N)? ") == 'y'
+    end
+    rhelper.perform_action(options) if continue
+  end
+
   desc "config", "configuration setup"
   method_option :conf_file, :aliases => "-c", :desc => "Where to write config to"
   def config
@@ -42,7 +61,7 @@ class Ssvm < Thor
     conf = {}
     conf[:username] = ask("Username: ")
     conf[:password] = ask("Password: ", true)
-    conf[:server] = ask("Chefkitchen server: ")
+    conf[:server] = ask("SSVM server: ")
     File.open(config_file, 'w') do |f|
       YAML.dump(conf, f)
     end

data/lib/request_helper.rb

@@ -18,6 +18,7 @@ class RequestHelper
     update_owner_field(options)
     handle_multi_instances(options)
     handle_custom_request(options)
+    handle_location_specs(options)
 
     begin
       response = RestClient.post("http://#{@server}/vm",
@@ -34,6 +35,24 @@ class RequestHelper
     end
   end
 
+  def perform_action(options)
+    action = options['action']
+    hosts = options['hosts'] 
+    begin
+      response = RestClient.post("http://#{@server}/vm/#{action}",
+                                 {:id => hosts.first},
+                                 {:cookies => @cookies, :content_type => 'application/json', :accept => :json})
+      result = JSON.parse(response)
+      if options[:json]
+        puts JSON.pretty_generate(result)
+      else
+        puts result['message']
+      end
+    rescue => e
+      puts e.inspect
+    end
+  end
+
   private
   def update_owner_field(options)
     if options['owners']
@@ -63,4 +82,22 @@ class RequestHelper
   def handle_custom_request(options)
     options['processor_count'] = options.delete('cpu')
   end
+
+  def handle_location_specs(options)
+    location_specs = options.delete('location_specs')
+    return unless location_specs
+
+    location_specs.each_with_index do |location_spec, index|
+      tokens = location_spec.split(':')
+      raise "Invalid location specs" if tokens.size != 3
+      if tokens[0] != "include" and tokens[0] != "exclude"
+        raise "Invalid location specs - modifier must be #{include} or #{exclude} but it's #{tokens[0]} instead" 
+      end
+      raise "Invalid location specs" unless ['node', 'chassis', 'rack'].include? tokens[1]
+      
+      options["location_specs_attributes[#{index}][modifier]"] = tokens[0]
+      options["location_specs_attributes[#{index}][loc_type]"] = tokens[1]
+      options["location_specs_attributes[#{index}][value]"] = tokens[2]
+    end
+  end
 end

data/ssvm_cli.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = "ssvm_cli"
-  s.version     = "0.0.2"
+  s.version     = "0.0.3"
   s.platform    = Gem::Platform::RUBY
   s.authors     = ["Darren Dao"]
   s.email       = ["darrendao@gmail.com"]

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: ssvm_cli
 version: !ruby/object:Gem::Version 
-  hash: 27
+  hash: 25
   prerelease: 
   segments: 
   - 0
   - 0
-  - 2
-  version: 0.0.2
+  - 3
+  version: 0.0.3
 platform: ruby
 authors: 
 - Darren Dao
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-05-30 00:00:00 Z
+date: 2012-06-06 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: thor