veewee 0.3.0.alpha4 → 0.3.0.alpha5

data/lib/veewee/provider/core/box/ssh.rb

@@ -0,0 +1,95 @@
+require 'veewee/provider/core/helper/ssh'
+module Veewee
+  module Provider
+    module  Core
+      module BoxCommand
+
+
+        def ssh(command=nil,options={})
+
+          raise Veewee::Error,"Box is not running" unless self.running?
+
+          if (options[:interactive]==true)
+            # Command line options
+            extended_command="#{command}"
+            host_ip=self.ip_address
+
+            unless host_ip.nil? || host_ip==""
+              ssh_command="ssh #{ssh_commandline_options(options)} #{host_ip} \"#{extended_command}\""
+
+              fg_exec(ssh_command,options)
+
+            else
+              env.ui.error "Can't ssh into '#{@name} as we couldn't figure out it's ip-address"
+            end
+          else
+            ssh_options={:user => definition.ssh_user,:password => definition.ssh_password, :port => definition.ssh_host_port}
+            ssh_execute(host_ip,command,ssh_options)
+          end
+
+        end
+
+        private
+        def ssh_commandline_options(options)
+
+          command_options = [
+            #"-q", #Suppress warning messages
+            #            "-T", #Pseudo-terminal will not be allocated because stdin is not a terminal.
+            "-p #{ssh_options[:port]}",
+            "-o UserKnownHostsFile=/dev/null",
+            "-t -o StrictHostKeyChecking=no",
+            "-o IdentitiesOnly=yes",
+            "-o VerifyHostKeyDNS=no"
+          ]
+          if !(definition.ssh_key.nil? ||  definition.ssh_key.length!="")
+            command_options << "-i #{definition.ssh_key}"
+          end
+          commandline_options="#{command_options.join(" ")} ".strip
+
+          user_option=definition.ssh_user.nil? ? "" : "-l #{definition.ssh_user}"
+
+          return "#{commandline_options} #{user_option}"
+        end
+
+
+        def fg_exec(ssh_command,options)
+          # Some hackery going on here. On Mac OS X Leopard (10.5), exec fails
+          # (GH-51). As a workaround, we fork and wait. On all other platforms,
+          # we simply exec.
+          pid = nil
+          pid = fork if Platform.leopard? || Platform.tiger?
+
+          env.logger.info "Executing internal ssh command"
+          env.logger.info ssh_command
+          Kernel.exec ssh_command if pid.nil?
+          Process.wait(pid) if pid
+        end
+
+        # Shameless copy of vagrant
+        class Platform
+          class << self
+            def tiger?
+              platform.include?("darwin8")
+            end
+
+            def leopard?
+              platform.include?("darwin9")
+            end
+
+            [:darwin, :bsd, :linux].each do |type|
+              define_method("#{type}?") do
+                platform.include?(type.to_s)
+              end
+            end
+
+            def platform
+              RbConfig::CONFIG["host_os"].downcase
+            end
+          end
+        end
+
+      end # Module
+    end # Module
+  end # Module
+end # Module
+

data/lib/veewee/provider/core/box/sudo.rb

@@ -0,0 +1,21 @@
+module Veewee
+  module Provider
+    module Core
+      module BoxCommand
+
+        def sudo(scriptname)
+          if definition.ssh_user=="root"
+            return "#{scriptname}"
+          else
+            command=definition.sudo_cmd
+            newcommand=command.gsub(/%p/,"#{definition.ssh_password}")
+            newcommand.gsub!(/%u/,"#{definition.ssh_user}")
+            newcommand.gsub!(/%f/,"#{scriptname}")
+            return newcommand
+          end
+        end
+
+      end #Module
+    end #Module
+  end #Module
+end #Module

data/lib/veewee/{builder → provider}/core/box/vnc.rb

@@ -5,14 +5,14 @@ require 'net/vnc'
 require 'net/vnc/vnc.rb'
 
 module Veewee
-  module Builder
+  module Provider
     module Core
       module BoxCommand
 
       def vnc_type(sequence,host,display=20)
         counter=0
         env.logger.info "Opening VNC #{host} on display #{display}"
