virtuoso-prototype 0.0.1

data/LICENSE

@@ -0,0 +1,14 @@
+ Copyright (C) 2010 3Crowd Technologies, Inc.
+
+ This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.

data/README

@@ -0,0 +1,2 @@
+README for Virtuoso prototype
+=============================

data/bin/virtuoso-prototype

@@ -0,0 +1,8 @@
+#!/usr/bin/env ruby
+
+$:.unshift(File.expand_path('..',File.dirname(__FILE__))) unless $:.include?(File.expand_path('..',File.dirname(__FILE__)))
+
+require 'lib/virtuoso'
+
+cli = Virtuoso::CLI.new
+cli.run(ARGV)

data/lib/virtuoso.rb

@@ -0,0 +1,80 @@
+$:.unshift(File.expand_path(File.dirname(__FILE__))) unless $:.include?(File.expand_path(File.dirname(__FILE__)))
+
+require 'virtuoso/errors'
+require 'virtuoso/virtualbox'
+
+#Choice, a command line option parser
+require 'choice'
+require 'logger'
+
+module Virtuoso
+
+class CLI
+
+  def initialize
+    @log = Logger.new(STDOUT)
+    @log.level = Logger::DEBUG
+    @vm_interface = VirtualBox::Interface.new
+  end
+
+  def run(arguments=[])
+    setup_cli_arguments!
+    options = parse_arguments(arguments)
+    ensure_virtualbox_installed
+    perform_action!(options[:operation], options[:vm_name], options[:bond_interface])
+  end
+
+  private
+
+  def perform_action!(operation, vm_name, bond_interface)
+    case operation
+      when 'create' then
+        @vm_interface.create!(vm_name, bond_interface)
+      when 'destroy' then
+        @vm_interface.destroy!(vm_name, :delete)
+      else
+        raise RuntimeError.new("Got operation option that should not be valid: #{operation}")
+    end
+  end
+
+  def ensure_virtualbox_installed
+    unless @vm_interface.installed?
+      raise Error::VirtualBoxNotInstalledError
+    end
+  end
+
+  def setup_cli_arguments!
+    Choice.options do
+      header 'Virtuoso Virtual Manager'
+      header 'Specific options:'
+
+      option :operation do
+        short '-o'
+        long '--operation=OPERATION'
+        desc 'The operation you wish to perform (create, or destroy) (default: create)'
+        valid %w[create destroy]
+        default 'create'
+      end
+
+      option :vm_name, :required => true do
+        short '-n'
+        long '--vmname=VMNAME'
+        desc 'The virtual machine you wish to operate on (required)'
+      end
+
+      option :bond_interface, :required => true do
+        short '-i'
+        long '--bond_interface=INTERFACE0'
+        desc 'The network interface on the host machine to which the virtual network interface should bond'
+      end
+    end
+  end
+
+  def parse_arguments(arguments)
+    @log.debug("Got choices: #{ Choice.choices.inspect }")
+    Choice.choices
+  end
+
+end
+
+end

data/lib/virtuoso/errors.rb

@@ -0,0 +1,4 @@
+require 'virtuoso/errors/disk_not_created'
+require 'virtuoso/errors/virtualbox_not_found'
+require 'virtuoso/errors/vm_not_found'
+require 'virtuoso/errors/vm_not_created'

data/lib/virtuoso/errors/disk_not_created.rb

@@ -0,0 +1,8 @@
+module Virtuoso
+module Errors
+
+class DiskNotCreatedError < StandardError
+end
+
+end
+end

data/lib/virtuoso/errors/virtualbox_not_found.rb

@@ -0,0 +1,8 @@
+module Virtuoso
+module Errors
+
+class VirtualBoxNotFoundError < StandardError
+end
+
+end
+end

data/lib/virtuoso/errors/vm_not_created.rb

@@ -0,0 +1,8 @@
+module Virtuoso
+module Errors
+
+  class VMNotCreatedError < StandardError
+  end
+
+end
+end

data/lib/virtuoso/errors/vm_not_found.rb

@@ -0,0 +1,8 @@
+module Virtuoso
+module Errors
+
+class VMNotFoundError < StandardError
+end
+
+end
+end

data/lib/virtuoso/virtualbox.rb

@@ -0,0 +1 @@
+require 'virtuoso/virtualbox/interface'

data/lib/virtuoso/virtualbox/interface.rb

