vagrant-windows-hyperv 1.0.1

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

@@ -0,0 +1,145 @@
+#-------------------------------------------------------------------------
+# 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]$guest_ip = $(throw "-guest_ip is required."),
+    [string]$username = $(throw "-guest_username is required."),
+    [string]$password = $(throw "-guest_password is required."),
+    [string]$host_path = $(throw "-host_path is required."),
+    [string]$guest_path = $(throw "-guest_path is required.")
+ )
+
+# Include the following modules
+$presentDir = Split-Path -parent $PSCommandPath
+$modules = @()
+$modules += $presentDir + "\utils\write_messages.ps1"
+$modules += $presentDir + "\utils\create_session.ps1"
+forEach ($module in $modules) { . $module }
+
+function Get-file-hash($source_path, $delimiter) {
+    $source_files = @()
+    (Get-ChildItem $source_path -rec -force | ForEach-Object -Process {
+      Get-FileHash -Path $_.FullName -Algorithm MD5 } ) |
+        ForEach-Object -Process {
+          $source_files += $_.Path.Replace($source_path, "") + $delimiter + $_.Hash
+        }
+    $source_files
+}
+
+function Get-remote-file-hash($source_path, $delimiter, $session) {
+    Invoke-Command -Session $session -ScriptBlock ${function:Get-file-hash} -ArgumentList $source_path, $delimiter
+    # TODO:
+    # Check if remote PS Scripting errors
+}
+
+function Sync-Remote-Machine($machine, $remove_files, $copy_files, $host_path, $guest_path) {
+    ForEach ($item in $copy_files) {
+      $from = $host_path + $item
+      $to = $guest_path + $item
+      # Copy VM can also take a VM object
+      Copy-VMFile  -VM $machine -SourcePath $from -DestinationPath $to -CreateFullPath -FileSource Host -Force
+    }
+}
+
+function Create-Remote-Folders($empty_source_folders, $guest_path) {
+
+    ForEach ($item in $empty_source_folders) {
+        $new_name =  $guest_path + $item
+        New-Item "$new_name" -type directory -Force
+    }
+}
+
+function Create-Guest-Folder($guest_path) {
+  try {
+    if (Test-Path $guest_path) {
+      $junction = Get-Item $guest_path
+      $junction.Delete()
+    }
+  }
+  # Catch any [IOException]
+   catch {
+     Remove-Item "$guest_path" -Force -Recurse
+   }
+   New-Item "$guest_path" -type directory -Force
+}
+
+function Get-Empty-folders-From-Source($host_path) {
+  Get-ChildItem $host_path -recurse |
+        Where-Object {$_.PSIsContainer -eq $True} |
+            Where-Object {$_.GetFiles().Count -eq 0} |
+                Select-Object FullName | ForEach-Object -Process {
+                    $empty_source_folders += ($_.FullName.Replace($host_path, ""))
+                }
+}
+
+$delimiter = " || "
+
+$machine = Get-VM -Id $vm_id
+
+# FIXME: PowerShell guys please fix this.
+# The below script checks for all VMIntegrationService which are not enabled
+# and will enable this.
+# When when all the services are enabled this throws an error.
+# Enable VMIntegrationService to true
+try {
+  Get-VM -Id $vm_id | Get-VMIntegrationService -Name "Guest Service Interface" | Enable-VMIntegrationService -Passthru
+  }
+  catch { }
+
+
+$response = Create-Remote-Session $guest_ip $username $password
+if (!$response["session"] -and $response["error"]) {
+  $errortHash = @{
+    type = "PowerShellError"
+    error = $response["error"]
+  }
+  Write-Error-Message $errortHash
+  return
+}
+
+$session = $response["session"]
+# Create the guest folder if not exist
+$result = Invoke-Command -Session $session -ScriptBlock ${function:Create-Guest-Folder} -ArgumentList $guest_path
+
+
+$source_files = Get-file-hash $host_path $delimiter
+$destination_files = Get-remote-file-hash $guest_path $delimiter $session
+
+if (!$destination_files) {
+  $destination_files = @()
+}
+if (!$source_files) {
+  $source_files = @()
+}
+
+# Compare source and destination files
+$remove_files = @()
+$copy_files = @()
+
+
+Compare-Object -ReferenceObject $source_files -DifferenceObject $destination_files | ForEach-Object {
+  if ($_.SideIndicator -eq '=>') {
+      $remove_files += $_.InputObject.Split($delimiter)[0]
+  } else {
+      $copy_files += $_.InputObject.Split($delimiter)[0]
+  }
+}
+
+# Update the files to remote machine
+Sync-Remote-Machine $machine $remove_files $copy_files $host_path $guest_path
+
+# Create any empty folders which missed to sync to remote machine
+$empty_source_folders = @()
+$directories = Get-Empty-folders-From-Source $host_path
+
+$result = Invoke-Command -Session $session -ScriptBlock ${function:Create-Remote-Folders} -ArgumentList $empty_source_folders, $guest_path
+# Always remove the connection after Use
+Remove-PSSession -Id $session.Id
+
+$resultHash = @{
+  message = "OK"
+}
+Write-Output-Message $resultHash

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

