ssvm_cli 0.0.1

data/README.md

@@ -0,0 +1,4 @@
+chefkitchen_cli
+===============
+
+CLI for talking to chefkitchen to do quick chef tasks

data/bin/ssvm

@@ -0,0 +1,70 @@
+#!/usr/bin/env ruby
+$:.unshift File.join(File.dirname(__FILE__), "..", "lib")
+
+require 'fileutils'
+require 'yaml'
+require 'rubygems'
+require 'thor'
+require 'cookie_jar'
+require 'request_helper'
+
+class Ssvm < Thor
+  CONFIG_FILE = File.expand_path('~/.ssvm.yml')
+
+  desc 'create_vm', 'submit a VM creation request'
+  method_option :json, :desc => "Return result in json format"
+  method_option :desired_hostname, :aliases => '-n', :required => true, :desc => "Name for the VM"
+  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 :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
+  method_option :optimal_spread, :desc => "Optimal spread for multi instances request", :default => "Rack"
+  method_option :minimal_spread, :desc => "Minimal spread for multi instances request", :default => "Rack"
+  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 :justifications, :aliases => '-j', :desc => "Justification for any special request"
+  def create_vm
+    conf = load_conf(options[:conf_file] || CONFIG_FILE)
+    rhelper = RequestHelper.new(conf)
+    rhelper.create_vm(options)
+  end
+
+  desc "config", "configuration setup"
+  method_option :conf_file, :aliases => "-c", :desc => "Where to write config to"
+  def config
+    config_file = File.expand_path(options[:conf_file] || CONFIG_FILE)
+    puts "Provide the following inputs so ssvm_cli can save them to #{config_file} for future execution"
+    conf = {}
+    conf[:username] = ask("Username: ")
+    conf[:password] = ask("Password: ", true)
+    conf[:server] = ask("Chefkitchen server: ")
+    File.open(config_file, 'w') do |f|
+      YAML.dump(conf, f)
+    end
+    FileUtils.chmod 0700, config_file 
+    puts "#{config_file} has been created"
+  end
+
+  private
+  def ask(str,mask=false)
+    begin
+      print str
+      system 'stty -echo;' if mask
+      input = STDIN.gets.chomp
+    ensure
+      system 'stty echo; echo ""'
+    end
+    return input
+  end
+
+  def load_conf(conf_file)
+    YAML::load(File.open(File.expand_path(conf_file)))
+  end
+end
+
+Ssvm.start

data/lib/cookie_jar.rb

@@ -0,0 +1,16 @@
+module CookieJar
+  def get_cookies
+    params = {:login => @username, :password => @password}
+    cookies = nil
+    RestClient.post("https://#{@server}/login/login", params) do |response, request, result, &block|
+      if response.code == 200
+        cookies = response.cookies      
+      elsif response.code == 302 && response.headers[:location] !~ /login/
+        cookies = response.cookies      
+      else
+        raise "Failed to authenticate"
+      end
+    end 
+    cookies
+  end
+end

data/lib/request_helper.rb

@@ -0,0 +1,66 @@
+require 'rubygems'
+require 'rest-client'
+require 'json'
+require 'cookie_jar'
+
+class RequestHelper
+  include CookieJar
+
+  def initialize(conf)
+    @server = conf[:server] || "localhost"
+    @username = conf[:username]
+    @password = conf[:password]
+    @cookies = get_cookies
+  end
+
+  def create_vm(options)
+    options = options.dup
+    update_owner_field(options)
+    handle_multi_instances(options)
+    handle_custom_request(options)
+
+    begin
+      response = RestClient.post("http://#{@server}/vm",
+                                 {:vm_request => options},
+                                 {: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']
+      options['primary_user'] = options['owners'].shift
+      options['secondary_user'] = options['owners']
+      options.delete('owners')
+    else
+      options['primary_user'] = @username
+    end
+  end
+
+  def handle_multi_instances(options)
+    if options['instances']
+      options['multi_instances'] = true
+      options['multi_instances_spec_attributes[optimal_spread]'] = options['optimal_spread']
+      options['multi_instances_spec_attributes[minimal_spread]'] = options['minimal_spread']
+      options['multi_instances_spec_attributes[num_instances]'] = options['instances']
+      options['multi_instances_spec_attributes[start_at]'] = options['start_at']
+    end
+
+    options.delete('optimal_spread')
+    options.delete('minimal_spread')
+    options.delete('start_at')
+    options.delete('instances')
+  end
+
+  def handle_custom_request(options)
+    options['processor_count'] = options.delete('cpu')
+  end
+end

data/ssvm_cli.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+
+$LOAD_PATH.push File.expand_path("../lib", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "ssvm_cli"
+  s.version     = "0.0.1"
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Darren Dao"]
+  s.email       = ["darrendao@gmail.com"]
+  s.homepage    = "https://github.com/darrendao/ssvm_cli"
+  s.summary     = %q{Ruby script to talk to ssvm web services.}
+  s.description     = %q{Ruby script to talk to ssvm web services.}
+
+  s.add_dependency 'thor'
+  s.add_dependency 'json'
+  s.add_dependency 'rest-client'
+  s.add_development_dependency 'yard', '~> 0.7'
+
+  s.files            = `git ls-files`.split("\n")
+  s.executables      = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths    = ["lib"]
+  s.extra_rdoc_files = ["README.md"]
+end
+

metadata

@@ -0,0 +1,127 @@
+--- !ruby/object:Gem::Specification 
+name: ssvm_cli
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Darren Dao
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-05-30 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thor
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: json
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rest-client
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: yard
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 5
+        segments: 
+        - 0
+        - 7
+        version: "0.7"
+  type: :development
+  version_requirements: *id004
+description: Ruby script to talk to ssvm web services.
+email: 
+- darrendao@gmail.com
+executables: 
+- ssvm
+extensions: []
+
+extra_rdoc_files: 
+- README.md
+files: 
+- README.md
+- bin/ssvm
+- lib/cookie_jar.rb
+- lib/request_helper.rb
+- ssvm_cli.gemspec
+homepage: https://github.com/darrendao/ssvm_cli
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Ruby script to talk to ssvm web services.
+test_files: []
+
+has_rdoc: