veewee 0.2.3 → 0.3.0.alpha1

data/lib/veewee/util/scancode.rb

@@ -0,0 +1,112 @@
+module Veewee
+  module Util
+    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
+
+

data/lib/veewee/util/shell.rb

@@ -0,0 +1,42 @@
+module Veewee
+  module Util
+
+    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 = {})
+        result=ShellResult.new("","",-1)
+        env.ui.info "Executing #{command}" unless options[:mute]
+        escaped_command=command
+        #        env.ui.info "#{escaped_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]
+          }
+          result.status=Process.waitpid2(p.pid)[1].exitstatus
+          if result.status!=0
+            env.ui.error "Exit status was not 0 but #{result.status}" unless options[:mute]
+          end
+        }
+        return result
+      end
+
+
+    end #Module
+  end #Module
+end #Module

data/lib/veewee/util/ssh.rb

@@ -0,0 +1,158 @@
+module Veewee
+  module Util
+
+    class SshResult
+      attr_accessor :stdout
+      attr_accessor :stderr
+      attr_accessor :status
+
+      def initialize(stdout,stderr,status)
+        @stdout=stdout
+        @stderr=stderr
+        @status=status
+      end
+    end
+
+    module Ssh
+
+      require 'net/ssh'
+      require 'net/scp'
+
+      def when_ssh_login_works(ip="localhost", options = {  } , &block)
+
+        defaults={ :port => '22', :timeout => 20000 , :user => 'vagrant', :password => 'vagrant'}
+
+        options=defaults.merge(options)
+
+        env.ui.info  "Waiting for ssh login on #{ip} with user #{options[:user]} to sshd on port => #{options[:port]} to work, timeout=#{options[:timeout]} sec"
+
+        begin
+          Timeout::timeout(options[:timeout]) do
+            connected=false
+            while !connected do
+              begin
+                env.ui.info ".",{:new_line => false}
+                Net::SSH.start(ip, options[:user], { :port => options[:port] , :password => options[:password], :paranoid => false , :timeout => options[:timeout] }) do |ssh|
+                  env.ui.info "\n"
+
+                  block.call(ip);
+                  env.ui.info ""
+                  return true
+                end
+              rescue Net::SSH::Disconnect,Errno::ECONNREFUSED, Errno::EHOSTUNREACH, Errno::ECONNABORTED, Errno::ECONNRESET, Errno::ENETUNREACH,Errno::ETIMEDOUT, Errno::ENETUNREACH
+                sleep 5
+              end
+            end
+          end
+        rescue Timeout::Error
+          env.ui.error "Ssh timeout #{options[:timeout]} min has been reached."
+          exit -1
+        end
+        env.ui.info ""
+        return false
+      end
+
+
+      def ssh_transfer_file(host,filename,destination = '.' , options = {})
+
+        defaults={ :paranoid => false }
+        options=defaults.merge(options)
+
+        Net::SSH.start( host,options[:user],options ) do |ssh|
+          env.ui.info "Transferring #{filename} to #{destination} "
+          ssh.scp.upload!( filename, destination ) do |ch, name, sent, total|
+            #   print "\r#{destination}: #{(sent.to_f * 100 / total.to_f).to_i}%"
+            env.ui.info ".",{:new_line => false}
+
+          end
+        end
+        env.ui.info ""
+      end
+
+
+      def ssh_exec(host,command, options = { :progress => "on"} )
+        defaults= { :port => "22", :exitcode => "0", :user => "root"}
+        options=defaults.merge(options)
+        pid=""
+        stdin=command
+        stdout=""
+        stderr=""
+        status=-99999
+
+        env.ui.info "Executing command: #{command}"
+
+        Net::SSH.start(host, options[:user], { :port => options[:port], :password => options[:password], :paranoid => false }) do |ssh|
+
+          # open a new channel and configure a minimal set of callbacks, then run
+          # the event loop until the channel finishes (closes)
+          channel = ssh.open_channel do |ch|
+
+            #request pty for sudo stuff and so
+            ch.request_pty do |ch, success|
+              raise "Error requesting pty" unless success
+            end
+
+            ch.exec "#{command}" do |ch, success|
+              raise "could not execute command" unless success
+
+
+              # "on_data" is called when the process writes something to stdout
+              ch.on_data do |c, data|
+                stdout+=data
+
+                env.ui.info data,{:new_line => false}
+
+              end
+
+              # "on_extended_data" is called when the process writes something to stderr
+              ch.on_extended_data do |c, type, data|
+                stderr+=data
+
+                env.ui.info data,{:new_line => false}
+
+              end
+
+              #exit code
+              #http://groups.google.com/group/comp.lang.ruby/browse_thread/thread/a806b0f5dae4e1e2
+              channel.on_request("exit-status") do |ch, data|
+                exit_code = data.read_long
+                status=exit_code
+                if exit_code > 0
+                  env.ui.info "ERROR: exit code #{exit_code}"
+                else
+                  #env.ui.info "Successfully executed"
+                end
+              end
+
+              channel.on_request("exit-signal") do |ch, data|
+                env.ui.info "SIGNAL: #{data.read_long}"
+              end
+
+              ch.on_close {
+                #env.ui.info "done!"
+              }
+              #status=ch.exec "echo $?"
+            end
+          end
+          channel.wait
+        end
+
+
+        if (status.to_s != options[:exitcode] )
+          if (options[:exitcode]=="*")
+            #its a test so we don't need to worry
+          else
+            raise "Exitcode was not what we expected"
+          end
+
+        end
+
+        return Veewee::Util::SshResult.new(stdout,stderr,status)
+
+      end
+
+
+
+    end #Class
+  end #Module
+end #Module

