fog-azure-rm-downgraded 0.3.0

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

@@ -0,0 +1,40 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def create_container(name, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "Creating container: #{name}. options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            container = @blob_client.create_container(name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Container #{name} created successfully."
+          container
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def create_container(*)
+          {
+            'name' => 'test_container',
+            'public_access_level' => nil,
+            'metadata' => {},
+            'properties' => {
+              'last_modified' => 'Mon, 04 Jul 2016 02:50:20 GMT',
+              'etag' => '0x8D3A3B5F017F52D',
+              'lease_status' => 'unlocked',
+              'lease_state' => 'available'
+            }
+          }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,62 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        # Create a disk in Azure storage.
+        #
+        # @param disk_name [String] Name of disk
+        # @param disk_size_in_gb [Integer] Disk size in GiB. Value range: [1, 1023]
+        # @param options  [Hash]
+        # @option options [String] container_name Sets name of the container which contains the disk. Default is 'vhds'.
+        #
+        # @return [Boolean]
+        #
+        def create_disk(disk_name, disk_size_in_gb, options = {})
+          msg = "Creating disk(#{disk_name}, #{disk_size_in_gb}). options: #{options}"
+          Fog::Logger.debug msg
+
+          raise ArgumentError, "disk_size_in_gb #{disk_size_in_gb} must be an integer" unless disk_size_in_gb.is_a?(Integer)
+          raise ArgumentError, "Azure minimum disk size is 1 GiB: #{disk_size_in_gb}" if disk_size_in_gb < 1
+          raise ArgumentError, "Azure maximum disk size is 1023 GiB: #{disk_size_in_gb}" if disk_size_in_gb > 1023
+
+          container_name = options.delete(:container_name)
+          container_name = 'vhds' if container_name.nil?
+          blob_name = "#{disk_name}.vhd"
+          vhd_size = disk_size_in_gb * 1024 * 1024 * 1024
+          blob_size = vhd_size + 512
+
+          opts = {
+            type:   :fixed,
+            name: '/tmp/footer.vhd', # Only used to initialize vhd, no local file is generated.
+            size: disk_size_in_gb
+          }
+          vhd_footer = Vhd::Library.new(opts).footer.values.join
+
+          begin
+            create_page_blob(container_name, blob_name, blob_size, options)
+            put_blob_pages(container_name, blob_name, vhd_size, blob_size - 1, vhd_footer, options)
+          rescue
+            begin
+              delete_blob(container_name, blob_name)
+            rescue => ex
+              Fog::Logger.debug "Cannot delete the blob: #{container_name}/#{blob_name} after create_disk failed. #{ex.inspect}"
+            end
+            raise
+          end
+
+          Fog::Logger.debug "Created a disk #{disk_name} successfully."
+          true
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def create_disk(*)
+          Fog::Logger.debug 'Disk created successfully.'
+          true
+        end
+      end
+    end
+  end
+end

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/create_page_blob.rb

@@ -0,0 +1,31 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def create_page_blob(container_name, blob_name, blob_size, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "create_page_blob #{blob_name} to the container #{container_name}. blob_size: #{blob_size}, options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            @blob_client.create_page_blob(container_name, blob_name, blob_size, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Page blob #{blob_name} created successfully."
+          true
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def create_page_blob(*)
+          Fog::Logger.debug 'Page blob created successfully.'
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,64 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def create_storage_account(storage_account_hash)
+          msg = "Creating Storage Account: #{storage_account_hash[:name]} in Resource Group #{storage_account_hash[:resource_group]}."
+          Fog::Logger.debug msg
+          storage_account_params = get_storage_account_params(storage_account_hash[:sku_name],
+                                                              storage_account_hash[:location],
+                                                              storage_account_hash[:replication],
+                                                              storage_account_hash[:encryption])
+          begin
+            storage_account = @storage_mgmt_client.storage_accounts.create(storage_account_hash[:resource_group],
+                                                                           storage_account_hash[:name],
+                                                                           storage_account_params)
+          rescue MsRestAzure::AzureOperationError => ex
+            raise_azure_exception(ex, msg)
+          end
+          Fog::Logger.debug 'Storage Account created successfully.'
+          storage_account
+        end
+
+        private
+
+        def get_storage_account_params(sku_name, location, replication, encryption_enabled)
+          params = Azure::ARM::Storage::Models::StorageAccountCreateParameters.new
+          sku = Azure::ARM::Storage::Models::Sku.new
+          sku.name = "#{sku_name}_#{replication}"
+          params.sku = sku
+          params.kind = Azure::ARM::Storage::Models::Kind::Storage
+          params.location = location
+          unless encryption_enabled.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 = encryption_enabled
+            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 create_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

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

@@ -0,0 +1,32 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def delete_blob(container_name, blob_name, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "Deleting blob: #{blob_name} in container #{container_name}. options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            @blob_client.delete_blob(container_name, blob_name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            return true if ex.message.include?('(404)')
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Blob #{blob_name} deleted successfully."
+          true
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def delete_blob(*)
+          Fog::Logger.debug 'Blob deleted successfully.'
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,31 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def delete_container(name)
+          options = { request_id: SecureRandom.uuid }
+          msg = "Deleting container: #{name}. options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            @blob_client.delete_container(name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            return true if ex.message.include?('(404)')
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Container #{name} deleted successfully."
+          true
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def delete_container(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,36 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        # Delete a disk in Azure storage.
+        #
+        # @param disk_name [String] Name of disk
+        # @param options  [Hash]
+        # @option options [String] container_name Sets name of the container which contains the disk. Default is 'vhds'.
+        #
+        # @return [Boolean]
+        #
+        def delete_disk(disk_name, options = {})
+          msg = "Deleting disk(#{disk_name}). options: #{options}"
+          Fog::Logger.debug msg
+
+          container_name = options.delete(:container_name)
+          container_name = 'vhds' if container_name.nil?
+          delete_blob(container_name, "#{disk_name}.vhd")
+
+          Fog::Logger.debug "Successfully deleted Disk: #{disk_name}."
+          true
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def delete_disk(*)
+          Fog::Logger.debug 'Successfully deleted Disk'
+          true
+        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/delete_storage_account.rb

@@ -0,0 +1,27 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def delete_storage_account(resource_group, name)
+          msg = "Deleting Storage Account: #{name} in Resource Group #{resource_group}."
+          Fog::Logger.debug msg
+          begin
+            @storage_mgmt_client.storage_accounts.delete(resource_group, name)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          Fog::Logger.debug "Storage Account #{name} deleted successfully."
+          true
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def delete_storage_account(resource_group, name)
+          Fog::Logger.debug "Storage Account #{name} from Resource group #{resource_group} 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