vagrant-vcenter 0.0.2.pre.dev

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

@@ -0,0 +1,37 @@
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class reads the IP info for the VM that the Vagrant provider is
+      # managing using VMware Tools.
+      class ReadSSHInfo
+        # FIXME: More work needed here for vCenter logic (vApp, VM IPs, etc.)
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::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)
+          return nil if env[:machine].id.nil?
+
+          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)
+
+          @logger.debug("IP Address: #{vm.guest.ipAddress}")
+
+          { :host => vm.guest.ipAddress, :port => 22 }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,52 @@
+require 'log4r'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This Class read the power state of the VM that Vagrant is managing.
+      class ReadState
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::read_state')
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env)
+          @app.call env
+        end
+
+        def read_state(env)
+          # FIXME: this part needs some cleanup
+          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)
+
+          @logger.debug("Current power state: #{vm.runtime.powerState}")
+          vm_name = env[:machine].name
+
+          if env[:machine].id.nil?
+            @logger.info("VM [#{vm_name}] is not created yet")
+            return :not_created
+          end
+
+          if vm.runtime.powerState == 'poweredOff'
+            @logger.info("VM [#{vm_name}] is stopped")
+            return :stopped
+          elsif vm.runtime.powerState == 'poweredOn'
+            @logger.info("VM [#{vm_name}] is running")
+            return :running
+          elsif vm.runtime.powerState == 'suspended'
+            @logger.info("VM [#{vm_name}] is suspended")
+            return :suspended
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class resumes the VM when it's suspended.
+      class Resume
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::resume')
+        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 on VM...')
+          vm.PowerOnVM_Task.wait_for_completion
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+require 'i18n'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class suspends the VM when it's powered on.
+      class Suspend
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::suspend')
+        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('Suspending VM...')
+          vm.SuspendVM_Task.wait_for_completion
+
+          @app.call env
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,135 @@
+# The MIT License (MIT)
+# Copyright (c) 2013 Mitchell Hashimoto
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of  this software and associated documentation files (the "Software"), to
+# deal in  the Software without restriction, including without limitation the
+# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
+# sell copies of the Software, and to permit persons to whom the Software is
+# furnished to do  so, subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR  COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER  IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+# FROM, OUT OF OR IN  CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+# IN THE SOFTWARE.
+
+require 'log4r'
+require 'vagrant/util/subprocess'
+require 'vagrant/util/scoped_hash_override'
+require 'vagrant/util/which'
+
+module VagrantPlugins
+  module VCenter
+    module Action
+      # This class syncs Vagrant folders using RSYNC, this code has been ported
+      # from vagrant-aws (https://github.com/mitchellh/vagrant-aws)
+      class SyncFolders
+        include Vagrant::Util::ScopedHashOverride
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new('vagrant_vcenter::action::sync_folders')
+        end
+
+        def call(env)
+          @app.call(env)
+
+          ssh_info = env[:machine].ssh_info
+
+          unless Vagrant::Util::Which.which('rsync')
+            env[:ui].warn(I18n.t('vagrant_vcenter.sync.rsync_not_found_warning',
+                                 :side => 'host'))
+            return
+          end
+
+          if env[:machine].communicate.execute('which rsync',
+                                               :error_check => false) != 0
+            env[:ui].warn(I18n.t('vagrant_vcenter.sync.rsync_not_found_warning',
+                                 :side => 'guest'))
+            return
+          end
+
+          env[:machine].config.vm.synced_folders.each do |id, data|
+            data = scoped_hash_override(data, :vCenter)
+
+            # Ignore disabled shared folders
+            next if data[:disabled]
+
+            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 !~ /\/$/
+
+            # on windows rsync.exe requires cygdrive-style paths
+            if Vagrant::Util::Platform.windows?
+              hostpath = hostpath.gsub(/^(\w):/) { "/cygdrive/\1" }
+            end
+
+            env[:ui].info(I18n.t('vagrant_vcenter.sync.rsync_folder',
+                                 :hostpath => hostpath,
+                                 :guestpath => guestpath))
+
+            # Create the host path if it doesn't exist and option flag is set
+            if data[:create]
+              begin
+                FileUtils.mkdir_p(hostpath)
+              rescue => err
+                raise Errors::MkdirError,
+                      :hostpath => hostpath,
+                      :err => err
+              end
+            end
+
+            # Create the guest path
+            env[:machine].communicate.sudo("mkdir -p '#{guestpath}'")
+            env[:machine].communicate.sudo(
+              "chown -R #{ssh_info[:username]} '#{guestpath}'")
+
+            # collect rsync excludes specified :rsync_excludes=>['path1',...]
+            # in synced_folder options
+            excludes = ['.vagrant/', 'Vagrantfile',
+                        *Array(data[:rsync_excludes])].uniq
+
+            # Rsync over to the guest path using the SSH info
+            command = [
+              'rsync', '--verbose', '--archive', '-z',
+              *excludes.map { |e|['--exclude', e] }.flatten,
+              '-e', "ssh -p #{ssh_info[:port]} -o StrictHostKeyChecking=no " +
+              "#{ssh_key_options(ssh_info)}", hostpath,
+              "#{ssh_info[:username]}@#{ssh_info[:host]}:#{guestpath}"]
+
+            # we need to fix permissions when using rsync.exe on windows, see
+            # http://stackoverflow.com/questions/5798807/rsync-permission-
+            # denied-created-directories-have-no-permissions
+            if Vagrant::Util::Platform.windows?
+              command.insert(1, '--chmod', 'ugo=rwX')
+            end
+
+            r = Vagrant::Util::Subprocess.execute(*command)
+            if r.exit_code != 0
+              fail Errors::RsyncError,
+                   :guestpath => guestpath,
+                   :hostpath => hostpath,
+                   :stderr => r.stderr
+            end
+          end
+        end
+
+        private
+
+        def ssh_key_options(ssh_info)
+          # Ensure that `private_key_path` is an Array (for Vagrant < 1.4)
+          Array(ssh_info[:private_key_path]).map { |path| "-i '#{path}' " }.join
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/config.rb