@@ -0,0 +1,109 @@
+require 'rubygems'
+require 'virtualbox'
+require 'systemu'
+require 'logger'
+require 'uuid'
+
+module Virtuoso
+module VirtualBox         
+
+  class Interface
+
+    def initialize
+      @log = Logger.new(STDOUT)
+      @virtualbox = Hash.new
+      @virtualbox[:version] = ::VirtualBox.version
+    end
+
+    def create!(vm_name, bond_interface)
+      vm = create_vm!(vm_name)
+      vm = setup_vm!(vm)
+      vm.reload
+      storage_controller = add_storage_controller!(vm)
+      disk = create_disk!(vm_name)
+      attach_disk!(vm, storage_controller, disk)
+      add_bridged_network_interface!(vm, 1, bond_interface)
+    end
+
+    def destroy!(vm_name, physically_delete_media=true)
+      vm = ::VirtualBox::VM.find(vm_name)
+      raise Errors::VMNotFoundError if vm.nil?
+      vm.destroy(:destroy_medium => physically_delete_media )
+    end
+
+    def installed?
+      @virtualbox[:version].nil? ? false : true
+    end
+
+    private
+
+    def add_bridged_network_interface!(vm, interface_number, bridge_interface)
+      status, stdout, stderr = systemu "VBoxManage modifyvm --nic#{interface_number} bridged --bridgeadapter#{interface_number} #{bridge_interface}"
+      @log.debug("Ran VBoxManage add network interface: output (#{status}):")
+      @log.debug("Standard out: " + stdout)
+      @log.debug("Standard err: " + stderr)
+      @ Reload the vm config, to pick up command line changes
+      vm.reload
+      vm.network_interfaces[interface_number-1]
+    end
+
+    def add_storage_controller!(vm)
+      uuid = UUID.new.generate
+      #FIXME: we have to shell out here since the virtualbox gem does not provide an easy method for creating storage controllers from scratch
+      status, stdout, stderr = systemu "VBoxManage storagectl #{vm.uuid} --name #{uuid} --add sata"
+      @log.debug("Ran VBoxManage add storage controller: output (#{status}):")
+      @log.debug("Standard out: " + stdout)
+      @log.debug("Standard err: " + stderr)
+      # Reload the vm config, to pick up the command line changes
+      vm.reload
+      vm.storage_controllers.select{|controller| controller.name == uuid }.first
+    end
+
+    def attach_disk!(vm, storage_controller, disk)
+      @log.debug("Got storage_controller #{storage_controller.inspect}, Got Disk #{disk.inspect}")
+      #FIXME: we have to shell out here since the virtualbox gem does not provide an easy to find method for attaching disk images to VMs
+      command = "VBoxManage storageattach #{vm.uuid} --storagectl #{storage_controller.name} --port 0 --device 0 --type hdd --medium #{disk.uuid} "
+      @log.debug("Running: #{command}")
+      status, disk_attach_stdout, disk_attach_stderr = systemu(command)
+      @log.debug("Ran VBoxManage storageattach output (#{status}):")
+      @log.debug("Standard out: " + disk_attach_stdout)
+      @log.debug("Standard err: " + disk_attach_stderr)
+    end
+
+    def setup_vm!(vm)
+      vm.vram_size = 24 #megabytes
+      vm.memory_size = 512 #megabytes
+      vm.save
+      return vm
+    end
+
+    def create_vm!(vm_name)
+      #FIXME: we have to shell out here since the virtualbox gem does not provide a method for creating new VMs directly through the library
+      status, vm_create_output_stdout, vm_create_output_stderr = systemu "VBoxManage createvm --name #{vm_name} --register"
+      @log.debug("Ran VBoxManage createvm output (#{status}):")
+      @log.debug("Standard out: " + vm_create_output_stdout)
+      @log.debug("Standard err: " + vm_create_output_stderr)
+      unless status != 0
+        return ::VirtualBox::VM.find(vm_name)
+      else
+        raise Errors::VMNotCreatedError
+      end
+    end
+
+    def create_disk!(disk_name)
+      disk = ::VirtualBox::HardDrive.new
+      disk.format = "VDI"
+      disk.location = "#{disk_name}.vdi"
+      disk.logical_size = 8000 # in megabytes
+      if disk.save
+        return disk
+      else
+        raise Errors::DiskNotCreatedError.new(disk.inspect)
+      end
+    end
+
+  end
+
+end
+end
+

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification 
+name: virtuoso-prototype
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Justin Lynn
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-09-22 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: A quick and dirty prototype for managing virtualbox virtual machines, only scripts VBoxManage to eliminate bringup of a raw VM
+email: 
+- eng@3crowd.com
+executables: 
+- virtuoso-prototype
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- bin/virtuoso-prototype
+- lib/virtuoso.rb
+- lib/virtuoso/virtualbox/interface.rb
+- lib/virtuoso/errors/vm_not_found.rb
+- lib/virtuoso/errors/disk_not_created.rb
+- lib/virtuoso/errors/virtualbox_not_found.rb
+- lib/virtuoso/errors/vm_not_created.rb
+- lib/virtuoso/virtualbox.rb
+- lib/virtuoso/errors.rb
+- LICENSE
+- README
+has_rdoc: true
+homepage: http://github.com/3Crowd/Virtuoso-prototype
+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: 23
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A protoype for managing virtualbox VMs, scripts VBoxManage
+test_files: []
+