vagrant-vmware-esxi 1.0.1

data/lib/vagrant-vmware-esxi/action/destroy.rb

@@ -0,0 +1,58 @@
+require 'log4r'
+require 'net/ssh/simple'
+
+module VagrantPlugins
+  module ESXi
+    module Action
+      # This action will Destroy VM. unregister and delete the VM from disk.
+      class Destroy
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_vmware_esxi::action::destroy')
+        end
+
+        def call(env)
+          destroy(env)
+          @app.call(env)
+        end
+
+        def destroy(env)
+          @logger.info('vagrant-vmware-esxi, destroy: start...')
+
+          # Get config.
+          machine = env[:machine]
+          config = env[:machine].provider_config
+
+          @logger.info("vagrant-vmware-esxi, destroy: machine id: #{machine.id}")
+          @logger.info('vagrant-vmware-esxi, destroy: current state: '\
+                       "#{env[:machine_state]}")
+
+          if env[:machine_state].to_s == 'not_created'
+            env[:ui].info I18n.t('vagrant_vmware_esxi.already_destroyed')
+          elsif env[:machine_state].to_s != 'powered_off'
+            raise Errors::ESXiError,
+                  message: 'Guest VM should have been powered off...'
+          else
+            Net::SSH::Simple.sync(
+              user:     config.esxi_username,
+              password: config.esxi_password,
+              port:     config.esxi_hostport,
+              keys:     config.esxi_private_keys
+            ) do
+
+              r = ssh config.esxi_hostname,
+                      "vim-cmd vmsvc/destroy #{machine.id}"
+              if r.exit_code != 0
+                raise Errors::ESXiError,
+                      message => "Unable to destroy the VM:\n"\
+                                 "  #{r.stdout}\n#{r.stderr}"
+              end
+              env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                   message: 'VM has been destroyed...')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/action/halt.rb

@@ -0,0 +1,52 @@
+require 'log4r'
+require 'net/ssh/simple'
+
+module VagrantPlugins
+  module ESXi
+    module Action
+      # This action will halt (power off) the VM
+      class Halt
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_vmware_esxi::action::halt')
+        end
+
+        def call(env)
+          halt(env)
+          @app.call(env)
+        end
+
+        def halt(env)
+          @logger.info('vagrant-vmware-esxi, halt: start...')
+
+          # Get config.
+          machine = env[:machine]
+          config = env[:machine].provider_config
+
+          if env[:machine_state].to_s == 'powered_off'
+            env[:ui].info I18n.t('vagrant_vmware_esxi.already_powered_off')
+          elsif env[:machine_state].to_s == 'not_created'
+            env[:ui].info I18n.t('vagrant_vmware_esxi.already_destroyed')
+          else
+            Net::SSH::Simple.sync(
+              user:     config.esxi_username,
+              password: config.esxi_password,
+              port:     config.esxi_hostport,
+              keys:     config.esxi_private_keys
+            ) do
+
+              r = ssh config.esxi_hostname,
+                      "vim-cmd vmsvc/power.off #{machine.id}"
+              if r.exit_code != 0
+                raise Errors::ESXiError,
+                      message: "Unable to power off the VM:\n"
+                               "  #{r.stdout}\n#{r.stderr}"
+              end
+              env[:ui].info I18n.t('vagrant_vmware_esxi.states.powered_off.short')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/action/read_ssh_info.rb

