lab 0.1.5 → 0.2.0

data/lib/lab/driver/fog_driver.rb

@@ -8,156 +8,156 @@ module Lab
 module Drivers
 class FogDriver < VmDriver
 
-	def initialize(config,fog_config)
-		
-		super(config)
-		@fog_config = fog_config
-
-		# Soft dependency
-		begin
-			require 'fog'
-		rescue LoadError
-			raise "WARNING: Library fog not found. Could Not Create Driver"
-		end
-
-		if @fog_config['fog_type'] == "ec2"
-
-			# AWS / EC2 Base Credential Configuration
-			@aws_cert_file = IO.read(fog_config['fog_aws_cert_file']).chomp if fog_config['fog_aws_cert_file']
-			@aws_private_key_file = IO.read(fog_config['fog_aws_private_key_file']).chomp if fog_config['fog_aws_private_key_file']
-			@ec2_access_key_file = IO.read(fog_config['fog_ec2_access_key_file']).chomp if fog_config['fog_ec2_access_key_file']
-			@ec2_secret_access_key_file = IO.read(fog_config['fog_ec2_secret_access_key_file']).chomp if fog_config['fog_ec2_secret_access_key_file']
-			
-			# Instance Keys
-			@ec2_instance_public_key_file = IO.read(fog_config['fog_ec2_instance_public_key_file']).chomp if fog_config['fog_ec2_instance_public_key_file']
-			@ec2_instance_private_key_file = IO.read(fog_config['fog_ec2_instance_private_key_file']).chomp if fog_config['fog_ec2_instance_private_key_file']
-			
-			# Instance Details
-			@ec2_base_ami = fog_config['fog_ec2_base_ami']
-			@ec2_flavor = fog_config['fog_ec2_flavor']
-			@ec2_user = fog_config['fog_ec2_user']
-			@ec2_region = fog_config['fog_ec2_region']
-					
-			# Set up a connection
-			@compute = Fog::Compute.new(
-				:provider => "Aws",
-				:aws_access_key_id => @aws_access_key_file,
-				:aws_secret_access_key => @aws_secret_access_key_file )
-		else
-			raise "Unsupported Fog Type"
-		end
-	end
-
-	def start
-		ec2_settings = {
-			:image_id => @ec2_base_ami, 
-			:flavor_id =>  @ec2_flavor,
-			:public_key_path => @ec2_instance_public_key_file,
-			:private_key_path => @ec2_instance_private_key_file,
-			:username => @ec2_user}
-			
-		begin
-			@fog_server = @compute.servers.bootstrap(ec2_settings)
-		rescue Fog::Compute::AWS::Error => e
-			raise "Couldn't authenticate to AWS - did you place keys in the creds/ directory?"
-			exit
-		end
-	end
-
-	def stop
-		@fog_server.destroy
-	end
-
-	def suspend
-		raise "unimplemented"
-	end
-
-	def pause
-		raise "unimplemented"
-	end
-
-	def reset
-		raise "unimplemented"
-	end
-
-	def create_snapshot(snapshot)
-		raise "unimplemented"
-	end
-
-	def revert_snapshot(snapshot)
-		raise "unimplemented"
-	end
-
-	def delete_snapshot(snapshot)
-		raise "unimplemented"
-	end
+  def initialize(config,fog_config)
+    
+    super(config)
+    @fog_config = fog_config
+
+    # Soft dependency
+    begin
+      require 'fog'
+    rescue LoadError
+      raise "WARNING: Library fog not found. Could Not Create Driver"
+    end
+
+    if @fog_config['fog_type'] == "ec2"
+
+      # AWS / EC2 Base Credential Configuration
+      @aws_cert_file = IO.read(fog_config['fog_aws_cert_file']).chomp if fog_config['fog_aws_cert_file']
+      @aws_private_key_file = IO.read(fog_config['fog_aws_private_key_file']).chomp if fog_config['fog_aws_private_key_file']
+      @ec2_access_key_file = IO.read(fog_config['fog_ec2_access_key_file']).chomp if fog_config['fog_ec2_access_key_file']
+      @ec2_secret_access_key_file = IO.read(fog_config['fog_ec2_secret_access_key_file']).chomp if fog_config['fog_ec2_secret_access_key_file']
+      
+      # Instance Keys
+      @ec2_instance_public_key_file = IO.read(fog_config['fog_ec2_instance_public_key_file']).chomp if fog_config['fog_ec2_instance_public_key_file']
+      @ec2_instance_private_key_file = IO.read(fog_config['fog_ec2_instance_private_key_file']).chomp if fog_config['fog_ec2_instance_private_key_file']
+      
+      # Instance Details
+      @ec2_base_ami = fog_config['fog_ec2_base_ami']
+      @ec2_flavor = fog_config['fog_ec2_flavor']
+      @ec2_user = fog_config['fog_ec2_user']
+      @ec2_region = fog_config['fog_ec2_region']
+          
+      # Set up a connection
+      @compute = Fog::Compute.new(
+        :provider => "Aws",
+        :aws_access_key_id => @aws_access_key_file,
+        :aws_secret_access_key => @aws_secret_access_key_file )
+    else
+      raise "Unsupported Fog Type"
+    end
+  end
+
+  def start
+    ec2_settings = {
+      :image_id => @ec2_base_ami, 
+      :flavor_id =>  @ec2_flavor,
+      :public_key_path => @ec2_instance_public_key_file,
+      :private_key_path => @ec2_instance_private_key_file,
+      :username => @ec2_user}
+      
+    begin
+      @fog_server = @compute.servers.bootstrap(ec2_settings)
+    rescue Fog::Compute::AWS::Error => e
+      raise "Couldn't authenticate to AWS - did you place keys in the creds/ directory?"
+      exit
+    end
+  end
+
+  def stop
+    @fog_server.destroy
+  end
+
+  def suspend
+    raise "unimplemented"
+  end
+
+  def pause
+    raise "unimplemented"
+  end
+
+  def reset
+    raise "unimplemented"
+  end
+
+  def create_snapshot(snapshot)
+    raise "unimplemented"
+  end
+
+  def revert_snapshot(snapshot)
+    raise "unimplemented"
+  end
+
+  def delete_snapshot(snapshot)
+    raise "unimplemented"
+  end
 
 =begin
 
