vagrant-libvirt 0.0.1

data/lib/vagrant-libvirt/action/message_already_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Libvirt
+    module Action
+      class MessageAlreadyCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_libvirt.already_created"))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/action/message_not_created.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module Libvirt
+    module Action
+      class MessageNotCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_libvirt.not_created"))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/action/read_ssh_info.rb

@@ -0,0 +1,51 @@
+require "log4r"
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+      # This action reads the SSH info for the machine and puts it into the
+      # `:machine_ssh_info` key in the environment.
+      class ReadSSHInfo
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_libvirt::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(
+            env[:libvirt_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+
+        def read_ssh_info(libvirt, machine)
+          return nil if machine.id.nil?
+
+          # Find the machine
+          server = libvirt.servers.get(machine.id)
+          if server.nil?
+            # The machine can't be found
+            @logger.info("Machine couldn't be found, assuming it got destroyed.")
+            machine.id = nil
+            return nil
+          end
+
+          # Get ip address of machine
+          ip_address = server.public_ip_address
+          ip_address = server.private_ip_address if ip_address == nil
+          return nil if ip_address == nil
+
+          # Return the info
+          # TODO: Some info should be configurable in Vagrantfile
+          return {
+            :host => ip_address,
+            :port => 22,
+            :username => 'root',
+          }
+        end
+ 
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/action/read_state.rb

@@ -0,0 +1,38 @@
+require "log4r"
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+      # This action reads the state of the machine and puts it in the
+      # `:machine_state_id` key in the environment.
+      class ReadState
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_libvirt::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env[:libvirt_compute], env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_state(libvirt, machine)
+          return :not_created if machine.id.nil?
+
+          # Find the machine
+          server = libvirt.servers.get(machine.id)
+          if server.nil? || [:"shutting-down", :terminated].include?(server.state.to_sym)
+            # The machine can't be found
+            @logger.info("Machine not found or terminated, assuming it got destroyed.")
+            machine.id = nil
+            return :not_created
+          end
+
+          # Return the state
+          return server.state.to_sym
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/action/set_name_of_domain.rb

@@ -0,0 +1,31 @@
+module VagrantPlugins
+  module Libvirt
+    module Action
+
+      # Setup name for domain and domain volumes.
+      class SetNameOfDomain
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:domain_name] = env[:root_path].basename.to_s.dup
+          env[:domain_name].gsub!(/[^-a-z0-9_]/i, "")
+          env[:domain_name] << "_#{Time.now.to_i}"
+
+          # Check if the domain name is not already taken
+          domain = Libvirt::Util::Collection.find_matching(
+            env[:libvirt_compute].servers.all, env[:domain_name])
+          if domain != nil
+            raise Vagrant::Errors::DomainNameExists,
+              :domain_name => env[:domain_name]
+          end
+
+          @app.call(env)
+        end
+      end
+
+    end
+  end
+end
+

data/lib/vagrant-libvirt/action/start_domain.rb

@@ -0,0 +1,27 @@
+require 'log4r'
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+
+      # Just start the domain.
+      class StartDomain
+        def initialize(app, env)
+          @logger = Log4r::Logger.new("vagrant_libvirt::action::start_domain")
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_libvirt.starting_domain"))
+
+          domain = env[:libvirt_compute].servers.get(env[:machine].id.to_s)
+          raise Errors::NoDomainError if domain == nil
+          domain.start
+
+          @app.call(env)
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant-libvirt/action/sync_folders.rb

@@ -0,0 +1,58 @@
+require "log4r"
+require "vagrant/util/subprocess"
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+      # This middleware uses `rsync` to sync the folders over to the
+      # libvirt domain.
+      class SyncFolders
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_libvirt::action::sync_folders")
+        end
+
+        def call(env)
+          @app.call(env)
+
+          ssh_info = env[:machine].ssh_info
+
+          env[:machine].config.vm.synced_folders.each do |id, data|
+            hostpath  = File.expand_path(data[:hostpath], env[:root_path])
+            guestpath = data[:guestpath]
+
+            # Make sure there is a trailing slash on the host path to
+            # avoid creating an additional directory with rsync
+            hostpath = "#{hostpath}/" if hostpath !~ /\/$/
+
+            env[:ui].info(I18n.t("vagrant_libvirt.rsync_folder",
+                                :hostpath => hostpath,
+                                :guestpath => guestpath))
+
+            # Create the guest path
+            env[:machine].communicate.sudo("mkdir -p '#{guestpath}'")
+            env[:machine].communicate.sudo(
+              "chown #{ssh_info[:username]} '#{guestpath}'")
+
+            # Rsync over to the guest path using the SSH info
+            command = [
+              "rsync", "--verbose", "--archive", "-z",
+              "--exclude", ".vagrant/",
+              "-e", "ssh -p #{ssh_info[:port]} -o StrictHostKeyChecking=no -i '#{ssh_info[:private_key_path]}'",
+              hostpath,
+              "#{ssh_info[:username]}@#{ssh_info[:host]}:#{guestpath}"]
+
+            r = Vagrant::Util::Subprocess.execute(*command)
+            if r.exit_code != 0
+              raise Errors::RsyncError,
+                :guestpath => guestpath,
+                :hostpath => hostpath,
+                :stderr => r.stderr
+            end
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/vagrant-libvirt/action/timed_provision.rb

@@ -0,0 +1,21 @@
+require "vagrant-libvirt/util/timer"
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+      # This is the same as the builtin provision except it times the
+      # provisioner runs.
+      class TimedProvision < Vagrant::Action::Builtin::Provision
+        def run_provisioner(env, p)
+          timer = Util::Timer.time do
+            super
+          end
+
+          env[:metrics] ||= {}
+          env[:metrics]["provisioner_times"] ||= []
+          env[:metrics]["provisioner_times"] << [p.class.to_s, timer]
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/action/wait_till_up.rb

@@ -0,0 +1,96 @@
+require 'log4r'
+require 'vagrant-libvirt/util/timer'
+require 'vagrant/util/retryable'
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+
+      # Wait till domain is started, till it obtains an IP address and is
+      # accessible via ssh.
+      class WaitTillUp
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @logger = Log4r::Logger.new("vagrant_libvirt::action::wait_till_up")
+          @app = app
+        end
+
+        def call(env)
+          # Initialize metrics if they haven't been
+          env[:metrics] ||= {}
+
+          # Get domain object
+          domain = env[:libvirt_compute].servers.get(env[:machine].id.to_s)
+          raise NoDomainError if domain == nil
+
+          # Wait for domain to obtain an ip address. Ip address is searched
+          # from arp table, either localy or remotely via ssh, if libvirt
+          # connection was done via ssh.
+          env[:ip_address] = nil
+          env[:metrics]["instance_ip_time"] = Util::Timer.time do
+            env[:ui].info(I18n.t("vagrant_libvirt.waiting_for_ip"))
+            retryable(:on => Fog::Errors::TimeoutError, :tries => 300) do
+              # If we're interrupted don't worry about waiting
+              next if env[:interrupted]
+
+              # Wait for domain to obtain an ip address
+              domain.wait_for(2) {
+                addresses.each_pair do |type, ip|
+                  env[:ip_address] = ip[0]
+                end
+                env[:ip_address] != nil
+              }
+            end
+          end
+          terminate(env) if env[:interrupted]
+          @logger.info("Got IP address #{env[:ip_address]}")
+          @logger.info("Time for getting IP: #{env[:metrics]["instance_ip_time"]}")
+
+          # Machine has ip address assigned, now wait till we are able to
+          # connect via ssh.
+          env[:metrics]["instance_ssh_time"] = Util::Timer.time do
+            env[:ui].info(I18n.t("vagrant_libvirt.waiting_for_ssh"))
+            retryable(:on => Fog::Errors::TimeoutError, :tries => 60) do
+              # If we're interrupted don't worry about waiting
+              next if env[:interrupted]
+
+              # Wait till we are able to connect via ssh.
+              while true
+                # If we're interrupted then just back out
+                break if env[:interrupted]
+                break if env[:machine].communicate.ready?
+                sleep 2
+              end            
+            end
+          end
+          terminate(env) if env[:interrupted]
+          @logger.info("Time for SSH ready: #{env[:metrics]["instance_ssh_time"]}")
+
+          # Booted and ready for use.
+          #env[:ui].info(I18n.t("vagrant_libvirt.ready"))
+          
+          @app.call(env)
+        end
+
+        def recover(env)
+          return if env["vagrant.error"].is_a?(Vagrant::Errors::VagrantError)
+
+          if env[:machine].provider.state.id != :not_created
+            # Undo the import
+            terminate(env)
+          end
+        end
+
+        def terminate(env)
+          destroy_env = env.dup
+          destroy_env.delete(:interrupted)
+          destroy_env[:config_validate] = false
+          destroy_env[:force_confirm_destroy] = true
+          env[:action_runner].run(Action.action_destroy, destroy_env)        
+        end
+      end
+    end
+  end
+end
+

data/lib/vagrant-libvirt/action.rb

@@ -0,0 +1,94 @@
+require 'vagrant/action/builder'
+
+module VagrantPlugins
+  module Libvirt
+    module Action
+      # Include the built-in modules so we can use them as top-level things.
+      include Vagrant::Action::Builtin
+
+      # This action is called to bring the box up from nothing.
+      def self.action_up
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectLibvirt
+          b.use Call, IsCreated do |env, b2|
+            if env[:result]
+              b2.use MessageAlreadyCreated
+              next
+            end
+
+            b2.use SetNameOfDomain
+            b2.use HandleStoragePool
+            b2.use HandleBoxImage
+            b2.use CreateDomainVolume
+            b2.use CreateDomain
+            b2.use CreateNetworkInterfaces
+          end
+
+          b.use TimedProvision
+          b.use StartDomain
+          b.use WaitTillUp
+          b.use SyncFolders
+        end
+      end
+
+      # This is the action that is primarily responsible for completely
+      # freeing the resources of the underlying virtual machine.
+      def self.action_destroy
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use Call, IsCreated do |env, b2|
+            if !env[:result]
+              b2.use MessageNotCreated
+              next
+            end
+
+            b2.use ConnectLibvirt
+            b2.use DestroyDomain
+          end
+        end
+      end
+
+      # This action is called to read the state of the machine. The resulting
+      # state is expected to be put into the `:machine_state_id` key.
+      def self.action_read_state
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectLibvirt
+          b.use ReadState
+        end
+      end
+
+      # This action is called to read the SSH info of the machine. The
+      # resulting state is expected to be put into the `:machine_ssh_info`
+      # key.
+      def self.action_read_ssh_info
+        Vagrant::Action::Builder.new.tap do |b|
+          b.use ConfigValidate
+          b.use ConnectLibvirt
+          b.use ReadSSHInfo
+        end
+      end
+
+      action_root = Pathname.new(File.expand_path("../action", __FILE__))
+      autoload :ConnectLibvirt, action_root.join("connect_libvirt")
+      autoload :IsCreated, action_root.join("is_created")
+      autoload :MessageAlreadyCreated, action_root.join("message_already_created")
+      autoload :MessageNotCreated, action_root.join("message_not_created")
+      autoload :HandleStoragePool, action_root.join("handle_storage_pool")
+      autoload :HandleBoxImage, action_root.join("handle_box_image")
+      autoload :SetNameOfDomain, action_root.join("set_name_of_domain")
+      autoload :CreateDomainVolume, action_root.join("create_domain_volume")
+      autoload :CreateDomain, action_root.join("create_domain")
+      autoload :CreateNetworkInterfaces, action_root.join("create_network_interfaces")
+      autoload :DestroyDomain, action_root.join("destroy_domain")
+      autoload :StartDomain, action_root.join("start_domain")
+      autoload :ReadState, action_root.join("read_state")
+      autoload :ReadSSHInfo, action_root.join("read_ssh_info")
+      autoload :TimedProvision, action_root.join("timed_provision")
+      autoload :WaitTillUp, action_root.join("wait_till_up")
+      autoload :SyncFolders, action_root.join("sync_folders")
+    end
+  end
+end
+

data/lib/vagrant-libvirt/config.rb

@@ -0,0 +1,48 @@
+require 'vagrant'
+
+module VagrantPlugins
+  module Libvirt
+    class Config < Vagrant.plugin('2', :config)
+      # A hypervisor name to access via Libvirt.
+      attr_accessor :driver
+
+      # The name of the server, where libvirtd is running.
+      attr_accessor :host
+
+      # If use ssh tunnel to connect to Libvirt.
+      attr_accessor :connect_via_ssh
+
+      # The username to access Libvirt.
+      attr_accessor :username
+
+      # Password for Libvirt connection.
+      attr_accessor :password
+
+      # Libvirt storage pool name, where box image and instance snapshots will
+      # be stored.
+      attr_accessor :storage_pool_name
+
+      def initialize
+        @driver            = UNSET_VALUE
+        @host              = UNSET_VALUE
+        @connect_via_ssh   = UNSET_VALUE
+        @username          = UNSET_VALUE
+        @password          = UNSET_VALUE
+        @storage_pool_name = UNSET_VALUE
+      end
+
+      def finalize!
+        @driver   = 'qemu' if @driver == UNSET_VALUE
+        @host     = nil if @host     == UNSET_VALUE
+        @connect_via_ssh = false if @connect_via_ssh == UNSET_VALUE
+        @username = nil if @username == UNSET_VALUE
+        @password = nil if @password == UNSET_VALUE
+        @storage_pool_name = 'default' if @storage_pool_name == UNSET_VALUE
+      end
+
+      def validate(machine)
+      end
+    end
+  end
+end
+

data/lib/vagrant-libvirt/errors.rb

@@ -0,0 +1,90 @@
+require 'vagrant'
+
+module VagrantPlugins
+  module Libvirt
+    module Errors
+      class VagrantLibvirtError < Vagrant::Errors::VagrantError
+        error_namespace("vagrant_libvirt.errors")
+      end
+
+      # Storage pools and volumes exceptions
+      class NoStoragePool < VagrantLibvirtError
+        error_key(:no_storage_pool)
+      end
+
+      class DomainVolumeExists < VagrantLibvirtError
+        error_key(:domain_volume_exists)
+      end
+
+      class NoDomainVolume < VagrantLibvirtError
+        error_key(:no_domain_volume)
+      end
+
+      class CreatingStoragePoolError < VagrantLibvirtError
+        error_key(:creating_storage_pool_error)
+      end
+
+      class ImageUploadError < VagrantLibvirtError
+        error_key(:image_upload_error_error)
+      end
+
+
+      # Box exceptions
+      class NoBoxVolume < VagrantLibvirtError
+        error_key(:no_box_volume)
+      end
+
+      class NoBoxVirtualSizeSet < VagrantLibvirtError
+        error_key(:no_box_virtual_size_error)
+      end
+
+      class NoBoxFormatSet < VagrantLibvirtError
+        error_key(:no_box_format_error)
+      end
+
+      class WrongBoxFormatSet < VagrantLibvirtError
+        error_key(:wrong_box_format_error)
+      end
+
+
+      # Fog libvirt exceptions
+      class FogLibvirtConnectionError < VagrantLibvirtError
+        error_key(:fog_libvirt_connection_error)
+      end
+
+      class FogCreateVolumeError < VagrantLibvirtError
+        error_key(:fog_create_volume_error)
+      end
+
+      class FogCreateDomainVolumeError < VagrantLibvirtError
+        error_key(:fog_create_domain_volume_error)
+      end
+
+      class FogCreateServerError < VagrantLibvirtError
+        error_key(:fog_create_server_error)
+      end
+
+
+      # Other exceptions
+      class InterfaceSlotNotAvailable < VagrantLibvirtError
+        error_key(:interface_slot_not_available)
+      end
+
+      class RsyncError < VagrantLibvirtError
+        error_key(:rsync_error)
+      end
+
+      class DomainNameExists < VagrantLibvirtError
+        error_key(:domain_name_exists_error)
+      end
+
+      class NoDomainError < VagrantLibvirtError
+        error_key(:no_domain_error)
+      end
+
+      class AttachDeviceError < VagrantLibvirtError
+        error_key(:attach_device_error)
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/plugin.rb

@@ -0,0 +1,77 @@
+begin
+  require 'vagrant'
+rescue LoadError
+  raise "The Vagrant Libvirt plugin must be run within Vagrant."
+end
+
+
+# This is a sanity check to make sure no one is attempting to install
+# this into an early Vagrant version.
+if Vagrant::VERSION < '1.1.0'
+  raise "The Vagrant Libvirt plugin is only compatible with Vagrant 1.1+"
+end
+
+module VagrantPlugins
+  module Libvirt
+    class Plugin < Vagrant.plugin('2')
+      name "libvirt"
+      description <<-DESC
+      Vagrant plugin to manage VMs in libvirt.
+      DESC
+
+
+      config("libvirt", :provider) do
+        require_relative "config"
+        Config
+      end
+
+      provider "libvirt" do
+        # Setup logging and i18n
+        setup_logging
+        setup_i18n
+
+        require_relative "provider"
+        Provider
+      end
+
+
+      # This initializes the internationalization strings.
+      def self.setup_i18n
+        I18n.load_path << File.expand_path("locales/en.yml", Libvirt.source_root)
+        I18n.reload!
+      end
+
+
+      # This sets up our log level to be whatever VAGRANT_LOG is.
+      def self.setup_logging
+        require "log4r"
+
+        level = nil
+        begin
+          level = Log4r.const_get(ENV["VAGRANT_LOG"].upcase)
+        rescue NameError
+          # This means that the logging constant wasn't found,
+          # which is fine. We just keep `level` as `nil`. But
+          # we tell the user.
+          level = nil
+        end
+
+        # Some constants, such as "true" resolve to booleans, so the
+        # above error checking doesn't catch it. This will check to make
+        # sure that the log level is an integer, as Log4r requires.
+        level = nil if !level.is_a?(Integer)
+
+        # Set the logging level on all "vagrant" namespaced
+        # logs as long as we have a valid level.
+        if level
+          logger = Log4r::Logger.new("vagrant_libvirt")
+          logger.outputters = Log4r::Outputter.stderr
+          logger.level = level
+          logger = nil
+        end
+      end
+
+    end
+  end
+end
+