fog-azure-rm 0.1.2 → 0.2.0

data/lib/fog/azurerm/requests/application_gateway/check_ag_exists.rb

@@ -0,0 +1,31 @@
+module Fog
+  module ApplicationGateway
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_ag_exists(resource_group_name, application_gateway_name)
+          msg = "Checking Application Gateway: #{application_gateway_name}"
+          Fog::Logger.debug msg
+          begin
+            @network_client.application_gateways.get(resource_group_name, application_gateway_name)
+            Fog::Logger.debug "Application Gateway #{application_gateway_name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Application Gateway #{application_gateway_name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_ag_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/application_gateway/create_or_update_application_gateway.rb

@@ -82,6 +82,12 @@ module Fog
             frontend_ip_configuration.private_ipaddress = fic[:private_ip_address]
             frontend_ip_configuration.private_ipallocation_method = fic[:private_ip_allocation_method]
 
+            if fic[:subnet_id]
+              subnet = Azure::ARM::Network::Models::Subnet.new
+              subnet.id = fic[:subnet_id]
+              frontend_ip_configuration.subnet = subnet
+            end
+
             unless fic[:public_ip_address_id].nil?
               pip = Azure::ARM::Network::Models::PublicIPAddress.new
               pip.id = fic[:public_ip_address_id]

data/lib/fog/azurerm/requests/compute/check_availability_set_exists.rb

@@ -0,0 +1,31 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_availability_set_exists(resource_group, name)
+          msg = "Checking Availability set: #{name}"
+          Fog::Logger.debug msg
+          begin
+            @compute_mgmt_client.availability_sets.get(resource_group, name)
+            Fog::Logger.debug "Availability set #{name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Availability set #{name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_availability_set_exists(_resource_group, _name)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/check_vm_exists.rb

@@ -0,0 +1,31 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_vm_exists(resource_group, name)
+          msg = "Checking Virtual Machine #{name}"
+          Fog::Logger.debug msg
+          begin
+            @compute_mgmt_client.virtual_machines.get(resource_group, name, 'instanceView')
+            Fog::Logger.debug "Virtual machine #{name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Virtual machine #{name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_vm_exists(_resource_group, _name)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/check_vm_extension_exists.rb

@@ -0,0 +1,31 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_vm_extension_exists(resource_group_name, virtual_machine_name, vm_extension_name)
+          msg = "Checking Virtual Machine Extension #{vm_extension_name}"
+          Fog::Logger.debug msg
+          begin
+            @compute_mgmt_client.virtual_machine_extensions.get(resource_group_name, virtual_machine_name, vm_extension_name)
+            Fog::Logger.debug "Virtual Machine Extension #{vm_extension_name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Virtual machine #{vm_extension_name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_vm_extension_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/check_record_set_exists.rb

@@ -0,0 +1,31 @@
+module Fog
+  module DNS
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_record_set_exists(resource_group, name, zone_name, record_type)
+          msg = "Checking Record set #{name}"
+          Fog::Logger.debug msg
+          begin
+            @dns_client.record_sets.get(resource_group, zone_name, name, record_type)
+            Fog::Logger.debug "Record set #{name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['code'] == 'NotFound'
+              Fog::Logger.debug "Record set #{name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_record_set_exists(_resource_group, _name)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/{check_for_zone.rb → check_zone_exists.rb}

@@ -3,13 +3,15 @@ module Fog
     class AzureRM
       # Real class for DNS Request
       class Real
-        def check_for_zone(resource_group, name)
+        def check_zone_exists(resource_group, name)
           msg = "Getting Zone #{name} from Resource Group #{resource_group}."
           Fog::Logger.debug msg
           begin
             zone = @dns_client.zones.get(resource_group, name)
           rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
+          rescue => e
+            Fog::Logger.debug e[:error][:code]
           end
           !zone.nil?
         end
@@ -17,7 +19,7 @@ module Fog
 
       # Mock class for DNS Request
       class Mock
-        def check_for_zone(*)
+        def check_zone_exists(*)
           Fog::Logger.debug 'Zone name name is available.'
           true
         end

