vagrant-openvz 0.0.2

data/lib/vagrant-openvz/action/created.rb

@@ -0,0 +1,20 @@
+module VagrantPlugins
+  module Openvz 
+    module Action
+      class Created
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+
+          # Set the result to be true if the machine is created.
+	      env[:result] = true 
+		  env[:result] = false if env[:machine].state.id == :not_created
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/destroy.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module Openvz
+    module Action
+      class Destroy
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+
+          env[:ui].info "Destroying!" 
+          env[:machine].provider.driver.destroy(config.vzctid)
+
+          env[:machine].id = nil
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/destroy_confirm.rb

@@ -0,0 +1,15 @@
+require "vagrant/action/builtin/confirm"
+
+module VagrantPlugins
+  module Openvz
+    module Action
+      class DestroyConfirm < Vagrant::Action::Builtin::Confirm
+        def initialize(app, env)
+          force_key = :force_confirm_destroy
+          message   = "Destory Confirmation? [y/n] "
+          super(app, env, message, force_key)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/fetch_ip.rb

@@ -0,0 +1,19 @@
+module VagrantPlugins
+  module Openvz
+    module Action
+      class FetchIp
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+		  config = env[:machine].provider_config
+
+          env[:machine_ip] ||= env[:machine].provider.driver.fetch_ip(config.vzctid)
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/forced_halt.rb

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module Openvz
+    module Action
+      class ForcedHalt
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+		  config = env[:machine].provider_config
+
+          if env[:machine].provider.state.id == :exist_mounted_running
+			env[:machine].provider.driver.forced_halt(config.vzctid)
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/is_running.rb

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module Openvz
+    module Action
+      class IsRunning
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+
+          env[:result] = env[:machine].state.id == :exist_mounted_running
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/message.rb

@@ -0,0 +1,22 @@
+module VagrantPlugins
+  module Openvz
+    module Action
+      class Message
+        def initialize(app, env, msg_key, type = :info)
+          @app     = app
+          @msg_key = msg_key
+          @type    = type
+        end
+
+        def call(env)
+          machine = env[:machine]
+          message = "#{machine.name}: #{@msg_key}"
+
+          env[:ui].send @type, message
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/action/share_folders.rb

@@ -0,0 +1,56 @@
+module VagrantPlugins
+  module Openvz
+	module Action
+	  class ShareFolders
+		def initialize(app, env)
+		  @app = app
+		end
+
+		def call(env)
+		  @env = env
+		  prepare_folders
+		  @app.call env
+		  add_folders
+		end
+
+		def shared_folders
+		  {}.tap do |result|
+			@env[:machine].config.vm.synced_folders.each do |id, data|
+			  # This to prevent overwriting the actual shared folders data
+			  result[id] = data.dup
+			end
+		  end
+		end
+
+		# Prepares the shared folders by verifying they exist and creating them
+		# if they don't.
+		def prepare_folders
+		  shared_folders.each do |id, options|
+			hostpath = Pathname.new(options[:hostpath]).expand_path(@env[:root_path])
+			if !hostpath.directory? && options[:create]
+			  # Host path doesn't exist, so let's create it.
+			  begin
+				hostpath.mkpath
+			  rescue Errno::EACCES
+				raise Vagrant::Errors::SharedFolderCreateFailed,
+				  :path => hostpath.to_s
+			  end
+			end
+		  end
+		end
+
+		def add_folders
+		  folders = []
+		  shared_folders.each do |id, data|
+			folders << {
+			  :name      => id,
+			  :hostpath  => File.expand_path(data[:hostpath], @env[:root_path]),
+			  :guestpath => data[:guestpath]
+			}
+		  end
+		  @env[:machine].provider.driver.share_folders(folders,@env[:machine].provider_config.vzctid)
+		end
+	  end
+	end
+  end
+end

data/lib/vagrant-openvz/command/version.rb

@@ -0,0 +1,11 @@
+
+module VagrantPlugins
+  module  Openvz
+	class VzctlVersion < Vagrant.plugin("2", "command")
+	  def execute
+		puts `vzctl --version`
+		return 0
+	  end
+	end
+  end
+end

data/lib/vagrant-openvz/config.rb

