veewee-atlassian 0.3.11

data/lib/veewee/provider/virtualbox/box/export_vagrant.rb

@@ -0,0 +1,100 @@
+require 'pathname'
+module Veewee
+  module Provider
+    module Virtualbox
+      module BoxCommand
+
+        #    Shellutil.execute("vagrant package --base #{vmname} --include /tmp/Vagrantfile --output /tmp/#{vmname}.box", {:progress => "on"})
+
+        def export_vagrant(options)
+
+          # Check if box already exists
+          unless self.exists?
+            ui.info "#{name} is not found, maybe you need to build it first?"
+            exit
+          end
+
+          if File.exists?("#{name}.box")
+            if options["force"]
+              env.logger.debug("#{name}.box exists, but --force was provided")
+              env.logger.debug("removing #{name}.box first")
+              FileUtils.rm("#{name}.box")
+              env.logger.debug("#{name}.box removed")
+            else
+              raise Veewee::Error, "export file #{name}.box already exists. Use --force option to overwrite."
+            end
+          end
+
+
+          # We need to shutdown first
+          if self.running?
+            ui.info "Vagrant requires the box to be shutdown, before it can export"
+            ui.info "Sudo also needs to work for user #{definition.ssh_user}"
+            ui.info "Performing a clean shutdown now."
+
+            self.halt
+
+            #Wait for state poweroff
+            while (self.running?) do
+              ui.info ".",{:new_line => false}
+              sleep 1
+            end
+            ui.info ""
+            ui.info "Machine #{name} is powered off cleanly"
+          end
+
+          #Vagrant requires a relative path for output of boxes
+
+          #4.0.x. not using boxes as a subdir
+          boxdir=Pathname.new(Dir.pwd)
+
+          full_path=File.join(boxdir,name+".box")
+          path1=Pathname.new(full_path)
+          path2=Pathname.new(Dir.pwd)
+          box_path=path1.relative_path_from(path2).to_s
+
+          if File.exists?("#{box_path}")
+            raise Veewee::Error, "box #{name}.box already exists"
+          end
+
+          ui.info "Executing vagrant voodoo:"
+          export_command="vagrant package --base '#{name}' --output '#{box_path}'"
+          export_command += " --include #{options["include"].join(',')}" unless options["include"].empty?
+          export_command += " --vagrantfile #{options["vagrantfile"].join(' ')}" unless options["vagrantfile"].empty?
+          ui.info "#{export_command}"
+          shell_exec("#{export_command}") #hmm, needs to get the gem_home set?
+          ui.info ""
+
+          #add_ssh_nat_mapping back!!!!
+          #vagrant removes the mapping
+          #we need to restore it in order to be able to login again
+          self.add_ssh_nat_mapping
+
+          ui.info "To import it into vagrant type:"
+          ui.info "vagrant box add '#{name}' '#{box_path}'"
+          ui.info ""
+          ui.info "To use it:"
+          ui.info "vagrant init '#{name}'"
+          ui.info "vagrant up"
+          ui.info "vagrant ssh"
+        end
+
+      end #Module
+    end #Module
+  end #Module
+end #Module
+
+
+#      #currently vagrant has a problem with the machine up, it calculates the wrong port to ssh to poweroff the system
+#      thebox.execute("shutdown -h now")
+#      thebox.wait_for_state("poweroff")
+
+
+#      Shellutil.execute("echo 'Vagrant::Config.run do |config|' > /tmp/Vagrantfile")
+#      Shellutil.execute("echo '   config.ssh.forwarded_port_key = \"ssh\"' >> /tmp/Vagrantfile")
+#      Shellutil.execute("echo '   config.raw.forward_port(\"ssh\",22,#{host_port})' >> /tmp/Vagrantfile")
+#      Shellutil.execute("echo 'end' >> /tmp/Vagrantfile")
+
+
+#vagrant export disables the machine
+#      thebox.ssh_enable_vmachine({:hostport => host_port , :guestport => 22} )

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

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

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

