veewee-atlassian 0.3.11

data/lib/veewee/provider/vmfusion/box/build.rb

@@ -0,0 +1,15 @@
+
+
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        def build(options)
+          super(options)
+          add_share_from_defn
+        end
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/create.rb

@@ -0,0 +1,64 @@
+require 'erb'
+
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        # When we create a new box
+        # We assume the box is not running
+        def create(options)
+          create_vm
+          create_disk
+          self.create_floppy("virtualfloppy.img")
+        end
+
+
+        def create_disk
+          #Disk types:
+          #    0                   : single growable virtual disk
+          #    1                   : growable virtual disk split in 2GB files
+          #    2                   : preallocated virtual disk
+          #    3                   : preallocated virtual disk split in 2GB files
+          #    4                   : preallocated ESX-type virtual disk
+          #    5                   : compressed disk optimized for streaming
+          #    6                   : thin provisioned virtual disk - ESX 3.x and above
+          disk_type=1
+          current_dir=FileUtils.pwd
+          FileUtils.chdir(vm_path)
+          env.ui.info "Creating disk"
+          command="#{File.dirname(vmrun_cmd).shellescape}/vmware-vdiskmanager -c -s #{definition.disk_size}M -a lsilogic -t #{disk_type} #{name}.vmdk"
+          shell_results=shell_exec("#{command}",{:mute => true})
+          FileUtils.chdir(current_dir)
+        end
+
+        def fusion_os_type(type_id)
+          env.logger.info "Translating #{type_id} into fusion type"
+          fusiontype=env.ostypes[type_id][:fusion]
+          env.logger.info "Found fusion type #{fusiontype}"
+          return fusiontype
+        end
+
+        def create_vm
+          fusion_definition=definition.dup
+
+          fusion_definition.os_type_id=fusion_os_type(definition.os_type_id)
+ 
+          FileUtils.mkdir_p(vm_path)
+          current_dir=FileUtils.pwd
+          FileUtils.chdir(vm_path)
+       
+          unless definition.vmdk_file.nil?
+            src = "#{File.join(definition.path,definition.vmdk_file)}"
+            FileUtils.cp(src, vm_path)
+          end
+
+          aFile = File.new(vmx_file_path, "w")
+          aFile.write(vmx_template(fusion_definition))
+          aFile.close
+          FileUtils.chdir(current_dir)
+        end
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/destroy.rb

@@ -0,0 +1,19 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        def destroy(options={})
+          unless raw.exists?
+            raise Veewee::Error, "Error:: You tried to destroy a non-existing box '#{name}'"
+          end
+
+          raw.halt if raw.state=="running"
+          ::Fission::VM.new(name).delete
+          # remove it from memory
+          @raw=nil
+        end
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/export_ova.rb

@@ -0,0 +1,49 @@
+require 'tempfile'
+
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+        # This function 'exports' the box based on the definition
+        def export_ova(options)
+          debug="--X:logToConsole=true --X:logLevel=\"verbose\""
+          debug=""
+          flags="--compress=9"
+
+          if File.exists?("#{name}.ova")
+            if options["force"]
+              env.logger.debug("#{name}.ova exists, but --force was provided")
+              env.logger.debug("removing #{name}.ova first")
+              FileUtils.rm("#{name}.ova")
+              env.logger.debug("#{name}.ova removed")
+            else
+              raise Veewee::Error, "export file #{name}.ova already exists. Use --force option to overwrite."
+            end
+          end
+
+          # Need to check binary first
+          if self.running?
+            # Wait for the shutdown to complete
+            begin
+              Timeout::timeout(20) do
+                self.halt(options)
+                status=self.running?
+                unless status
+                  return
+                end
+                sleep 4
+              end
+            rescue TimeoutError::Error => ex
+              raise Veewee::Error,ex
+            end
+          end
+
+          # before exporting the system needs to be shut down
+
+          # otherwise the debug log will show - The specified virtual disk needs repair
+          shell_exec("#{File.dirname(vmrun_cmd).shellescape}/ovftool/ovftool.bin #{debug} #{flags} #{vmx_file_path.shellescape} #{name}.ova")
+        end
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/halt.rb