-	def run_command(command)
-		## vm_driver will need a little patching for this to work, as 
-		## amis use keys for auth. i think it's just a matter of not passing the 
-		## password to ssh_exec. So maybe the thing to do is have a ssh_key_exec
-		## function in vm_driver.rb that does the right thing. 
-
-		script_rand_name = rand(10000)
-
-		if @os == "windows"
-			local_tempfile_path = "/tmp/lab_script_#{script_rand_name}.bat"
-			remote_tempfile_path = "C:\\\\lab_script_#{script_rand_name}.bat"
-			remote_run_command = remote_tempfile_path
-		else
-			local_tempfile_path = "/tmp/lab_script_#{script_rand_name}.sh"
-			remote_tempfile_path = "/tmp/lab_script_#{script_rand_name}.sh"
-			remote_run_command = "/bin/sh #{remote_tempfile_path}"
-		end
-
-		# write out our script locally
-		File.open(local_tempfile_path, 'w') {|f| f.write(command) }
-
-		# since we can't copy easily w/o tools, let's just run it directly :/
-		if @os == "linux"
-			output_file = "/tmp/lab_command_output_#{rand(1000000)}"
-			
-			scp_to(local_tempfile_path, remote_tempfile_path)
-			ssh_exec(remote_run_command + "> #{output_file}")
-			scp_from(output_file, output_file)
-			ssh_exec("rm #{output_file}")
-			ssh_exec("rm #{remote_tempfile_path}")
-			
-			# Ghettohack!
-			string = File.open(output_file,"r").read
-			`rm #{output_file}`
-			
-		else
-			raise "zomgwtfbbqnotools"
-		end
-	end
-	
-	def copy_from(from, to)
-		raise "unimplemented"
-	end
-
-	def copy_to(from, to)
-		raise "unimplemented"
-	end
-
-	def check_file_exists(file)
-		raise "unimplemented"
-	end
-
-	def create_directory(directory)
-		raise "unimplemented"
-	end
+  def run_command(command)
+    ## vm_driver will need a little patching for this to work, as 
+    ## amis use keys for auth. i think it's just a matter of not passing the 
+    ## password to ssh_exec. So maybe the thing to do is have a ssh_key_exec
+    ## function in vm_driver.rb that does the right thing. 
+
+    script_rand_name = rand(10000)
+
+    if @os == "windows"
+      local_tempfile_path = "/tmp/lab_script_#{script_rand_name}.bat"
+      remote_tempfile_path = "C:\\\\lab_script_#{script_rand_name}.bat"
+      remote_run_command = remote_tempfile_path
+    else
+      local_tempfile_path = "/tmp/lab_script_#{script_rand_name}.sh"
+      remote_tempfile_path = "/tmp/lab_script_#{script_rand_name}.sh"
+      remote_run_command = "/bin/sh #{remote_tempfile_path}"
+    end
+
+    # write out our script locally
+    File.open(local_tempfile_path, 'w') {|f| f.write(command) }
+
+    # since we can't copy easily w/o tools, let's just run it directly :/
+    if @os == "linux"
+      output_file = "/tmp/lab_command_output_#{rand(1000000)}"
+      
+      scp_to(local_tempfile_path, remote_tempfile_path)
+      ssh_exec(remote_run_command + "> #{output_file}")
+      scp_from(output_file, output_file)
+      ssh_exec("rm #{output_file}")
+      ssh_exec("rm #{remote_tempfile_path}")
+      
+      # Ghettohack!
+      string = File.open(output_file,"r").read
+      `rm #{output_file}`
+      
+    else
+      raise "zomgwtfbbqnotools"
+    end
+  end
+  
+  def copy_from(from, to)
+    raise "unimplemented"
+  end
+
+  def copy_to(from, to)
+    raise "unimplemented"
+  end
+
+  def check_file_exists(file)
+    raise "unimplemented"
+  end
+
+  def create_directory(directory)
+    raise "unimplemented"
+  end
 =end
 
