vagrant-windows-hyperv 1.0.1

data/lib/vagrant-windows-hyperv/provider.rb

@@ -0,0 +1,30 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "#{Vagrant::source_root}/plugins/providers/hyperv/provider"
+
+module VagrantPlugins
+  module VagrantHyperV
+    class Provider < VagrantPlugins::HyperV::Provider
+
+      def machine_id_changed
+        @driver = Driver.new(@machine)
+      end
+
+      def action(name)
+        # Attempt to get the action method from the Action class if it
+        # exists, otherwise return nil to show that we don't support the
+        # given action.
+        action_method = "action_#{name}"
+        if Action.respond_to?(action_method)
+          return Action.send(action_method)
+        else
+          super
+        end
+      end
+
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/provisioner/chef_solo.rb

@@ -0,0 +1,181 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "fileutils"
+require "tempfile"
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Provisioner
+      class ChefSolo
+        attr_reader :provisioner
+        def initialize(env)
+          @env = env
+          @provisioner = env[:provisioner]
+        end
+
+        def provision_for_windows
+          # TODO
+          # Verify that the proper shared folders exist.
+
+          # Copy the chef cookbooks roles data bags and environment folders to Guest
+          copy_folder_to_guest(provisioner.cookbook_folders)
+          copy_folder_to_guest(provisioner.role_folders)
+          copy_folder_to_guest(provisioner.data_bags_folders)
+          copy_folder_to_guest(provisioner.environments_folders)
+
+          # Upload Encrypted data bag
+          upload_encrypted_data_bag_secret if config.encrypted_data_bag_secret_key_path
+          setup_json
+          setup_solo_config
+          run_chef_solo
+          delete_encrypted_data_bag_secret
+        end
+
+        def setup_json
+          @env[:machine].env.ui.info I18n.t("vagrant.provisioners.chef.json")
+
+          # Get the JSON that we're going to expose to Chef
+          json = config.json
+          json[:run_list] = config.run_list if !config.run_list.empty?
+          json = JSON.pretty_generate(json)
+
+          # Create a temporary file to store the data so we
+          # can upload it
+          temp = Tempfile.new("vagrant")
+          temp.write(json)
+          temp.close
+
+          remote_file = File.join(config.provisioning_path, "dna.json")
+          @env[:machine].provider.driver.upload(temp.path, remote_file)
+        end
+
+        def setup_solo_config
+          cookbooks_path = guest_paths(provisioner.cookbook_folders)
+          roles_path = guest_paths(provisioner.role_folders)
+          data_bags_path = guest_paths(provisioner.data_bags_folders).first
+          environments_path = guest_paths(provisioner.environments_folders).first
+          source_path = "#{VagrantPlugins::VagrantHyperV.source_root}"
+          template_path = source_path + "/templates/provisioners/chef-solo/solo"
+          setup_config(template_path, "solo.rb", {
+            :cookbooks_path => cookbooks_path,
+            :recipe_url => config.recipe_url,
+            :roles_path => roles_path,
+            :data_bags_path => data_bags_path,
+            :environments_path => environments_path
+          })
+        end
+
+        def setup_config(template, filename, template_vars)
+          # If we have custom configuration, upload it
+          remote_custom_config_path = nil
+          if config.custom_config_path
+            expanded = File.expand_path(
+              config.custom_config_path, @machine.env.root_path)
+            remote_custom_config_path = File.join(
+              config.provisioning_path, "custom-config.rb")
+
+            @env[:machine].provider.driver.upload(expanded, remote_custom_config_path)
+          end
+
+          config_file = Vagrant::Util::TemplateRenderer.render(template, {
+            :custom_configuration => remote_custom_config_path,
+            :file_cache_path => config.file_cache_path,
+            :file_backup_path => config.file_backup_path,
+            :log_level        => config.log_level.to_sym,
+            :verbose_logging  => config.verbose_logging,
+            :http_proxy       => config.http_proxy,
+            :http_proxy_user  => config.http_proxy_user,
+            :http_proxy_pass  => config.http_proxy_pass,
+            :https_proxy      => config.https_proxy,
+            :https_proxy_user => config.https_proxy_user,
+            :https_proxy_pass => config.https_proxy_pass,
+            :no_proxy         => config.no_proxy,
+            :formatter        => config.formatter
+          }.merge(template_vars))
+
+          # Create a temporary file to store the data so we
+          # can upload it
+          temp = Tempfile.new("vagrant")
+          temp.write(config_file)
+          temp.close
+
+          remote_file = File.join(config.provisioning_path, filename)
+          @env[:machine].provider.driver.upload(temp.path, remote_file)
+        end
+
+        def run_chef_solo
+          if config.run_list && config.run_list.empty?
+            @env[:machine].ui.warn(I18n.t("vagrant.chef_run_list_empty"))
+          end
+
+          options = [
+            "-c #{config.provisioning_path}/solo.rb",
+            "-j #{config.provisioning_path}/dna.json"
+          ]
+
+          command_env = config.binary_env ? "#{config.binary_env} " : ""
+          command_args = config.arguments ? " #{config.arguments}" : ""
+          command = "#{command_env}#{chef_binary_path("chef-solo")} " +
+            "#{options.join(" ")} #{command_args}"
+          config.attempts.times do |attempt|
+            if attempt == 0
+              @env[:machine].env.ui.info I18n.t("vagrant.provisioners.chef.running_solo")
+            else
+              @env[:machine].env.ui.info I18n.t("vagrant.provisioners.chef.running_solo_again")
+            end
+
+            @env[:machine].provider.driver.run_remote_ps(command) do |type, data|
+              # Output the data with the proper color based on the stream.
+              if (type == :stdout || type == :stderr)
+                @env[:ui].detail data
+              end
+            end
+          end
+
+        end
+
+        def delete_encrypted_data_bag_secret
+          # Run remote command to delete
+          # config.encrypted_data_bag_secret
+        end
+
+        def upload_encrypted_data_bag_secret
+          @machine.env.ui.info I18n.t("vagrant.provisioners.chef.upload_encrypted_data_bag_secret_key")
+          @env[:machine].provider.driver.upload(encrypted_data_bag_secret_key_path,
+                        config.encrypted_data_bag_secret)
+        end
+
+        def encrypted_data_bag_secret_key_path
+          File.expand_path(config.encrypted_data_bag_secret_key_path, @env[:machine].env.root_path)
+        end
+
+        def config
+          provisioner.config
+        end
+
+        def guest_paths(folders)
+          folders.map { |parts| parts[2] }
+        end
+
+        # Returns the path to the Chef binary, taking into account the
+        # `binary_path` configuration option.
+        def chef_binary_path(binary)
+          return binary if !config.binary_path
+          return File.join(config.binary_path, binary)
+        end
+
+        def copy_folder_to_guest(folders)
+          folders.each do |type, local_path, remote_path|
+            if type == :host
+              @env[:machine].provider.driver.upload(local_path, remote_path)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/provisioner/puppet.rb