@@ -0,0 +1,28 @@
+module Veewee
+  module Provider
+    module Virtualbox
+      module BoxCommand
+
+        def build_info
+          info=super
+          info << { :filename => ".vbox_version",
+                    :content => "#{self.vboxga_version}" }
+        end
+
+         # Transfer information provide by the provider to the box
+         #
+         #
+         def transfer_buildinfo(options)
+           super(options)
+           # with windows, we just use the mounted volume 
+           if not (definition.winrm_user && definition.winrm_password)
+             iso_image="VBoxGuestAdditions_#{self.vboxga_version}.iso"
+             env.logger.info "About to transfer virtualbox guest additions iso to the box #{name} - #{ip_address} - #{ssh_options}"
+             self.copy_to_box("#{File.join(env.config.veewee.iso_dir,iso_image)}",File.basename(iso_image))
+           end
+         end
+
+      end
+    end
+  end
+end

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

@@ -0,0 +1,62 @@
+require 'veewee/provider/core/helper/scancode'
+require 'veewee/provider/core/helper/tcp'
+require 'veewee/provider/core/helper/shell'
+
+module Veewee
+  module Provider
+    module Virtualbox
+      module BoxCommand
+        def console_type(sequence)
+          send_virtualbox_sequence(sequence)
+        end
+
+        def send_virtualbox_sequence(sequence)
+
+          ui.info ""
+
+          counter=0
+          sequence.each { |s|
+            counter=counter+1
+
+            ui.info "Typing:[#{counter}]: "+s
+
+            keycodes=Veewee::Provider::Core::Helper::Scancode.string_to_keycode(s)
+
+            # VBox seems to have issues with sending the scancodes as one big
+            # .join()-ed string. It seems to get them out or order or ignore some.
+            # A workaround is to send the scancodes one-by-one.
+            codes=""
+            for keycode in keycodes.split(' ') do
+              unless keycode=="wait"
+                send_keycode(keycode)
+                sleep 0.01
+              else
+                sleep 1
+              end
+            end
+            #sleep after each sequence (needs to be param)
+            sleep 0.5
+          }
+
+          ui.info "Done typing."
+          ui.info ""
+
+        end
+
+        def send_keycode(keycode)
+          command= "#{@vboxcmd} controlvm \"#{name}\" keyboardputscancode #{keycode}"
+          env.logger.info "#{command}"
+          sshresult=shell_exec("#{command}",{:mute => true})
+          unless sshresult.stdout.index("E_ACCESSDENIED").nil?
+            error= "There was an error typing the commands on the console"
+            error+="Probably the VM did not get started."
+            error+= ""
+            error+= "#{sshresult.stdout}"
+            raise Veewee::Error, error
+          end
+        end
+
+      end #Module
+    end #Module
+  end #Module
+end #Module

data/lib/veewee/provider/virtualbox/box/helper/create.rb

