fog-azure-rm 0.0.9 → 0.1.0

data/lib/fog/azurerm/requests/storage/get_backup_container.rb

@@ -0,0 +1,53 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def get_backup_container(resource_group, rv_name, vm_name)
+          msg = "Getting backup container from Recovery Vault #{rv_name}"
+          Fog::Logger.debug msg
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupProtectionContainers?api-version=2016-05-01&$filter=backupManagementType eq 'AzureIaasVM' and status eq 'Registered' and friendlyName eq '#{vm_name}'"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            response = RestClient.get(
+              response = resource_url,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Successfully retrieved backup container from Recovery Vault #{rv_name}"
+          JSON.parse(response)['value']
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def get_backup_container(*)
+          body = '{
+            "value": [{
+              "id": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault/backupFabrics/Azure/protectionContainers/IaasVMContainer;iaasvmcontainerv2;fog-test-vm-rg;fog-test-vm",
+              "name": "IaasVMContainer;iaasvmcontainerv2;fog-test-vm-rg;fog-test-vm",
+              "type": "Microsoft.RecoveryServices/vaults/backupFabrics/protectionContainers",
+              "properties": {
+                "virtualMachineId": "/subscriptions/########-####-####-####-############/resourceGroups/TestRG/providers/Microsoft.Compute/virtualMachines/TestVM",
+                "virtualMachineVersion": "Compute",
+                "resourceGroup": "fog-test-vm-rg",
+                "friendlyName": "fog-test-vm",
+                "backupManagementType": "AzureIaasVM",
+                "registrationStatus": "Registered",
+                "healthStatus": "Healthy",
+                "containerType": "Microsoft.Compute/virtualMachines",
+                "protectableObjectType": "Microsoft.Compute/virtualMachines"
+              }
+            }]
+          }'
+          JSON.parse(body)['value']
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/get_backup_item.rb

@@ -0,0 +1,58 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def get_backup_item(resource_group, rv_name)
+          msg = "Getting backup item from Recovery Vault #{rv_name}"
+          Fog::Logger.debug msg
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupProtectedItems?api-version=2016-05-01&$filter=backupManagementType eq 'AzureIaasVM' and itemType eq 'VM'"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            response = RestClient.get(
+              response = resource_url,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Successfully retrieved backup item from Recovery Vault #{rv_name}"
+          JSON.parse(response)['value']
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def get_backup_item(*)
+          body = '{
+            "value": [{
+              "id": "/Subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault/backupFabrics/Azure/protectionContainers/IaasVMContainer;iaasvmcontainerv2;testrg;testvm/protectedItems/VM;fog-test-container-name",
+              "name": "iaasvmcontainerv2;fog-test-vm-rg;fog-test-vm",
+              "type": "Microsoft.RecoveryServices/vaults/backupFabrics/protectionContainers/protectedItems",
+              "properties": {
+                "friendlyName": "fog-test-vm",
+                "virtualMachineId": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-vm-rg/providers/Microsoft.Compute/virtualMachines/fog-test-vm",
+                "protectionStatus": "Healthy",
+                "protectionState": "Protected",
+                "lastBackupStatus": "Completed",
+                "lastBackupTime": "2016-10-17T10:30:47.2289274Z",
+                "protectedItemType": "Microsoft.Compute/virtualMachines",
+                "backupManagementType": "AzureIaasVM",
+                "workloadType": "VM",
+                "containerName": "iaasvmcontainerv2;fog-test-vm-rg;fog-test-vm",
+                "sourceResourceId": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-vm-rg/providers/Microsoft.Compute/virtualMachines/fog-test-vm",
+                "policyId": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault/backupPolicies/DefaultPolicy",
+                "policyName": "DefaultPolicy",
+                "lastRecoveryPoint": "2016-10-17T10:32:38.4666692Z"
+              }
+            }]
+          }'
+          JSON.parse(body)['value']
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/get_backup_job_for_vm.rb