@@ -0,0 +1,99 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "fileutils"
+require "tempfile"
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Provisioner
+      class Puppet
+        attr_reader :provisioner
+        def initialize(env)
+          @env = env
+          @provisioner = env[:provisioner]
+        end
+
+        def provision_for_windows
+
+          options = [config.options].flatten
+          @module_paths = provisioner.instance_variable_get("@module_paths")
+          @hiera_config_path = provisioner.instance_variable_get("@hiera_config_path")
+          @manifest_file = provisioner.instance_variable_get("@manifest_file")
+
+          # Copy the manifests directory to the guest
+          if config.manifests_path[0].to_sym == :host
+            @env[:machine].provider.driver.upload(
+              File.expand_path(config.manifests_path[1], @env[:machine].env.root_path),
+              provisioner.manifests_guest_path)
+          end
+
+          # Copy the module paths to the guest
+          @module_paths.each do |from, to|
+            @env[:machine].provider.driver.upload(from, to)
+          end
+
+
+          module_paths = @module_paths.map { |_, to| to }
+          if !@module_paths.empty?
+            # Prepend the default module path
+            module_paths.unshift("/ProgramData/PuppetLabs/puppet/etc/modules")
+
+            # Add the command line switch to add the module path
+            options << "--modulepath '#{module_paths.join(':')}'"
+          end
+
+          if @hiera_config_path
+            options << "--hiera_config=#{@hiera_config_path}"
+
+            # Upload Hiera configuration if we have it
+            local_hiera_path   = File.expand_path(config.hiera_config_path,
+              @env[:machine].env.root_path)
+            @env[:machine].provider.driver.upload(local_hiera_path, @hiera_config_path)
+          end
+
+          options << "--manifestdir #{provisioner.manifests_guest_path}"
+          options << "--detailed-exitcodes"
+          options << @manifest_file
+          options = options.join(" ")
+
+          # Build up the custom facts if we have any
+          facter = ""
+          if !config.facter.empty?
+            facts = []
+            config.facter.each do |key, value|
+              facts << "FACTER_#{key}='#{value}'"
+            end
+
+            facter = "#{facts.join(" ")} "
+          end
+
+          command = "#{facter}puppet apply #{options}"
+          if config.working_directory
+            command = "cd #{config.working_directory} && #{command}"
+          end
+
+          @env[:ui].info I18n.t("vagrant.provisioners.puppet.running_puppet",
+                                      :manifest => config.manifest_file)
+
+          @env[:ui].info "Executing puppet script in Guest"
+          @env[:machine].provider.driver.run_remote_ps(command) do |type, data|
+            # Output the data with the proper color based on the stream.
+            if (type == :stdout || type == :stderr)
+              @env[:ui].detail data
+            end
+          end
+        end
+
+        protected
+
+        def config
+          provisioner.config
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/provisioner/shell.rb

