azure-armrest 0.0.1 → 0.0.2

data/lib/azure/armrest/{subnet_manager.rb → subnet_service.rb}

@@ -3,12 +3,12 @@ module Azure
   # Armrest namespace
   module Armrest
     # Base class for managing subnets
-    class SubnetManager < VirtualNetworkManager
+    class SubnetService < VirtualNetworkService
 
-      # Create and return a new SubnetManager instance. Most methods for a
-      # SubnetManager instance will return one or Subnet instances.
+      # Create and return a new SubnetService instance. Most methods for a
+      # SubnetService instance will return one or Subnet instances.
       #
-      def initialize(options = {})
+      def initialize(_armrest_configuration, _options = {})
         super
 
         @base_url += "/subnets"

data/lib/azure/armrest/template_deployment_service.rb

@@ -0,0 +1,97 @@
+module Azure::Armrest
+# Base class for managing templates and deployments
+class TemplateDeploymentService < ArmrestService
+
+  def initialize(_armrest_configuration, _options = {})
+    super
+
+    # has to be hard coded for now
+    set_service_api_version({'api_version' => '2014-04-01-preview'}, '')
+  end
+
+  # Get names of all deployments in a resource group
+  def list_names(resource_group = armrest_configuration.resource_group)
+    list(resource_group).map {|e| e['name']}
+  end
+
+  # Get all deployments in a resource group
+  # If the resource group is nil, then return deployments in all groups
+  def list(resource_group = armrest_configuration.resource_group)
+    if resource_group
+      url = build_deployment_url(resource_group)
+      JSON.parse(rest_get(url))['value']
+    else
+      threads = []
+      array = []
+      mutex = Mutex.new
+
+      resource_groups.each do |rg|
+        threads << Thread.new(rg['name']) do |group|
+          url = build_deployment_url(group)
+          results = JSON.parse(rest_get(url))['value']
+          if results && !results.empty?
+            mutex.synchronize {
+              results.each { |hash| hash['resourceGroup'] = group }
+              array << results
+            }
+          end
+        end
+      end
+
+      threads.each(&:join)
+      array.flatten
+    end
+  end
+
+  # Get the deployment in a resource group
+  def get(deploy_name, resource_group = armrest_configuration.resource_group)
+    url = build_deployment_url(resource_group, deploy_name)
+    JSON.parse(rest_get(url))
+  end
+
+  # Get all operations of a deployment in a resource group
+  def list_deployment_operations(deploy_name, resource_group = armrest_configuration.resource_group)
+    url = build_deployment_url(resource_group, deploy_name, 'operations')
+    JSON.parse(rest_get(url))['value']
+  end
+
+  # Get the operation of a deployment in a resource group
+  def get_deployment_operation(deploy_name, op_id, resource_group = armrest_configuration.resource_group)
+    url = build_deployment_url(resource_group, deploy_name, 'operations', op_id)
+    JSON.parse(rest_get(url))
+  end
+
+  # Create a template deployment
+  # The template and parameters should be provided through the options hash
+  def create(deploy_name, options, resource_group = armrest_configuration.resource_group)
+    url = build_deployment_url(resource_group, deploy_name)
+    body = options.has_key?('properties') ? options.to_json : {'properties' => options}.to_json
+    JSON.parse(rest_put(url, body))
+  end
+
+  # Delete a deployment
+  def delete(deploy_name, resource_group = armrest_configuration.resource_group)
+    url = build_deployment_url(resource_group, deploy_name)
+    rest_delete(url)
+  end
+
+  private
+
+
+  # Builds a URL based on subscription_id an resource_group and any other
+  # arguments provided, and appends it with the api_version.
+  #
+  def build_deployment_url(resource_group, *args)
+    url = File.join(
+      Azure::Armrest::COMMON_URI,
+      armrest_configuration.subscription_id,
+      'resourceGroups',
+      resource_group,
+      'deployments',
+    )
+
+    url = File.join(url, *args) unless args.empty?
+    url << "?api-version=#{@api_version}"
+  end
+end
+end

data/lib/azure/armrest/version.rb

@@ -1,5 +1,5 @@
 module Azure
   module Armrest