data/lib/veewee/util/tcp.rb

@@ -0,0 +1,88 @@
+require 'socket'
+require 'timeout'
+
+module Veewee
+  module Util
+    module Tcp
+
+      def is_tcp_port_open?(ip, port)
+        begin
+          Timeout::timeout(1) do
+            begin
+              s = TCPSocket.new(ip, port)
+              s.close
+              return true
+            rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
+              return false
+            end
+          end
+        rescue Timeout::Error
+        end
+
+        return false
+      end
+
+      # This tries to guess a local free tcp port 
+      def guess_free_port(min_port,max_port)
+        env.ui.info "Received port hint - #{min_port}"
+
+        guessed_port=nil
+
+        for port in (min_port..max_port)
+          unless is_tcp_port_open?(get_local_ip, port)
+            guessed_port=port
+            break
+          end
+        end
+
+        if guessed_port.nil?
+          env.ui.info "No free port available: tried #{min_port}..#{max_port}"
+          exit -1
+        else
+          env.ui.info "Found port #{guessed_port} available"
+        end
+
+        return guessed_port
+      end
+
+      def execute_when_tcp_available(ip="localhost", options = { } , &block)
+
+        defaults={ :port => 22, :timeout => 2 , :pollrate => 5}
+
+        options=defaults.merge(options)
+
+        begin
+          Timeout::timeout(options[:timeout]) do
+            connected=false
+            while !connected do
+              begin
+                env.ui.info "trying connection"
+                s = TCPSocket.new(ip, options[:port])
+                s.close
+                block.call(ip);
+                return true
+              rescue Errno::ECONNREFUSED, Errno::EHOSTUNREACH
+                sleep options[:pollrate]
+              end
+            end
+          end
+        rescue Timeout::Error
+          raise 'timeout connecting to port'
+        end
+
+        return false
+      end
+
+      def get_local_ip
+        orig, Socket.do_not_reverse_lookup = Socket.do_not_reverse_lookup, true  # turn off reverse DNS resolution temporarily
+
+        UDPSocket.open do |s|
+          s.connect '64.233.187.99', 1
+          s.addr.last
+        end
+      ensure
+        Socket.do_not_reverse_lookup = orig
+      end
+    end #Module
+  end #Module
+end #Module

data/lib/veewee/util/web.rb

@@ -0,0 +1,63 @@
+module Veewee
+  module Util
+      require 'webrick'
+      include WEBrick
+    module Servlet
+
+
+      class FileServlet < WEBrick::HTTPServlet::AbstractServlet
+
+        attr_reader :env
+
+        def initialize(server,localfile,env)
+          super(server)
+          @server=server
+          @localfile=localfile
+          @env=env
+        end
+
+        def do_GET(request,response)
+          response['Content-Type']='text/plain'
+          response.status = 200
+          env.ui.info "Serving file #{@localfile}"
+          displayfile=File.open(@localfile,'r')
+          content=displayfile.read()
+          response.body=content
+          #If we shut too fast it might not get the complete file
+          sleep 2
+          @server.shutdown
+        end
+      end
+
+    end
+    module Web
+
+      def wait_for_http_request(filename,options={:timeout => 10, :web_dir => "", :port => 7125})
+
+
+        # Calculate the OS equivalent of /dev/null , on windows this is NUL:
+        # http://www.ruby-forum.com/topic/115472
+        fn = test(?e, '/dev/null') ? '/dev/null' : 'NUL:'
+
+        webrick_logger=WEBrick::Log.new(fn, WEBrick::Log::INFO)
+
+        web_dir=options[:web_dir]
+        filename=filename
+        s= ::WEBrick::HTTPServer.new(
+          :Port => options[:port],
+          :Logger => webrick_logger,
+          :AccessLog => webrick_logger
+        )
+        env.logger.debug ("mounting file /#{filename}")
+        s.mount("/#{filename}", Veewee::Util::Servlet::FileServlet,File.join(web_dir,filename),env)
+        trap("INT"){
+          s.shutdown
+          env.ui.info "Stopping webserver"
+          exit
+        }
+        s.start
+      end
+
+    end #Class
+  end #Module
+end #Module