vagrant-skytap 0.1.1a

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

@@ -0,0 +1,46 @@
+module VagrantPlugins
+  module Skytap
+    module Action
+      class PrepareNFSSettings
+        attr_reader :env, :machine
+
+        def initialize(app,env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::prepare_nfs_settings")
+        end
+
+        def call(env)
+          @machine = env[:machine]
+
+          if using_nfs?
+            env[:nfs_host_ip] = read_host_ip
+            env[:nfs_machine_ip] = read_machine_ip
+          end
+
+          @app.call(env)
+        end
+
+        def using_nfs?
+          @machine.config.vm.synced_folders.any? { |_, opts| opts[:type] == :nfs }
+        end
+
+        # Returns the IP address of the host, preferring one on an interface which
+        # the client can route to.
+        def read_host_ip
+          UDPSocket.open do |s|
+            s.connect(machine.ssh_info[:host], 1)
+            s.addr.last
+          end.tap do |ret|
+            @logger.debug "read_host_ip returning #{ret}"
+          end
+        end
+
+        def read_machine_ip
+          machine.ssh_info[:host]
+        end
+      end
+    end
+  end
+end
+

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

@@ -0,0 +1,28 @@
+require 'log4r'
+require_relative "mixin_machine_index"
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Creates a list of vm ids from all Skytap VMs in Vagrant's global machine
+      # index. Any Vagrant machine not on this list will have its NFS entry
+      # pruned by the SyncedFolderCleanup action.
+      # NOTE: Unfortunately, Vagrant providers have no way to know about each
+      # other's machines. https://github.com/mitchellh/vagrant/issues/6439
+      class PrepareNFSValidIds
+        include MixinMachineIndex
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::prepare_nfs_valid_ids")
+        end
+
+        def call(env)
+          env[:nfs_valid_ids] = machine_index.collect{|entry| entry.extra_data['vm_id']}.compact
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+require "log4r"
+require 'vagrant-skytap/api/environment'
+require 'vagrant-skytap/vm_properties'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # This action reads the SSH info for the machine and puts it into the
+      # `:machine_ssh_info` key in the environment.
+      class ReadSSHInfo
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_skytap::action::read_ssh_info")
+        end
+
+        def call(env)
+          env[:machine_ssh_info] = VmProperties.new(env[:machine].data_dir).ssh_info
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,42 @@
+require "log4r"
+require 'vagrant-skytap/environment_properties'
+require 'vagrant-skytap/vm_properties'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # This action reads the state of the machine and puts it in the
+      # `:machine_state_id` key in the environment.
+      class ReadState
+        attr_reader :env
+
+        def initialize(app, env)
+          @app    = app
+          @env    = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::read_state")
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env[:machine])
+
+          @app.call(env)
+        end
+
+        def read_state(machine)
+          if VmProperties.read(machine.data_dir)
+            if environment = env[:environment]
+              environment.reload
+            elsif props = API::Environment.properties(env)
+              @logger.info("env[:environment] was not set!")
+            end
+
+            if vm = environment.try(:current_vm)
+              return vm.runstate.to_sym
+            end
+          end
+          :not_created
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/run_environment.rb

@@ -0,0 +1,53 @@
+require "log4r"
+require 'json'
+
+require 'vagrant/util/retryable'
+
+require 'vagrant-skytap/util/timer'
+require 'vagrant-skytap/api/environment'
+
+require 'net/https'
+require 'uri'
+require 'base64'
+require 'json'
+require 'timeout'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Runs the Skytap environment and waits until running.
+      class RunEnvironment
+        include Vagrant::Util::Retryable
+
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::run_environment")
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_skytap.launching_instance"))
+
+          environment.run!
+
+          wait_for_ssh_ready
+
+          @app.call(env)
+        end
+
+        def environment
+          env[:environment]
+        end
+
+        def wait_for_ssh_ready
+          loop do
+            break if env[:interrupted] || env[:machine].communicate.ready?
+            sleep 1
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/run_vm.rb

@@ -0,0 +1,51 @@
+require "log4r"
+require 'json'
+
+require 'vagrant/util/retryable'
+
+require 'vagrant-skytap/util/timer'
+require 'vagrant-skytap/api/environment'
+
+require 'net/https'
+require 'uri'
+require 'base64'
+require 'json'
+require 'timeout'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Runs the Skytap VM and waits until running.
+      class RunVm
+        include Vagrant::Util::Retryable
+
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::run_vm")
+        end
+
+        def call(env)
+          env[:ui].info(I18n.t("vagrant_skytap.launching_vm"))
+          environment.current_vm.run!
+          wait_for_ssh_ready
+
+          @app.call(env)
+        end
+
+        def environment
+          env[:environment]
+        end
+
+        def wait_for_ssh_ready
+          loop do
+            break if env[:interrupted] || env[:machine].communicate.ready?
+            sleep 1
+          end
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/set_hostname.rb

@@ -0,0 +1,31 @@
+require "log4r"
+#require 'vagrant'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # sets VM hostname via the API
+      class SetHostname
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_skytap::action::set_hostname")
+        end
+
+        def call(env)
+          if hostname = env[:machine].config.vm.hostname
+            if vm = env[:environment].current_vm
+              if nic = vm.interfaces.first
+                if hostname != nic.get_api_attribute('hostname')
+                  @logger.info("Updating hostname: #{hostname}")
+                  nic.update_with_retry(hostname: hostname)
+                end
+              end
+            end
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/set_up_vm.rb

