vagrant-azure 1.0.1

data/lib/vagrant-azure/driver.rb

@@ -0,0 +1,79 @@
+#---------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'json'
+require "#{Vagrant::source_root}/plugins/providers/hyperv/driver"
+
+module VagrantPlugins
+  module WinAzure
+    class Driver < VagrantPlugins::HyperV::Driver
+      def initialize(machine)
+        @id = machine.id
+        @machine = machine
+      end
+
+      def ssh_info
+        @ssh_info ||= @machine.provider.winrm_info
+        @ssh_info[:username] ||= @machine.config.ssh.username
+        @ssh_info[:password] ||= @machine.config.ssh.password
+        @ssh_info
+      end
+
+      def remote_credentials
+        @remote_credentials ||= {
+          guest_ip: ssh_info[:host],
+          guest_port: ssh_info[:port],
+          username: ssh_info[:username],
+          password: ssh_info[:password]
+        }
+      end
+
+      def run_remote_ps(command, &block)
+        options = remote_credentials.merge(command: command)
+        script_path = local_script_path('run_in_remote.ps1')
+
+        ps_options = []
+
+        options.each do |key, value|
+          ps_options << "-#{key}"
+          ps_options << "'#{value}'"
+        end
+
+        ps_options << '-ErrorAction' << 'Stop'
+        opts = { notify: [:stdout, :stderr, :stdin] }
+        Vagrant::Util::PowerShell.execute(
+          script_path,
+          *ps_options,
+          **opts,
+          &block
+        )
+      end
+
+      def upload(from, to)
+        options = {
+          host_path: windows_path(from),
+          guest_path: windows_path(to)
+        }.merge(remote_credentials)
+
+        script_path = local_script_path('upload_file.ps1')
+        execute(script_path, options)
+      end
+
+      protected
+
+      def local_script_path(path)
+        lib_path = Pathname.new(File.expand_path('../scripts', __FILE__))
+        windows_path(lib_path.join(path).to_s)
+      end
+
+      def windows_path(path)
+        if path
+          path = path.gsub('/', "\\")
+          path = "c:#{path}" if path =~ /^\\/
+        end
+        path
+      end
+    end
+  end
+end

data/lib/vagrant-azure/plugin.rb

@@ -0,0 +1,87 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+begin
+  require 'vagrant'
+rescue LoadError
+  raise 'The Vagrant Azure plugin must be run within Vagrant.'
+end
+
+# This is a sanity check to make sure no one is attempting to install this into
+# an early Vagrant version.
+if Vagrant::VERSION < '1.2.0'
+  raise 'The Vagrant Azure plugin is only compatible with Vagrant 1.2+'
+end
+
+module VagrantPlugins
+  module WinAzure
+    class Plugin < Vagrant.plugin('2')
+      name 'azure'
+      description <<-DESC
+      This plugin installs a provider that allows Vagrant to manage
+      machines in Windows Azure.
+      DESC
+
+      config(:azure, :provider) do
+        require_relative 'config'
+        Config
+      end
+
+      provider(:azure, parallel: true) do
+        # Setup logging and i18n
+        setup_logging
+        setup_i18n
+
+        # Return the provider
+        require_relative 'provider'
+        Provider
+      end
+
+      command 'rdp' do
+        require_relative 'command/rdp/command'
+        Command
+      end
+
+      def self.setup_i18n
+        I18n.load_path << File.expand_path(
+          'locales/en.yml',
+          WinAzure.source_root
+        )
+        I18n.load_path << File.expand_path(
+          'templates/locales/providers_hyperv.yml',
+          Vagrant.source_root
+        )
+        I18n.reload!
+      end
+
+      def self.setup_logging
+        require 'log4r'
+
+        level = nil
+        begin
+          level = Log4r.const_get(ENV['VAGRANT_LOG'].upcase)
+        rescue NameError
+          # This means that the logging constant wasn't found,
+          # which is fine. We just keep `level` as `nil`. But
+          # we tell the user.
+          level = nil
+        end
+
+        # Some constants, such as "true" resolve to booleans, so the
+        # above error checking doesn't catch it. This will check to make
+        # sure that the log level is an integer, as Log4r requires.
+        level = nil if !level.is_a?(Integer)
+
+        # Set the logging level on all "vagrant" namespaced logs as long as
+        # we have a valid level
+        if level
+          logger = Log4r::Logger.new("vagrant_azure")
+          logger.outputters = Log4r::Outputter.stderr
+          logger.level = level
+          logger = nil
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/provider.rb

@@ -0,0 +1,70 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+require 'vagrant'
+
+module VagrantPlugins
+  module WinAzure
+    class Provider < Vagrant.plugin('2', :provider)
+      attr_reader :driver
+
+      def initialize(machine)
+        @machine = machine
+
+        # Load the driver
+        machine_id_changed
+      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}"
+        return Action.send(action_method) if Action.respond_to?(action_method)
+        nil
+      end
+
+      def machine_id_changed
+        @driver = Driver.new(@machine)
+      end
+
+      def ssh_info
+        # Run a custom action called "read_ssh_info" which does what it
+        # says and puts the resulting SSH info into the `:machine_ssh_info`
+        # key in the environment.
+        env = @machine.action('read_ssh_info')
+        env[:machine_ssh_info]
+      end
+
+      def rdp_info
+        env = @machine.action('read_rdp_info')
+        env[:machine_ssh_info]
+      end
+
+      def winrm_info
+        env = @machine.action('read_winrm_info')
+        env[:machine_ssh_info]
+      end
+
+      def state
+        # Run a custom action we define called "read_state" which does what it
+        # says. It puts the state in the `:machine_state_id` key in the env
+        env = @machine.action('read_state')
+        state_id = env[:machine_state_id]
+
+        short = "Machine's current state is #{state_id}"
+        long = ""
+
+        # Return the MachineState object
+        Vagrant::MachineState.new(state_id, short, long)
+      end
+
+      def to_s
+        id = @machine.id.nil? ? 'new' : @machine.id
+        "Azure (#{id})"
+      end
+    end
+  end
+end

data/lib/vagrant-azure/provisioner/puppet.rb

@@ -0,0 +1,109 @@
+#---------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#---------------------------------------------------------------------------
+require 'fileutils'
+require 'tempfile'
+
+module VagrantPlugins
+  module WinAzure
+    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_s, to)
+          end
+
+          module_paths = @module_paths.map { |_, to| to }
+          unless 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 custome facts if we have any
+          facter = ''
+          unless 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_azure.provisioners.puppet.running_puppet',
+            manifest: config.manifest_file
+          )
+          @env[:ui].info 'Executing puppet script in Windows Azure VM'
+          @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-azure/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-azure/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
+  }

data/lib/vagrant-azure/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