@@ -0,0 +1,56 @@
+require "vagrant"
+
+module VagrantPlugins
+  module Openvz
+	class Config < Vagrant.plugin("2", :config)
+	  attr_accessor :physpages
+	  attr_accessor :hostname
+	  attr_accessor :ipadd
+	  attr_accessor :nameserver
+	  attr_accessor :vzctid
+	  attr_accessor :pubkey
+	  attr_accessor :diskspace
+	  attr_accessor :diskinodes
+	  attr_accessor :quotatime
+
+	  # A String that points to a file that acts as a wrapper for sudo commands.
+	  # This allows us to have a single entry when whitelisting NOPASSWD commands
+	  # on /etc/sudoers
+	  attr_accessor :sudo_wrapper
+
+	  def initialize
+		@customizations = []
+		@sudo_wrapper   = UNSET_VALUE
+	  end
+
+	  def finalize!
+		@sudo_wrapper = nil if @sudo_wrapper == UNSET_VALUE
+	  end
+
+	  def validate(machine)
+		errors = []
+
+		errors << "config.physpages" if physpages.nil?
+		errors << "config.hostname" if hostname.nil?
+		errors << "config.ipadd" if ipadd.nil?
+		errors << "config.nameserver" if nameserver.nil?
+		errors << "config.vzctid" if vzctid.nil?
+		errors << "config.pubkey" if pubkey.nil?
+		errors << "config.diskspace" if diskspace.nil?
+		errors << "config.diskinodes" if diskinodes.nil?
+		errors << "config.quotatime" if quotatime.nil?
+
+		if @sudo_wrapper
+		  hostpath = Pathname.new(@sudo_wrapper).expand_path(machine.env.root_path)
+		  if ! hostpath.file?
+			errors << 'vagrant_openvz.sudo_wrapper_not_found'
+		  elsif ! hostpath.executable?
+			errors << 'vagrant_openvz.sudo_wrapper_not_executable'
+		  end
+		end
+
+		{ "openvz provider" => errors }
+	  end
+	end
+  end
+end

data/lib/vagrant-openvz/driver.rb