@@ -0,0 +1,13 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        def halt(options={})
+           super(options)
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/buildinfo.rb

@@ -0,0 +1,62 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        def build_info
+          info=super
+          output=IO.popen("#{vmrun_cmd.shellescape}").readlines
+          info << {:filename => ".vmfusion_version",:content => @provider.fusion_version }
+        end
+
+
+       def guest_iso_directory
+          # use vmware fusion 3.x as default path
+          iso_images_dir="/Library/Application Support/VMware Fusion/isoimages"
+
+          # if path doesn't exist check for vmware fusion >= 4.x path
+          if( ! File.exists?(iso_images_dir) )
+            iso_images_dir="/Applications/VMware Fusion.app/Contents/Library/isoimages"
+          end
+          return iso_images_dir
+        end
+
+        # Determine the iso of the guest additions
+        def guest_iso_path
+          # So we begin by transferring the ISO file of the vmware tools
+          iso_image=File.join(guest_iso_directory, "linux.iso")
+          iso_image=File.join(guest_iso_directory, "darwin.iso") if definition.os_type_id=~/^Darwin/
+          iso_image=File.join(guest_iso_directory, "freebsd.iso") if definition.os_type_id=~/^Free/
+          iso_image=File.join(guest_iso_directory, "windows.iso") if definition.os_type_id=~/^Win/
+          return iso_image
+        end
+
+        # Transfer information provide by the provider to the box
+        #
+        #
+        def transfer_buildinfo(options)
+          super(options)
+
+          # Initialize download_tools to true if null
+          if definition.vmfusion[:vm_options]['download_tools'].nil?
+            definition.vmfusion[:vm_options]['download_tools'] = true 
+          end
+
+          # When we get here, ssh is available and no postinstall scripts have been executed yet
+          # So we begin by transferring the ISO file of the vmware tools
+          if !(definition.winrm_user && definition.winrm_password) && definition.vmfusion[:vm_options]['download_tools']
+            # with windows, we just use the mounted volume
+            env.logger.info "About to transfer vmware tools iso buildinfo to the box #{name} - #{ip_address} - #{ssh_options}"
+            iso_image=guest_iso_path
+            if File.exists?(iso_image)
+              self.copy_to_box(iso_image,File.basename(iso_image))
+            else
+              raise Veewee::Error, "We could not find the file #{iso_image}. In newer versions of Fusion, you might have to download the Guest Additions yourself. You can do this by first manually creating a vm and than 'installing the guest additions'"
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/console_type.rb

@@ -0,0 +1,18 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        # Type on the console
+        def console_type(sequence,type_options={})
+          if vnc_enabled?
+            vnc_type(sequence,"127.0.0.1",vnc_display_port)
+          else
+            raise Veewee::Error, "VNC is not enabled"
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/ip.rb

