virtualmaster 0.0.3

data/.gitignore

@@ -0,0 +1,6 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+vendor/bundle
+vendor/cache

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in vmaster.gemspec
+gemspec

data/README.mdown

@@ -0,0 +1,55 @@
+# VirtualMaster command line interface
+
+Proof of concept of VirtualMaster command line interface
+
+    # virtualmaster config
+    Running virtualmaster for first time.
+    Setting up environment.
+
+    Your API credentials are available from http://www.virtualmaster.cz/en/api#settings
+
+    Enter your API username:
+    Enter your API password:
+
+    Settings stored within ~/.virtualmaster
+
+    # virtualmaster create demo1 --image lucid
+    Using image 'lucid' with id (999).
+    Creating 'micro' instance (512 MB memory/5 GB storage).
+    Instance launch request accepted (instance id 44444).
+
+    Default password 'AdVc:PBi8&7L'.
+
+    # virtualmaster instances
+    +-------+---------+----------------+
+    | name  | state   | ip_address     |
+    +-------+---------+----------------+
+    | demo1 | RUNNING | 195.140.253.88 |
+    +-------+---------+----------------+
+
+## Install
+
+The current version of `virtualmaster` command line interface should work with Ruby 1.8.7+.
+
+## Automatically install SSH keys
+
+VirtualMaster CLI can install your SSH keys to a remote machine automatically using `--copy-id` switch.
+
+
+	virtualmaster create demo1 --image ubuntu_lucid --copy-id 
+	Using image 'ubuntu_lucid' with ID 124
+	Creating 'micro' instance (512 MB memory/10 GB storage)
+	Instance launch request accepted. Instance ID 45387
+
+	Default password 'vBK7i!kK'
+	Waiting for instance............
+	Loading identity file
+
+	Instance ready!
+	Try to login using `ssh root@195.140.253.130'
+	
+If you want to specify other key (ie. not ~/.ssh/id_rsa) use option `--identity IDENTITY_FILE`.
+
+## More information	
+
+Additional topics are available in [the wiki](https://github.com/Virtualmaster/virtualmaster-cli/wiki).

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/virtualmaster

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+
+begin
+  require 'vmaster'
+rescue LoadError
+  vmaster_path = File.expand_path('../../lib', __FILE__)
+
+  $:.unshift(vmaster_path)
+
+  require 'vmaster'
+end
+
+program :name, "virtualmaster"
+program :version, VirtualMaster::VERSION
+program :description, "VirtualMaster command line interface"
+program :help_formatter, :compact
+
+default_command :help
+
+VirtualMaster::CLI.run do
+end
+

data/lib/vmaster.rb

@@ -0,0 +1,9 @@
+require "vmaster/version"
+require "vmaster/helpers"
+require "vmaster/cli"
+require "commander/import"
+
+# include commands
+require 'vmaster/config_command'
+require 'vmaster/server_commands'
+

data/lib/vmaster/cli.rb

@@ -0,0 +1,32 @@
+require 'yaml'
+require 'deltacloud'
+
+module VirtualMaster
+  class CLI
+    @@api = nil
+    @@config = nil
+
+    def self.run
+      # load config
+      config_file = File.join(ENV["HOME"], ".virtualmaster")
+      if File.exists? config_file
+        config = YAML::load(File.open(config_file))
+
+        @@config = config.inject({}){|memo,(k,v)| memo[k.to_sym] = v; memo}
+
+        @@api = DeltaCloud.new(@@config[:username], @@config[:password], VirtualMaster::DEFAULT_URL)
+      end
+
+      yield
+    end
+
+    def self.api
+      abort "No configuration available! Please run 'virtualmaster config' first!" unless @@api
+      @@api
+    end
+
+    def self.config
+      @@config
+    end
+  end
+end

data/lib/vmaster/config_command.rb

@@ -0,0 +1,50 @@
+require 'yaml'
+
+
+command :config do |c|
+  c.description = "Configure VirtualMaster API username and password"
+
+  c.option '--username NAME', String, 'API username'
+  c.option '--password PASSWD', String, 'API password'
+
+  c.action do |args, options|
+    config_file = File.join(ENV['HOME'], VirtualMaster::CONFIG_FILE)
+    if File.exists?(config_file)
+      abort "Default configuration file already exists: #{config_file}"
+    else
+      say "Running virtualmaster for first time"
+    end
+
+    unless options.username && options.password
+      say "\n"
+      say "Your API credentials are available from http://www.virtualmaster.cz/en/api#settings"
+      say "\n"
+    end
+
+    options.username = ask("Enter API username:") unless options.username
+    options.password = password("Enter API password:", "*") unless options.password
+
+    # verify and store credentials
+    begin
+      api = DeltaCloud.new(options.username, options.password, VirtualMaster::DEFAULT_URL)
+
+      config = {
+        'username' => options.username,
+        'password' => options.password,
+        'default_image' => VirtualMaster::DEFAULT_IMAGE,
+        'images' => VirtualMaster::IMAGES
+      }
+
+      File.open(config_file, 'w') do |f|
+        f.puts YAML.dump(config)
+      end
+
+      say "Setting stored under #{config_file}"
+    rescue DeltaCloud::API::BackendError => e
+      say "Unable to connect to VirtualMaster API: #{e.message}"
+    rescue Exception => e
+      say "Unable to configure environment: #{e.message}"
+    end
+
+  end
+end

data/lib/vmaster/helpers.rb

@@ -0,0 +1,71 @@
+module VirtualMaster
+
+  CONFIG_FILE = ".virtualmaster"
+  DEFAULT_URL = "https://www.virtualmaster.cz/services/deltacloud"
+
+  DEFAULT_IMAGE = 124
+  DEFAULT_PROFILE = "micro"
+
+  # pre-defined list of images
+  IMAGES = {
+    :ubuntu_lucid => 124,
+    :debian_squeeze => 1741,
+    :centos_6 => 1743
+  }
+
+  PROFILES = {
+    :nano => {
+      :memory => 256,
+      :storage => 3840
+    },
+    :micro => {
+      :memory => 512,
+      :storage => 6144
+    },
+    :milli => {
+      :memory => 1024,
+      :storage => 20480,
+    },
+    :small => {
+      :memory => 2048,
+      :storage => 30720
+    },
+    :medium => {
+      :memory => 4096,
+      :storage => 40960
+    }
+    #:large => {
+    #  :memory => 8192,
+    #  :storage => 8192
+    #}
+  }
+
+
+  module Helpers
+    def self.get_instances
+      VirtualMaster::CLI.api.instances
+    end
+
+    def self.get_instance(name)
+      get_instances.each do |instance|
+        return instance if instance.name == name
+      end
+
+      nil
+    end
+
+    def self.get_hw_profile(memory, storage)
+      api = VirtualMaster::CLI.api
+
+      profile_list = api.hardware_profiles.reject { |p| p.memory.value.to_i != memory && p.storage.value.to_i != storage }
+
+      profile_list.first
+    end
+
+    def self.create_instance(name, image_id, profile_id)
+      api = VirtualMaster::CLI.api
+
+      api.create_instance(image_id, :name => name, :hwp_id => profile_id)
+    end
+  end
+end

data/lib/vmaster/server_commands.rb

@@ -0,0 +1,163 @@
+require 'terminal-table'
+require 'net/ssh'
+require 'logger'
+require "base64"
+require 'openssl'
+
+log = Logger.new(STDOUT)
+log.level = Logger::WARN
+
+# SSH copy id
+# FIXME how to select the right key
+
+command :create do |c|
+  c.description = "Launch new server instance"
+  c.option '--image TEMPLATE', String, 'instance template to use'
+  c.option '--profile PROFILE', String, 'instance hardware profile'
+  c.option '--copy-id', 'install public key on a machine'
+  c.option '--identity IDENTITY', String, 'SSH identity to use (with --copy-id)'
+  c.option '--wait', 'wait for instance to become operational'
+  c.action do |args, options|
+    # default values
+    options.default :identity => File.join(ENV['HOME'], '.ssh/id_rsa')
+
+    name = args.shift || abort('Server name required')
+
+    # verify server name
+    abort("Virtual server with name #{name} already exists!") if VirtualMaster::Helpers.get_instance(name)
+
+    # image 
+    image_name = nil
+    image_id = VirtualMaster::CLI.config[:default_image] || VirtualHost::DEFAULT_IMAGE
+
+    if options.image
+      image_name = options.image
+
+      if image_name.match /^id:/
+        # use image_id directly
+        image_id = image_name[3..-1].to_i
+
+        image_name = nil
+      else
+        # lookup predefined images
+        image_id = VirtualMaster::CLI.config[:images][image_name.to_sym]
+
+        abort "Image '#{image_name}' not recognized!" unless image_id
+      end
+    end
+
+    say image_name ? "Using image '#{image_name}' with ID #{image_id}" : "Using image with ID #{image_id}"
+
+    # instance hardware profile
+    profile_name = options.profile || VirtualMaster::DEFAULT_PROFILE
+
+    profile = VirtualMaster::PROFILES[profile_name.to_sym]
+    abort "Image name '#{options.profile}' not recognized!" unless profile
+
+    hwp = VirtualMaster::Helpers.get_hw_profile(profile[:memory], profile[:storage])
+    abort "Internal error: hardware profile not available" unless hwp
+
+    say "Creating '#{profile_name}' instance (#{profile[:memory]} MB memory/#{profile[:storage]/1024} GB storage)"
+
+    instance = VirtualMaster::Helpers.create_instance(name, image_id, hwp.id)
+
+    # TODO handle exceptions (invalid image/profile, limits, etc.)
+
+    say "Instance launch request accepted. Instance ID #{instance.id}"
+
+    # FIXME authentication is missrepresented within Ruby object
+    password = instance.authentication[:username]
+    say "\n"
+    say "Default password '#{password}'"
+
+    # copy-id implies waiting for instance to become operational
+    if options.wait || options.copy_id
+      print 'Waiting for instance'
+
+      while (instance = VirtualMaster::Helpers.get_instance(name)).state != "RUNNING" do
+        print '.'
+
+        sleep(5)
+      end
+      
+      puts
+
+      # copy ssh id
+      if options.copy_id
+        authorized_key = nil
+
+        abort "Specified identity file #{options.identity} doesn't exists!" unless File.exist?(options.identity)
+        
+        say "Loading identity file\n"
+        key = OpenSSL::PKey::RSA.new File.read options.identity
+        
+        # build authorized key output string
+        authtype = key.class.to_s.split('::').last.downcase
+        b64pub = ::Base64.encode64(key.to_blob).strip.gsub(/[\r\n]/, '')
+        authorized_key = "ssh-%s %s\n" % [authtype, b64pub]  # => ssh-rsa AAAAB3NzaC1...=
+        
+        Net::SSH.start(instance.public_addresses.first[:address], 'root', :password => password) do |ssh|
+          # TODO exception handling
+          output = ssh.exec!("mkdir ~/.ssh")
+          output = ssh.exec!("echo '#{authorized_key}' >>~/.ssh/authorized_keys")
+        end
+      end
+      
+      puts
+      puts "Instance ready!"
+      puts "Try to login using `ssh root@#{instance.public_addresses.first[:address]}'"      
+    end
+  end
+end
+
+command :list do |c|
+  c.description = "List all running servers"
+  c.action do |args, options|
+    instances = []
+
+    VirtualMaster::Helpers.get_instances.each do |instance|
+      unless instance.public_addresses.first.nil?
+        ip_address = instance.public_addresses.first[:address]
+      else
+        ip_address = "(not assigned)"
+      end
+
+      instances << [instance.name, instance.state, ip_address]
+    end
+
+    abort "No instances found" if instances.empty?
+
+    table = Terminal::Table.new :headings => ['name','state','ip_address'], :rows => instances
+    puts table
+  end
+end
+
+def instance_action(action, args)
+  name = args.shift || abort('server name required')
+
+  instance = VirtualMaster::Helpers.get_instance(name)
+  instance.send("#{action}!")
+end
+
+%w{start reboot stop shutdown destroy}.each do |cmd|
+  command cmd do |c|
+    c.syntax = "virtualmaster #{c.name} SERVER"
+
+    case c.name
+    when "start"
+      c.description = "Start server (when stopped)"
+    when "reboot"
+      c.description = "Reboot server"
+    when "stop"
+      c.description = "Stop server"
+    when "shutdown" 
+      c.description = "Shutdown server (ACPI)"
+    when "destroy"
+      c.description = "Remove server"
+    end
+
+    c.action do |args, options|
+      instance_action(c.name, args)
+    end
+  end
+end