@@ -0,0 +1,90 @@
+require "vagrant-openvz/errors"
+require "vagrant-openvz/driver/cli"
+require "etc"
+
+require "log4r"
+
+
+module VagrantPlugins
+  module Openvz
+	class Driver
+	  
+	  # This is raised if the container can't be found when initializing it with
+	  # a name.
+	  class ContainerNotFound < StandardError; end
+
+	  # Root folder where container configs are stored
+	  CONTAINERS_PATH = '/vz/root/' 
+
+	  attr_reader :container_name, :customizations
+
+	  def generate_guest_path(path,vzctlid)
+        Pathname.new("#{CONTAINERS_PATH}#{vzctlid}").join(path.gsub(/^\//, ''))
+	  end
+
+	  def initialize(container_name, sudo_wrapper, cli = nil)
+        @logger    = Log4r::Logger.new("vagrant::provider::openvz")
+		@container_name = container_name
+        @logger.debug("In Driver::initialize, @container_name=#{@container_name}")
+		@sudo_wrapper   = sudo_wrapper
+		@cli            = cli || CLI.new(sudo_wrapper, container_name)
+		# @logger         = Log4r::Logger.new("vagrant::provider::lxc::driver")
+		@customizations = []
+	  end
+
+	  def share_folders(folders,vzctlid)
+        folders.each do |folder|
+          guestpath = generate_guest_path(folder[:guestpath],vzctlid) 
+          @logger.debug("In Driver::share_folders, guestpath=#{guestpath}")
+          @logger.debug("In Driver::create, guestpath.directory?=#{guestpath.directory?}")
+		  unless guestpath.directory?
+            begin
+              @logger.debug("Guest path doesn't exist, creating: #{guestpath}")
+              @sudo_wrapper.run("mkdir", '-p', guestpath.to_s)
+            rescue Errno::EACCES
+              raise Vagrant::Errors::SharedFolderCreateFailed, :path => guestpath.to_s
+            end
+          end
+          @cli.share_folder(folder[:hostpath],guestpath)
+        end
+      end
+
+	  def validate!
+		raise ContainerNotFound if @container_name && ! @cli.list.include?(@container_name)
+	  end
+
+	  def create(name,vzctlid,settings={})
+        @logger.debug("In Driver::create, parameter name=#{name}")
+		@cli.name = @container_name = name
+		@logger.debug("In Driver::create, @container_name=#{@container_name}")
+
+		@cli.create vzctlid, settings 
+	  end
+
+	  def start(vzctlid,pubkey)
+		@logger.info('Starting container...')
+		@cli.start(vzctlid,pubkey)
+	  end
+	  
+      def forced_halt(vzctlid)
+        @logger.info('Shutting down container...')
+		@cli.stop(vzctlid)
+	  end
+
+	  def destroy(vzctlid)
+        @cli.destroy(vzctlid)
+      end
+    
+	  def state(vzctlid)
+        if @container_name
+           @cli.status(vzctlid)
+        end
+      end
+
+	  def fetch_ip(vzctlid)
+		@cli.fetch_ip(vzctlid)
+	  end
+
+	end
+  end
+end

data/lib/vagrant-openvz/driver/cli.rb

@@ -0,0 +1,81 @@
+
+module VagrantPlugins
+  module Openvz
+	class Driver
+	  class CLI
+		attr_accessor :name
+
+		def initialize(sudo_wrapper, name = nil)
+		  @sudo_wrapper = sudo_wrapper
+		  @name         = name
+		end
+
+		def list
+		  run(:vzlist,"-a").split(/\s+/).uniq
+		end
+
+		def fetch_ip(vzctlid)
+		  run(:vzlist,"-a","-H","-t","-o","ip","#{vzctlid}")
+		end
+
+		def create(vzctlid,settings={})
+
+		  # run :rsync, settings[:box_location],settings[:template_location] 
+
+		  run :vzctl, 'create', vzctlid, '--ostemplate', settings[:template_name]
+
+		  settings.delete(:box_location)
+		  settings.delete(:template_location)
+		  settings.delete(:template_name)
+
+		  settings.each do |key,value|
+			run :vzctl,'set',vzctlid,"--#{key}","#{value}","--save"  
+		  end
+
+		end
+
+		def start(vzctlid,pubkey)
+		  run :vzctl, 'start', "#{vzctlid}"
+		  add_vagrant_user(vzctlid,pubkey)
+		end
+
+		def destroy(vzctlid)
+		  run :vzctl, 'destroy', "#{vzctlid}"
+		end
+
+		def stop(vzctlid)
+		  run :vzctl, 'stop',  "#{vzctlid}"
+		end
+
+		def status(vzctlid)
+		  if @name && run(:vzctl, 'status', "#{vzctlid}") =~ /[a-z]+\s[0-9]+\s([a-z]+)\s([a-z]+)\s([a-z]+)/i
+			status = "#{$1}_#{$2}_#{$3}".downcase.to_sym
+			status
+		  elsif @name
+			:unknown
+		  end
+		end
+
+		def share_folder(source,destination) 
+          run :mount, "-o", "bind", "#{source}", "#{destination}"
+		end
+
+		private
+
+		def add_vagrant_user(vzctlid,pubkey)
+		  run :vzctl, "exec",  "#{vzctlid}", "if [[ ! \`grep vagrant /etc/passwd\` ]]; then adduser vagrant; fi"
+		  run :vzctl, "exec",  "#{vzctlid}", "echo 'vagrant:vagrant' | chpasswd"
+		  run :vzctl, "exec",  "#{vzctlid}", "mkdir -p /home/vagrant/.ssh" 
+		  run :vzctl, "exec",  "#{vzctlid}", "echo '#{pubkey}' > /home/vagrant/.ssh/authorized_keys"
+		  run :vzctl, "exec",  "#{vzctlid}", "chown -R vagrant:vagrant /home/vagrant/.ssh"
+		  run :vzctl, "exec",  "#{vzctlid}", "echo 'vagrant    ALL=NOPASSWD:   ALL' > /etc/sudoers.d/vagrant"
+		  run :vzctl, "exec",  "#{vzctlid}", "sed -i -e 's/Defaults    requiretty/#Defaults requiretty/' /etc/sudoers"
+		end
+
+		def run(command, *args)
+		  @sudo_wrapper.run("#{command}", *args)
+		end
+	  end
+	end
+  end
+end