sahara 0.0.1

data/.gitignore

@@ -0,0 +1,6 @@
+.DS_Store
+gems/*
+pkg/*
+tmp/*
+.vagrant
+Vagrantfile

data/.rvmrc

@@ -0,0 +1,5 @@
+rvm use ruby-1.8.7
+rvm_gemset_create_on_use_flag=1
+rvm gemset use sahara
+alias vagrant="bundle exec vagrant"
+alias irb="bundle exec irb"

data/Gemfile

@@ -0,0 +1,5 @@
+source "http://rubygems.org"
+
+gem "sahara", :path => "."
+
+#gem "vagrant", :git => "git://github.com/mitchellh/vagrant.git"

data/Gemfile.lock

@@ -0,0 +1,48 @@
+PATH
+  remote: .
+  specs:
+    sahara (0.0.1)
+      popen4 (~> 0.1.2)
+      thor (~> 0.14.6)
+      vagrant (~> 0.7.0)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    Platform (0.4.0)
+    abstract (1.0.0)
+    archive-tar-minitar (0.5.2)
+    erubis (2.6.6)
+      abstract (>= 1.0.0)
+    ffi (0.6.3)
+      rake (>= 0.8.7)
+    i18n (0.5.0)
+    json (1.5.1)
+    mario (0.0.6)
+    net-scp (1.0.4)
+      net-ssh (>= 1.99.1)
+    net-ssh (2.1.4)
+    open4 (1.0.1)
+    popen4 (0.1.2)
+      Platform (>= 0.4.0)
+      open4 (>= 0.4.0)
+    rake (0.8.7)
+    thor (0.14.6)
+    vagrant (0.7.3)
+      archive-tar-minitar (= 0.5.2)
+      erubis (~> 2.6.6)
+      i18n (~> 0.5.0)
+      json (~> 1.5.1)
+      mario (~> 0.0.6)
+      net-scp (~> 1.0.4)
+      net-ssh (~> 2.1.0)
+      thor (~> 0.14.6)
+      virtualbox (~> 0.8.3)
+    virtualbox (0.8.3)
+      ffi (~> 0.6.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  sahara!

data/README.md

@@ -0,0 +1,19 @@
+# Description
+
+Sahara allows vagrant to operate in sandbox mode
+
+Typical usage:
+
+- Enter sandbox mode: <pre>vagrant sandbox on</pre>
+- Do some stuff: <pre>vagrant ssh </pre>
+- If satisfied apply the changes: <pre>vagrant sandbox commit</pre>
+- If not satisfied you can rollback: <pre>vagrant sandbox rollback</pre>
+- To leave sandbox mod: <pre>vagrant sandbox off</pre>
+
+Many kudos go to the creators of [vagrant](http://vagrantup.com)
+
+# Installation
+
+This is now available as gem:
+
+<pre>gem install sahara</pre>

data/Rakefile

@@ -0,0 +1,5 @@
+require 'rubygems'
+require 'bundler'
+require 'bundler/setup'
+Bundler::GemHelper.install_tasks
+

data/lib/sahara.rb

@@ -0,0 +1,2 @@
+require 'vagrant'
+require 'sahara/command'

data/lib/sahara/command.rb

@@ -0,0 +1,33 @@
+require 'sahara/session'
+
+module Sahara
+	class Command < Vagrant::Command::GroupBase
+	  register "sandbox","Manages a sandbox"
+
+	  desc "status [NAME]", "Shows the status of the sandbox"
+	  def status(boxname=nil)
+	    Sahara::Session.status(boxname)
+	  end
+
+	  desc "on [NAME]", "Enters sandbox state"
+	  def on(boxname=nil)
+	    Sahara::Session.on(boxname)
+	  end
+
+	  desc "commit [NAME]", "Commits changes - moves sandbox initial state to currentstate"
+	  def commit(boxname=nil)
+	    Sahara::Session.commit(boxname)
+	  end
+
+	  desc "rollback [NAME]", "Rollback changes since sandbox state was entered "
+	  def rollback(boxname=nil)
+	    Sahara::Session.rollback(boxname)
+	  end
+	  
+	  desc "off [NAME] ", "Leaves sandbox state"
+	  def off(boxname=nil)
+	    Sahara::Session.off(boxname)
+	  end
+    
+	end
+end

data/lib/sahara/session.rb

@@ -0,0 +1,178 @@
+require 'pp'
+
+require 'drift/shell'
+
+module Drift
+  module Session
+
+    def self.determine_vboxcmd
+      return "VBoxManage"
+    end
+
+    def self.initialize
+      @vagrant_env=Vagrant::Environment.new
+      @vboxcmd=determine_vboxcmd
+      @sandboxname="drift-sandbox"
+    end
+
+    def self.status(selection)
+      self.initialize
+
+      on_selected_vms(selection) do |boxname|
+        if is_snapshot_mode_on?(boxname)
+          puts "[#{boxname}] - snapshot mode is on"
+        else
+          puts "[#{boxname}] - snapshot mode is off"
+        end
+      end
+
+    end
+
+    def self.on(selection)
+      self.initialize
+
+      on_selected_vms(selection) do |boxname|
+        if is_snapshot_mode_on?(boxname)
+          puts "[#{boxname}] - snapshot mode is already on"
+        else
+          instance_name="#{@vagrant_env.vms[boxname.to_sym].vm.name}"
+
+          #Creating a snapshot
+          puts "[#{boxname}] - Enabling sandbox"
+
+          execute("#{@vboxcmd} snapshot '#{instance_name}' take '#{@sandboxname}'")
+        end
+
+      end
+    end
+
+    def self.commit(selection)
+
+      self.initialize
+      on_selected_vms(selection) do |boxname|
+
+
+        if !is_snapshot_mode_on?(boxname)
+          puts "[#{boxname}] - this requires that sandbox mode is on."
+        else
+          instance_name="#{@vagrant_env.vms[boxname.to_sym].vm.name}"
+
+          #Discard snapshot so current state is the latest state
+          puts "[#{boxname}] - unwinding sandbox"
+          execute("#{@vboxcmd} snapshot '#{instance_name}' delete '#{@sandboxname}'")
+
+          #Now retake the snapshot
+          puts "[#{boxname}] - fastforwarding sandbox"
+
+          execute("#{@vboxcmd} snapshot '#{instance_name}' take '#{@sandboxname}'")
+          
+        end
+
+      end
+
+    end
+
+    def self.rollback(selection)
+      self.initialize
+
+      on_selected_vms(selection) do |boxname|
+
+        if !is_snapshot_mode_on?(boxname)
+          puts "[#{boxname}] - this requires that sandbox mode is on."
+        else
+          instance_name="#{@vagrant_env.vms[boxname.to_sym].vm.name}"
+
+          puts "[#{boxname}] - powering off machine"
+
+          #Poweroff machine
+          execute("#{@vboxcmd} controlvm '#{instance_name}' poweroff")
+
+          puts "[#{boxname}] - roll back machine"
+
+          #Rollback until snapshot
+          execute("#{@vboxcmd} snapshot '#{instance_name}' restore '#{@sandboxname}'")
+
+          puts "[#{boxname}] - starting the machine again"
+
+          #Startvm again
+          execute("#{@vboxcmd} startvm --type headless '#{instance_name}' ")
+          
+        end
+
+      end
+
+
+    end
+
+    def self.off(selection)
+      self.initialize
+
+      on_selected_vms(selection) do |boxname|
+
+
+        instance_name="#{@vagrant_env.vms[boxname.to_sym].vm.name}"
+
+        if !is_snapshot_mode_on?(boxname)
+          puts "[#{boxname}] - this requires that sandbox mode is on."
+        else
+          puts "[#{boxname}] - switching sandbox off"
+
+          # We might wanna check for sandbox changes or not
+
+          #Discard snapshot
+          execute("#{@vboxcmd} snapshot '#{instance_name}' delete '#{@sandboxname}' ")
+
+        end
+
+      end
+    end
+
+    def self.execute(command)
+      #puts "#{command}"
+      output=Sahara::Shell.execute("#{command}")
+      return output
+    end
+
+    def self.is_vm_created?(boxname)
+      return !@vagrant_env.vms[boxname.to_sym].vm.nil?
+    end
+
+    def self.list_snapshots(boxname)
+
+      instance_name="#{@vagrant_env.vms[boxname.to_sym].vm.name}"
+      snapshotlist=Array.new
+      snapshotresult=execute("#{@vboxcmd} showvminfo --machinereadable '#{instance_name}' |grep ^SnapshotName| cut -d '=' -f 2")
+      snapshotresult.each do |result|
+        clean=result.gsub(/\"/,'').chomp
+        snapshotlist << clean
+      end
+      return snapshotlist
+    end
+
+    def self.is_snapshot_mode_on?(boxname)
+      snapshots=list_snapshots(boxname)
+      return snapshots.include?(@sandboxname)
+    end
+
+    def self.on_selected_vms(selection,&block)
+      if selection.nil?
+        #puts "no selection was done"
+        @vagrant_env.vms.each do |name,vm|
+          #puts "Processing #{name}"
+          yield name
+        end
+      else
+        yield selection        
+      end
+    end
+  end
+end
+
+
+#command="#{@vboxcmd} unregistervm  '#{boxname}' --delete"    
+#puts command
+#puts "Deleting vm #{boxname}"
+
+#Exec and system stop the execution here
+#Veewee::Shell.execute("#{command}")
+

data/lib/sahara/shell.rb

@@ -0,0 +1,58 @@
+#require 'open4'
+
+module Sahara
+  class Shell
+ 
+    def self.execute(command,options = {})
+      STDOUT.sync = true
+      output=nil
+      result=IO.popen("#{command}") {|f| output=f.readlines}
+      #{ |f| print "#{f}, #{f.class}" }
+#      output=result.read
+      return output
+    end
+    
+    #pty allows you to gradually see the output of a local command
+    #http://www.shanison.com/?p=415
+      def self.execute2(command, options = {} )
+        require "pty"
+            begin
+              PTY.spawn( command ) do |r, w, pid|
+                begin
+                  #r.each { }      
+                  r.each { |line| print line;}      
+
+               rescue Errno::EIO        
+               end      
+             end  
+           rescue PTY::ChildExited => e
+              puts "The child process exited!"
+           end
+      end
+
+      #occassinally fails with 'no child processes
+      def self.execute3(command, options = {} )
+        defaults= { :port => "22", :exitcode => "0", :user => "root"}
+          options=defaults.merge(options) 
+
+          status = POpen4::popen4(command) do |stdout,stderr,stdin|
+            stdout.each do |line|
+              puts line
+            end
+          end
+          
+        @status=status.to_i
+  
+        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
+
+      end
+
+  end #Class
+end #Module

data/lib/sahara/version.rb

@@ -0,0 +1,3 @@
+module Sahara
+  VERSION = "0.0.1"
+end

data/lib/vagrant_init.rb

@@ -0,0 +1,6 @@
+begin
+  require 'sahara'
+rescue LoadError
+  require 'rubygems'
+  require 'sahara'
+end

data/sahara.gemspec

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path("../lib/sahara/version", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "sahara"
+  s.version     = Sahara::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Patrick Debois"]
+  s.email       = ["patrick.debois@jedi.be"]
+  s.homepage    = "http://github.com/jedi4ever/sahara/"
+  s.summary     = %q{Vagrant box creation}
+  s.description = %q{Allows you to sandbox your vagrant}
+
+  s.required_rubygems_version = ">= 1.3.6"
+  s.rubyforge_project         = "sahara"
+
+  s.add_dependency "vagrant", "~> 0.7.0"
+  s.add_dependency "popen4", "~> 0.1.2"
+  s.add_dependency "thor", "~> 0.14.6"
+
+  s.add_development_dependency "bundler", ">= 1.0.0"
+
+  s.files        = `git ls-files`.split("\n")
+  s.executables  = `git ls-files`.split("\n").map{|f| f =~ /^bin\/(.*)/ ? $1 : nil}.compact
+  s.require_path = 'lib'
+end
+

metadata

@@ -0,0 +1,145 @@
+--- !ruby/object:Gem::Specification 
+name: sahara
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Patrick Debois
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-04-28 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        - 7
+        - 0
+        version: 0.7.0
+  type: :runtime
+  requirement: *id001
+  name: vagrant
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 31
+        segments: 
+        - 0
+        - 1
+        - 2
+        version: 0.1.2
+  type: :runtime
+  requirement: *id002
+  name: popen4
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 43
+        segments: 
+        - 0
+        - 14
+        - 6
+        version: 0.14.6
+  type: :runtime
+  requirement: *id003
+  name: thor
+  prerelease: false
+- !ruby/object:Gem::Dependency 
+  version_requirements: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 23
+        segments: 
+        - 1
+        - 0
+        - 0
+        version: 1.0.0
+  type: :development
+  requirement: *id004
+  name: bundler
+  prerelease: false
+description: Allows you to sandbox your vagrant
+email: 
+- patrick.debois@jedi.be
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- .rvmrc
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- lib/sahara.rb
+- lib/sahara/command.rb
+- lib/sahara/session.rb
+- lib/sahara/shell.rb
+- lib/sahara/version.rb
+- lib/vagrant_init.rb
+- sahara.gemspec
+has_rdoc: true
+homepage: http://github.com/jedi4ever/sahara/
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 23
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: sahara
+rubygems_version: 1.4.2
+signing_key: 
+specification_version: 3
+summary: Vagrant box creation
+test_files: []
+