vagrant-openvz 0.0.2

data/lib/vagrant-openvz/errors.rb

@@ -0,0 +1,29 @@
+require 'vagrant/errors'
+
+module VagrantPlugins
+  module Openvz 
+    module Errors
+
+      class ExecuteError < Vagrant::Errors::VagrantError
+        error_key(:openvz_execute_error)
+      end
+
+      class OpenvzNotInstalled < Vagrant::Errors::VagrantError
+        error_key(:openvz_not_installed)
+      end
+
+      # Box related errors
+      class TemplateFileMissing < Vagrant::Errors::VagrantError
+        error_key(:openvz_template_file_missing)
+      end
+
+      class RootFSTarballMissing < Vagrant::Errors::VagrantError
+        error_key(:openvz_invalid_box_version)
+      end
+
+      class IncompatibleBox < Vagrant::Errors::VagrantError
+        error_key(:openvz_incompatible_box)
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/plugin.rb

@@ -0,0 +1,32 @@
+require "vagrant"
+
+module VagrantPlugins
+  module Openvz
+	class Plugin < Vagrant.plugin("2")
+
+	  name "vagrant-openvz"
+
+	  description <<-DESC
+	  This plugin installs a provider that allows vagrant to manage
+	  Openvz machine instances.
+	  DESC
+	  
+	  config(:openvz, :provider) do
+		require_relative 'config'
+		Config
+	  end
+
+	  command "vzctl-version" do
+		require_relative "command/version"
+		VzctlVersion	
+	  end
+
+	  provider(:openvz) do
+		require_relative "provider"
+		Provider
+	  end
+
+	end
+  end
+end
+

data/lib/vagrant-openvz/provider.rb

@@ -0,0 +1,108 @@
+begin
+  require "vagrant"
+rescue LoadError
+  raise "This plugin must run within Vagrant."
+end
+
+require "log4r"
+
+require "vagrant-openvz/action"
+require "vagrant-openvz/driver"
+require "vagrant-openvz/sudo_wrapper"
+
+module VagrantPlugins
+  module Openvz
+	class Provider < Vagrant.plugin("2",:provider)
+	  attr_reader :driver
+
+	  def initialize(machine)
+        @logger    = Log4r::Logger.new("vagrant::provider::openvz")
+		@machine = machine
+
+		ensure_openvz_installed!
+		machine_id_changed
+	  end
+
+	  def sudo_wrapper
+		@shell ||= begin
+	      wrapper = @machine.provider_config.sudo_wrapper
+	      wrapper = Pathname(wrapper).expand_path(@machine.env.root_path).to_s if wrapper
+		  SudoWrapper.new(wrapper)
+		end
+	  end
+
+	  def action(name)
+		# Attempt to get the action method from the Action class if it
+		# exists, otherwise return nil to show that we don't support the
+		# given action.
+		action_method = "action_#{name}"
+		return Action.send(action_method) if Action.respond_to?(action_method)
+		nil
+	  end
+
+	  def ensure_openvz_installed!
+		unless system("which vzctl > /dev/null")
+		  raise Errors::OpenvzNotInstalled
+		end
+	  end
+
+	  # If the machine ID changed, then we need to rebuild our underlying
+	  # container.
+	  def machine_id_changed
+		id = @machine.id
+        @logger.debug("In Provider::machine_id_changed, id=#{id}")
+
+		begin
+          @logger.debug("Instantiating the container for: #{id.inspect}")
+
+		  @driver = Driver.new(id, self.sudo_wrapper)
+		  @driver.validate!
+		rescue Driver::ContainerNotFound
+		  # The container doesn't exist, so we probably have a stale
+		  # ID. Just clear the id out of the machine and reload it.
+		  @logger.debug("Container not found! Clearing saved machine ID and reloading.")
+		  id = nil
+		  retry
+		end
+	  end
+ 
+	  # Returns the SSH info for accessing the Container.
+      def ssh_info
+        # If the Container is not created then we cannot possibly SSH into it, so
+        # we return nil.
+        return nil if state == :not_created
+
+        # Run a custom action called "fetch_ip" which does what it says and puts
+        # the IP found into the `:machine_ip` key in the environment.
+        env = @machine.action("fetch_ip")
+
+        # If we were not able to identify the container's IP, we return nil
+        # here and we let Vagrant core deal with it ;)
+        return nil unless env[:machine_ip]
+
+        {
+          :host => env[:machine_ip],
+          :port => @machine.config.ssh.guest_port
+        }
+      end
+
+	  def state
+        @logger.debug("In Provider::state, @driver.container_name=#{@driver.container_name}")
+		@logger.debug("In Provider::state, @machine.provider_config.vzctid=#{@machine.provider_config.vzctid}")
+
+        state_id = nil
+
+        state_id = :not_created if !@driver.container_name
+        state_id = @driver.state(@machine.provider_config.vzctid) if !state_id
+        state_id = :unknown if !state_id
+
+        short = state_id.to_s.gsub("_", " ")
+        long  = state_id.to_s.gsub("_", " ")
+
+		@logger.debug("In Provider::state, state_id=#{state_id}")
+        Vagrant::MachineState.new(state_id, short, long)
+      end
+
+	end
+  end
+end

data/lib/vagrant-openvz/sudo_wrapper.rb