@@ -0,0 +1,21 @@
+require 'log4r'
+require 'json'
+require 'vagrant-skytap/setup_helper'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      class SetUpVm
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new("vagrant_skytap::action::set_up_vm")
+        end
+
+        def call(env)
+          SetupHelper.run!(env, env[:environment])
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/stop_environment.rb

@@ -0,0 +1,43 @@
+require "log4r"
+require 'json'
+
+require 'vagrant-skytap/util/timer'
+require 'vagrant-skytap/api/environment'
+
+require 'net/https'
+require 'uri'
+require 'base64'
+require 'json'
+require 'timeout'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Stops the Skytap environment and waits until stopped.
+      class StopEnvironment
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::stop_environment")
+        end
+
+        def call(env)
+          environment = env[:environment]
+          if env[:force_halt]
+            environment.poweroff!
+          else
+            begin
+              environment.stop!
+            rescue Errors::InstanceReadyTimeout
+              @logger.info(I18n.t("vagrant_skytap.graceful_halt_environment_failed"))
+              environment.poweroff!
+            end
+          end
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/stop_vm.rb

@@ -0,0 +1,43 @@
+require "log4r"
+require 'json'
+
+require 'vagrant-skytap/util/timer'
+require 'vagrant-skytap/api/environment'
+
+require 'net/https'
+require 'uri'
+require 'base64'
+require 'json'
+require 'timeout'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Stops the Skytap VM and waits until stopped.
+      class StopVm
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::stop_vm")
+        end
+
+        def call(env)
+          vm = env[:environment].current_vm
+          if env[:force_halt]
+            vm.poweroff!
+          else
+            begin
+              vm.stop!
+            rescue Errors::InstanceReadyTimeout
+              @logger.info(I18n.t("vagrant_skytap.graceful_halt_vm_failed"))
+              vm.poweroff!
+            end
+          end
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/store_extra_data.rb

@@ -0,0 +1,35 @@
+require 'log4r'
+require_relative "mixin_machine_index"
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Stores some provider-specific data in Vagrant's global machine index.
+      class StoreExtraData
+        include MixinMachineIndex
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::store_extra_data")
+        end
+
+        def call(env)
+          entry = machine_index_entry
+          entry.extra_data.merge!(provider_extra_data)
+          entry = machine_index.set(entry)
+          machine_index.release(entry)
+
+          @app.call(env)
+        end
+
+        def provider_extra_data
+          {
+            'vm_id' => env[:machine].id
+          }
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/suspend_environment.rb

@@ -0,0 +1,32 @@
+require 'json'
+
+require 'vagrant-skytap/util/timer'
+require 'vagrant-skytap/api/environment'
+
+require 'net/https'
+require 'uri'
+require 'base64'
+require 'json'
+require 'timeout'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Suspends the Skytap environment and waits until suspended.
+      class SuspendEnvironment
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::suspend_environment")
+        end
+
+        def call(env)
+          env[:environment].suspend!
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/suspend_vm.rb

@@ -0,0 +1,32 @@
+require 'json'
+
+require 'vagrant-skytap/util/timer'
+require 'vagrant-skytap/api/environment'
+
+require 'net/https'
+require 'uri'
+require 'base64'
+require 'json'
+require 'timeout'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # Suspends the Skytap VM and waits until suspended.
+      class SuspendVm
+        attr_reader :env
+
+        def initialize(app, env)
+          @app = app
+          @env = env
+          @logger = Log4r::Logger.new("vagrant_skytap::action::suspend_vm")
+        end
+
+        def call(env)
+          env[:environment].current_vm.suspend!
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/timed_provision.rb

@@ -0,0 +1,21 @@
+require "vagrant-skytap/util/timer"
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # This is the same as the builtin provision except it times the
+      # provisioner runs.
+      class TimedProvision < Vagrant::Action::Builtin::Provision
+        def run_provisioner(env, name, p)
+          timer = Util::Timer.time do
+            super
+          end
+
+          env[:metrics] ||= {}
+          env[:metrics]["provisioner_times"] ||= []
+          env[:metrics]["provisioner_times"] << [name, timer]
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-skytap/action/update_hardware.rb

@@ -0,0 +1,37 @@
+require "log4r"
+#require 'vagrant'
+require 'vagrant-skytap/api/vm'
+
+module VagrantPlugins
+  module Skytap
+    module Action
+      # This action reads the SSH info for the machine and puts it into the
+      # `:machine_ssh_info` key in the environment.
+      class UpdateHardware
+        def initialize(app, env)
+          @app    = app
+          @logger = Log4r::Logger.new("vagrant_skytap::action::update_hardware")
+        end
+
+        def call(env)
+          if vm = env[:environment].current_vm
+            provider_config = vm.machine.provider_config
+            hardware_info = {
+              cpus:            provider_config.cpus,
+              cpus_per_socket: provider_config.cpuspersocket,
+              ram:             provider_config.ram,
+              guestOS:         provider_config.guestos,
+            }.reject{|k, v| v.nil? || v == vm.hardware[k.to_s]}
+
+            if hardware_info.present?
+              @logger.info("Updating hardware properties: #{hardware_info}")
+              vm.update_with_retry(hardware: hardware_info)
+            end
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end