@@ -0,0 +1,81 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module VagrantPlugins
+  module VagrantHyperV
+    module Provisioner
+      class Shell
+        attr_reader :provisioner
+        def initialize(env)
+          @env = env
+          @provisioner = env[:provisioner]
+        end
+
+        def provision_for_windows
+          arguments = ""
+          arguments = " #{config.args}" if config.args
+          with_windows_script_file do |path|
+
+            guest_path = if File.extname(config.upload_path) == ""
+              "#{config.upload_path}#{File.extname(path.to_s)}"
+            else
+              config.upload_path
+            end
+
+            response = @env[:machine].provider.driver.upload(path, guest_path)
+
+            command = "powershell.exe #{guest_path} #{arguments}"
+            @env[:machine].provider.driver.run_remote_ps(command) do |type, data|
+              if type == :stdout || type == :stderr
+                @env[:ui].detail data
+              end
+            end
+          end
+        end
+
+        protected
+
+        def config
+          provisioner.config
+        end
+
+        # This method yields the path to a script to upload and execute
+        # on the remote server. This method will properly clean up the
+        # script file if needed.
+        def with_windows_script_file
+          if config.remote?
+            download_path = @env[:machine].env.tmp_path.join("#{@env[:machine].id}-remote-script#{File.extname(config.path)}")
+            download_path.delete if download_path.file?
+
+            begin
+              Vagrant::Util::Downloader.new(config.path, download_path).download!
+              yield download_path
+            ensure
+              download_path.delete
+            end
+
+          elsif config.path
+            # Just yield the path to that file...
+            yield config.path
+          else
+            # Otherwise we have an inline script, we need to Tempfile it,
+            # and handle it specially...
+            file = Tempfile.new(['vagrant-powershell', '.ps1'])
+
+            begin
+              file.write(config.inline)
+              file.fsync
+              file.close
+              yield file.path
+            ensure
+              file.close
+            end
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/vagrant-windows-hyperv/scripts/check_winrm.ps1

@@ -0,0 +1,41 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+param (
+    [string]$guest_ip = $(throw "-guest_ip is required."),
+    [string]$username = $(throw "-guest_username is required."),
+    [string]$password = $(throw "-guest_password is required.")
+)
+
+# Include the following modules
+$presentDir = Split-Path -parent $PSCommandPath
+. ([System.IO.Path]::Combine($presentDir, "utils\write_messages.ps1"))
+. ([System.IO.Path]::Combine($presentDir, "utils\create_session.ps1"))
+
+try {
+  $response = Create-Remote-Session $guest_ip $username $password
+  if (!$response["session"] -and $response["error"]) {
+      Write-Host $response["error"]
+      return
+  }
+    function Remote-Execute() {
+      $winrm_state = ""
+      get-service winrm | ForEach-Object {
+        $winrm_state = $_.status
+      }
+      return "$winrm_state"
+    }
+    $result = Invoke-Command -Session $response["session"] -ScriptBlock ${function:Remote-Execute} -ErrorAction "stop"
+    $resultHash = @{
+      message = "$result"
+    }
+    Write-Output-Message $resultHash
+  } catch {
+    $errortHash = @{
+      type = "PowerShellError"
+      error ="Failed to copy file $_"
+    }
+    Write-Error-Message $errortHash
+    return
+  }

data/lib/vagrant-windows-hyperv/scripts/export_vm.ps1

@@ -0,0 +1,31 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+param (
+    [string]$vm_id = $(throw "-vm_id is required."),
+    [string]$path = $(throw "-path is required.")
+)
+
+# Include the following modules
+$presentDir = Split-Path -parent $PSCommandPath
+. ([System.IO.Path]::Combine($presentDir, "utils\write_messages.ps1"))
+
+
+# Export the Virtual Machine
+try {
+  $vm = Get-Vm -Id $vm_id
+  $vm  | Export-VM -Path $path -ErrorAction "stop"
+  $name = $vm.name
+  $resultHash = @{
+    name = "$name"
+  }
+  Write-Output-Message $resultHash
+  } catch {
+    $errortHash = @{
+      type = "PowerShellError"
+      error = "Failed to export a  VM $_"
+    }
+    Write-Error-Message $errortHash
+  }