vagrant-vmware-desktop 0.0.1 → 3.0.0

data/lib/vagrant-vmware-desktop/action/package_vagrantfile.rb

@@ -0,0 +1,46 @@
+require 'vagrant/util/template_renderer'
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      class PackageVagrantfile
+        # For TemplateRenderer
+        include Vagrant::Util
+
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          @env = env
+          create_metadata
+          @app.call(env)
+        end
+
+        # This method creates the auto-generated Vagrantfile at the root of the
+        # box. This Vagrantfile contains the MAC address so that the user doesn't
+        # have to worry about it.
+        #
+        # @note This is deprecated as the base mac is no longer required to be set.
+        #       The method (and template) are preserved to easily allow enabling
+        #       this functionality in the future for a different purpose if requried.
+        def create_vagrantfile
+          mac_addresses = @env[:machine].provider.driver.read_mac_addresses
+          base_mac = mac_addresses[mac_addresses.keys.min]
+          File.open(File.join(@env["export.temp_dir"], "Vagrantfile"), "w") do |f|
+            f.write(TemplateRenderer.render("package_Vagrantfile", {
+              base_mac: base_mac
+            }))
+          end
+        end
+
+        # Creates a metadata.json file which includes provider information
+        def create_metadata
+          File.open(File.join(@env["export.temp_dir"], "metadata.json"), "w") do |f|
+            f.write({provider: "vmware_desktop"}.to_json)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/prepare_forwarded_port_collision_params.rb

@@ -0,0 +1,28 @@
+require "log4r"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This prepares some variables for the environment hash so that
+      # forwarded port collision detection works properly with VMware.
+      class PrepareForwardedPortCollisionParams
+        include Common
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::prepare_fp_collision")
+        end
+
+        def call(env)
+          # Get the forwared ports for all NAT devices and mark them as in use.
+          env[:port_collision_extra_in_use] = env[:machine].provider.driver.all_forwarded_ports
+
+          # Repair the forwarded port collisions
+          env[:port_collision_repair] = true
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/prepare_nfs_settings.rb

@@ -0,0 +1,43 @@
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      class PrepareNFSSettings
+        include Common
+
+        def initialize(app,env)
+          @app = app
+          @logger = Log4r::Logger.new("vagrant::action::vm::nfs")
+        end
+
+        def call(env)
+          @app.call(env)
+
+          env[:nfs_machine_ip] = env[:machine].provider.driver.read_ip(
+            env[:machine].provider_config.enable_vmrun_ip_lookup
+          )
+
+          # We just assume the machine IP is the first 3 octets
+          # with "1" for the last octet. Poor assumption but can
+          # be fixed later.
+          if env[:nfs_machine_ip]
+            host_ip = env[:nfs_machine_ip].split(".").map(&:to_i)
+            host_ip[3] = 1
+            env[:nfs_host_ip] = host_ip.join(".")
+          end
+
+          using_nfs = false
+          env[:machine].config.vm.synced_folders.each do |id, opts|
+            if opts[:nfs]
+              using_nfs = true
+              break
+            end
+          end
+
+          if using_nfs
+            raise Errors::NFSNoNetwork if !env[:nfs_machine_ip]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/prepare_synced_folder_cleanup.rb

@@ -0,0 +1,19 @@
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      class PrepareSyncedFolderCleanup
+        include Common
+
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:nfs_valid_ids] = env[:machine].provider.driver.read_running_vms
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/prune_forwarded_ports.rb

@@ -0,0 +1,30 @@
+require "log4r"
+
+require "vagrant-vmware-desktop/helper/lock"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This action prunes out forwarded ports that aren't in use any longer,
+      # usually by machines that don't exist anymore.
+      class PruneForwardedPorts
+        include Common
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::prune_forwarded_ports")
+        end
+
+        def call(env)
+          @logger.info("Pruning forwarded ports...")
+          Helper::Lock.lock(env[:machine], "vmware-network") do
+            env[:machine].provider.driver.prune_forwarded_ports
+          end
+
+          # Carry on
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/prune_nfs_exports.rb

@@ -0,0 +1,22 @@
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      class PruneNFSExports
+        include Common
+
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          if env[:host]
+            vms = env[:machine].provider.driver.read_running_vms
+            env[:host].nfs_prune(vms)
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/running.rb

@@ -0,0 +1,20 @@
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This can be used with "Call" built-in to check if the machine
+      # is running.
+      class Running
+        include Common
+
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          env[:result] = env[:machine].state.id == :running
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/set_display_name.rb

@@ -0,0 +1,37 @@
+require "log4r"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This class sets the default displayName for the VM.
+      class SetDisplayName
+        include Common
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::vmx_modify")
+        end
+
+        def call(env)
+          default_name = "#{env[:root_path].basename.to_s}: #{env[:machine].name}"
+
+          @logger.info("Setting the default display name: #{default_name}")
+          env[:machine].provider.driver.vmx_modify do |vmx|
+            # Delete any alternate casings
+            vmx.keys.each do |key|
+              if key.downcase == "displayname"
+                vmx.delete(key)
+              end
+            end
+
+            # Set the displayName
+            vmx["displayname"] = default_name
+          end
+
+          # Carry on
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/share_folders.rb