@@ -0,0 +1,63 @@
+require 'log4r'
+
+module VagrantPlugins
+  module ESXi
+    module Action
+      # This action will get the SSH "availability".
+      class ReadSSHInfo
+        def initialize(app, _env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vmware_esxi::action::read_ssh_info')
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = read_ssh_info(env)
+          @app.call(env)
+        end
+
+        def read_ssh_info(env)
+          @logger.info('vagrant-vmware-esxi, read_ssh_info: start...')
+
+          # Get config.
+          machine = env[:machine]
+          config = env[:machine].provider_config
+
+          return :not_created if machine.id.nil?
+
+          @logger.info("vagrant-vmware-esxi, read_ssh_info: machine id: #{machine.id}")
+          @logger.info('vagrant-vmware-esxi, read_ssh_info: current state:'\
+                       " #{env[:machine_state]}")
+
+          #  Figure out vm_ipaddress
+          Net::SSH::Simple.sync(
+            user:     config.esxi_username,
+            password: config.esxi_password,
+            port:     config.esxi_hostport,
+            keys:     config.esxi_private_keys
+          ) do
+
+            @logger = Log4r::Logger.new('vagrant_vmware_esxi::action::'\
+                                        'read_ssh_info-net_ssh')
+
+            #  ugly, but it works...
+            ssh_execute_cmd = "vim-cmd vmsvc/get.guest #{machine.id} |"
+            ssh_execute_cmd << 'grep -i "^   ipAddress"|'
+            ssh_execute_cmd << 'grep -oE "((1?[0-9][0-9]?|2[0-4][0-9]|25[0-5])\.){3}(1?[0-9][0-9]?|2[0-4][0-9]|25[0-5])"'
+            r = ssh config.esxi_hostname, ssh_execute_cmd
+
+            ipaddress = r.stdout.strip
+            @logger.info('vagrant-vmware-esxi, read_ssh_info: ipaddress: '\
+                         "#{ipaddress}")
+
+            return nil if (ipaddress == '') || (r.exit_code != 0)
+
+            return {
+              host: ipaddress,
+              port: 22
+            }
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/action/read_state.rb

@@ -0,0 +1,67 @@
+require 'log4r'
+
+module VagrantPlugins
+  module ESXi
+    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_vmware_esxi::action::read_state')
+        end
+
+        def call(env)
+          env[:machine_state] = read_state(env)
+          @app.call(env)
+        end
+
+        def read_state(env)
+          @logger.info('vagrant-vmware-esxi, read_state: start...')
+
+          # Get config.
+          machine = env[:machine]
+          config = env[:machine].provider_config
+
+          return :not_created if machine.id.to_i < 1
+
+          @logger.info("vagrant-vmware-esxi, read_state: machine id: #{machine.id}")
+          @logger.info("vagrant-vmware-esxi, read_state: current state: #{env[:machine_state]}")
+
+          Net::SSH::Simple.sync(
+            user:     config.esxi_username,
+            password: config.esxi_password,
+            port:     config.esxi_hostport,
+            keys:     config.esxi_private_keys
+          ) do
+
+            r = ssh config.esxi_hostname,
+                    "vim-cmd vmsvc/getallvms|grep -q \"^#{machine.id} \" && "\
+                    "vim-cmd vmsvc/power.getstate #{machine.id} || return 254"
+            power_status = r.stdout
+
+            return :not_created if r.exit_code == 254
+            if power_status == "" or r.exit_code != 0
+              raise Errors::ESXiError,
+                    message: 'Unable to get VM Power State!'
+            end
+
+            if (power_status.include? "Powered on") && !env[:machine].ssh_info.nil?
+              return :running
+            elsif power_status.include? "Powered on"
+              return :powered_on
+            elsif power_status.include? "Powered off"
+              return :powered_off
+            elsif power_status.include? "Suspended"
+              return :suspended
+            end
+
+            return nil
+          end
+
+          return :not_created
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/action/resume.rb

@@ -0,0 +1,63 @@
+require 'log4r'
+require 'net/ssh/simple'
+
+module VagrantPlugins
+  module ESXi
+    module Action
+      # This action will Resume (power on) the VM
+      class Resume
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_vmware_esxi::action::resume')
+        end
+
+        def call(env)
+          resume(env)
+          @app.call(env)
+        end
+
+        def resume(env)
+          @logger.info('vagrant-vmware-esxi, resume: start...')
+
+          # Get config.
+          machine = env[:machine]
+          config = env[:machine].provider_config
+
+          @logger.info("vagrant-vmware-esxi, resume: machine id: #{machine.id}")
+          @logger.info("vagrant-vmware-esxi, resume: current state: #{env[:machine_state]}")
+
+          if (env[:machine_state].to_s == 'powered_on') ||
+             (env[:machine_state].to_s == 'running')
+            env[:ui].info I18n.t('vagrant_vmware_esxi.already_powered_on')
+          elsif env[:machine_state].to_s == 'not_created'
+           env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                message: 'Cannot resume in this state')
+          elsif (env[:machine_state].to_s == 'powered_off') ||
+                (env[:machine_state].to_s == 'suspended')
+            env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                 message: 'Attempting to resume')
+
+            Net::SSH::Simple.sync(
+              user:     config.esxi_username,
+              password: config.esxi_password,
+              port:     config.esxi_hostport,
+              keys:     config.esxi_private_keys
+            ) do
+
+              r = ssh config.esxi_hostname, "vim-cmd vmsvc/power.on #{machine.id}"
+              if r.exit_code != 0
+                raise Errors::ESXiError,
+                      message: "Unable to resume the VM:\n"\
+                               "  #{r.stdout}\n#{r.stderr}"
+              end
+              env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                   message: 'VM has been resumed...')
+            end
+          else
+            raise Errors::ESXiError, message: 'Unknown state to resume...'
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/action/suspend.rb

