vagrant-vcenter 0.0.2.pre.dev

data/lib/vagrant-vcenter/action/connect_vcenter.rb

@@ -0,0 +1,37 @@
+require 'rbvmomi'
+require 'log4r'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class connects the vagrant-vcenter provider to vCenter.
+      class ConnectvCenter
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new(
+                      'vagrant_vcenter::action::connect_vCenter')
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+          # Avoid recreating a new session each time.
+          unless config.vcenter_cnx
+            @logger.info('Connecting to vCenter...')
+
+            @logger.debug("config.hostname: #{config.hostname}")
+            @logger.debug("config.username: #{config.username}")
+            @logger.debug("config.password: #{config.password}")
+
+            # FIXME: fix the insecure flag, catch the exception
+            config.vcenter_cnx = RbVmomi::VIM.connect(
+                                  host: config.hostname,
+                                  user: config.username,
+                                  password: config.password,
+                                  insecure: true)
+          end
+          @app.call env
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class destroy the VM created by the Vagrant provider.
+      class Destroy
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::destroy')
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+          # FIXME: Raise a correct exception
+          dc = config.vcenter_cnx.serviceInstance.find_datacenter(
+               config.datacenter_name) or abort 'datacenter not found'
+          root_vm_folder = dc.vmFolder
+          vm = root_vm_folder.findByUuid(env[:machine].id)
+
+          # Poweron VM
+          env[:ui].info('Destroying VM...')
+          vm.Destroy_Task.wait_for_completion
+          env[:machine].id = nil
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/disconnect_vcenter.rb

@@ -0,0 +1,29 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class disconnects the vagrant-vcenter provider from vCenter.
+      class DisconnectvCenter
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new(
+                    'vagrant_vcenter::action::disconnect_vcenter')
+        end
+
+        def call(env)
+          @logger.info('Disconnecting from vCenter ...')
+
+          config = env[:machine].provider_config
+
+          if !config.vcenter_cnx
+            @logger.info('Session not open, impossible to disconnect')
+          else
+            config.vcenter_cnx.close
+            @logger.info('Succesfully disconnected from vCenter...')
+          end
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/inventory_check.rb

@@ -0,0 +1,132 @@
+require 'etc'
+require 'log4r'
+require 'rbvmomi'
+require 'rbvmomi/utils/deploy'
+require 'rbvmomi/utils/admission_control'
+require 'yaml'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This Class inspect the vCenter inventory for templates and uploads it
+      # if needed.
+      class InventoryCheck
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new(
+                    'vagrant_vcenter::action::inventory_check')
+        end
+
+        def call(env)
+          vcenter_check_inventory(env)
+          @app.call env
+        end
+
+        def vcenter_upload_box(env)
+          config = env[:machine].provider_config
+
+          box_dir = env[:machine].box.directory.to_s
+          box_file = env[:machine].box.name.to_s
+
+          box_ovf = "file://#{box_dir}/#{box_file}.ovf"
+
+          # Still relying on ruby-progressbar because report_progress basically
+          # sucks.
+
+          @logger.debug("OVF File: #{box_ovf}")
+
+          env[:ui].info("Adding [#{env[:machine].box.name.to_s}]")
+
+          # FIXME: Raise a correct exception
+          dc = config.vcenter_cnx.serviceInstance.find_datacenter(
+               config.datacenter_name) or fail 'datacenter not found'
+
+          root_vm_folder = dc.vmFolder
+          vm_folder = root_vm_folder
+          if config.template_folder_name.nil?
+            vm_folder = root_vm_folder.traverse(config.template_folder_name,
+                                                RbVmomi::VIM::Folder)
+          end
+          template_folder = root_vm_folder.traverse!(
+                            config.template_folder_name,
+                            RbVmomi::VIM::Folder)
+
+          template_name = box_file
+
+          # FIXME: Raise a correct exception
+          datastore = dc.find_datastore(
+                      config.datastore_name) or fail 'datastore not found'
+          # FIXME: Raise a correct exception
+          computer = dc.find_compute_resource(
+                      config.computer_name) or fail 'Host not found'
+
+          network = computer.network.find { |x| x.name == config.network_name }
+
+          deployer = CachedOvfDeployer.new(
+            config.vcenter_cnx,
+            network,
+            computer,
+            template_folder,
+            vm_folder,
+            datastore
+          )
+
+          deployer_opts = {
+            :run_without_interruptions => true,
+            :simple_vm_name => true
+          }
+
+          deployer.upload_ovf_as_template(
+                                          box_ovf,
+                                          template_name,
+                                          deployer_opts)
+          # FIXME: Progressbar??
+        end
+
+        def vcenter_check_inventory(env)
+          # Will check each mandatory config value against the vcenter
+          # Instance and will setup the global environment config values
+          config = env[:machine].provider_config
+          # FIXME: Raise a correct exception
+          dc = config.vcenter_cnx.serviceInstance.find_datacenter(
+               config.datacenter_name) or fail 'datacenter not found'
+
+          if config.template_folder_name.nil?
+            box_to_search = env[:machine].box.name.to_s
+          else
+            box_to_search = config.template_folder_name +
+                            '/' + env[:machine].box.name.to_s
+          end
+
+          @logger.debug("This is the box we're looking for: #{box_to_search}")
+
+          config.template_id = dc.find_vm(box_to_search)
+
+          if config.template_id.nil?
+            env[:ui].warn("Template [#{env[:machine].box.name.to_s}] " +
+                          'does not exist!')
+
+            user_input = env[:ui].ask(
+              "Would you like to upload the [#{env[:machine].box.name.to_s}]" +
+              " box?\nChoice (yes/no): "
+            )
+
+            if user_input.downcase == 'yes' || user_input.downcase == 'y'
+              env[:ui].info("Uploading [#{env[:machine].box.name.to_s}]...")
+              vcenter_upload_box(env)
+            else
+              env[:ui].error('Template not uploaded, exiting...')
+
+              # FIXME: wrong error message
+              fail VagrantPlugins::VCenter::Errors::VCenterError,
+                   :message => 'Catalog not available, exiting...'
+
+            end
+          else
+            @logger.debug("Template found at #{box_to_search}")
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,26 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class verifies if the VM has been created.
+      class IsCreated
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::is_created')
+        end
+
+        def call(env)
+          vm_id = env[:machine].id
+          if vm_id
+            @logger.info("VM has been created and ID is: [#{vm_id}]")
+            env[:result] = true
+          else
+            @logger.warn('VM has not been created')
+            env[:result] = false
+          end
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/is_paused.rb

