fog-azure-rm-temp 0.0.3 → 0.0.4

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

@@ -0,0 +1,54 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def create_or_update_recovery_vault(resource_group, location, name)
+          msg = "Creating/Updating Recovery Vault #{name} in Resource Group #{resource_group}"
+          Fog::Logger.debug msg
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.RecoveryServices/vaults/#{name}?api-version=#{REST_CLIENT_API_VERSION[1]}"
+          body = {
+            location: location,
+            tags: {},
+            sku: { name: 'standard' },
+            properties: {}
+          }
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            response = RestClient.put(
+              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 "Recovery Vault #{name} created/updated successfully"
+          Fog::JSON.decode(response)
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def create_or_update_recovery_vault(*)
+          recovery_vault = '{
+            "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"
+            }
+          }'
+          Fog::JSON.decode(recovery_vault)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def delete_recovery_vault(resource_group, name)
+          msg = "Deleting 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/#{name}?api-version=#{REST_CLIENT_API_VERSION[1]}"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            RestClient.delete(
+              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} in Resource Group #{resource_group} deleted successfully"
+          true
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def delete_recovery_vault(*)
+          Fog::Logger.debug 'Recovery Vault TestVault in Resource Group TestRG deleted successfully'
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,60 @@
+PROTECTION_STOPPED = 'ProtectionStopped'.freeze
+
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def disable_backup_protection(rv_name, rv_resource_group, vm_name, vm_resource_group)
+          msg = "Disabling protection for VM #{vm_name} in Recovery Vault #{rv_name}"
+          Fog::Logger.debug msg
+
+          set_recovery_vault_context(rv_resource_group, rv_name)
+          vm_id = get_virtual_machine_id(vm_resource_group, vm_name)
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}/protectedItems/vm;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}?api-version=#{REST_CLIENT_API_VERSION[1]}"
+          body = {
+            properties: {
+              protectedItemType: 'Microsoft.Compute/virtualMachines',
+              policyId: '',
+              sourceResourceId: vm_id,
+              protectionState: PROTECTION_STOPPED
+            },
+            tags: {}
+          }
+
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            RestClient.put(
+              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 disabled protection for VM #{vm_name} in Recovery Vault #{rv_name}"
+          true
+        end
+
+        private
+        
+        def get_virtual_machine_id(vm_resource_group, vm_name)
+          compute_service = Fog::Compute::AzureRM.new(tenant_id: @tenant_id, client_id: @client_id, client_secret: @client_secret, subscription_id: @subscription_id)
+          compute_service.get_virtual_machine(vm_resource_group, vm_name).id
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def disable_backup_protection(*)
+          Fog::Logger.debug 'Successfully disabled protection for VM {vm_name} in Recovery Vault {rv_name}'
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,61 @@
+CONFIGURE_BACKUP = 'ConfigureBackup'.freeze
+
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def enable_backup_protection(rv_name, rv_resource_group, vm_name, vm_resource_group)
+          msg = "Enabling backup protection for VM #{vm_name} in Resource Group #{vm_resource_group}"
+          Fog::Logger.debug msg
+
+          set_recovery_vault_context(rv_resource_group, rv_name)
+          backup_protection_policy = get_backup_protection_policy(rv_resource_group, rv_name)
+          policy = backup_protection_policy.select { |item| item['name'].eql? 'DefaultPolicy' }[0]
+          vm_id = get_virtual_machine_id(vm_resource_group, vm_name)
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupFabrics/Azure/protectionContainers/iaasvmcontainer;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}/protectedItems/vm;iaasvmcontainerv2;#{vm_resource_group.downcase};#{vm_name.downcase}?api-version=#{REST_CLIENT_API_VERSION[1]}"
+          body = {
+            properties: {
+              protectedItemType: 'Microsoft.Compute/virtualMachines',
+              policyId: policy['id'],
+              sourceResourceId: vm_id
+            },
+            tags: {}
+          }
+
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            RestClient.put(
+              resource_url,
+              body.to_json,
+              accept: 'application/json',
+              content_type: 'application/json',
+              authorization: token
+            )
+          rescue RestClient::Exception => e
+            raise_azure_exception(e, msg)
+          end
+
+          @job = get_backup_job_for_vm(rv_name, rv_resource_group, vm_name, vm_resource_group, CONFIGURE_BACKUP)
+
+          until @job.nil?
+            sleep 10
+            @job = get_backup_job_for_vm(rv_name, rv_resource_group, vm_name, vm_resource_group, CONFIGURE_BACKUP)
+          end
+
+          Fog::Logger.debug "Successfully enabled backup protection for VM #{vm_name} in Resource Group #{vm_resource_group}"
+          true
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def enable_backup_protection(*)
+          Fog::Logger.debug 'Successfully enabled backup protection for VM {vm_name} in Resource Group {vm_resource_group}'
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,56 @@
+module Fog
+  module Storage
+    class AzureRM
+      # Real class for Recovery Vault request
+      class Real
+        def get_all_backup_jobs(rv_name, rv_resource_group)
+          msg = "Getting all backup jobs for Recovery Vault #{rv_name}"
+          Fog::Logger.debug msg
+
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{rv_resource_group}/providers/Microsoft.RecoveryServices/vaults/#{rv_name}/backupJobs?api-version=#{REST_CLIENT_API_VERSION[1]}&$filter=status eq 'InProgress'"
+          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 jobs for Recovery Vault #{rv_name}"
+          Fog::JSON.decode(response)['value']
+        end
+      end
+
+      # Mock class for Recovery Vault request
+      class Mock
+        def get_all_backup_jobs(*)
+          body = '{
+            "value": [{
+              "id": "/subscriptions/########-####-####-####-############/resourceGroups/fog-test-rg/providers/Microsoft.RecoveryServices/vaults/fog-test-vault/backupJobs/########-####-####-####-############",
+              "name": "########-####-####-####-############",
+              "type": "Microsoft.RecoveryServices/vaults/backupJobs",
+              "properties": {
+                "jobType": "AzureIaaSVMJob",
+                "duration": "XX:XX:XX.XXXXXXX",
+                "actionsInfo": [
+                  1
+                ],
+                "virtualMachineVersion": "Compute",
+                "entityFriendlyName": "fog-test-vm",
+                "backupManagementType": "AzureIaasVM",
+                "operation": "Backup",
+                "status": "InProgress",
+                "startTime": "2016-10-19T07:49:31.1466534Z",
+                "activityId": "########-####-####-####-############"
+              }
+            }]
+          }'
+          Fog::JSON.decode(body)['value']
+        end
+      end
+    end
+  end
+end

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=#{REST_CLIENT_API_VERSION[1]}&$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}"
+          Fog::JSON.decode(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"
+              }
+            }]
+          }'
+          Fog::JSON.decode(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=#{REST_CLIENT_API_VERSION[1]}&$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}"
+          Fog::JSON.decode(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"
+              }
+            }]
+          }'
+          Fog::JSON.decode(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"
+            }
+          }'
+          Fog::JSON.decode(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=#{REST_CLIENT_API_VERSION[1]}&$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}"
+          Fog::JSON.decode(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
+            }
+          }]
+        }'
+        Fog::JSON.decode(body)['value']
+      end
+    end
+  end
+end