@@ -0,0 +1,25 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+# Include the following modules
+$presentDir = Split-Path -parent $PSCommandPath
+. ([System.IO.Path]::Combine($presentDir, "utils\write_messages.ps1"))
+
+try {
+$hostname =  $(whoami)
+$ip = (Get-WmiObject -class win32_NetworkAdapterConfiguration -Filter 'ipenabled = "true"').ipaddress[0]
+  $resultHash = @{
+    host_name = "$username"
+    host_ip = "$ip"
+  }
+  Write-Output-Message $resultHash
+}
+catch {
+  $errortHash = @{
+    type = "PowerShellError"
+    error = "$_"
+  }
+  Write-Error-Message $errortHash
+}

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

@@ -0,0 +1,36 @@
+#-------------------------------------------------------------------------
+# 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]$command = ""
+ )
+
+# Include the following modules
+$presentDir = Split-Path -parent $PSCommandPath
+. ([System.IO.Path]::Combine($presentDir, "utils\write_messages.ps1"))
+
+try {
+  $vm = Get-VM -Id $vm_id -ErrorAction "stop"
+  switch ($command) {
+    "start" { Start-VM $vm }
+    "stop" { Stop-VM $vm }
+    "suspend" { Suspend-VM $vm }
+    "resume" { Resume-VM $vm }
+  }
+
+  $state = $vm.state
+  $status = $vm.status
+  $name = $vm.name
+  } catch [Microsoft.HyperV.PowerShell.VirtualizationOperationFailedException] {
+    $state = "not_created"
+    $status = "Not Created"
+  }
+  $resultHash = @{
+    state = "$state"
+    status = "$status"
+    name = "$name"
+  }
+  Write-Output-Message $resultHash

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

@@ -0,0 +1,31 @@
+#-------------------------------------------------------------------------
+# 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."),
+    [string]$command = ""
+)
+
+# 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($command) {
+    cmd /c $command
+  }
+  Invoke-Command -Session $response["session"] -ScriptBlock ${function:Remote-Execute} -ArgumentList $command -ErrorAction "stop"
+} catch {
+  Write-Host $_
+  return
+}

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