@@ -0,0 +1,80 @@
+module VagrantPlugins
+  module Openvz
+    class SudoWrapper
+
+      # Include this so we can use `Subprocess` more easily.
+      include Vagrant::Util::Retryable
+
+      def initialize(wrapper_path = nil)
+        @wrapper_path = wrapper_path
+        # @logger       = Log4r::Logger.new("vagrant::lxc::shell")
+      end
+
+      def run(*command)
+        command.unshift @wrapper_path if @wrapper_path
+        execute *(['sudo'] + command)
+      end
+
+      def su_c(command)
+        su_command = if @wrapper_path
+            "#{@wrapper_path} \"#{command}\""
+          else
+            "su root -c \"#{command}\""
+          end
+        @logger.debug "Running 'sudo #{su_command}'"
+        system "sudo #{su_command}"
+      end
+
+      private
+
+      # TODO: Review code below this line, it was pretty much a copy and
+      #       paste from VirtualBox base driver and has no tests
+      def execute(*command, &block)
+        # Get the options hash if it exists
+        opts = {}
+        opts = command.pop if command.last.is_a?(Hash)
+
+        tries = 0
+        tries = 3 if opts[:retryable]
+
+        sleep = opts.fetch(:sleep, 1)
+
+        # Variable to store our execution result
+        r = nil
+
+        retryable(:on => Openvz::Errors::ExecuteError, :tries => tries, :sleep => sleep) do
+          # Execute the command
+          r = raw(*command, &block)
+
+          # If the command was a failure, then raise an exception that is
+          # nicely handled by Vagrant.
+          if r.exit_code != 0
+            if @interrupted
+              @logger.info("Exit code != 0, but interrupted. Ignoring.")
+            else
+              raise Openvz::Errors::ExecuteError, :command => command.inspect
+            end
+          end
+        end
+
+        # Return the output, making sure to replace any Windows-style
+        # newlines with Unix-style.
+        r.stdout.gsub("\r\n", "\n")
+      end
+
+      def raw(*command, &block)
+        int_callback = lambda do
+          @interrupted = true
+          @logger.info("Interrupted.")
+        end
+
+        # Append in the options for subprocess
+        command << { :notify => [:stdout, :stderr] }
+
+        Vagrant::Util::Busy.busy(int_callback) do
+          Vagrant::Util::Subprocess.execute(*command, &block)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-openvz/version.rb

@@ -0,0 +1,5 @@
+module VagrantPlugins
+  module Openvz
+	VERSION = "0.0.2"
+  end
+end

data/vagrant-openvz.gemspec

@@ -0,0 +1,23 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'vagrant-openvz/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "vagrant-openvz"
+  spec.version       = VagrantPlugins::Openvz::VERSION
+  spec.authors       = ["Aslan Brooke"]
+  spec.email         = ["aslandbrooke@yahoo.com"]
+  spec.description   = %q{Vagrant plugin for using OpenVZ as a provider.}
+  spec.summary       = %q{Vagrant plugin for using OpenVZ as a provider.}
+  spec.homepage      = "https://github.com/abrooke-forks/vagrant-openvz"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+end

data/vg.sh

@@ -0,0 +1,4 @@
+#!/bin/bash
+export VAGRANT_DEFAULT_PROVIDER=openvz
+export VAGRANT_LOG=DEBUG
+/usr/bin/time bundle exec vagrant $*

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification
+name: vagrant-openvz
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Aslan Brooke
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-10-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Vagrant plugin for using OpenVZ as a provider.
+email:
+- aslandbrooke@yahoo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- Vagrantfile
+- box/centos-6-x86_64/VagrantFile
+- box/centos-6-x86_64/centos-6-x86_64.box
+- box/centos-6-x86_64/make_box.sh
+- box/centos-6-x86_64/metadata.json
+- lib/vagrant-openvz.rb
+- lib/vagrant-openvz/action.rb
+- lib/vagrant-openvz/action/boot.rb
+- lib/vagrant-openvz/action/check_created.rb
+- lib/vagrant-openvz/action/check_running.rb
+- lib/vagrant-openvz/action/create.rb
+- lib/vagrant-openvz/action/created.rb
+- lib/vagrant-openvz/action/destroy.rb
+- lib/vagrant-openvz/action/destroy_confirm.rb
+- lib/vagrant-openvz/action/fetch_ip.rb
+- lib/vagrant-openvz/action/forced_halt.rb
+- lib/vagrant-openvz/action/is_running.rb
+- lib/vagrant-openvz/action/message.rb
+- lib/vagrant-openvz/action/share_folders.rb
+- lib/vagrant-openvz/command/version.rb
+- lib/vagrant-openvz/config.rb
+- lib/vagrant-openvz/driver.rb
+- lib/vagrant-openvz/driver/cli.rb
+- lib/vagrant-openvz/errors.rb
+- lib/vagrant-openvz/plugin.rb
+- lib/vagrant-openvz/provider.rb
+- lib/vagrant-openvz/sudo_wrapper.rb
+- lib/vagrant-openvz/version.rb
+- vagrant-openvz.gemspec
+- vg.sh
+homepage: https://github.com/abrooke-forks/vagrant-openvz
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Vagrant plugin for using OpenVZ as a provider.
+test_files: []