-    VERSION = "0.0.1"
+    VERSION = "0.0.2"
   end
 end

data/lib/azure/armrest/{virtual_machine_extension_manager.rb → virtual_machine_extension_service.rb}

@@ -3,7 +3,11 @@ module Azure
   # Armrest namespace
   module Armrest
     # Base class for managing virtual machine extensions
-    class VirtualMachineExtensionManager < VirtualMachineManager
+    class VirtualMachineExtensionService < VirtualMachineService
+
+      def initialize(_armrest_configuration, _options = {})
+        super
+      end
 
       # Creates a new virtual machine extension for +vmname+ with the given
       # +extension_name+, and the given +options+. Possible options are:
@@ -20,7 +24,7 @@ module Azure
         #type = options.fetch(:type)
         #type_handler_version = options.fetch(:type_handler_version)
 
-        url = @uri + "/#{vmname}/extensions/#{extension_name}?#{api_version}"
+        url = @uri + "/#{vmname}/extensions/#{extension_name}?#{@api_version}"
         url
       end
 
@@ -31,7 +35,7 @@ module Azure
       # DELETE
       #
       def delete(vmname, extension_name)
-        url = @uri + "/#{vmname}/extensions/#{extension_name}?#{api_version}"
+        url = @uri + "/#{vmname}/extensions/#{extension_name}?#{@api_version}"
         url
       end
 
@@ -43,7 +47,7 @@ module Azure
       def get(vmname, instance_view = false)
         url = @uri + "/#{vmname}/extensions/#{extension_name}?"
         url += "$expand=instanceView," if instance_view
-        url += "#{api_version}"
+        url += "#{@api_version}"
         url
       end
 
@@ -51,7 +55,7 @@ module Azure
       def list(vmname, instance_view = false)
         url = @uri + "/#{vmname}/extensions"
         url += "$expand=instanceView," if instance_view
-        url += "#{api_version}"
+        url += "#{@api_version}"
         url
       end
     end

data/lib/azure/armrest/{virtual_machine_image_manager.rb → virtual_machine_image_service.rb}

@@ -3,7 +3,7 @@ module Azure
   # Armrest namespace
   module Armrest
     # Base class for managing virtual machine images
-    class VirtualMachineImageManager < ArmrestManager
+    class VirtualMachineImageService < ArmrestService
       # The location used in requests when gathering VM image information.
       attr_accessor :location
 
@@ -13,23 +13,20 @@ module Azure
       # The publisher used in requests when gathering VM image information.
       attr_accessor :publisher
 
-      # Create and return a new VirtualMachineImageManager (VMIM) instance.
+      # Create and return a new VirtualMachineImageService (VMIM) instance.
       #
       # This subclass accepts the additional :location, :provider, and
       # :publisher options as well. The default provider is set to
       # 'Microsoft.Compute'.
       #
-      def initialize(options = {})
+      def initialize(_armrest_configuration, options = {})
         super
 
         @location  = options[:location]
         @provider  = options[:provider] || 'Microsoft.Compute'
         @publisher = options[:publisher]
 
-        # Typically only empty in testing.
-        unless @@providers.empty?
-          @api_version = @@providers[@provider]['locations/publishers']['api_version']
-        end
+        set_service_api_version(options, 'locations/publishers')
       end
 
       # Set a new provider to use the default for other methods. This may alter
@@ -37,8 +34,8 @@ module Azure
       # 'Microsoft.Compute' or 'Microsoft.ClassicCompute' should be used.
       #
       def provider=(name)
-        @api_version = @@providers[name]['locations/publishers']['api_version']
         @provider = name
+        set_service_api_version({}, 'locations/publishers')
       end
 
       # Return a list of VM image offers from the given +publisher+ and +location+.