@@ -0,0 +1,53 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def get_backup_job_for_vm(rv_name, rv_resource_group, vm_name, vm_resource_group, operation)
+          msg = "Getting backup job for VM #{vm_name} in Resource Group #{vm_resource_group}"
+          Fog::Logger.debug msg
+
+          backup_jobs = get_all_backup_jobs(rv_name, rv_resource_group)
+          backup_jobs = backup_jobs.select do |job|
+            (job['properties']['status'].eql? 'InProgress') &&
+            (job['properties']['entityFriendlyName'].eql? vm_name.downcase) &&
+            (job['properties']['operation'].eql? operation)
+          end
+
+          backup_jobs[0]
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def get_backup_job_for_vm(*)
+          body = '{
+            "id": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault/backupJobs/########-####-####-####-############",
+            "name": "########-####-####-####-############",
+            "type": "Microsoft.RecoveryServices/vaults/backupJobs",
+            "properties": {
+              "jobType": "AzureIaaSVMJob",
+              "duration": "00:00:52.3309441",
+              "virtualMachineVersion": "Compute",
+              "extendedInfo": {
+                "tasksList": [],
+                "propertyBag": {
+                  "VM Name": "fog-test-vm",
+                  "Policy Name": "DefaultPolicy"
+                }
+              },
+              "entityFriendlyName": "fog-test-vm",
+              "backupManagementType": "AzureIaasVM",
+              "operation": "ConfigureBackup",
+              "status": "Completed",
+              "startTime": "2016-10-13T09:55:49.1168243Z",
+              "endTime": "2016-10-13T09:56:41.4477684Z",
+              "activityId": "383f05d9-a4bf-4b95-bb41-d39849b3a86e-2016-10-13 09:55:53Z-PS"
+            }
+          }'
+          JSON.parse(body)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/get_backup_protection_policy.rb

@@ -0,0 +1,64 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def get_backup_protection_policy(resource_group, name)
+          msg = "Get backup protection policy from Resource Group #{resource_group}"
+          Fog::Logger.debug msg
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}/backupPolicies?api-version=2016-05-01&$filter=backupManagementType eq 'AzureIaasVM'"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            response = RestClient.get(
+              resource_url,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Successfully retrieved backup protection policy from Resource Group #{resource_group}"
+          JSON.parse(response)['value']
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        body = '{
+          "value": [{
+            "id": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault/backupPolicies/DefaultPolicy",
+            "name": "DefaultPolicy",
+            "type": "Microsoft.RecoveryServices/vaults/backupPolicies",
+            "properties": {
+              "backupManagementType": "AzureIaasVM",
+              "schedulePolicy": {
+                "schedulePolicyType": "SimpleSchedulePolicy",
+                "scheduleRunFrequency": "Daily",
+                "scheduleRunTimes": [
+                  "2016-10-13T19:30:00Z"
+                ],
+                "scheduleWeeklyFrequency": 0
+              },
+              "retentionPolicy": {
+                "retentionPolicyType": "LongTermRetentionPolicy",
+                "dailySchedule": {
+                  "retentionTimes": [
+                    "2016-10-13T19:30:00Z"
+                  ],
+                  "retentionDuration": {
+                    "count": 30,
+                    "durationType": "Days"
+                  }
+                }
+              },
+              "protectedItemsCount": 0
+            }
+          }]
+        }'
+        JSON.parse(body)['value']
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/get_recovery_vault.rb

@@ -0,0 +1,49 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def get_recovery_vault(resource_group, name)
+          msg = "Retrieving Recovery Vault #{name} from Resource Group #{resource_group}"
+          Fog::Logger.debug msg
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults?api-version=2016-05-01"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            recovery_vault_response = RestClient.get(
+              resource_url,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Recovery Vault #{name} from Resource Group #{resource_group} retrieved successfully"
+          recovery_vault = JSON.parse(recovery_vault_response)['value']
+          recovery_vault.select { |vault| vault['name'].eql? name }[0]
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def get_recovery_vault(*)
+          body = '{
+            "value": [{
+              "location": "westus",
+              "name": "fog-test-vault",
+              "properties": {
+                "provisioningState": "Succeeded"
+              },
+              "id": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault",
+              "type": "Microsoft.RecoveryServices/vaults",
+              "sku": {
+                "name": "standard"
+              }
+            }]
+          }'
+          JSON.parse(body)['value'][0]
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/list_recovery_vaults.rb

@@ -0,0 +1,48 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def list_recovery_vaults(resource_group)
+          msg = "Listing Recovery Vaults in Resource Group #{resource_group}"
+          Fog::Logger.debug msg
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults?api-version=2016-05-01"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            recovery_vaults_response = RestClient.get(
+              resource_url,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Recovery Vaults in Resource Group #{resource_group} listed successfully"
+          JSON.parse(recovery_vaults_response)['value']
+        end
+      end
+
+      # Mock class for Recovery Vault Request
+      class Mock
+        def list_recovery_vaults(*)
+          body = '{
+            "value": [{
+              "location": "westus",
+              "name": "fog-test-vault",
+              "properties": {
+                "provisioningState": "Succeeded"
+              },
+              "id": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault",
+              "type": "Microsoft.RecoveryServices/vaults",
+              "sku": {
+                "name": "standard"
+              }
+            }]
+          }'
+          JSON.parse(body)['value']
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/set_recovery_vault_context.rb