@@ -0,0 +1,97 @@
+require "log4r"
+
+require "vagrant/util/scoped_hash_override"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This takes the configured synced folders on the VM and shares them
+      # into the VMware guest.
+      class ShareFolders
+        include Common
+
+        include Vagrant::Util::ScopedHashOverride
+
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::shared_folders")
+        end
+
+        def call(env)
+          # Move on first so that the machine is powered on
+          @app.call(env)
+
+          shared_folders = {}
+          env[:machine].config.vm.synced_folders.each do |id, data|
+            data      = scoped_hash_override(data, :vmware)
+
+            # Ignore disabled shared folders
+            if data[:disabled]
+              @logger.info("Disabled shared folder, ignoring: #{id}")
+              next
+            end
+
+            # Ignore NFS shared folders as well
+            next if data[:nfs]
+
+            # Use it!
+            shared_folders[id] = data
+          end
+
+          if shared_folders.empty?
+            @logger.info("No shared folders. Doing nothing.")
+            return
+          end
+
+          # Verify that the machine can actually support sharing folders
+          # if we can.
+          if env[:machine].guest.capability?(:verify_vmware_hgfs)
+            @logger.info("Verifying HGFS is on the guest...")
+            if !env[:machine].guest.capability(:verify_vmware_hgfs)
+              raise Errors::GuestMissingHGFS
+            end
+          end
+
+          # Get the SSH info which we'll use later
+          ssh_info = env[:machine].ssh_info
+
+          # short guestpaths first, so we don't step on ourselves
+          shared_folders = shared_folders.dup.sort_by do |id, data|
+            if data[:guestpath]
+              data[:guestpath].length
+            else
+              # A long enough path to just do this at the end.
+              10000
+            end
+          end
+
+          @logger.info("Preparing shared folders with VMX...")
+          env[:ui].info I18n.t("hashicorp.vagrant_vmware_desktop.sharing_folders")
+          env[:machine].provider.driver.enable_shared_folders
+          shared_folders.each do |id, data|
+            id        = id.gsub('/', env[:machine].provider_config.shared_folder_special_char)
+            path      = File.expand_path(data[:hostpath], env[:root_path])
+            guestpath = data[:guestpath]
+
+            env[:ui].info(I18n.t("hashicorp.vagrant_vmware_desktop.sharing_folder_single",
+                                 hostpath: path,
+                                 guestpath: guestpath))
+
+            env[:machine].provider.driver.share_folder(id, path)
+
+            # Remove trailing slashes
+            guestpath = guestpath.gsub(/\/+$/, "")
+
+            # Calculate the owner and group
+            data[:owner] ||= ssh_info[:username]
+            data[:group] ||= ssh_info[:username]
+
+            # Mount it!
+            env[:machine].guest.capability(
+              :mount_vmware_shared_folder, id, data[:guestpath], data)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/snapshot_delete.rb

@@ -0,0 +1,26 @@
+require "log4r"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This snapshots the VMware machine.
+      class SnapshotDelete
+        include Common
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::snapshot_delete")
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t(
+            "hashicorp.vagrant_vmware_desktop.snapshot_deleting",
+            name: env[:snapshot_name]))
+          env[:machine].provider.driver.snapshot_delete(env[:snapshot_name])
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/snapshot_restore.rb

@@ -0,0 +1,26 @@
+require "log4r"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This snapshots the VMware machine.
+      class SnapshotRestore
+        include Common
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::snapshot_restore")
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t(
+            "hashicorp.vagrant_vmware_desktop.snapshot_restoring",
+            name: env[:snapshot_name]))
+          env[:machine].provider.driver.snapshot_revert(env[:snapshot_name])
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-vmware-desktop/action/snapshot_save.rb

@@ -0,0 +1,26 @@
+require "log4r"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This snapshots the VMware machine.
+      class SnapshotSave
+        include Common
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::snapshot_save")
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t(
+            "hashicorp.vagrant_vmware_desktop.snapshot_saving",
+            name: env[:snapshot_name]))
+          env[:machine].provider.driver.snapshot_take(env[:snapshot_name])
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,26 @@
+require "log4r"
+
+module HashiCorp
+  module VagrantVMwareDesktop
+    module Action
+      # This suspends the VMware machine.
+      class Suspend
+        include Common
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("hashicorp::provider::vmware::suspend")
+        end
+
+        def call(env)
+          if env[:machine].provider.state.id == :running
+            env[:ui].info(I18n.t("hashicorp.vagrant_vmware_desktop.suspending"))
+            env[:machine].provider.driver.suspend
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end