@@ -0,0 +1,95 @@
+#-------------------------------------------------------------------------
+# 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]$host_path = $(throw "-host_path is required."),
+    [string]$guest_path = $(throw "-guest_path is required."),
+    [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 {
+  # Enable Guest Service Interface if they are disabled
+  try {
+    Get-VM -Id $vm_id | Get-VMIntegrationService -Name "Guest Service Interface" | Enable-VMIntegrationService -Passthru
+    }
+    catch { }
+
+  function Upload-FIle-To-VM($host_path, $guest_path, $machine) {
+    Write-Host $host_path
+    Write-Host $guest_path
+    Copy-VMFile  -VM $machine -SourcePath $host_path -DestinationPath $guest_path -CreateFullPath -FileSource Host -Force -ErrorAction stop
+  }
+
+  function Prepare-Guest-Folder($guest_ip, $username, $password) {
+    $response = Create-Remote-Session $guest_ip $username $password
+    if (!$response["session"] -and $response["error"]) {
+      $errortHash = @{
+        type = "PowerShellError"
+        message = $response["error"]
+      }
+      Write-Error-Message $errorResult
+      return
+    }
+    $session = $response["session"]
+    # Create the guest folder if not exist
+    $result = Invoke-Command -Session $session -ScriptBlock ${function:Create-Guest-Folder} -ArgumentList $guest_path
+  }
+
+  function Create-Guest-Folder($guest_path) {
+    try {
+      if (Test-Path $guest_path) {
+        # First attempt to remove a Junction drive. The fall back to removing a
+        # folder
+        $junction = Get-Item $guest_path
+        $junction.Delete()
+      }
+    }
+    # Catch any [IOException]
+     catch {
+       Remove-Item "$guest_path" -Force -Recurse
+     }
+     New-Item "$guest_path" -type directory -Force
+  }
+
+  $machine = Get-VM -Id $vm_id
+  # When Host path is a folder.
+  # Find all files within it and copy to the Guest
+  if (Test-Path $host_path -pathtype container) {
+    # Open a remote PS Session with the guest
+    Prepare-Guest-Folder $guest_ip $username $password
+    # Copy all files from Host path to Guest Path
+    Get-ChildItem $host_path -rec |
+      Where-Object {$_.PSIsContainer -eq $false} |
+        ForEach-Object -Process {
+          $file_name = $_.Fullname.Replace($host_path, "")
+          $from = $host_path + $file_name
+          $to = $guest_path + $file_name
+          # Write-Host $from
+          # Write-Host $to
+          Upload-FIle-To-VM $from $to $machine
+        }
+  } elseif (Test-Path $host_path) {
+    Upload-FIle-To-VM $host_path $guest_path $machine
+  }
+  $resultHash = @{
+    message = "OK"
+  }
+  Write-Output-Message $resultHash
+} catch {
+  $errortHash = @{
+    type = "PowerShellError"
+    error ="Failed to copy file $_"
+  }
+  Write-Error-Message $errortHash
+  return
+}

data/lib/vagrant-windows-hyperv/scripts/utils/create_session.ps1

@@ -0,0 +1,34 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+function Get-Remote-Session($guest_ip, $username, $password) {
+    $secstr = convertto-securestring -AsPlainText -Force -String $password
+    $cred = new-object -typename System.Management.Automation.PSCredential -argumentlist $username, $secstr
+    New-PSSession -ComputerName $guest_ip -Credential $cred -ErrorAction "stop"
+}
+
+function Create-Remote-Session($guest_ip, $username, $password) {
+    $count = 0
+    $session_error = ""
+    $session = ""
+    do {
+        $count++
+        try {
+            $session = Get-Remote-Session $guest_ip $username $password
+            $session_error = ""
+        }
+        catch {
+            Start-Sleep -s 1
+            $session_error = $_
+            $session = ""
+        }
+    }
+    while (!$session -and $count -lt 20)
+
+    return  @{
+        session = $session
+        error = $session_error
+    }
+}

data/lib/vagrant-windows-hyperv/scripts/utils/write_messages.ps1

@@ -0,0 +1,18 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+function Write-Error-Message($message) {
+  $result = ConvertTo-Json $message
+  Write-Host "===Begin-Error==="
+  Write-Host $result
+  Write-Host "===End-Error==="
+}
+
+function Write-Output-Message($message) {
+  $result = ConvertTo-Json $message
+  Write-Host "===Begin-Output==="
+  Write-Host $result
+  Write-Host "===End-Output==="
+}

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

@@ -0,0 +1,10 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+module VagrantPlugins
+  module VagrantHyperV
+    VERSION = "1.0.1"
+  end
+end

data/locales/en.yml

@@ -0,0 +1,15 @@
+en:
+  vagrant_win_hyperv:
+    generating_rdp: |-
+      Generating a RDP file to establish a remote session. You will soon be prompted to authenticate the
+      RDP session and enter the Virtual Machine's password.
+    message_rdp_not_ready: |-
+        Hyper-V machine isn't running. Can't RDP in!
+    errors:
+      ssh_not_available: |-
+        SSH is available for non windows guest. Vagrant detected guest to be %{guest}
+      rdp_not_available: |-
+        RDP is available for windows guest. Vagrant detected guest to be %{guest}
+      win_rm_not_ready: |-
+        WinRM is not enabled in the remot guest. Vagrant-Windows-HyperV uses remote PowerShell scripts
+        to communicate to the remote machine. Please check the same and try again.

data/spec/hyper-v/config_spec.rb

@@ -0,0 +1,36 @@
+#-------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+
+require "vagrant-windows-hyperv/config"
+require_relative "spec_helper"
+
+describe VagrantPlugins::HyperV::Config do
+  let(:instance) { VagrantPlugins::HyperV::Config.new }
+  let(:machine) { Object.new }
+  describe "Check host share config" do
+    it "Should return the error messages under a key HyperV" do
+      instance.host_config do |share|
+        share.username = "vagrant"
+      end
+      error = instance.validate(machine)
+      assert(true, error.has_key?("HyperV"))
+    end
+    it "Should raise errors for missing properties" do
+      instance.host_config do |share|
+        share.username = "vagrant"
+      end
+      error = instance.validate(machine)
+      assert(true, !error["HyperV"].empty?)
+    end
+    it "Should have no errors when all properties are passed" do
+      instance.host_config do |share|
+        share.username = "vagrant"
+        share.password = "my_secret_password"
+      end
+      error = instance.validate(machine)
+      assert(true, error["HyperV"].empty?)
+    end
+  end
+end