data/lib/fog/azurerm/requests/dns/list_record_sets.rb

@@ -7,7 +7,7 @@ module Fog
           msg = 'Getting list of Record sets.'
           Fog::Logger.debug msg
           begin
-            zones = @dns_client.record_sets.list_all_in_resource_group(resource_group, zone_name)
+            zones = @dns_client.record_sets.list_by_dns_zone(resource_group, zone_name)
           rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end

data/lib/fog/azurerm/requests/dns/list_zones.rb

@@ -7,7 +7,7 @@ module Fog
           msg = 'Getting list of Zones.'
           Fog::Logger.debug msg
           begin
-            zones = @dns_client.zones.list_in_subscription
+            zones = @dns_client.zones.list
           rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end
@@ -20,7 +20,7 @@ module Fog
           msg = "Getting list of Zones in Resource Group #{resource_group}."
           Fog::Logger.debug msg
           begin
-            zones = @dns_client.zones.list_in_resource_group(resource_group)
+            zones = @dns_client.zones.list_by_resource_group(resource_group)
           rescue MsRestAzure::AzureOperationError => e
             raise_azure_exception(e, msg)
           end

data/lib/fog/azurerm/requests/key_vault/check_vault_exists.rb

@@ -0,0 +1,31 @@
+module Fog
+  module KeyVault
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_vault_exists(resource_group, vault_name)
+          msg = "Checking Vault #{vault_name}"
+          Fog::Logger.debug msg
+          begin
+            @key_vault_client.vaults.get(resource_group, vault_name)
+            Fog::Logger.debug "Vault #{vault_name} exists."
+            true
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Vault #{vault_name} doesn't exist."
+              false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_vault_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/key_vault/create_or_update_vault.rb

@@ -0,0 +1,105 @@
+module Fog
+  module KeyVault
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def create_or_update_vault(vault_hash)
+          msg = "Creating Vault: #{vault_hash[:name]}."
+          Fog::Logger.debug msg
+          vault_parameters = get_vault_param_object(vault_hash)
+          begin
+            vault = @key_vault_client.vaults.create_or_update(vault_hash[:resource_group], vault_hash[:name], vault_parameters)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Vault: #{vault_hash[:name]} created successfully."
+          vault
+        end
+
+        private
+
+        def get_vault_param_object(vault_hash)
+          vault_param = Azure::ARM::KeyVault::Models::VaultCreateOrUpdateParameters.new
+          vault_param.location = vault_hash[:location]
+          vault_param.properties = get_vault_properties_object(vault_hash)
+          vault_param
+        end
+
+        def get_vault_properties_object(vault_hash)
+          vault_properties = Azure::ARM::KeyVault::Models::VaultProperties.new
+          vault_properties.tenant_id = vault_hash[:tenant_id]
+          vault_properties.sku = get_sku_object(vault_hash)
+          vault_properties.access_policies = get_access_policies_object(vault_hash[:access_policies])
+          vault_properties
+        end
+
+        def get_sku_object(vault_hash)
+          sku = Azure::ARM::KeyVault::Models::Sku.new
+          sku.family = vault_hash[:sku_family]
+          sku.name = vault_hash[:sku_name]
+          sku
+        end
+
+        def get_access_policies_object(access_policies_hash)
+          access_policies_arr = []
+          unless access_policies_hash.nil?
+            access_policies_hash.each do |access_policy_hash|
+              access_policies_arr.push(get_access_policy_entry_object(access_policy_hash))
+            end
+          end
+          access_policies_arr
+        end
+
+        def get_access_policy_entry_object(access_policy_hash)
+          access_policy_entry = Azure::ARM::KeyVault::Models::AccessPolicyEntry.new
+          access_policy_entry.tenant_id = access_policy_hash[:tenant_id]
+          access_policy_entry.object_id = access_policy_hash[:object_id]
+          access_policy_entry.permissions = get_permission_object(access_policy_hash[:permissions])
+          access_policy_entry
+        end
+
+        def get_permission_object(permission_hash)
+          permission = Azure::ARM::KeyVault::Models::Permissions.new
+          permission.keys = permission_hash[:keys]
+          permission.secrets = permission_hash[:secrets]
+          permission.certificates = permission_hash[:certificates]
+          permission
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def create_or_update_vault(*)
+          vault = {
+            'id' => '/subscriptions/<AZURE_SUBSCRIPTION_ID>/resourceGroups/RubySDKTest_azure_mgmt_kv/providers/Microsoft.KeyVault/vaults/sampleVault758347',
+            'name' => 'sampleVault758347',
+            'type' => 'Microsoft.KeyVault/vaults',
+            'location' => 'westus',
+            'tags' => {},
+            'properties' => {
+              'sku' => {
+                'family' => 'A',
+                'name' => 'standard'
+              },
+              'tenantId' => '<AZURE_TENANT_ID>',
+              'accessPolicies' => [
+                {
+                  'tenantId' => '<AZURE_TENANT_ID>',
+                  'objectId' => '<AZURE_OBJECT_ID>',
+                  'permissions' => {
+                    'keys' => ['all'],
+                    'secrets' => ['all']
+                  }
+                }
+              ],
+              'enabledForDeployment' => false,
+              'vaultUri' => 'https =>//samplevault758347.vault.azure.net/'
+            }
+          }
+          vault_mapper = Azure::ARM::KeyVault::Models::Vault.mapper
+          @key_vault_client.deserialize(vault_mapper, vault, 'result.body')
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/key_vault/delete_vault.rb