@@ -119,7 +116,7 @@ module Azure
       def build_url(location, *args)
         url = File.join(
           Azure::Armrest::COMMON_URI,
-          subscription_id,
+          armrest_configuration.subscription_id,
           'providers',
           provider,
           'locations',
@@ -130,6 +127,6 @@ module Azure
         url << "?api-version=#{api_version}"
       end
 
-    end # VirtualMachineImageManager
+    end # VirtualMachineImageService
   end # Armrest
 end # Azure

data/lib/azure/armrest/{virtual_machine_manager.rb → virtual_machine_service.rb}

@@ -3,12 +3,12 @@ module Azure
   # Armrest namespace
   module Armrest
     # Base class for managing virtual machines
-    class VirtualMachineManager < ArmrestManager
+    class VirtualMachineService < ArmrestService
 
       # The provider used in requests when gathering VM information.
       attr_reader :provider
 
-      # Create and return a new VirtualMachineManager (VMM) instance. Most
+      # Create and return a new VirtualMachineService (VMM) instance. Most
       # methods for a VMM instance will return one or more VirtualMachine
       # instances.
       #
@@ -16,39 +16,34 @@ module Azure
       # default is 'Microsoft.ClassicCompute'. You may need to set this to
       # 'Microsoft.Compute' for your purposes.
       #
-      def initialize(options = {})
+      def initialize(_armrest_configuration, options = {})
         super
-
         @provider = options[:provider] || 'Microsoft.Compute'
-
-        # Typically only empty in testing.
-        unless @@providers.empty?
-          @api_version = @@providers[@provider]['virtualMachines']['api_version']
-        end
+        set_service_api_version(options, 'virtualMachines')
       end
 
       # Set a new provider to use the default for other methods. This may alter
       # the api_version used for future requests. In practice, only
       # 'Microsoft.Compute' or 'Microsoft.ClassicCompute' should be used.
       #
-      def provider=(name)
-        @api_version = @@providers[name]['virtualMachines']['api_version']
+      def provider=(name, options = {})
         @provider = name
+        set_service_api_version(options, 'virtualMachines')
       end
 
       # Return a list of available VM series (aka sizes, flavors, etc), such
       # as "Basic_A1", though information is included as well.
       #
       def series(location)
-        unless @@providers[@provider] && @@providers[@provider]['locations/vmSizes']
+        unless @@providers_hash[@provider.downcase] && @@providers_hash[@provider.downcase]['locations/vmSizes']
           raise ArgumentError, "Invalid provider '#{provider}'"
         end
 
-        version = @@providers[@provider]['locations/vmSizes']['api_version']
+        version = @@providers_hash[@provider.downcase]['locations/vmSizes']['api_version']
 
         url = url_with_api_version(
-          version, @base_url, 'subscriptions', subscription_id, 'providers',
-          provider, 'locations', location, 'vmSizes'
+          version, @base_url, 'subscriptions', armrest_configuration.subscription_id,
+          'providers', provider, 'locations', location, 'vmSizes'
         )
 
         JSON.parse(rest_get(url))['value']
@@ -72,8 +67,9 @@ module Azure
       # p JSON.parse(resp.body)["value"][0]["properties"]["hardwareProfile"]
       # p JSON.parse(resp.body)["value"][0]["properties"]["storageProfile"]
       #
-      def list(group = nil)
+      def list(group = armrest_configuration.resource_group)
         array = []
+
         if group
           url = build_url(group)
           array << JSON.parse(rest_get(url))['value']
@@ -81,63 +77,67 @@ module Azure
           threads = []
           mutex = Mutex.new
 
-          resource_groups.each do |group|
-            url = build_url(group['name'])
-
-            threads << Thread.new(url) do |thread_url|
-              response = rest_get(thread_url)
+          resource_groups.each do |rg|
+            threads << Thread.new(rg['name']) do |rgroup|
+              response = rest_get(build_url(rgroup))
               result = JSON.parse(response)['value']
-              mutex.synchronize{ array << result if result }
+              if result
+                mutex.synchronize{
+                  result.each{ |hash| hash['resourceGroup'] = rgroup }
+                  array << result
+                }
+              end
             end
           end
 
           threads.each(&:join)
 
           array = array.flatten
+
+          if provider.downcase == 'microsoft.compute'
+            add_network_profile(array)
+            add_power_status(array)
+          end
         end
 
-        add_network_profile(array)
+        array
       end
 
       alias get_vms list
 
-      def add_network_profile(vms)
-        vms.each { |vm|
-          vm['properties']['networkProfile']['networkInterfaces'].each { |net|
-            get_nic_profile(net)
-          }
-        }
-      end
-
-      def get_nic_profile(nic)
-        url = File.join(
-          Azure::Armrest::RESOURCE,
-          nic['id'],
-          "?api-version=#{api_version}"
-        )
+      # Captures the +vmname+ and associated disks into a reusable CSM template.
+      # The 3rd argument is a hash of options that supports the following keys:
+      #
+      # * prefix    - The prefix in the name of the blobs.
+      # * container - The name of the container inside which the image will reside.
+      # * overwrite - Boolean that indicates whether or not to overwrite any VHD's
+      #               with the same prefix. The default is false.
+      #
+      # The :prefix and :container options are mandatory. You may optionally
+      # specify the :resource_group as an option, or as the 3rd argument, but
+      # it is also mandatory if not already set in the constructor.
+      #
+      # Note that this is a long running operation. You are expected to
+      # poll on the client side to check the status of the operation.
+      #
+      def capture(vmname, options = {}, group = nil)
+        prefix = options.fetch(:prefix)
+        container = options.fetch(:container)
+        overwrite = options[:overwrite] || false
+        group ||= options[:resource_group] || armrest_configuration.resource_group
 
-        nic['properties'] = JSON.parse(rest_get(url))['properties']['ipConfigurations']
-        nic['properties'].each do |n|
-          next if n['properties']['publicIPAddress'].nil?
+        raise ArgumentError, "no resource group provided" unless group
 
-          # public IP is a URI so we need to make another rest call to get it.
-          url = File.join(
-            Azure::Armrest::RESOURCE,
-            n['properties']['publicIPAddress']['id'],
-            "?api-version=#{api_version}"
-          )
+        body = {
+          :vhdPrefix => prefix,
+          :destinationContainer => container,
+          :overwriteVhds => overwrite
+        }.to_json
 
-          public_ip = JSON.parse(rest_get(url))['properties']['ipAddress']
-          n['properties']['publicIPAddress'] = public_ip
-        end
-      end
+        url = build_url(group, 'capture')
 
-      # Captures the +vmname+ and associated disks into a reusable CSM template.
-      #--
-      # POST
-      def capture(vmname, action = 'capture')
-        uri = @uri + "/#{vmname}/#{action}?api-version=#{api_version}"
-        uri
+        response = rest_post(url, body)
+        response.return!
       end
 
       # Creates a new virtual machine (or updates an existing one). Pass a hash
@@ -202,7 +202,7 @@ module Azure
       #
       # Example:
       #
-      #   vmm = VirtualMachineManager.new(x, y, z)
+      #   vmm = VirtualMachineService.new(x, y, z)
       #
       #   vm = vmm.create(
       #     :name            => 'test1',
@@ -224,23 +224,16 @@ module Azure
       #   )
       #--
       # PUT operation
-      #
-      def create(options = {})
-        #name = options.fetch(:name)
-        #location = options.fetch(:location)
-        #tags = option[:tags]
-        vmsize = options.fetch(:vmsize)
-
-        unless VALID_VM_SIZES.include?(vmsize)
-          raise ArgumentError, "Invalid vmsize '#{vmsize}'"
-        end
-      end
+      # TODO: Implement
+      #def create(options = {})
+      #end
 
-      alias update create
+      #alias update create
 
       # Stop the VM +vmname+ in +group+ and deallocate the tenant in Fabric.
       #
-      def deallocate(vmname, group = vmname)
+      def deallocate(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         url = build_url(group, vmname, 'deallocate')
         response = rest_post(url)
         response.return!
@@ -249,7 +242,8 @@ module Azure
       # Deletes the +vmname+ in +group+ that you specify. Note that associated
       # disks are not deleted.
       #
-      def delete(vmname, group = vmname)
+      def delete(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         url = build_url(group, vmname)
         response = rest_delete(url)
         response.return!
@@ -257,7 +251,8 @@ module Azure
 
       # Sets the OSState for the +vmname+ in +group+ to 'Generalized'.
       #
-      def generalize(vmname, group = vmname)
+      def generalize(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         url = build_url(group, vmname, 'generalize')
         response = rest_post(url)
         response.return!
@@ -270,7 +265,9 @@ module Azure
       # parameter is false, it will retrieve an instance view. The difference is
       # in the details of the information retrieved.
       #
-      def get(vmname, group = vmname, model_view = true)
+      def get(vmname, group = armrest_configuration.resource_group, model_view = true)
+        raise ArgumentError, "no resource group provided" unless group
+
         if model_view
           url = build_url(group, vmname)
         else
@@ -283,14 +280,16 @@ module Azure
       # Convenient wrapper around the get method that retrieves the model view
       # for +vmname+ in resource_group +group+.
       #
-      def get_model_view(vmname, group = vmname)
+      def get_model_view(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         get(vmname, group, true)
       end
 
       # Convenient wrapper around the get method that retrieves the instance view
       # for +vmname+ in resource_group +group+.
       #
-      def get_instance_view(vmname, group = vmname)
+      def get_instance_view(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         get(vmname, group, false)
       end
 
@@ -300,7 +299,8 @@ module Azure
       # This is an asynchronous operation that returns a response object
       # which you can inspect, such as response.code or response.headers.
       #
-      def restart(vmname, group = vmname)
+      def restart(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         url = build_url(group, vmname, 'restart')
         response = rest_post(url)
         response.return!
@@ -312,7 +312,8 @@ module Azure
       # This is an asynchronous operation that returns a response object
       # which you can inspect, such as response.code or response.headers.
       #
-      def start(vmname, group = vmname)
+      def start(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         url = build_url(group, vmname, 'start')
         response = rest_post(url)
         response.return!
@@ -324,7 +325,8 @@ module Azure
       # This is an asynchronous operation that returns a response object
       # which you can inspect, such as response.code or response.headers.
       #
-      def stop(vmname, group = vmname)
+      def stop(vmname, group = armrest_configuration.resource_group)
+        raise ArgumentError, "no resource group provided" unless group
         url = build_url(group, vmname, 'powerOff')
         response = rest_post(url)
         response.return!
@@ -332,6 +334,45 @@ module Azure
 
       private
 
+      def add_network_profile(vms)
+        vms.each { |vm|
+          vm['properties']['networkProfile']['networkInterfaces'].each { |net|
+            get_nic_profile(net)
+          }
+        }
+      end
+
+      def get_nic_profile(nic)
+        url = File.join(
+          Azure::Armrest::RESOURCE,
+          nic['id'],
+          "?api-version=#{@api_version}"
+        )
+
+        nic['properties'] = JSON.parse(rest_get(url))['properties']['ipConfigurations']
+        nic['properties'].each do |n|
+          next if n['properties']['publicIPAddress'].nil?
+
+          # public IP is a URI so we need to make another rest call to get it.
+          url = File.join(
+            Azure::Armrest::RESOURCE,
+            n['properties']['publicIPAddress']['id'],
+            "?api-version=#{@api_version}"
+          )
+
+          public_ip = JSON.parse(rest_get(url))['properties']['ipAddress']
+          n['properties']['publicIPAddress'] = public_ip
+        end
+      end
+
+      def add_power_status(vms)
+        vms.each do |vm|
+          i_view            = get_instance_view(vm["name"], vm["resourceGroup"])
+          powerstatus_hash  = i_view["statuses"].find {|h| h["code"].include? "PowerState"}
+          vm["powerStatus"] = powerstatus_hash['displayStatus'] unless powerstatus_hash.nil?
+        end
+      end
+
       # If no default subscription is set, then use the first one found.
       def set_default_subscription
         @subscription_id ||= subscriptions.first['subscriptionId']
@@ -343,7 +384,7 @@ module Azure
       def build_url(resource_group, *args)
         url = File.join(
           Azure::Armrest::COMMON_URI,
-          subscription_id,
+          armrest_configuration.subscription_id,
           'resourceGroups',
           resource_group,
           'providers',
@@ -352,7 +393,7 @@ module Azure
         )
 
         url = File.join(url, *args) unless args.empty?
-        url << "?api-version=#{api_version}"
+        url << "?api-version=#{@api_version}"
       end
     end
   end