@@ -0,0 +1,61 @@
+require 'log4r'
+require 'net/ssh/simple'
+
+module VagrantPlugins
+  module ESXi
+    module Action
+      # This action will Suspend the VM
+      class Suspend
+        def initialize(app, _env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_vmware_esxi::action::suspend')
+        end
+
+        def call(env)
+          suspend(env)
+          @app.call(env)
+        end
+
+        def suspend(env)
+          @logger.info('vagrant-vmware-esxi, suspend: start...')
+
+          # Get config.
+          machine = env[:machine]
+          config = env[:machine].provider_config
+
+          @logger.info("vagrant-vmware-esxi, suspend: machine id: #{machine.id}")
+          @logger.info("vagrant-vmware-esxi, suspend: current state: #{env[:machine_state]}")
+
+          if env[:machine_state].to_s == 'suspended'
+            env[:ui].info I18n.t('vagrant_vmware_esxi.already_suspended')
+          elsif (env[:machine_state].to_s == 'powered_off') ||
+                (env[:machine_state].to_s == 'not_created')
+           env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                message: 'Cannot suspend in this state')
+          else
+            env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                 message: 'Attempting to suspend')
+
+            Net::SSH::Simple.sync(
+              user:     config.esxi_username,
+              password: config.esxi_password,
+              port:     config.esxi_hostport,
+              keys:     config.esxi_private_keys,
+              timeout:  300
+            ) do
+
+              r = ssh config.esxi_hostname, "vim-cmd vmsvc/power.suspend #{machine.id}"
+              if r.exit_code != 0
+                raise Errors::ESXiError,
+                      message: "Unable to suspend the VM:\n"\
+                               "  #{r.stdout}\n#{r.stderr}"
+              end
+              env[:ui].info I18n.t('vagrant_vmware_esxi.support')
+              env[:ui].info I18n.t('vagrant_vmware_esxi.states.suspended.short')
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/action/wait_for_state.rb

@@ -0,0 +1,40 @@
+require 'log4r'
+require 'timeout'
+
+module VagrantPlugins
+  module ESXi
+    module Action
+      # This action will wait for a machine to reach a specific state or quit
+      # by timeout
+      class WaitForState
+        # env[:result] will be false in case of timeout.
+        # @param [Symbol] state Target machine state.
+        # @param [Number] timeout Timeout in seconds.
+        def initialize(app, _env, state, timeout)
+          @app     = app
+          @logger  = Log4r::Logger.new('vagrant_vmware_esxi::action::wait_for_state')
+          @state   = state
+          @timeout = timeout
+        end
+
+        def call(env)
+          env[:result] = 'True'
+          if env[:machine].state.id != @state
+            env[:ui].info I18n.t('vagrant_vmware_esxi.vagrant_vmware_esxi_message',
+                                 message: "Waiting for state \"#{@state}\"")
+            begin
+              Timeout.timeout(@timeout) do
+                until env[:machine].state.id == @state
+                  sleep 4
+                end
+              end
+            rescue Timeout::Error
+              env[:result] = 'False' # couldn't reach state in time
+            end
+          end
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-esxi/config.rb

@@ -0,0 +1,56 @@
+#  Config
+
+module VagrantPlugins
+  module ESXi
+    # Config class
+    class Config < Vagrant.plugin('2', :config)
+      attr_accessor :esxi_hostname
+      attr_accessor :esxi_hostport
+      attr_accessor :esxi_username
+      attr_accessor :esxi_password
+      attr_accessor :esxi_private_keys
+      attr_accessor :ssh_username
+      attr_accessor :private_key_path
+      attr_accessor :vmname
+      attr_accessor :vmname_prefix
+      attr_accessor :vm_disk_store
+      attr_accessor :virtual_network
+      attr_accessor :resource_pool
+      attr_accessor :memsize
+      attr_accessor :numvcpus
+      attr_accessor :custom_vmx_settings
+      attr_accessor :allow_overwrite
+      attr_accessor :system_private_keys_path
+      def initialize
+        @esxi_hostname = nil
+        @esxi_hostport = 22
+        @esxi_username = 'root'
+        @esxi_password = nil
+        @esxi_private_keys = UNSET_VALUE
+        @ssh_username = 'vagrant'
+        @private_key_path = UNSET_VALUE
+        @vmname = nil
+        @vmname_prefix = 'V-'
+        @vm_disk_store = nil
+        @virtual_network = nil
+        @resource_pool = nil
+        @memsize = UNSET_VALUE
+        @numvcpus = UNSET_VALUE
+        @custom_vmx_settings = UNSET_VALUE
+        @allow_overwrite = 'False'
+        @system_private_keys_path = [
+          '~/.ssh/id_rsa',
+          '~/.ssh/id_ecdsa',
+          '~/.ssh/id_ed25519',
+          '~/.ssh/id_dsa'
+        ]
+      end
+
+      def finalize!
+        @private_key_path = nil if @private_key_path == UNSET_VALUE
+        @ssh_username = nil if @ssh_username == UNSET_VALUE
+        @esxi_private_keys = @system_private_keys_path if @esxi_private_keys == UNSET_VALUE
+      end
+    end
+  end
+end