lab 0.1.5 → 0.2.0

data/lib/lab/driver/remote_workstation_driver.rb

@@ -9,187 +9,187 @@ module Drivers
 
 class RemoteWorkstationDriver < VmDriver
 
-	attr_accessor :location # among other things
-
-	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'])
-	end
-
-	def start
-		remote_system_command("vmrun -T ws start \'#{@location}\' nogui")
-	end
-
-	def stop
-		remote_system_command("vmrun -T ws stop \'#{@location}\' nogui")
-	end
-
-	def suspend
-		remote_system_command("vmrun -T ws suspend \'#{@location}\' nogui")
-	end
-
-	def pause
-		remote_system_command("vmrun -T ws pause \'#{@location}\' nogui")
-	end
-
-	def reset
-		remote_system_command("vmrun -T ws reset \'#{@location}\' nogui")
-	end
-
-	def create_snapshot(snapshot)
-		snapshot = filter_input(snapshot)
-		remote_system_command("vmrun -T ws snapshot \'#{@location}\' #{snapshot} nogui")
-	end
-
-	def revert_snapshot(snapshot)
-		snapshot = filter_input(snapshot)
-		remote_system_command("vmrun -T ws revertToSnapshot \'#{@location}\' #{snapshot} nogui")
-	end
-
-	def delete_snapshot(snapshot)
-		snapshot = filter_input(snapshot)
-		remote_system_command("vmrun -T ws deleteSnapshot \'#{@location}\' #{snapshot} nogui" )
-	end
-	
-	def run_command(command)
-		# generate local & remote script paths
-		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) }
-
-		# we really can't filter command, so we're gonna stick it in a script
-		if @tools
-			# copy it to the vm host - this is because we're a remote driver
-			remote_copy_command = "scp #{local_tempfile_path} #{@user}@#{@host}:#{local_tempfile_path}"
-			system_command(remote_copy_command)
-
-			# we have it on the vm host, copy it to the vm guest
-			vmrunstr = "ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
-					"copyFileFromHostToGuest \'#{@location}\' \'#{local_tempfile_path}\' " +
-					"\'#{remote_tempfile_path}\' nogui\""
-			system_command(vmrunstr)
-
-			# now run it on the guest
-			vmrunstr = "ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " + 
-					"runProgramInGuest \'#{@location}\' -noWait -activeWindow \'#{remote_run_command}\'"
-			system_command(vmrunstr)
-
-			## CLEANUP
-			# delete it on the guest
-			vmrunstr = "ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " + 
-					"deleteFileInGuest \'#{@location}\' \'#{remote_tempfile_path}\'"
-			system_command(vmrunstr)
-
-			# and delete it on the vm host
-			vmhost_delete_command = "ssh #{@user}@#{@host} rm #{local_tempfile_path}"
-			system_command(vmhost_delete_command)
-
-			# delete it locally
-			local_delete_command = "rm #{local_tempfile_path}"
-			system_command(local_delete_command)
-		else
-			# since we can't copy easily w/o tools, let's just run it directly :/
-			if @os == "linux"
-				scp_to(local_tempfile_path, remote_tempfile_path)
-				ssh_exec(remote_run_command)
-				ssh_exec("rm #{remote_tempfile_path}")
-			else
-				raise "Not Implemented - Install VmWare Tools"
-			end	
-		end
-	end
-	
-	def copy_from(from, to)
-		from = filter_input(from)
-		to = filter_input(to)
-		
-		# copy it to the vm host - this is because we're a remote driver
-		remote_copy_command = "scp #{from} #{@user}@#{@host}:#{from}"
-		system_command(remote_copy_command)
-
-		if @tools 
-			remote_system_command("ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
-					"copyFileFromGuestToHost \'#{@location}\' \'#{from}\' \'#{to}\' nogui")
-		else
-			scp_from(to,from)
-		end
-	end
-
-	def copy_to(from, to)
-	
-		from = filter_input(from)
-		to = filter_input(to)
-		
-		# copy it to the vm host - this is because we're a remote driver
-		remote_copy_command = "scp #{from} #{@user}@#{@host}:#{from}"
-		system_command(remote_copy_command)
-		
-		if @tools
-			remote_system_command("vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
-					"copyFileFromHostToGuest \'#{@location}\' \'#{from}\' \'#{to}\' nogui")
-		else
-			scp_to(from,to)
-		end
-	end
-
-	def check_file_exists(file)
-		
-		if @tools
-			file = filter_input(file)
-			remote_system_command("vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
-					"fileExistsInGuest \'#{@location}\' \'{file}\' nogui")
-		else
-			raise "Not Implemented - Install VmWare Tools"
-		end
-	end
-
-	def create_directory(directory)
-		directory = filter_input(directory)
-	
-		if @tools
-			emote_system_command("ssh #{@user}@#{@host} vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
-					"createDirectoryInGuest \'#{@location}\' \'#{directory}\' nogui")
-			system_command(vmrunstr)
-		else
-			raise "Not Implemented - Install VmWare Tools"
-		end
-	end
-
-	def cleanup
-
-	end
-
-	def running?
-		## Get running VMs
-		running = `ssh #{@user}@#{@host} \"vmrun list nogui\"`
-		running_array = running.split("\n")
-		running_array.shift
-
-		running_array.each do |vmx|
-			if vmx.to_s == @location.to_s
-				return true
-			end
-		end
-
-		false
-	end
+  attr_accessor :location # among other things
+
+  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'])
+  end
+
+  def start
+    remote_system_command("vmrun -T ws start \'#{@location}\' nogui")
+  end
+
+  def stop
+    remote_system_command("vmrun -T ws stop \'#{@location}\' nogui")
+  end
+
+  def suspend
+    remote_system_command("vmrun -T ws suspend \'#{@location}\' nogui")
+  end
+
+  def pause
+    remote_system_command("vmrun -T ws pause \'#{@location}\' nogui")
+  end
+
+  def reset
+    remote_system_command("vmrun -T ws reset \'#{@location}\' nogui")
+  end
+
+  def create_snapshot(snapshot)
+    snapshot = filter_input(snapshot)
+    remote_system_command("vmrun -T ws snapshot \'#{@location}\' #{snapshot} nogui")
+  end
+
+  def revert_snapshot(snapshot)
+    snapshot = filter_input(snapshot)
+    remote_system_command("vmrun -T ws revertToSnapshot \'#{@location}\' #{snapshot} nogui")
+  end
+
+  def delete_snapshot(snapshot)
+    snapshot = filter_input(snapshot)
+    remote_system_command("vmrun -T ws deleteSnapshot \'#{@location}\' #{snapshot} nogui" )
+  end
+  
+  def run_command(command)
+    # generate local & remote script paths
+    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) }
+
+    # we really can't filter command, so we're gonna stick it in a script
+    if @tools
+      # copy it to the vm host - this is because we're a remote driver
+      remote_copy_command = "scp #{local_tempfile_path} #{@user}@#{@host}:#{local_tempfile_path}"
+      system_command(remote_copy_command)
+
+      # we have it on the vm host, copy it to the vm guest
+      vmrunstr = "ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
+          "copyFileFromHostToGuest \'#{@location}\' \'#{local_tempfile_path}\' " +
+          "\'#{remote_tempfile_path}\' nogui\""
+      system_command(vmrunstr)
+
+      # now run it on the guest
+      vmrunstr = "ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " + 
+          "runProgramInGuest \'#{@location}\' -noWait -activeWindow \'#{remote_run_command}\'"
+      system_command(vmrunstr)
+
+      ## CLEANUP
+      # delete it on the guest
+      vmrunstr = "ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " + 
+          "deleteFileInGuest \'#{@location}\' \'#{remote_tempfile_path}\'"
+      system_command(vmrunstr)
+
+      # and delete it on the vm host
+      vmhost_delete_command = "ssh #{@user}@#{@host} rm #{local_tempfile_path}"
+      system_command(vmhost_delete_command)
+
+      # delete it locally
+      local_delete_command = "rm #{local_tempfile_path}"
+      system_command(local_delete_command)
+    else
+      # since we can't copy easily w/o tools, let's just run it directly :/
+      if @os == "linux"
+        scp_to(local_tempfile_path, remote_tempfile_path)
+        ssh_exec(remote_run_command)
+        ssh_exec("rm #{remote_tempfile_path}")
+      else
+        raise "Not Implemented - Install VmWare Tools"
+      end  
+    end
+  end
+  
+  def copy_from(from, to)
+    from = filter_input(from)
+    to = filter_input(to)
+    
+    # copy it to the vm host - this is because we're a remote driver
+    remote_copy_command = "scp #{from} #{@user}@#{@host}:#{from}"
+    system_command(remote_copy_command)
+
+    if @tools 
+      remote_system_command("ssh #{@user}@#{@host} \"vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
+          "copyFileFromGuestToHost \'#{@location}\' \'#{from}\' \'#{to}\' nogui")
+    else
+      scp_from(to,from)
+    end
+  end
+
+  def copy_to(from, to)
+  
+    from = filter_input(from)
+    to = filter_input(to)
+    
+    # copy it to the vm host - this is because we're a remote driver
+    remote_copy_command = "scp #{from} #{@user}@#{@host}:#{from}"
+    system_command(remote_copy_command)
+    
+    if @tools
+      remote_system_command("vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
+          "copyFileFromHostToGuest \'#{@location}\' \'#{from}\' \'#{to}\' nogui")
+    else
+      scp_to(from,to)
+    end
+  end
+
+  def check_file_exists(file)
+    
+    if @tools
+      file = filter_input(file)
+      remote_system_command("vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
+          "fileExistsInGuest \'#{@location}\' \'{file}\' nogui")
+    else
+      raise "Not Implemented - Install VmWare Tools"
+    end
+  end
+
+  def create_directory(directory)
+    directory = filter_input(directory)
+  
+    if @tools
+      emote_system_command("ssh #{@user}@#{@host} vmrun -T ws -gu #{@vm_user} -gp #{@vm_pass} " +
+          "createDirectoryInGuest \'#{@location}\' \'#{directory}\' nogui")
+      system_command(vmrunstr)
+    else
+      raise "Not Implemented - Install VmWare Tools"
+    end
+  end
+
+  def cleanup
+
+  end
+
+  def running?
+    ## Get running VMs
+    running = `ssh #{@user}@#{@host} \"vmrun list nogui\"`
+    running_array = running.split("\n")
+    running_array.shift
+
+    running_array.each do |vmx|
+      if vmx.to_s == @location.to_s
+        return true
+      end
+    end
+
+    false
+  end
 
 end
 