@@ -0,0 +1,28 @@
+module Fog
+  module KeyVault
+    class AzureRM
+      # Real class for KeyVault Request
+      class Real
+        def delete_vault(resource_group, vault_name)
+          msg = "Deleting Vault: #{vault_name}."
+          Fog::Logger.debug msg
+          begin
+            @key_vault_client.vaults.delete(resource_group, vault_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Vault: #{vault_name} deleted successfully."
+          true
+        end
+      end
+
+      # Mock class for KeyVault Request
+      class Mock
+        def delete_vault(resource_group, vault_name)
+          Fog::Logger.debug "Vault #{vault_name} from Resource group #{resource_group} deleted successfully."
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/key_vault/get_vault.rb

@@ -0,0 +1,54 @@
+module Fog
+  module KeyVault
+    class AzureRM
+      # Real class for KeyVault Request
+      class Real
+        def get_vault(resource_group, vault_name)
+          msg = "Getting Vault => #{vault_name} from Resource Group => #{resource_group}..."
+          Fog::Logger.debug msg
+          begin
+            vault = @key_vault_client.vaults.get(resource_group, vault_name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Vault fetched successfully from Resource Group => #{resource_group}"
+          vault
+        end
+      end
+
+      # Mock class for KeyVault Request
+      class Mock
+        def get_vault(*)
+          vault = {
+            'id' => '/subscriptions/<AZURE_SUBSCRIPTION_ID>/resourceGroups/RubySDKTest_azure_mgmt_kv/providers/Microsoft.KeyVault/vaults/sampleVault758347',
+            'name' => 'sampleVault758347',
+            'type' => 'Microsoft.KeyVault/vaults',
+            'location' => 'westus',
+            'tags' => {},
+            'properties' => {
+              'sku' => {
+                'family' => 'A',
+                'name' => 'standard'
+              },
+              'tenantId' => '<AZURE_TENANT_ID>',
+              'accessPolicies' => [
+                {
+                  'tenantId' => '<AZURE_TENANT_ID>',
+                  'objectId' => '<AZURE_OBJECT_ID>',
+                  'permissions' => {
+                    'keys' => ['all'],
+                    'secrets' => ['all']
+                  }
+                }
+              ],
+              'enabledForDeployment' => false,
+              'vaultUri' => 'https =>//samplevault758347.vault.azure.net/'
+            }
+          }
+          vault_mapper = Azure::ARM::KeyVault::Models::Vault.mapper
+          @key_vault_client.deserialize(vault_mapper, vault, 'result.body')
+        end
+      end
+    end
+  end
+end