@@ -0,0 +1,36 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def set_recovery_vault_context(resource_group, name)
+          msg = "Set context for Recovery Vault #{name}"
+          Fog::Logger.debug msg
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=2016-05-01"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            RestClient.get(
+              resource_url,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Successfully set context for Recovery Vault #{name}"
+          true
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def set_recovery_vault_context(*)
+          Fog::Logger.debug 'Successfully set context for Recovery Vault {name}'
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/start_backup.rb

@@ -0,0 +1,54 @@
+IAAS_VM_BACKUP_REQUEST = 'IaasVMBackupRequest'.freeze
+
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def start_backup(resource_group, name, vm_name, vm_resource_group)
+          msg = "Starting backup for VM #{vm_name}"
+          Fog::Logger.debug msg
+
+          job = get_backup_job_for_vm(name, resource_group, vm_name, vm_resource_group, 'Backup')
+
+          unless job.nil?
+            Fog::Logger.debug "Backup already in progress for VM #{vm_name} in Recovery Vault #{name}"
+            return false
+          end
+
+          backup_items = get_backup_item(resource_group, name, vm_name)
+          backup_item = backup_items.select { |item| (item['properties']['friendlyName'].eql? vm_name.downcase) && (vm_resource_group.eql? get_resource_group_from_id(item['properties']['virtualMachineId'])) }[0]
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;#{backup_item['name']}/protectedItems/vm;#{backup_item['name']}/backup?api-version=2016-05-01"
+          body = {
+            properties: {
+              objectType: IAAS_VM_BACKUP_REQUEST
+            }
+          }
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            RestClient.post(
+              resource_url,
+              body.to_json,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Successfully started backup for VM #{vm_name} in Recovery Vault #{name}"
+          true
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def start_backup(*)
+          Fog::Logger.debug 'Successfully started backup for VM {vm_name} in Recovery Vault {name}'
+          true
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/storage/update_storage_account.rb

@@ -0,0 +1,59 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def update_storage_account(storage_account_hash)
+          msg = "Updating Storage Account: #{storage_account_hash[:name]} in Resource Group #{storage_account_hash[:resource_group]}."
+          Fog::Logger.debug msg
+          storage_account_params = get_storage_account_update_params(storage_account_hash)
+          begin
+            storage_account = @storage_mgmt_client.storage_accounts.update(storage_account_hash[:resource_group],
+                                                                           storage_account_hash[:name],
+                                                                           storage_account_params)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug 'Storage Account updated successfully.'
+          storage_account
+        end
+
+        private
+
+        def get_storage_account_update_params(storage_account_hash)
+          params = Azure::ARM::Storage::Models::StorageAccountUpdateParameters.new
+          sku = Azure::ARM::Storage::Models::Sku.new
+          sku.name = "#{storage_account_hash[:sku_name]}_#{storage_account_hash[:replication]}"
+          params.sku = sku
+          unless storage_account_hash[:encryption].nil?
+            encryption = Azure::ARM::Storage::Models::Encryption.new
+            encryption_services = Azure::ARM::Storage::Models::EncryptionServices.new
+            encryption_service = Azure::ARM::Storage::Models::EncryptionService.new
+            encryption_service.enabled = storage_account_hash[:encryption]
+            encryption_service.last_enabled_time = Time.new if encryption_service.enabled
+            encryption_services.blob = encryption_service
+            encryption.services = encryption_services
+            params.encryption = encryption
+          end
+          params
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def update_storage_account(*)
+          storage_account_hash = {
+            'id' => '/subscriptions/67f2116d-4ea2-4c6c-b20a-f92183dbe3cb/resourceGroups/fog_test_rg/providers/Microsoft.Storage/storageAccounts/fogtestsasecond',
+            'name' => 'fog-test-storage-account',
+            'location' => 'west us',
+            'sku' =>
+              {
+                'name' => 'Standard_LRS'
+              }
+          }
+          storage_account_mapper = Azure::ARM::Storage::Models::StorageAccount.mapper
+          storage_mgmt_client.deserialize(storage_account_mapper, storage_account_hash, 'hash')
+        end
+      end
+    end
+  end
+end