-	def cleanup
-		@fog_server.destroy
-	end
+  def cleanup
+    @fog_server.destroy
+  end
 
-	def running?
-		return true #TODO
-	end
+  def running?
+    return true #TODO
+  end
 
 end
 end 

data/lib/lab/driver/remote_esxi_driver.rb

@@ -0,0 +1,177 @@
+require 'vm_driver'
+
+##
+## $Id$
+##
+
+# This driver was built against: 
+# VMware ESXi Host Agent 4.1.0 build-348481
+
+module Lab
+module Drivers
+
+class RemoteEsxiDriver < VmDriver
+
+  def initialize(config)
+    unless config['user'] then raise ArgumentError, "Must provide a username" end
+    unless config['host'] then raise ArgumentError, "Must provide a hostname" end
+
+    super(config)
+    
+    @user = filter_command(config['user'])
+    @host = filter_command(config['host'])
+    @port = config['port']    
+  end
+
+  def start
+    remote_system_command("vim-cmd vmsvc/power.on #{@vmid}")
+  end
+
+  def stop
+    remote_system_command("vim-cmd vmsvc/power.off #{@vmid}")
+  end
+
+  def suspend
+    remote_system_command("vim-cmd vmsvc/power.suspend #{@vmid}")
+  end
+
+  def pause
+    remote_system_command("vim-cmd vmsvc/power.suspend #{@vmid}")
+  end
+
+  def resume
+    remote_system_command("vim-cmd vmsvc/power.suspendResume #{@vmid}")
+  end
+
+  def reset
+    remote_system_command("vim-cmd vmsvc/power.reset #{@vmid}")
+  end
+
+  def create_snapshot(snapshot)
+    snapshot = filter_input(snapshot)
+    
+    remote_system_command("vim-cmd vmsvc/snapshot.create #{@vmid} #{snapshot} \'lab created snapshot\' 1 true")
+  end
+
+  def revert_snapshot(snapshot)
+
+    snapshots = get_snapshots
+    
+    # Look through our snapshot list, choose the right one based on display_name    
+    snapshots.each do |snapshot_obj|
+    
+      #puts "DEBUG: checking #{snapshot_obj}"
+    
+      if snapshot_obj[:display_name].downcase == snapshot.downcase
+        snapshot_identifier = snapshot_obj[:name].join(" ")
+        
+        #puts "DEBUG: I would revert to #{snapshot_obj}"
+        remote_system_command("vim-cmd vmsvc/snapshot.revert #{@vmid} 0 #{snapshot_identifier}")
+        return true
+      end
+    end
+  
+    # If we got here, the snapshot didn't exist
+    raise "Invalid Snapshot Name"
+  end
+
+  def delete_snapshot(snapshot, remove_children=false)
+    snapshots = get_snapshots
+    
+    # Look through our snapshot list, choose the right one based on display_name    
+    snapshots.each do |snapshot_obj|
+    
+      #puts "DEBUG: checking #{snapshot_obj}"
+    
+      if snapshot_obj[:display_name].downcase == snapshot.downcase
+        snapshot_identifier = snapshot_obj[:name].join(" ")
+        remote_system_command("vim-cmd vmsvc/snapshot.remove #{@vmid} #{remove_children} #{snapshot_identifier}")
+        return true
+      end
+    end
+  
+    # If we got here, the snapshot didn't exist
+    raise "Invalid Snapshot Name"
+  end
+  
+  def delete_all_snapshots
+    remote_system_command("vim-cmd vmsvc/snapshot.removeall #{@vmid}")
+  end
+    
+  def run_command(command)
+    raise "Not Implemented"
+  end
+  
+  def copy_from(from, to)
+    if @os == "linux"
+      scp_from(from, to)
+    else
+      raise "Unimplemented"
+    end
+  end
+
+  def copy_to(from, to)
+    if @os == "linux"
+      scp_to(from, to)
+    else
+      raise "Unimplemented"
+    end
+  end
+
+  def check_file_exists(file)
+    raise "Not Implemented"
+  end
+
+  def create_directory(directory)
+    raise "Not Implemented"
+  end
+
+  def cleanup
+
+  end
+
+  def running?
+    power_status_string = `ssh #{@user}@#{@host} \"vim-cmd vmsvc/power.getstate #{@vmid}\"`
+    return true if power_status_string =~ /Powered on/
+  false
+  end 
+
+  def get_snapshots
+    # Command take the format: 
+    # vmware-vim-cmd vmsvc/snapshot.revert [vmid: int] [snapshotlevel: int] [snapshotindex: int]
+    output = `ssh #{@user}@#{@host} \"vim-cmd vmsvc/snapshot.get #{@vmid}\"`
+
+    # this keeps track of the snapshots, takes the form:
+    #[ {:name => [0,0], :display_name => "String containing the snapshotname}, 
+    #  {:name => [0,1], :display_name => "String containing the snapshotname}, ]
+    #  ... 
+    snapshots = []
+    
+    # Use these to keep track of the parsing...    
+    current_tree = -1
+    current_num = 0
+    count = 0
+    
+    # Do the parsing & stick the snapshots in the snapshots array
+    output_lines = output.split("\n")
+    output_lines.each do |line|
+      if line.include?("|") # this is a new snapshot
+        if line.include?("ROOT") # it's a root
+          current_num = 0
+          current_tree = current_tree + 1 # new tree
+          snapshots << { :name => [current_num, current_tree], :display_name => output_lines[count+1].split(":").last.strip }
+        else
+          current_num = current_num + 1 # new snapshot in current tree
+          snapshots << { :name => [current_num, current_tree], :display_name => output_lines[count+1].split(":").last.strip }
+        end
+      end
+      count = count+1
+    end
+
+  snapshots
+  end
+
+end
+
+end 
+end