data/lib/lab/driver/virtualbox_driver.rb

@@ -6,137 +6,137 @@ require 'nokogiri'
 ##
 module Lab
 module Drivers
-	class VirtualBoxDriver < VmDriver
-
-		attr_accessor :location
-
-		def initialize(config)
-
-			super(config)
-
-			## Check to see if we already know this vm, if not, go on location
-			vmid_list = ::Lab::Controllers::VirtualBoxController::config_list
-			unless vmid_list.include? @vmid
-				raise "Error, no such vm: #{@vmid}" unless @location
-				
-				if !File.exist?(@location)
-					raise ArgumentError,"Error, no vm at: #{@location}"
-				end
-				
-				# Registering @location
-				@vmid = register_and_return_vmid
-			end
-			
-			vmInfo = `VBoxManage showvminfo \"#{@vmid}\" --machinereadable`
-			@location = vmInfo.scan(/CfgFile=\"(.*?)\"/).flatten[0].to_s
-
-			if !File.exist?(@location)
-				raise ArgumentError,"Couldn't find: " + @location
-			end
-
-		end
-
-		def register_and_return_vmid
-			
-			xml = Nokogiri::XML(File.new(@location))
-			vmid = xml.root.xpath("//Machine[@name]")
-			
-			## only register if we don't already know the vmid
-			if !::Lab::Controllers::VirtualBoxController::config_list.include? vmid
-				system_command("VBoxManage registervm \"#{@location}\"")
-			end
-			
-			return vmid
-			
-		end
-
-		def unregister
-			system_command("VBoxManage unregistervm \"#{@vmid}\"")
-		end
-
-		def start
-			system_command("VBoxManage startvm \"#{@vmid}\"")
-		end
-
-		def stop
-			system_command("VBoxManage controlvm \"#{@vmid}\" poweroff")
-		end
-
-		def suspend
-			system_command("VBoxManage controlvm \"#{@vmid}\" savestate")
-		end
-
-		def pause
-			system_command("VBoxManage controlvm \"#{@vmid}\" pause")
-		end
-
-		def reset
-			system_command("VBoxManage controlvm \"#{@vmid}\" reset")
-		end
-
-		def create_snapshot(snapshot)
-			snapshot = filter_input(snapshot)
-			system_command("VBoxManage snapshot \"#{@vmid}\" take #{snapshot}")
-		end
-
-		def revert_snapshot(snapshot)
-			snapshot = filter_input(snapshot)
-			system_command("VBoxManage snapshot \"#{@vmid}\" restore #{snapshot}")
-		end
-
-		def delete_snapshot(snapshot)
-			snapshot = filter_input(snapshot)
-			system_command("VBoxManage snapshot \"#{@vmid}\" delete  #{snapshot}")
-		end
-
-		def run_command(command, arguments=nil)
-			command = filter_input(command)
-			arguments = filter_input(arguments)
-
-			command = "VBoxManage guestcontrol exec \"#{@vmid}\" \"#{command}\" --username \"#{@vm_user}\"" +
-					 " --password \"#{@vm_pass}\" --arguments \"#{arguments}\""
-			system_command(command)
-		end
-	
-		def copy_from(from, to)
-			from = filter_input(from)
-			to = filter_input(to)
-
-			raise "Not supported by Virtual Box"
-		end
-
-		def copy_to(from, to)
-			from = filter_input(from)
-			to = filter_input(to)
-			
-			command = "VBoxManage guestcontrol copyto \"#{@vmid}\" \"#{from}\"  \"#{to}\" " +
-					 "--username \"#{@vm_user}\" --password \"#{@vm_pass}\""
-			system_command(command)
-		end
-
-		def check_file_exists(file)
-			file = filter_input(file)
-
-			raise "Not supported by Virtual Box"
-		end
-
-		def create_directory(directory)
-			directory = filter_input(directory)
-
-			command = "VBoxManage guestcontrol createdir \"#{@vmid}\" \"#{directory}\" " + 
-					 "--username \"#{@vm_user}\" --password \"#{@vm_pass}\""
-			system_command(command)
-		end
-
-		def cleanup
-
-		end
-
-		def running?
-			## Get running Vms
-			::Lab::Controllers::VirtualBoxController::running_list.include? @vmid
-		end
-		
-	end
+  class VirtualBoxDriver < VmDriver
+
+    attr_accessor :location
+
+    def initialize(config)
+
+      super(config)
+
+      ## Check to see if we already know this vm, if not, go on location
+      vmid_list = ::Lab::Controllers::VirtualBoxController::config_list
+      unless vmid_list.include? @vmid
+        raise "Error, no such vm: #{@vmid}" unless @location
+        
+        if !File.exist?(@location)
+          raise ArgumentError,"Error, no vm at: #{@location}"
+        end
+        
+        # Registering @location
+        @vmid = register_and_return_vmid
+      end
+      
+      vmInfo = `VBoxManage showvminfo \"#{@vmid}\" --machinereadable`
+      @location = vmInfo.scan(/CfgFile=\"(.*?)\"/).flatten[0].to_s
+
+      if !File.exist?(@location)
+        raise ArgumentError,"Couldn't find: " + @location
+      end
+
+    end
+
+    def register_and_return_vmid
+      
+      xml = Nokogiri::XML(File.new(@location))
+      vmid = xml.root.xpath("//Machine[@name]")
+      
+      ## only register if we don't already know the vmid
+      if !::Lab::Controllers::VirtualBoxController::config_list.include? vmid
+        system_command("VBoxManage registervm \"#{@location}\"")
+      end
+      
+      return vmid
+      
+    end
+
+    def unregister
+      system_command("VBoxManage unregistervm \"#{@vmid}\"")
+    end
+
+    def start
+      system_command("VBoxManage startvm \"#{@vmid}\"")
+    end
+
+    def stop
+      system_command("VBoxManage controlvm \"#{@vmid}\" poweroff")
+    end
+
+    def suspend
+      system_command("VBoxManage controlvm \"#{@vmid}\" savestate")
+    end
+
+    def pause
+      system_command("VBoxManage controlvm \"#{@vmid}\" pause")
+    end
+
+    def reset
+      system_command("VBoxManage controlvm \"#{@vmid}\" reset")
+    end
+
+    def create_snapshot(snapshot)
+      snapshot = filter_input(snapshot)
+      system_command("VBoxManage snapshot \"#{@vmid}\" take #{snapshot}")
+    end
+
+    def revert_snapshot(snapshot)
+      snapshot = filter_input(snapshot)
+      system_command("VBoxManage snapshot \"#{@vmid}\" restore #{snapshot}")
+    end
+
+    def delete_snapshot(snapshot)
+      snapshot = filter_input(snapshot)
+      system_command("VBoxManage snapshot \"#{@vmid}\" delete  #{snapshot}")
+    end
+
+    def run_command(command, arguments=nil)
+      command = filter_input(command)
+      arguments = filter_input(arguments)
+
+      command = "VBoxManage guestcontrol exec \"#{@vmid}\" \"#{command}\" --username \"#{@vm_user}\"" +
+           " --password \"#{@vm_pass}\" --arguments \"#{arguments}\""
+      system_command(command)
+    end
+  
+    def copy_from(from, to)
+      from = filter_input(from)
+      to = filter_input(to)
+
+      raise "Not supported by Virtual Box"
+    end
+
+    def copy_to(from, to)
+      from = filter_input(from)
+      to = filter_input(to)
+      
+      command = "VBoxManage guestcontrol copyto \"#{@vmid}\" \"#{from}\"  \"#{to}\" " +
+           "--username \"#{@vm_user}\" --password \"#{@vm_pass}\""
+      system_command(command)
+    end
+
+    def check_file_exists(file)
+      file = filter_input(file)
+
+      raise "Not supported by Virtual Box"
+    end
+
+    def create_directory(directory)
+      directory = filter_input(directory)
+
+      command = "VBoxManage guestcontrol createdir \"#{@vmid}\" \"#{directory}\" " + 
+           "--username \"#{@vm_user}\" --password \"#{@vm_pass}\""
+      system_command(command)
+    end
+
+    def cleanup
+
+    end
+
+    def running?
+      ## Get running Vms
+      ::Lab::Controllers::VirtualBoxController::running_list.include? @vmid
+    end
+    
+  end
 end
 end