@@ -0,0 +1,92 @@
+require 'vagrant'
+
+module VagrantPlugins
+  module VCenter
+    # Initialize Provider configuration parameters.
+    class Config < Vagrant.plugin('2', :config)
+      # login attributes
+
+      # The vCenter  hostname
+      #
+      # @return [String]
+      attr_accessor :hostname
+
+      # The username used to log in
+      #
+      # @return [String]
+      attr_accessor :username
+
+      # The password used to log in
+      #
+      # @return [String]
+      attr_accessor :password
+
+      # WIP on these
+
+      # Catalog Name where the item resides
+      #
+      # @return [String]
+      attr_accessor :folder_name
+
+      # Catalog Name where the item resides
+      #
+      # @return [String]
+      attr_accessor :template_folder_name
+
+      # Catalog Item to be used as a template
+      #
+      # @return [String]
+      attr_accessor :datastore_name
+
+      # Virtual Data Center to be used
+      #
+      # @return [String]
+      attr_accessor :datacenter_name
+
+      # Virtual Data Center to be used
+      #
+      # @return [String]
+      attr_accessor :computer_name
+
+      # Virtual Data Center Network to be used
+      #
+      # @return [String]
+      attr_accessor :network_name
+
+      # Virtual Data Center Network to be used
+      #
+      # @return [Bool]
+      attr_accessor :linked_clones
+
+      ##
+      ## vCenter  config runtime values
+      ##
+
+      # connection handle
+      attr_accessor :vcenter_cnx
+      attr_accessor :template_id
+
+      def validate(machine)
+        errors = _detected_errors
+
+        # TODO: add blank?
+        errors <<
+        I18n.t('vagrant_vcenter.config.hostname') if hostname.nil?
+        errors <<
+        I18n.t('vagrant_vcenter.config.username') if username.nil?
+        errors <<
+        I18n.t('vagrant_vcenter.config.password') if password.nil?
+        errors <<
+        I18n.t('vagrant_vcenter.config.datastore_name') if datastore_name.nil?
+        errors <<
+        I18n.t('vagrant_vcenter.config.datacenter_name') if datacenter_name.nil?
+        errors <<
+        I18n.t('vagrant_vcenter.config.computer_name') if computer_name.nil?
+        errors <<
+        I18n.t('vagrant_vcenter.config.network_name') if network_name.nil?
+
+        { 'vCenter Provider' => errors }
+      end
+    end
+  end
+end

data/lib/vagrant-vcenter/errors.rb

@@ -0,0 +1,88 @@
+require 'vagrant'
+
+module VagrantPlugins
+  module VCenter
+    module Errors
+      # Initialize main error class.
+      class VCenterError < Vagrant::Errors::VagrantError
+        error_namespace('vagrant_vcenter.errors')
+      end
+      # Set key for Rsync errors.
+      class RsyncError < VCenterError
+        error_key(:rsync_error)
+      end
+      # Set key for Mkdir errors.
+      class MkdirError < VCenterError
+        error_key(:mkdir_error)
+      end
+      # Set key for VCenterOldVersion errors.
+      class VCenterOldVersion < VCenterError
+        error_key(:vcenter_old_version)
+      end
+      # Set key for CatalogAddError errors.
+      class CatalogAddError < VCenterError
+        error_key(:catalog_add_error)
+      end
+      # Set key for HostNotFound errors.
+      class HostNotFound < VCenterError
+        error_key(:host_not_found)
+      end
+      # Set key for HostRedirect errors.
+      class HostRedirect < VCenterError
+        error_key(:host_redirect)
+      end
+      # Set key for UnauthorizedAccess errors.
+      class UnauthorizedAccess < VCenterError
+        error_key(:unauthorized_access)
+      end
+      # Set key for StopVAppError errors.
+      class StopVAppError < VCenterError
+        error_key(:stop_vapp_error)
+      end
+      # Set key for ComposeVAppError errors.
+      class ComposeVAppError < VCenterError
+        error_key(:compose_vapp_error)
+      end
+      # Set key for InvalidNetSpecification errors.
+      class InvalidNetSpecification < VCenterError
+        error_key(:invalid_network_specification)
+      end
+      # Set key for ForwardPortCollision errors.
+      class ForwardPortCollision < VCenterError
+        error_key(:forward_port_collision)
+      end
+      # Set key for SubnetErrors errors.
+      class SubnetErrors < VCenterError
+        error_namespace('vagrant_Vcenter.errors.subnet_errors')
+      end
+      # Set key for InvalidSubnet errors.
+      class InvalidSubnet < SubnetErrors
+        error_key(:invalid_subnet)
+      end
+      # Set key for SubnetTooSmall errors.
+      class SubnetTooSmall < SubnetErrors
+        error_key(:subnet_too_small)
+      end
+      # Set key for RestError errors.
+      class RestError < VCenterError
+        error_namespace('vagrant_Vcenter.errors.rest_errors')
+      end
+      # Set key for ObjectNotFound errors.
+      class ObjectNotFound < RestError
+        error_key(:object_not_found)
+      end
+      # Set key for InvalidConfigError errors.
+      class InvalidConfigError < RestError
+        error_key(:invalid_config_error)
+      end
+      # Set key for InvalidStateError errors.
+      class InvalidStateError < RestError
+        error_key(:invalid_state_error)
+      end
+      # Set key for SyncError errors.
+      class SyncError < VCenterError
+        error_key(:sync_error)
+      end
+    end
+  end
+end