@@ -0,0 +1,200 @@
+module Veewee
+  module Provider
+    module Virtualbox
+      module BoxCommand
+
+        def add_ide_controller
+          #unless => "${vboxcmd} showvminfo \"${vname}\" | grep \"IDE Controller\" "
+          command ="#{@vboxcmd} storagectl \"#{name}\" --name \"IDE Controller\" --add ide"
+          shell_exec("#{command}")
+        end
+
+        def add_sata_controller 
+          #unless => "${vboxcmd} showvminfo \"${vname}\" | grep \"SATA Controller\" ";
+          command ="#{@vboxcmd} storagectl \"#{name}\" --name \"SATA Controller\" --add sata --hostiocache #{definition.hostiocache} --sataportcount #{definition.disk_count}"
+          shell_exec("#{command}")
+        end
+
+        def attach_serial_console
+          command ="#{@vboxcmd} modifyvm \"#{name}\" --uart1 0x3F8 4"
+          shell_exec("#{command}")
+          command ="#{@vboxcmd} modifyvm \"#{name}\" --uartmode1 file \"#{File.join(FileUtils.pwd,name+"-serial-console"+".log")}\""
+          shell_exec("#{command}")
+        end
+
+        def add_ssh_nat_mapping
+
+          unless definition.nil?
+            unless definition.skip_nat_mapping == true
+              #Map SSH Ports
+              command="#{@vboxcmd} modifyvm \"#{name}\" --natpf#{self.natinterface} \"guestssh,tcp,,#{definition.ssh_host_port},,#{definition.ssh_guest_port}\""
+              shell_exec("#{command}")
+            end
+          end
+        end
+
+        def add_winrm_nat_mapping
+
+          unless definition.nil?
+            #Map SSH Ports
+            command="#{@vboxcmd} modifyvm \"#{name}\" --natpf1 'guestwinrm,tcp,,#{definition.winrm_host_port},,#{definition.winrm_guest_port}'"
+            shell_exec("#{command}")
+          end
+        end
+
+        def add_shared_folder
+          command="#{@vboxcmd} sharedfolder add  \"#{name}\" --name \"veewee-validation\" --hostpath \"#{File.expand_path(env.validation_dir)}\" --automount"
+          shell_exec("#{command}")
+        end
+
+        def get_vbox_home
+          command="#{@vboxcmd}  list  systemproperties"
+          shell_results=shell_exec("#{command}")
+          # On windows Default machine path would include a drive letter, then ':'.
+          # So here we tell to split no more than 2 elements to keep the full path
+          # This should work for all OS as we just need to separate the parameter name with first ':' from the value
+          location=shell_results.stdout.split(/\n/).grep(/Default machine/)[0].split(":", 2)[1].strip
+          return location
+        end
+
+
+        def suppress_messages
+          day=24*60*60
+          update_date=Time.now+365*day
+
+          extraData = [
+            ["GUI/RegistrationData","triesLeft=0"],
+            ["GUI/SuppressMessages","confirmInputCapture,remindAboutAutoCapture,remindAboutMouseIntegrationOff,remindAboutMouseIntegrationOn,remindAboutWrongColorDepth"],
+            ["GUI/UpdateDate", "1 d, #{update_date.year}-#{update_date.month.to_s.rjust(2,'0')}-#{update_date.day.to_s.rjust(2,'0')}, stable"],
+            ["GUI/UpdateCheckCount","60"]
+          ]
+          extraData.each do |data|
+            command="#{@vboxcmd} setextradata global \"#{data[0]}\" \"#{data[1]}\""
+            shell_results=shell_exec("#{command}")
+          end
+
+        end
+
+        def create_disk
+           place=get_vbox_home
+           1.upto(definition.disk_count) do |f|
+             ui.info "Creating new harddrive of size #{definition.disk_size.to_i}, format #{definition.disk_format}, variant #{definition.disk_variant} "
+             command ="#{@vboxcmd} createhd --filename \"#{File.join(place,name,name+"#{f}."+definition.disk_format.downcase)}\" --size \"#{definition.disk_size.to_i}\" --format #{definition.disk_format.downcase} --variant #{definition.disk_variant.downcase}"
+             shell_exec("#{command}")
+           end
+        end
+  
+        def attach_disk_common(storagectl, device_number)
+          place=get_vbox_home
+         
+          1.upto(definition.disk_count) do |f|
+            location=name+"#{f}."+definition.disk_format.downcase
+  
+            location="#{File.join(place,name,location)}"
+            ui.info "Attaching disk: #{location}"
+  
+            command ="#{@vboxcmd} storageattach \"#{name}\" --storagectl \"#{storagectl}\" --port #{f-1} --device #{device_number} --type hdd --medium \"#{location}\""
+            shell_exec("#{command}")
+          end
+        end
+
+        def attach_disk_ide(device_number=0)
+          self.attach_disk_common("IDE Controller", device_number)
+        end
+
+        def attach_disk_sata(device_number=0)
+          self.attach_disk_common("SATA Controller", device_number)
+        end
+
+
+        def attach_isofile(device_number=0)
+          full_iso_file=File.join(env.config.veewee.iso_dir,definition.iso_file)
+          ui.info "Mounting cdrom: #{full_iso_file}"
+          command ="#{@vboxcmd} storageattach \"#{name}\" --storagectl \"IDE Controller\" --type dvddrive --port 0 --device #{device_number} --medium \"#{full_iso_file}\""
+          shell_exec("#{command}")
+        end
+
+        def attach_guest_additions
+          full_iso_file=File.join(env.config.veewee.iso_dir,"VBoxGuestAdditions_#{self.vboxga_version}.iso")
+          ui.info "Mounting guest additions: #{full_iso_file}"
+          command ="#{@vboxcmd} storageattach \"#{name}\" --storagectl \"IDE Controller\" --type dvddrive --port 1 --device 0 --medium \"#{full_iso_file}\""
+          shell_exec("#{command}")
+        end
+
+
+        def add_floppy_controller
+          # Create floppy controller
+          unless definition.floppy_files.nil?
+
+            command="#{@vboxcmd} storagectl \"#{name}\" --name \"Floppy Controller\" --add floppy"
+            shell_exec("#{command}")
+          end
+        end
+
+
+        def attach_floppy
+          unless definition.floppy_files.nil?
+
+            # Attach floppy to machine (the vfd extension is crucial to detect msdos type floppy)
+            floppy_file=File.join(definition.path,"virtualfloppy.vfd")
+            command="#{@vboxcmd} storageattach \"#{name}\" --storagectl \"Floppy Controller\" --port 0 --device 0 --type fdd --medium \"#{floppy_file}\""
+            shell_exec("#{command}")
+          end
+        end
+
+        def vbox_os_type_id(veewee_type_id)
+          type=env.ostypes[veewee_type_id][:vbox]
+          env.logger.info("Using VBOX os_type_id #{type}")
+          return type
+        end
+
+        def create_vm
+          command="#{@vboxcmd} createvm --name \"#{name}\" --ostype \"#{vbox_os_type_id(definition.os_type_id)}\" --register"
+
+          #Exec and system stop the execution here
+          shell_exec("#{command}")
+
+          ui.info "Creating vm #{name} : #{definition.memory_size}M - #{definition.cpu_count} CPU - #{vbox_os_type_id(definition.os_type_id)}"
+
+          #setting cpu's
+          command="#{@vboxcmd} modifyvm \"#{name}\" --cpus #{definition.cpu_count}"
+          shell_exec("#{command}")
+
+          #setting memory size
+          command="#{@vboxcmd} modifyvm \"#{name}\" --memory #{definition.memory_size}"
+          shell_exec("#{command}")
+
+          #setting video memory size
+          command="#{@vboxcmd} modifyvm \"#{name}\" --vram #{definition.video_memory_size}"
+          shell_exec("#{command}")
+          
+          #setting bootorder
+          command="#{@vboxcmd} modifyvm \"#{name}\" --boot1 disk --boot2 dvd --boot3 none --boot4 none"
+          shell_exec("#{command}")
+
+          # Modify the vm to enable or disable hw virtualization extensions
+          vm_flags=%w{pagefusion acpi ioapic pae hpet hwvirtex hwvirtexcl nestedpaging largepages vtxvpid synthxcpu rtcuseutc}
+
+          vm_flags.each do |vm_flag|
+            if definition.instance_variable_defined?("@#{vm_flag}")
+              vm_flag_value=definition.instance_variable_get("@#{vm_flag}")
+              ui.info "Setting VM Flag #{vm_flag} to #{vm_flag_value}"
+              ui.warn "Used of #{vm_flag} is deprecated - specify your options in the definition file as \n :virtualbox => { :vm_options => [\"#{vm_flag}\" => \"#{vm_flag_value}\"]}"
+              command="#{@vboxcmd} modifyvm #{name} --#{vm_flag.to_s} #{vm_flag_value}"
+              shell_exec("#{command}")
+            end
+          end
+
+          unless definition.virtualbox[:vm_options][0].nil?
+            definition.virtualbox[:vm_options][0].each do |vm_flag,vm_flag_value|
+              ui.info "Setting VM Flag #{vm_flag} to #{vm_flag_value}"
+              command="#{@vboxcmd} modifyvm #{name} --#{vm_flag.to_s} #{vm_flag_value}"
+              shell_exec("#{command}")
+            end
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/veewee/provider/virtualbox/box/helper/forwarding.rb

@@ -0,0 +1,38 @@
+module Veewee
+  module Provider
+    module Virtualbox
+      module BoxCommand
+
+        def forwarding(name)
+          command="#{@vboxcmd} showvminfo --details --machinereadable \"#{self.name}\""
+          shell_results=shell_exec("#{command}")
+          rules=shell_results.stdout.split(/\n/).grep(/^Forward/)
+          result=nil
+          rules.each do |rule|
+            #Forwarding(0)
+            nr=rule.split('=')[0].split('(')[1].split(')')[0].to_i + 1
+            #  guestssh,tcp,,2222,,22
+            details=rule.split('=')[1].split('"')[1].split(',')
+            result = {
+              :nr => nr,
+              :name => details[0],
+              :protocol => details[1],
+              :host_ip => details[2],
+              :host_port => details[3],
+              :guest_ip => details[4],
+              :guest_port => details[5]
+            }
+          end
+          return result
+        end
+
+        def delete_forwarding(name)
+          forward=self.forwarding(name)
+          command="#{@vboxcmd} controlvm \"#{self.name}\" natpf#{self.natinterface} delete #{name}"
+          shell_results=shell_exec("#{command}")
+        end
+
+      end
+    end
+  end
+end