HeSYINUvSBZfxqA-veewee 0.2.0.2

data/lib/veewee/command.rb

@@ -0,0 +1,82 @@
+require 'veewee/session'
+
+#Load Veewee::Session libraries
+lib_dir= File.expand_path(File.join(File.dirname(__FILE__),"..","..", "lib"))
+Dir.glob(File.join(lib_dir, '**','*.rb')).each {|f| require f  }
+
+#Setup some base variables to use
+template_dir=File.expand_path(File.join(lib_dir,"..", "templates"))
+
+veewee_dir="."
+definition_dir= File.expand_path(File.join(veewee_dir, "definitions"))
+tmp_dir=File.expand_path(File.join(veewee_dir, "tmp"))
+iso_dir=File.expand_path(File.join(veewee_dir, "iso"))
+box_dir=File.expand_path(File.join(veewee_dir, "boxes"))
+validation_dir=File.expand_path(File.join(lib_dir, "..","validation"))
+
+#Initialize
+Veewee::Session.setenv({:veewee_dir => veewee_dir, :definition_dir => definition_dir,
+   :template_dir => template_dir, :iso_dir => iso_dir, :box_dir => box_dir, :tmp_dir => tmp_dir, :validation_dir => validation_dir})
+
+module Veewee
+class Command < Vagrant::Command::GroupBase
+  register "basebox","Commands to manage baseboxes"  
+
+  desc "templates", "List the currently available basebox templates"
+  def templates
+    Veewee::Session.list_templates
+  end
+
+  desc "define BOXNAME TEMPLATE", "Define a new basebox starting from a template"
+  method_option :force,:type => :boolean , :default => false, :aliases => "-f", :desc => "overwrite the definition"
+  def define(boxname, template)    
+    Veewee::Session.define(boxname,template,options)
+  end
+
+  desc "undefine BOXNAME", "Removes the definition of a basebox "
+  def undefine(boxname)    
+      Veewee::Session.undefine(boxname)
+  end
+
+  desc "build BOXNAME", "Build the box BOXNAME"
+  method_option :force,:type => :boolean , :default => false, :aliases => "-f", :desc => "overwrite the basebox"
+  method_option :nogui,:type => :boolean , :default => false, :aliases => "-n", :desc => "no gui"
+
+  def build(boxname)
+    Veewee::Session.build(boxname,options)
+  end
+
+  desc "ostypes", "List the available Operating System types"
+  def ostypes
+    Veewee::Session.list_ostypes
+  end
+  
+  desc "destroy BOXNAME", "Destroys the virtualmachine that was build for a basebox"
+  def destroy(boxname)
+     Veewee::Session.destroy_vm(boxname)
+  end
+  
+  desc "list", "Lists all defined baseboxes"
+  def list
+  Veewee::Session.list_definitions
+  end
+  
+  desc "export [NAME]", "Exports the basebox to the vagrant box format" 
+  method_options :force => :boolean  
+  def export(boxname)
+      if (!boxname.nil?)
+        Veewee::Session.export_box(boxname)
+      end
+  end
+  
+  desc "validate [NAME]", "Validates a box against vagrant compliancy rules"
+  method_option :user,:default => "vagrant", :aliases => "-u", :desc => "user to login with"
+  def validate(boxname)
+      if (!boxname.nil?)
+        Veewee::Session.validate_box(boxname,options)
+      end
+  end
+  
+end
+
+end

data/lib/veewee/config.rb

@@ -0,0 +1,5 @@
+module Veewee
+  class BaseBoxConfig < Vagrant::Config::Base
+    configures :basebox
+  end
+end

data/lib/veewee/export.rb

@@ -0,0 +1,81 @@
+require 'pathname'
+module Veewee
+  class Export
+
+#    Shellutil.execute("vagrant package --base #{vmname} --include /tmp/Vagrantfile --output /tmp/#{vmname}.box", {:progress => "on"})    
+    
+    def self.vagrant(boxname,boxdir,definition)
+      
+      #Check if box already exists
+      vm=VirtualBox::VM.find(boxname)
+      if vm.nil?
+        puts "#{boxname} is not found, maybe you need to build first?"
+        exit
+      end
+      #We need to shutdown first
+      if vm.running?
+        puts "Vagrant requires the box to be shutdown, before it can export"
+        puts "Sudo also needs to work for user #{definition[:ssh_user]}"
+        puts "Performing a clean shutdown now."
+        ssh_options={ :user => definition[:ssh_user], :port => definition[:ssh_host_port], :password => definition[:ssh_password],
+            :timeout => definition[:ssh_timeout]}       
+                    
+        Veewee::Ssh.execute("localhost","sudo #{definition[:shutdown_cmd]}",ssh_options)
+
+        #Wait for state poweroff
+        while (vm.running?) do 
+          print '.'
+          sleep 1
+        end
+        puts
+        puts "Machine #{boxname} 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,boxname+".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}")
+        puts "box #{boxname}.box already exists"
+        exit
+      end
+      
+      puts "Excuting vagrant voodoo:"
+      export_command="vagrant package --base '#{boxname}' --output '#{box_path}'"
+      puts "#{export_command}"
+      Veewee::Shell.execute("#{export_command}") #hmm, needs to get the gem_home set?
+      puts
+      
+      #add_ssh_nat_mapping back!!!!      
+      
+      puts "To import it into vagrant type:"
+      puts "vagrant box add '#{boxname}' '#{box_path}'"
+      puts ""
+      puts "To use it:"
+      puts "vagrant init '#{boxname}'"
+      puts "vagrant up"
+      puts "vagrant ssh"
+    end
+ 
+  end
+end
+
+
+#      #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.vm.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/scancode.rb

@@ -0,0 +1,151 @@
+module Veewee
+  class Scancode
+  
+    def self.send_sequence(vboxcmd,vname,sequence,webport)
+            puts
+            counter=0
+            sequence.each { |s|
+              counter=counter+1
+ 
+              s.gsub!(/%IP%/,Veewee::Session.local_ip);
+              s.gsub!(/%PORT%/,webport);
+              s.gsub!(/%NAME%/, vname);
+              puts "Typing:[#{counter}]: "+s
+
+              keycodes=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(vboxcmd,vname,keycode)    
+                   sleep 0.01                  
+                  else
+                    sleep 1
+                  end
+              end
+        	    #sleep after each sequence (needs to be param)
+        	    sleep 1
+    	    }
+
+          puts "Done typing."
+          puts
+  
+    end
+
+    def self.send_keycode(vboxcmd,vname,keycode)
+           command= "#{vboxcmd} controlvm '#{vname}' keyboardputscancode #{keycode}"
+           #puts "#{command}"
+                  IO.popen("#{command}") { |f| print '' }
+    end
+
+    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['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
+                puts "no scan code for #{thestring.slice(0,1)}"
+              end
+              #pop one
+              thestring=thestring.slice(1,thestring.length-1)
+            end
+          end
+
+            return keycodes
+    end
+
+  end
+end
+
+