@@ -0,0 +1,23 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class sets the state to suspended if the VM is suspended.
+      class IsPaused
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          # Set the result to be true if the machine is suspended.
+          env[:result] = env[:machine].state.id == :suspended
+
+          # Call the next if we have one (but we shouldn't, since this
+          # middleware is built to run with the Call-type middlewares)
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class sets the state to running if the VM is running.
+      class IsRunning
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          # Set the result to be true if the machine is running.
+          env[:result] = env[:machine].state.id == :running
+
+          # Call the next if we have one (but we shouldn't, since this
+          # middleware is built to run with the Call-type middlewares)
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/message_already_running.rb

@@ -0,0 +1,17 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # Prints out a message that the VM is already running.
+      class MessageAlreadyRunning
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_vcenter.power.vm_already_running'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/message_cannot_suspend.rb

@@ -0,0 +1,19 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # Prints out a message that the VM is already halted and cannot be
+      # suspended
+      class MessageCannotSuspend
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t(
+                        'vagrant_vcenter.power.vm_halted_cannot_suspend'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,17 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # Prints out a message that the VM has not yet been created.
+      class MessageNotCreated
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_vcenter.power.vm_not_created'))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,18 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # Prints out a message that the VM will not be destroyed.
+      class MessageWillNotDestroy
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t('vagrant_vcenter.power.will_not_destroy',
+                               name: env[:machine].name))
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/power_off.rb

@@ -0,0 +1,30 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class powers off the VM that the Vagrant provider is managing.
+      class PowerOff
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::poweroff')
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+          # FIXME: Raise a correct exception
+          dc = config.vcenter_cnx.serviceInstance.find_datacenter(
+               config.datacenter_name) or abort 'datacenter not found'
+          root_vm_folder = dc.vmFolder
+          vm = root_vm_folder.findByUuid(env[:machine].id)
+
+          # Poweroff VM
+          env[:ui].info('Powering off VM...')
+          vm.PowerOffVM_Task.wait_for_completion
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/action/power_on.rb

@@ -0,0 +1,30 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class powers on the VM that the Vagrant provider is managing.
+      class PowerOn
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::power_on')
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+          # FIXME: Raise a correct exception
+          dc = config.vcenter_cnx.serviceInstance.find_datacenter(
+               config.datacenter_name) or abort 'datacenter not found'
+          root_vm_folder = dc.vmFolder
+          vm = root_vm_folder.findByUuid(env[:machine].id)
+
+          # Poweron VM
+          env[:ui].info('Powering on VM...')
+          vm.PowerOnVM_Task.wait_for_completion
+          sleep(3) until env[:machine].communicate.ready?
+          @app.call env
+        end
+      end
+    end
+  end
+end