@@ -0,0 +1,64 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        # Retrieve the first mac address for a vm
+        # This will only retrieve the first auto generate mac address
+        def mac_address
+          raise ::Fission::Error,"VM #{name} does not exist" unless self.exists?
+
+          line=File.new(vmx_file_path).grep(/^ethernet0.generatedAddress =/)
+          if line.nil?
+            #Fission.ui.output "Hmm, the vmx file #{vmx_path} does not contain a generated mac address "
+            return nil
+          end
+          address=line.first.split("=")[1].strip.split(/\"/)[1]
+          return address
+        end
+
+        # Retrieve the ip address for a vm.
+        # This will only look for dynamically assigned ip address via vmware dhcp
+        def ip_address
+          # Does not work for now as the vmx path is not escape correctly by fission 0.4.0
+          #return raw.network_info.data.first['ip_address']
+          raise ::Fission::Error,"VM #{name} does not exist" unless self.exists?
+
+          unless mac_address.nil?
+            lease = Fission::Lease.find_by_mac_address(mac_address).data
+            return lease.ip_address unless lease.nil?
+            return nil
+          else
+            # No mac address was found for this machine so we can't calculate the ip-address
+            return nil
+          end
+        end
+
+        # http://www.thirdbit.net/articles/2008/03/04/dhcp-on-vmware-fusion/
+        def host_ip_as_seen_by_guest
+
+          # if File.exists?("/Library/Application Support/VMware Fusion/vmnet8/nat.conf")
+          #   file = "/Library/Application Support/VMware Fusion/vmnet8/nat.conf"
+          # end
+
+          # if File.exists?("/Library/Preferences/VMware Fusion/vmnet8/nat.conf")
+          #   file = "/Library/Preferences/VMware Fusion/vmnet8/nat.conf"
+          # end
+          # File.open(file).readlines.grep(/ip = /).first.split(" ")[2]
+
+          # The above is not always correct
+          # There seems also an entry for vmnet8 in the dhcpd.conf
+          # /Library/Preferences/VMware Fusion/vmnet8/dhcpd.conf
+          # host vmnet8 {
+          #   fixed-address
+
+          # The above is fancy but doesn't always agree, we need to do is ifconfig vmnet8
+          # Ifconfig never lies
+          shell_results = shell_exec("ifconfig vmnet8", { :mute => true})
+          shell_results.stdout.split(/\n/).grep(/inet /)[0].strip.split(/ /)[1]
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/ssh_options.rb

@@ -0,0 +1,20 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        # Translate the definition ssh options to ssh options that can be passed to Net::Ssh calls
+        def ssh_options
+          ssh_options={
+            :user => definition.ssh_user,
+            :port => 22,
+            :password => definition.ssh_password,
+            :timeout => definition.ssh_login_timeout.to_i
+          }
+          return ssh_options
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/status.rb

@@ -0,0 +1,20 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        # Check if box is running
+        def running?
+          return false if raw.nil?
+          return raw.running?
+        end
+
+        # Check if the box already exists
+        def exists?
+          return raw.exists?
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/vnc.rb

@@ -0,0 +1,61 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        def vnc_port
+           lines=File.readlines(vmx_file_path)
+           matches=lines.grep(/^RemoteDisplay.vnc.port/)
+           if matches.length==0
+              raise Veewee::Error,"No VNC port found, maybe it is not enabled?"
+           else
+              value=matches.first.split("\"")[1].to_i
+              return value
+           end
+        end
+
+        # This tries to guess a port for the VNC Display
+        def guess_vnc_port
+          min_port=5920
+          max_port=6000
+          guessed_port=nil
+
+          for port in (min_port..max_port)
+            unless is_tcp_port_open?("127.0.0.1", port)
+              guessed_port=port
+              break
+            end
+          end
+
+          if guessed_port.nil?
+            env.ui.info "No free VNC port available: tried #{min_port}..#{max_port}"
+            raise Veewee::Error, "No free VNC port available: tried #{min_port}..#{max_port}"
+          else
+            env.ui.info "Found VNC port #{guessed_port} available"
+          end
+
+          return guessed_port
+        end
+
+        def vnc_display_port
+          self.vnc_port - 5900
+        end
+
+        def vnc_enabled?
+           lines=File.readlines(vmx_file_path)
+           matches=lines.grep(/^RemoteDisplay.vnc.enabled/)
+           if matches.length==0
+              return false
+           else
+              if matches.first.split("\"")[1].downcase == 'true'
+                return true
+              else
+                return false
+              end
+           end
+        end
+
+      end
+    end
+  end
+end

data/lib/veewee/provider/vmfusion/box/helper/winrm_options.rb

@@ -0,0 +1,21 @@
+module Veewee
+  module Provider
+    module Vmfusion
+      module BoxCommand
+
+        def winrm_options
+          winrm_options={
+            :user => definition.winrm_user,
+            :pass => definition.winrm_password,
+            :port => definition.winrm_host_port,
+            :basic_auth_only => true,
+            :timeout => definition.winrm_login_timeout.to_i,
+            :operation_timeout => 600 # ten minutes
+          }
+          return winrm_options
+        end
+
+      end
+    end
+  end
+end