vagrant-azure 1.0.1

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

@@ -0,0 +1,51 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class ReadSSHInfo
+        def initialize(app, env, port = 22)
+          @app = app
+          @port = port
+          @logger = Log4r::Logger.new('vagrant_azure::action::read_ssh_info')
+        end
+
+        def call(env)
+          env[:ui].info "Looking for #{@port}"
+
+          env[:machine_ssh_info] = read_ssh_info(
+            env[:azure_vm_service],
+            env[:machine]
+          )
+
+          @app.call(env)
+        end
+
+        def read_ssh_info(azure, machine)
+          return nil if machine.id.nil?
+          machine.id =~ /@/
+          vm = azure.get_virtual_machine($`, $')
+
+          if vm.nil? || !vm.instance_of?(Azure::VirtualMachineManagement::VirtualMachine)
+            # Machine cannot be found
+            @logger.info 'Machine not found. Assuming it was destroyed'
+            machine.id = nil
+            return nil
+          end
+
+          vm.tcp_endpoints.each do |endpoint|
+            if endpoint[:local_port] == "#{@port}"
+              return { :host => "#{vm.cloud_service_name}.cloudapp.net", :port => endpoint[:public_port] }
+            end
+          end
+
+          return nil
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,46 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class ReadState
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_azure::action::read_state')
+        end
+
+        def call(env)
+          env[:machine_state_id] = read_state(env)
+
+          @app.call(env)
+        end
+
+        def read_state(env)
+          env[:machine].id = "#{env[:machine].provider_config.vm_name}@#{env[:machine].provider_config.cloud_service_name}" unless env[:machine].id
+
+          env[:machine].id =~ /@/
+
+          env[:ui].info "Attempting to read state for #{$`} in #{$'}"
+
+          vm = env[:azure_vm_service].get_virtual_machine($`, $')
+
+          if vm.nil? || \
+            !vm.instance_of?(Azure::VirtualMachineManagement::VirtualMachine) || \
+            [ :DeletingVM ].include?(vm.status.to_sym)
+            # Machine can't be found
+            @logger.info 'Machine cannot be found'
+            env[:machine].id = nil
+            return :NotCreated
+          end
+
+          env[:ui].info "VM Status: #{vm.status.to_sym}"
+          return vm.status.to_sym
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/action/restart_vm.rb

@@ -0,0 +1,27 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class RestartVM
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_azure::action::restart_vm')
+        end
+
+        def call(env)
+          env[:machine].id =~ /@/
+
+          env[:ui].info "Restarting #{$`} in #{$'}"
+          env[:azure_vm_service].restart_virtual_machine($`, $')
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/action/run_instance.rb

@@ -0,0 +1,111 @@
+#---------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+require 'json'
+require 'azure'
+
+require 'vagrant/util/retryable'
+
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class RunInstance
+        include Vagrant::Util::Retryable
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_azure::action::run_instance')
+        end
+
+        def call(env)
+          config = env[:machine].provider_config
+
+          # Add the mandatory parameters and options
+          params = {
+            vm_name: config.vm_name,
+            vm_user: config.vm_user,
+            image: config.vm_image
+          }
+
+          options = {
+            cloud_service_name: config.cloud_service_name
+          }
+
+
+          # Add the optional parameters and options if not nil
+          params[:password] = config.vm_password unless config.vm_password.nil?
+          params[:location] = config.vm_location unless config.vm_location.nil?
+          params[:affinity_group] = config.vm_affinity_group unless \
+            config.vm_affinity_group.nil?
+
+          options[:storage_account_name] = config.storage_acct_name unless \
+            config.storage_acct_name.nil?
+          options[:deployment_name] = config.deployment_name unless \
+            config.deployment_name.nil?
+          options[:tcp_endpoints] = config.tcp_endpoints unless \
+            config.tcp_endpoints.nil?
+          options[:private_key_file] = config.ssh_private_key_file unless \
+            config.ssh_private_key_file.nil?
+          options[:certificate_file] = config.ssh_certificate_file unless \
+            config.ssh_certificate_file.nil?
+          options[:ssh_port] = config.ssh_port unless \
+            config.ssh_port.nil?
+          options[:vm_size] = config.vm_size unless \
+            config.vm_size.nil?
+          options[:winrm_transport] = config.winrm_transport unless \
+            config.winrm_transport.nil?
+          options[:winrm_http_port] = config.winrm_http_port unless \
+            config.winrm_http_port.nil?
+          options[:winrm_https_port] = config.winrm_https_port unless \
+            config.winrm_https_port.nil?
+          options[:availability_set_name] = config.availability_set_name unless \
+            config.availability_set_name.nil?
+
+          add_role = false
+
+          env[:ui].info(params.inspect)
+          env[:ui].info(options.inspect)
+
+          # Check if the cloud service exists and if yes, does it contain
+          # a deployment.
+          if config.cloud_service_name && !config.cloud_service_name.empty?
+            begin
+              cloud_service = ManagementHttpRequest.new(
+                :get,
+                "/services/hostedservices/#{config.cloud_service_name}?embed-detail=true"
+              ).call
+
+              deployments = cloud_service.css 'HostedService Deployments Deployment'
+
+              # Lets see if any deployments exist. Set add_role = true if yes.
+              # We're not worried about deployment slots, because the SDK has
+              # hard coded 'Production' as deployment slot and you can have only
+              # one deployment per deployment slot.
+              add_role = deployments.length == 1
+            rescue Exception => e
+              add_role = false
+            end
+          end
+          env[:ui].info("Add Role? - #{add_role}")
+
+          server = env[:azure_vm_service].create_virtual_machine(
+            params, options, add_role
+          )
+
+          # TODO: Exception/Error Handling
+
+          if server.instance_of? String
+            env[:ui].info "Server not created. Error is: #{server}"
+            raise "#{server}"
+          end
+
+          env[:machine].id = "#{server.vm_name}@#{server.cloud_service_name}"
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/action/start_instance.rb

@@ -0,0 +1,35 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+
+# require 'vagrant/util/retryable'
+
+# Barebones basic implemenation. This a work in progress in very early stages
+module VagrantPlugins
+  module WinAzure
+    module Action
+      # This starts a stopped instance
+      class StartInstance
+        # include Vagrant:Util::Retryable
+
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_azure:action::start_instance')
+        end
+
+        def call(env)
+          env[:machine].id = "#{env[:machine].provider_config.vm_name}@#{env[:machine].provider_config.cloud_service_name}" unless env[:machine].id
+          env[:machine].id =~ /@/
+
+          env[:ui].info "Attempting to start '#{$`}' in '#{$'}'"
+
+          env[:azure_vm_service].start_virtual_machine($`, $')
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/action/stop_instance.rb

@@ -0,0 +1,38 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+
+# Barebones basic implemenation. This a work in progress in very early stages
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class StopInstance
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_azure::action::stop_instance')
+        end
+
+        def call(env)
+          if env[:machine].state.id == :StoppedDeallocated
+            env[:ui].info(
+              I18n.t('vagrant_azure.already_status', :status => 'stopped.')
+            )
+          else
+            env[:machine].id =~ /@/
+            env[:ui].info(
+              I18n.t(
+                'vagrant_azure.stopping',
+                :vm_name => $`,
+                :cloud_service_name => $'
+              )
+            )
+            env[:azure_vm_service].shutdown_virtual_machine($`, $')
+          end
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/action/terminate_instance.rb

@@ -0,0 +1,34 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class TerminateInstance
+        def initialize(app, env)
+          @app = app
+          @logger = Log4r::Logger.new('vagrant_azure::action::terminate_instance')
+        end
+
+        def call(env)
+          env[:machine].id =~ /@/
+
+          vm = env[:azure_vm_service].get_virtual_machine($`, $')
+
+          if vm.nil?
+            # machine not found. assuming it was not created or destroyed
+            env[:ui].info (I18n.t('vagrant_azure.not_created'))
+          else
+            env[:azure_vm_service].delete_virtual_machine($`, $')
+            env[:machine].id = nil
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/action/wait_for_state.rb

@@ -0,0 +1,49 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'log4r'
+require 'timeout'
+
+module VagrantPlugins
+  module WinAzure
+    module Action
+      class WaitForState
+        def initialize(app, env, state)
+          @app = app
+          @state = state
+          @logger = Log4r::Logger.new("vagrant_azure::action::wait_for_state")
+        end
+
+        def call(env)
+          env[:result] = true
+
+          if env[:machine].state.id == @state
+            @logger.info(
+              I18n.t('vagrant_azure.already_status', :status => @state)
+            )
+          else
+            timeout = env[:machine].provider_config.state_read_timeout
+
+            env[:ui].info "Waiting for machine to reach state #{@state}"
+            @logger.info("Waiting for machine to reach state #{@state}")
+
+            begin
+              Timeout.timeout(timeout)  do
+                until env[:machine].state.id == @state
+                  sleep 30
+                end
+              end
+              env[:ui].success "Machine reached state #{@state}"
+            rescue Timeout::Error
+              env[:ui].error "Machine failed to reached state '#{@state}' in '#{timeout}' seconds."
+              env[:result] = false # couldn't reach state in time
+            end
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-azure/command/rdp/command.rb

@@ -0,0 +1,21 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+module VagrantPlugins
+  module WinAzure
+    class Command < Vagrant.plugin('2', :command)
+      def self.synopsis
+        'Opens an RDP session for a vagrant machine'
+      end
+
+      def execute
+        with_target_vms do |vm|
+          vm.action(:rdp)
+        end
+
+        0
+      end
+    end
+  end
+end

data/lib/vagrant-azure/config.rb

@@ -0,0 +1,147 @@
+#--------------------------------------------------------------------------
+# Copyright (c) Microsoft Open Technologies, Inc.
+# All Rights Reserved. Licensed under the Apache 2.0 License.
+#--------------------------------------------------------------------------
+require 'vagrant'
+require 'azure'
+
+module VagrantPlugins
+  module WinAzure
+    class Config < Vagrant.plugin('2', :config)
+      attr_accessor :mgmt_certificate
+      attr_accessor :mgmt_endpoint
+      attr_accessor :subscription_id
+      attr_accessor :storage_acct_name
+      attr_accessor :storage_access_key
+
+      attr_accessor :vm_name
+      attr_accessor :vm_user
+      attr_accessor :vm_password
+      attr_accessor :vm_image
+      attr_accessor :vm_location
+      attr_accessor :vm_affinity_group
+
+      attr_accessor :cloud_service_name
+      attr_accessor :deployment_name
+      attr_accessor :tcp_endpoints
+      attr_accessor :ssh_private_key_file
+      attr_accessor :ssh_certificate_file
+      attr_accessor :ssh_port
+      attr_accessor :vm_size
+      attr_accessor :winrm_transport
+      attr_accessor :winrm_http_port
+      attr_accessor :winrm_https_port
+      attr_accessor :availability_set_name
+
+      attr_accessor :state_read_timeout
+
+      def initialize
+        @storage_acct_name = UNSET_VALUE
+        @storage_access_key = UNSET_VALUE
+        @mgmt_certificate = UNSET_VALUE
+        @mgmt_endpoint = UNSET_VALUE
+        @subscription_id = UNSET_VALUE
+
+        @vm_name = UNSET_VALUE
+        @vm_user = UNSET_VALUE
+        @vm_password = UNSET_VALUE
+        @vm_image = UNSET_VALUE
+        @vm_location = UNSET_VALUE
+        @vm_affinity_group = UNSET_VALUE
+
+        @cloud_service_name = UNSET_VALUE
+        @deployment_name = UNSET_VALUE
+        @tcp_endpoints = UNSET_VALUE
+        @ssh_private_key_file = UNSET_VALUE
+        @ssh_certificate_file = UNSET_VALUE
+        @ssh_port = UNSET_VALUE
+        @vm_size = UNSET_VALUE
+        @winrm_transport = UNSET_VALUE
+        @winrm_http_port = UNSET_VALUE
+        @winrm_https_port = UNSET_VALUE
+        @availability_set_name = UNSET_VALUE
+        @state_read_timeout = UNSET_VALUE
+      end
+
+      def finalize!
+        @storage_acct_name = ENV["AZURE_STORAGE_ACCOUNT"] if \
+          @storage_acct_name == UNSET_VALUE
+        @storage_access_key = ENV["AZURE_STORAGE_ACCESS_KEY"] if \
+          @storage_access_key == UNSET_VALUE
+        @mgmt_certificate = ENV["AZURE_MANAGEMENT_CERTIFICATE"] if \
+          @mgmt_certificate == UNSET_VALUE
+        @mgmt_endpoint = ENV["AZURE_MANAGEMENT_ENDPOINT"] if \
+          @mgmt_endpoint == UNSET_VALUE
+        @subscription_id = ENV["AZURE_SUBSCRIPTION_ID"] if \
+          @subscription_id == UNSET_VALUE
+
+        @vm_name = nil if @vm_name == UNSET_VALUE
+        @vm_user = 'vagrant' if @vm_user == UNSET_VALUE
+        @vm_password = nil if @vm_password == UNSET_VALUE
+        @vm_image = nil if @vm_image == UNSET_VALUE
+        @vm_location = nil if @vm_location == UNSET_VALUE
+        @vm_affinity_group = nil if @vm_affinity_group == UNSET_VALUE
+
+        @cloud_service_name = nil if @cloud_service_name == UNSET_VALUE
+        @deployment_name = nil if @deployment_name == UNSET_VALUE
+        @tcp_endpoints = nil if @tcp_endpoints == UNSET_VALUE
+        @ssh_private_key_file = nil if @ssh_private_key_file == UNSET_VALUE
+        @ssh_certificate_file = nil if @ssh_certificate_file == UNSET_VALUE
+        @ssh_port = nil if @ssh_port == UNSET_VALUE
+        @vm_size = nil if @vm_size == UNSET_VALUE
+        @winrm_transport = nil if @winrm_transport == UNSET_VALUE
+        @winrm_http_port = nil if @winrm_http_port == UNSET_VALUE
+        @winrm_https_port = nil if @winrm_https_port == UNSET_VALUE
+        @availability_set_name = nil if @availability_set_name == UNSET_VALUE
+
+        @state_read_timeout = 360 if @state_read_timeout == UNSET_VALUE
+
+        # This done due to a bug in Ruby SDK - it doesn't generate a storage 
+        # account name if add_role = true
+        if @storage_acct_name.nil? || @storage_acct_name.empty?
+          @storage_acct_name = Azure::Core::Utility.random_string(
+            "#{@vm_name}storage"
+          ).gsub(/[^0-9a-z ]/i, '').downcase[0..23]
+        end
+
+        if @cloud_service_name.nil? || @cloud_service_name.empty?
+          @cloud_service_name = Azure::Core::Utility.random_string(
+            "#{@vm_name}-service-"
+          )
+        end
+      end
+
+      def merge(other)
+        super.tap do |result|
+          result.mgmt_certificate = other.mgmt_certificate || \
+            self.mgmt_certificate
+          result.mgmt_endpoint = other.mgmt_endpoint || \
+            self.mgmt_endpoint
+          result.subscription_id = other.subscription_id || \
+            self.subscription_id
+          result.storage_account_name = other.storage_acct_name || \
+            self.storage_acct_name
+          result.storage_access_key = other.storage_access_key || \
+            self.storage_access_key
+        end
+      end
+
+      def validate(machine)
+        errors = _detected_errors
+
+        # Azure connection properties related validation.
+        errors << "vagrant_azure.subscription_id.required" if \
+          @subscription_id.nil?
+        errors << "vagrant_azure.mgmt_certificate.required" if \
+          @mgmt_certificate.nil?
+        errors << "vagrant_azure.mgmt_endpoint.required" if \
+          @mgmt_endpoint.nil?
+
+        # Azure Virtual Machine related validation
+        errors << "vagrant_azure.vm_name.required" if @vm_name.nil?
+
+        { "Windows Azure Provider" => errors }
+      end
+    end
+  end
+end