data/lib/vmaster/version.rb

@@ -0,0 +1,3 @@
+module VirtualMaster
+  VERSION = "0.0.3"
+end

data/spec/commands_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe "VirtualMaster commands" do
+  before :each do
+    @runner = Commander::Runner.instance
+  end
+
+  it "should have command :config" do
+    @runner.commands.should have_key('config')
+  end
+
+  it "should have command server commands" do
+    %w{create list start reboot stop shutdown destroy}.each do |cmd|
+      @runner.commands.should have_key(cmd)
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,13 @@
+$:.unshift File.dirname(__FILE__) + '/../lib'
+
+require 'vmaster'
+
+program :name, "virtualmaster"
+program :version, VirtualMaster::VERSION
+program :description, "VirtualMaster command line interface"
+program :help_formatter, :compact
+
+default_command :test
+
+command :test do |c|
+end

data/virtualmaster.gemspec

@@ -0,0 +1,31 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "vmaster/version"
+
+Gem::Specification.new do |s|
+  s.name        = "virtualmaster"
+  s.version     = VirtualMaster::VERSION
+  s.authors     = ["Radim Marek"]
+  s.email       = ["radim@laststation.net"]
+  s.homepage    = "https://github.com/virtualmaster/virtualmaster-cli"
+  s.summary     = %q{Command line interface to VirtualMaster}
+  s.description = %q{Command line interface to VirtualMaster. Control your virtual infrastructure.}
+
+  s.rubyforge_project = "vmaster"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+  
+  s.add_dependency "commander", "~> 4.1.2"
+  s.add_dependency "deltacloud-client", "~> 0.5.0"
+  s.add_dependency "terminal-table", "~> 1.4.4"
+  s.add_dependency "net-ssh", "~> 2.3.0"
+
+  s.add_development_dependency "rspec", "~> 2"
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: virtualmaster
+version: !ruby/object:Gem::Version
+  version: 0.0.3
+  prerelease: 
+platform: ruby
+authors:
+- Radim Marek
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-03-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: commander
+  requirement: &70220237000360 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 4.1.2
+  type: :runtime
+  prerelease: false
+  version_requirements: *70220237000360
+- !ruby/object:Gem::Dependency
+  name: deltacloud-client
+  requirement: &70220236995660 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.5.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70220236995660
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: &70220236993960 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.4.4
+  type: :runtime
+  prerelease: false
+  version_requirements: *70220236993960
+- !ruby/object:Gem::Dependency
+  name: net-ssh
+  requirement: &70220236992760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.3.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70220236992760
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70220236991620 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :development
+  prerelease: false
+  version_requirements: *70220236991620
+description: Command line interface to VirtualMaster. Control your virtual infrastructure.
+email:
+- radim@laststation.net
+executables:
+- virtualmaster
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- README.mdown
+- Rakefile
+- bin/virtualmaster
+- lib/vmaster.rb
+- lib/vmaster/cli.rb
+- lib/vmaster/config_command.rb
+- lib/vmaster/helpers.rb
+- lib/vmaster/server_commands.rb
+- lib/vmaster/version.rb
+- spec/commands_spec.rb
+- spec/spec_helper.rb
+- virtualmaster.gemspec
+homepage: https://github.com/virtualmaster/virtualmaster-cli
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: vmaster
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: Command line interface to VirtualMaster
+test_files:
+- spec/commands_spec.rb
+- spec/spec_helper.rb