vagrant-libvirt 0.0.41 → 0.0.42

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

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      # This can be used with "Call" built-in to check if the machine
+      # is created and branch in the middleware.
+      class IsCreated
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          env[:result] = env[:machine].state.id != :not_created
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      # This can be used with "Call" built-in to check if the machine
+      # is running and branch in the middleware.
+      class IsRunning
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          domain = env[:machine].provider.driver.connection.servers.get(env[:machine].id.to_s)
+          raise Errors::NoDomainError if domain.nil?
+          env[:result] = domain.state.to_s == 'running'
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      # This can be used with "Call" built-in to check if the machine
+      # is suspended and branch in the middleware.
+      class IsSuspended
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          domain = env[:machine].provider.driver.connection.servers.get(env[:machine].id.to_s)
+          raise Errors::NoDomainError if domain.nil?
+
+          config = env[:machine].provider_config
+          libvirt_domain = env[:machine].provider.driver.connection.client.lookup_domain_by_uuid(env[:machine].id)
+          if config.suspend_mode == 'managedsave'
+            if libvirt_domain.has_managed_save?
+              env[:result] = libvirt_domain.has_managed_save?
+            else
+              env[:result] = domain.state.to_s == 'paused'
+              if env[:result]
+                env[:ui].warn('One time switching to pause suspend mode, found a paused VM.')
+                config.suspend_mode = 'pause'
+              end
+            end
+          else
+            if libvirt_domain.has_managed_save?
+              env[:ui].warn('One time switching to managedsave suspend mode, state found.')
+              env[:result] = true
+              config.suspend_mode = 'managedsave'
+            else
+              env[:result] = domain.state.to_s == 'paused'
+            end
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    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 ProviderLibvirt
+    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/message_not_running.rb

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class MessageNotRunning
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_libvirt.not_running'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,16 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class MessageNotSuspended
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_libvirt.not_suspended'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,17 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class MessageWillNotDestroy
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info I18n.t("vagrant.commands.destroy.will_not_destroy",
+                              name: env[:machine].name)
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,105 @@
+require 'log4r'
+
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      # Action for create new box for libvirt provider
+      class PackageDomain
+        def initialize(app, env)
+          @logger = Log4r::Logger.new('vagrant_libvirt::action::package_domain')
+          @app = app
+          env['package.files'] ||= {}
+          env['package.output'] ||= 'package.box'
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_libvirt.package_domain'))
+          libvirt_domain = env[:machine].provider.driver.connection.client.lookup_domain_by_uuid(
+            env[:machine].id
+          )
+          domain = env[:machine].provider.driver.connection.servers.get(env[:machine].id.to_s)
+          root_disk = domain.volumes.select do |x|
+            x.name == libvirt_domain.name + '.img'
+          end.first
+          boxname = env['package.output']
+          raise "#{boxname}: Already exists" if File.exist?(boxname)
+          @tmp_dir = Dir.pwd + '/_tmp_package'
+          @tmp_img = @tmp_dir + '/box.img'
+          Dir.mkdir(@tmp_dir)
+          if File.readable?(root_disk.path)
+            backing = `qemu-img info "#{root_disk.path}" | grep 'backing file:' | cut -d ':' -f2`.chomp
+          else
+            env[:ui].error("Require set read access to #{root_disk.path}. sudo chmod a+r #{root_disk.path}")
+            FileUtils.rm_rf(@tmp_dir)
+            raise 'Have no access'
+          end
+          env[:ui].info('Image has backing image, copying image and rebasing ...')
+          FileUtils.cp(root_disk.path, @tmp_img)
+          `qemu-img rebase -p -b "" #{@tmp_img}`
+          # remove hw association with interface
+          # working for centos with lvs default disks
+          `virt-sysprep --no-logfile --operations defaults,-ssh-userdir -a #{@tmp_img}`
+          # add any user provided file
+          extra = ''
+          @tmp_include = @tmp_dir + '/_include'
+          if env['package.include']
+            extra = './_include'
+            Dir.mkdir(@tmp_include)
+            env['package.include'].each do |f|
+              env[:ui].info("Including user file: #{f}")
+              FileUtils.cp(f, @tmp_include)
+            end
+          end
+          if env['package.vagrantfile']
+            extra = './_include'
+            Dir.mkdir(@tmp_include) unless File.directory?(@tmp_include)
+            env[:ui].info('Including user Vagrantfile')
+            FileUtils.cp(env['package.vagrantfile'], @tmp_include + '/Vagrantfile')
+          end
+          Dir.chdir(@tmp_dir)
+          info = JSON.parse(`qemu-img info --output=json #{@tmp_img}`)
+          img_size = (Float(info['virtual-size'])/(1024**3)).ceil
+          File.write(@tmp_dir + '/metadata.json', metadata_content(img_size))
+          File.write(@tmp_dir + '/Vagrantfile', vagrantfile_content)
+          assemble_box(boxname, extra)
+          FileUtils.mv(@tmp_dir + '/' + boxname, '../' + boxname)
+          FileUtils.rm_rf(@tmp_dir)
+          env[:ui].info('Box created')
+          env[:ui].info('You can now add the box:')
+          env[:ui].info("vagrant box add #{boxname} --name any_comfortable_name")
+          @app.call(env)
+        end
+
+        def assemble_box(boxname, extra)
+          `tar cvzf "#{boxname}" --totals ./metadata.json ./Vagrantfile ./box.img #{extra}`
+        end
+
+        def vagrantfile_content
+          <<-EOF
+            Vagrant.configure("2") do |config|
+              config.vm.provider :libvirt do |libvirt|
+                libvirt.driver = "kvm"
+                libvirt.host = ""
+                libvirt.connect_via_ssh = false
+                libvirt.storage_pool_name = "default"
+              end
+            end
+
+            user_vagrantfile = File.expand_path('../_include/Vagrantfile', __FILE__)
+            load user_vagrantfile if File.exists?(user_vagrantfile)
+          EOF
+        end
+
+        def metadata_content(filesize)
+          <<-EOF
+            {
+              "provider": "libvirt",
+              "format": "qcow2",
+              "virtual_size": #{filesize}
+            }
+          EOF
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,94 @@
+require 'nokogiri'
+require 'socket'
+require 'timeout'
+
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class PrepareNFSSettings
+        include Vagrant::Action::Builtin::MixinSyncedFolders
+
+        def initialize(app, _env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant::action::vm::nfs')
+        end
+
+        def call(env)
+          @machine = env[:machine]
+          @app.call(env)
+
+          if using_nfs?
+            @logger.info('Using NFS, preparing NFS settings by reading host IP and machine IP')
+            env[:nfs_machine_ip] = read_machine_ip(env[:machine])
+            env[:nfs_host_ip]    = read_host_ip(env[:nfs_machine_ip])
+
+            @logger.info("host IP: #{env[:nfs_host_ip]} machine IP: #{env[:nfs_machine_ip]}")
+
+            raise Vagrant::Errors::NFSNoHostonlyNetwork if !env[:nfs_machine_ip] || !env[:nfs_host_ip]
+          end
+        end
+
+        # We're using NFS if we have any synced folder with NFS configured. If
+        # we are not using NFS we don't need to do the extra work to
+        # populate these fields in the environment.
+        def using_nfs?
+          !!synced_folders(@machine)[:nfs]
+        end
+
+        # Returns the IP address of the host
+        #
+        # @param [Machine] machine
+        # @return [String]
+        def read_host_ip(ip)
+          UDPSocket.open do |s|
+            if ip.is_a?(Array)
+              s.connect(ip.last, 1)
+            else
+              s.connect(ip, 1)
+            end
+            s.addr.last
+          end
+        end
+
+        # Returns the IP address of the guest
+        #
+        # @param [Machine] machine
+        # @return [String]
+        def read_machine_ip(machine)
+          # check host only ip
+          ssh_host = machine.ssh_info[:host]
+          return ssh_host if ping(ssh_host)
+
+          # check other ips
+          command = "ip=$(which ip); ${ip:-/sbin/ip} addr show | grep -i 'inet ' | grep -v '127.0.0.1' | tr -s ' ' | cut -d' ' -f3 | cut -d'/' -f 1"
+          result  = ''
+          machine.communicate.execute(command) do |type, data|
+            result << data if type == :stdout
+          end
+
+          ips = result.chomp.split("\n").uniq
+          @logger.info("guest IPs: #{ips.join(', ')}")
+          ips.each do |ip|
+            next if ip == ssh_host
+            return ip if ping(ip)
+          end
+        end
+
+        private
+
+        # Check if we can open a connection to the host
+        def ping(host, timeout = 3)
+          ::Timeout.timeout(timeout) do
+            s = TCPSocket.new(host, 'echo')
+            s.close
+          end
+          true
+        rescue Errno::ECONNREFUSED
+          true
+        rescue Timeout::Error, StandardError
+          false
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,17 @@
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class PrepareNFSValidIds
+        def initialize(app, _env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant::action::vm::nfs')
+        end
+
+        def call(env)
+          env[:nfs_valid_ids] = env[:machine].provider.driver.connection.servers.all.map(&:id)
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+require 'yaml'
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class PruneNFSExports
+        def initialize(app, _env)
+          @app = app
+        end
+
+        def call(env)
+          if env[:host]
+            uuid = env[:machine].id
+            # get all uuids
+            uuids = env[:machine].provider.driver.connection.servers.all.map(&:id)
+            # not exiisted in array will removed from nfs
+            uuids.delete(uuid)
+            env[:host].capability(
+              :nfs_prune, env[:machine].ui, uuids
+            )
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+require 'log4r'
+
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class ReadMacAddresses
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_libvirt::action::read_mac_addresses')
+        end
+
+        def call(env)
+          env[:machine_mac_addresses] = read_mac_addresses(env[:machine].provider.driver.connection, env[:machine])
+        end
+
+        def read_mac_addresses(libvirt, machine)
+          return nil if machine.id.nil?
+
+          domain = libvirt.client.lookup_domain_by_uuid(machine.id)
+
+          if domain.nil?
+            @logger.info('Machine could not be found, assuming it got destroyed')
+            machine.id = nil
+            return nil
+          end
+
+          xml = Nokogiri::XML(domain.xml_desc)
+          mac = xml.xpath('/domain/devices/interface/mac/@address')
+
+          return {} if mac.empty?
+
+          Hash[mac.each_with_index.map do |x, i|
+            @logger.debug("interface[#{i}] = #{x.value}")
+            [i, x.value]
+          end]
+        end
+      end
+    end
+  end
+end