-        vnc=Net::VNC.open("#{host}:#{display}",{:wait => 0.01})
+        vnc=Net::VNC.open("#{host}:#{display}",{:wait => 0.001})
         sequence.each { |s|
           counter=counter+1
 
@@ -38,9 +38,9 @@ module Veewee
 
         if keycode.is_a?(Symbol)
           vnc.key_press keycode
-          sleep 0.3
+          sleep 1
         else
-            vnc.type_string keycode,{:wait => 0.01}
+            vnc.type_string keycode,{:wait => 0.1}
         end
 
       end

data/lib/veewee/provider/core/helper/iso.rb

@@ -0,0 +1,142 @@
+module Veewee
+  module Provider
+    module Core
+      module Helper
+        module Iso
+
+          require 'open-uri'
+          require 'progressbar'
+          require 'highline/import'
+          require 'digest/md5'
+
+          def download_iso(url,filename)
+            if !File.exists?(env.config.veewee.iso_dir)
+              env.ui.info "Creating an iso directory"
+              FileUtils.mkdir(env.config.veewee.iso_dir)
+            end
+            env.ui.info "Checking if isofile #{filename} already exists."
+            full_path=File.join(env.config.veewee.iso_dir,filename)
+            env.ui.info "Full path: #{full_path}"
+            if File.exists?(full_path)
+              env.ui.info ""
+              env.ui.info "The isofile #{filename} already exists."
+            else
+              download_progress(url,full_path)
+            end
+          end
+
+          def download_progress(url,localfile)
+            pbar = nil
+            URI.parse(url).open(
+              :content_length_proc => lambda {|t|
+              if t && 0 < t
+                pbar = ProgressBar.new("Fetching file", t)
+                pbar.file_transfer_mode
+              end
+            },
+              :progress_proc => lambda {|s|
+              pbar.set s if pbar
+            }) { |src|
+              # We assume large 10K files, so this is tempfile object 
+              env.logger.info "#{src.class}"
+                env.ui.info "Moving #{src.path} to #{localfile}"
+                FileUtils.mv(src.path,localfile)
+                #open(localfile,"wb") { |dst|
+                  #dst.write(src.read)
+                #}
+            }
+          end
+
+          # Compute hash code
+          def hashsum(filename)
+            checksum=Digest::MD5.new
+            buflen=1024
+            open(filename, "r") do |io|
+              counter = 0
+              while (!io.eof)
+                readBuf = io.readpartial(buflen)
+                env.ui.info('.',{:new_line => false}) if ((counter+=1) % 20000 == 0)
+                checksum.update(readBuf)
+              end
+            end
+            return checksum.hexdigest
+          end
+
+
+          def verify_iso(options)
+            filename=self.iso_file
+            full_path=File.join(env.config.veewee.iso_dir,filename)
+
+            if File.exists?(full_path)
+              env.ui.info ""
+              env.ui.info "The isofile #{filename} already exists."
+            else
+
+              path1=Pathname.new(full_path)
+              path2=Pathname.new(Dir.pwd)
+              rel_path=path1.relative_path_from(path2).to_s
+
+              env.ui.info ""
+              env.ui.info "We did not find an isofile in <currentdir>/iso. \n\nThe definition provided the following download information:"
+              unless "#{self.iso_src}"==""
+                env.ui.info "- Download url: #{self.iso_src}"
+              end
+              env.ui.info "- Md5 Checksum: #{self.iso_md5}"
+              env.ui.info "#{self.iso_download_instructions}"
+              env.ui.info ""
+
+              if self.iso_src == ""
+                env.ui.info "Please follow the instructions above:"
+                env.ui.info "- to get the ISO"
+                env.ui.info" - put it in <currentdir>/iso"
+                env.ui.info "- then re-run the command"
+                env.ui.info ""
+                raise Veewee::Error, "No ISO src is available, can't download it automatically"
+              else
+                answer=nil
+                answer="yes" if options["auto"]==true
+                env.logger.info "Auto download enabled?#{answer} #{!options['auto'].nil?}"
+                if answer.nil?
+                  answer=env.ui.ask("Download? (Yes/No)") {|q| q.default="No"}
+                end
+
+                if answer.downcase == "yes"
+                  begin
+                    download_iso(self.iso_src,filename)
+                  rescue OpenURI::HTTPError => ex
+                    env.ui.error "There was an error downloading #{self.iso_src}:"
+                    env.ui.error "#{ex}"
+                    raise Veewee::Error, "There was an error downloading #{self.iso_src}:\n#{ex}"
+                  end
+                else
+                  env.ui.info "You have selected manual download: "
+                  env.ui.info "curl -C - -L '#{self.iso_src}' -o '#{rel_path}'"
+                  env.ui.info "md5 '#{rel_path}' "
+                  env.ui.info ""
+                  exit
+                end
+
+                unless File.readable?(full_path)
+                  raise Veewee::Error, "The provided iso #{full_path} is not readable"
+                end
+
+                env.ui.info "Verifying md5 checksum : #{self.iso_md5}"
+                file_md5=hashsum(full_path)
+
+                unless file_md5==self.iso_md5
+                  env.ui.error "The MD5 checksums for file #{filename } do not match: "
+                  env.ui.error "- #{file_md5} (current) vs #{self.iso_md5} (specified)"
+                  raise Veewee::Error, "The MD5 checksums for file #{filename } do not match: \n"+ "- #{file_md5} (current) vs #{self.iso_md5} (specified)"
+                end
+
+              end
+
+            end
+
+          end
+        end #Module
+
+      end #Module
+    end #Module
+  end #Module
+end #Module

data/lib/veewee/provider/core/helper/scancode.rb

@@ -0,0 +1,114 @@
+module Veewee
+  module Provider
+    module Core
+      module Helper
+        class Scancode
+
+          def self.string_to_keycode(thestring)
+
+            #http://www.win.tue.nl/~aeb/linux/kbd/scancodes-1.html
+
+            k=Hash.new
+            k['1'] = '02 82' ; k['2'] = '03 83' ; k['3'] = '04 84'; k['4']= '05 85' ;
+            k['5']='06 86'; k['6'] = '07 87' ; k['7'] = '08 88'; k['8'] = '09 89'; k['9']= '0a 8a';
+            k['0']='0b 8b'; k['-'] = '0c 8c'; k['='] = '0d 8d' ;
+            k['Tab'] = '0f 8f';
+            k['q']  = '10 90' ;       k['w']  = '11 91' ;       k['e']  = '12 92';
+            k['r'] = '13 93'       ; k['t'] = '14 94'       ; k['y'] = '15 95';
+            k['u']= '16 96'        ; k['i']='17 97';      k['o'] = '18 98'       ; k['p'] = '19 99' ;
+
+            k['Q']  = '2a 10 aa' ; k['W']  = '2a 11 aa' ; k['E']  = '2a 12 aa'; k['R'] = '2a 13 aa' ; k['T'] = '2a 14 aa' ; k['Y'] = '2a 15 aa'; k['U']= '2a 16 aa' ; k['I']='2a 17 aa'; k['O'] = '2a 18 aa' ; k['P'] = '2a 19 aa' ;
+
+            k['a'] = '1e 9e'; k['s']  = '1f 9f' ; k['d']  = '20 a0' ; k['f']  = '21 a1'; k['g'] = '22 a2' ; k['h'] = '23 a3' ; k['j'] = '24 a4';
+            k['k']= '25 a5' ; k['l']='26 a6';
+            k['A'] = '2a 1e aa 9e'; k['S']  = '2a 1f aa 9f' ; k['D']  = '2a 20 aa a0' ; k['F']  = '2a 21 aa a1';
+            k['G'] = '2a 22 aa a2' ; k['H'] = '2a 23 aa a3' ; k['J'] = '2a 24 aa a4'; k['K']= '2a 25 aa a5' ; k['L']='2a 26 aa a6';
+
+            k[';'] = '27 a7' ;k['"']='2a 28 aa a8';k['\'']='28 a8';
+
+            k['\\'] = '2b ab';   k['|'] = '2a 2b aa 8b';
+
+            k['[']='1a 9a'; k[']']='1b 9b';
+            k['<']='2a 33 aa b3'; k['>']='2a 34 aa b4';
+            k['$']='2a 05 aa 85';
+            k['+']='2a 0d aa 8d';
+
+            k['?']='2a 35 aa b5';
+            k['z'] = '2c ac'; k['x']  = '2d ad' ; k['c']  = '2e ae' ; k['v']  = '2f af'; k['b'] = '30 b0' ; k['n'] = '31 b1' ;
+            k['m'] = '32 b2';
+            k['Z'] = '2a 2c aa ac'; k['X']  = '2a 2d aa ad' ; k['C']  = '2a 2e aa ae' ; k['V']  = '2a 2f aa af';
+            k['B'] = '2a 30 aa b0' ; k['N'] = '2a 31 aa b1' ; k['M'] = '2a 32 aa b2';
+
+            k[',']= '33 b3' ; k['.']='34 b4'; k['/'] = '35 b5' ;k[':'] = '2a 27 aa a7';
+            k['%'] = '2a 06 aa 86';  k['_'] = '2a 0c aa 8c';
+            k['&'] = '2a 08 aa 88';
+            k['('] = '2a 0a aa 8a';
+            k[')'] = '2a 0b aa 8b';
+
+
+            special=Hash.new;
+            special['<Enter>'] = '1c 9c';
+            special['<Backspace>'] = '0e 8e';
+            special['<Spacebar>'] = '39 b9';
+            special['<Return>'] = '1c 9c'
+            special['<Esc>'] = '01 81';
+            special['<Tab>'] = '0f 8f';
+            special['<KillX>'] = '1d 38 0e';
+            special['<Wait>'] = 'wait';
+
+            special['<Up>'] = '48 c8';
+            special['<Down>'] = '50 d0';
+            special['<PageUp>'] = '49 c9';
+            special['<PageDown>'] = '51 d1';
+            special['<End>'] = '4f cf';
+            special['<Insert>'] = '52 d2';
+            special['<Delete>'] = '53 d3';
+            special['<Left>'] = '4b cb';
+            special['<Right>'] = '4d cd';
+            special['<Home>'] = '47 c7';
+
+            special['<F1>'] = '3b';
+            special['<F2>'] = '3c';
+            special['<F3>'] = '3d';
+            special['<F4>'] = '3e';
+            special['<F5>'] = '3f';
+            special['<F6>'] = '40';
+            special['<F7>'] = '41';
+            special['<F8>'] = '42';
+            special['<F9>'] = '43';
+            special['<F10>'] = '44';
+
+            keycodes=''
+            thestring.gsub!(/ /,"<Spacebar>")
+
+            until thestring.length == 0
+              nospecial=true;
+              special.keys.each { |key|
+                if thestring.start_with?(key)
+                  #take thestring
+                  #check if it starts with a special key + pop special string
+                  keycodes=keycodes+special[key]+' ';
+                  thestring=thestring.slice(key.length,thestring.length-key.length)
+                  nospecial=false;
+                  break;
+                end
+              }
+              if nospecial
+                code=k[thestring.slice(0,1)]
+                if !code.nil?
+                  keycodes=keycodes+code+' '
+                else
+                  env.ui.info "no scan code for #{thestring.slice(0,1)}"
+                end
+                #pop one
+                thestring=thestring.slice(1,thestring.length-1)
+              end
+            end
+
+            return keycodes
+          end
+        end #Class
+      end #Module
+    end #Module
+  end #Module
+end #Module

data/lib/veewee/provider/core/helper/shell.rb

@@ -0,0 +1,55 @@
+module Veewee
+  module Provider
+    module Core
+      module Helper
+
+        class ShellResult
+          attr_accessor :stdout
+          attr_accessor :stderr
+          attr_accessor :status
+
+          def initialize(stdout,stderr,status)
+            @stdout=stdout
+            @stderr=stderr
+            @status=status
+          end
+        end
+
+        module Shell
+
+          # http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-talk/185404
+          # This should work on windows too now
+          # This will result in a ShellResult structure with stdout, stderr and status
+          def shell_exec(command,options = {:mute => true,:status => 0})
+            defaults={:mute => true, :status => 0}
+            options=defaults.merge(options)
+            result=ShellResult.new("","",-1)
+            env.ui.info "Executing #{command}" unless options[:mute]
+            env.logger.debug "Command: \"#{command}\""
+            env.logger.debug "Output:"
+            env.logger.debug "-------"
+            escaped_command=command
+            IO.popen("#{escaped_command}"+ " 2>&1") { |p|
+              p.each_line{ |l|
+                result.stdout+=l
+                env.ui.info(l,{:new_line => false})  unless options[:mute]
+                env.logger.debug(l.chomp)
+              }
+              result.status=Process.waitpid2(p.pid)[1].exitstatus
+              if result.status.to_i!=options[:status]
+                env.ui.error "Error: We executed a shell command and the exit status was not #{options[:status]}"
+                env.ui.error "- Command :#{command}."
+                env.ui.error "- Exitcode :#{result.status}."
+                env.ui.error "- Output   :\n#{result.stdout}"
+                raise Veewee::Error,"Wrong exit code for command #{command}"
+              end
+            }
+            return result
+          end
+
+
+        end #Module
+      end #Module
+    end #Module
+  end #Module
+end #Module