fog-azure-rm-temp 0.0.1

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

@@ -0,0 +1,32 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def acquire_blob_lease(container_name, name, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "Leasing blob: #{name} of container #{container_name} options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            lease_id = @blob_client.acquire_blob_lease(container_name, name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Blob #{name} leased successfully."
+          lease_id
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def acquire_blob_lease(*)
+          {
+            'leaseId' => 'abc123'
+          }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,32 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def acquire_container_lease(name, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "Leasing container: #{name} options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            lease_id = @blob_client.acquire_container_lease(name, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Container #{name} leased successfully."
+          lease_id
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def acquire_container_lease(*)
+          {
+            'leaseId' => 'abc123'
+          }
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_container_exists(name)
+          msg = "Checking container #{name}."
+          Fog::Logger.debug msg
+          begin
+            get_container_properties(name)
+            Fog::Logger.debug "Container #{name} exists."
+            true
+          rescue Exception => e
+            if e.message.include? 'NotFound'
+              Fog::Logger.debug "The specified container #{name} does not exist."
+              false
+            end
+          end
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_container_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,37 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_storage_account_exists(resource_group_name, storage_account_name)
+          msg = "Checking Storage Account: #{storage_account_name}."
+          Fog::Logger.debug msg
+          begin
+            storage_account = @storage_mgmt_client.storage_accounts.get_properties(resource_group_name, storage_account_name)
+          rescue MsRestAzure::AzureOperationError => e
+            if e.body['error']['code'] == 'ResourceNotFound'
+              Fog::Logger.debug "Storage Account #{storage_account_name} doesn't exist."
+              return false
+            else
+              raise_azure_exception(e, msg)
+            end
+          end
+
+          if storage_account.provisioning_state == 'Succeeded'
+            Fog::Logger.debug "Storage Account #{storage_account_name} exists."
+            true
+          else
+            Fog::Logger.debug "Storage Account #{storage_account_name} doesn't exist."
+            false
+          end
+        end
+      end
+      # This class provides the mock implementation.
+      class Mock
+        def check_storage_account_exists(*)
+          true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,44 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def check_storage_account_name_availability(storage_account_name, storage_account_type)
+          msg = "Checking Storage Account Name availability: #{storage_account_name}."
+          Fog::Logger.debug msg
+
+          storage_account_params = create_storage_account_params(storage_account_name, storage_account_type)
+          begin
+            name_available_obj = @storage_mgmt_client.storage_accounts.check_name_availability(storage_account_params)
+          rescue MsRestAzure::AzureOperationError => e
+            raise_azure_exception(e, msg)
+          end
+          if name_available_obj.name_available
+            Fog::Logger.debug "Name: #{storage_account_name} is available."
+            true
+          else
+            Fog::Logger.debug "Name: #{storage_account_type} is not available."
+            Fog::Logger.debug "Reason: #{name_available_obj.reason}."
+            false
+          end
+        end
+
+        private
+
+        def create_storage_account_params(name, type)
+          params = Azure::ARM::Storage::Models::StorageAccountCheckNameAvailabilityParameters.new
+          params.name = name
+          params.type = type
+          params
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def check_storage_account_name_availability(params)
+          Fog::Logger.debug "Name: #{params.name} is available."
+          true
+        end
+      end
+    end
+  end
+end

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

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

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

@@ -0,0 +1,148 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def compare_container_blobs(container1, container2)
+          msg = "Comparing blobs from container #{container1} to container #{container2}"
+          Fog::Logger.debug msg
+          begin
+            identical_blobs = get_identical_blobs_from_containers(container1, container2)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+          identical_blobs
+        end
+
+        private
+
+        def get_identical_blobs_from_containers(container1, container2)
+          container1_blobs = list_blobs(container1)
+          container2_blobs = list_blobs(container2)
+
+          identical_blobs = []
+          container1_blobs[:blobs].each do |container1_blob|
+            container2_blobs[:blobs].each do |container2_blob|
+              if container1_blob.name == container2_blob.name && container1_blob.properties[:content_md5] == container2_blob.properties[:content_md5]
+                identical_blobs.push(container1_blob)
+              end
+            end
+          end
+          identical_blobs
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def compare_container_blobs(*)
+          [
+            {
+              'name' => 'test_blob1',
+              'metadata' => {},
+              'properties' => {
+                'last_modified' => 'Mon, 04 Jul 2016 02:50:20 GMT',
+                'etag' => '0x8D3A3B5F017F52D',
+                'lease_status' => 'unlocked',
+                'lease_state' => 'available',
+                'content_length' => 4_194_304,
+                'content_type' => 'application/octet-stream',
+                'content_encoding' => nil,
+                'content_language' => nil,
+                'content_disposition' => nil,
+                'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw==',
+                'cache_control' => nil,
+                'sequence_number' => 0,
+                'blob_type' => 'PageBlob',
+                'copy_id' => '095adc3b-e277-4c3d-97e0-0abca881f60c',
+                'copy_status' => 'success',
+                'copy_source' => 'https://testaccount.blob.core.windows.net/test_container/test_blob?snapshot=2016-02-04T08%3A35%3A50.3256874Z',
+                'copy_progress' => '4194304/4194304',
+                'copy_completion_time' => 'Thu, 04 Feb 2016 08:35:52 GMT',
+                'copy_status_description' => nil,
+                'accept_ranges' => 0
+              }
+            },
+            {
+              'name' => 'test_blob2',
+              'metadata' => {},
+              'properties' => {
+                'last_modified' => 'Tue, 04 Aug 2015 06:02:08 GMT',
+                'etag' => '0x8D29C92173526C8',
+                'lease_status' => 'unlocked',
+                'lease_state' => 'available',
+                'content_length' => 4_194_304,
+                'content_type' => 'application/octet-stream',
+                'content_encoding' => nil,
+                'content_language' => nil,
+                'content_disposition' => nil,
+                'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw==',
+                'cache_control' => nil,
+                'sequence_number' => 0,
+                'blob_type' => 'PageBlob',
+                'copy_id' => '0abcdc3b-4c3d-e277-97e0-0abca881f60c',
+                'copy_status' => 'success',
+                'copy_source' => 'https://testaccount.blob.core.windows.net/test_container/test_blob?snapshot=2016-02-04T08%3A35%3A55.3157696Z',
+                'copy_progress' => '4194304/4194304',
+                'copy_completion_time' => 'Thu, 04 Feb 2016 08:40:52 GMT',
+                'copy_status_description' => nil,
+                'accept_ranges' => 0
+              }
+            },
+            {
+              'name' => 'test_blob3',
+              'metadata' => {},
+              'properties' => {
+                'last_modified' => 'Tue, 04 Aug 2015 06:02:08 GMT',
+                'etag' => '0x8D29C92173526C8',
+                'lease_status' => 'unlocked',
+                'lease_state' => 'available',
+                'content_length' => 4_194_304,
+                'content_type' => 'application/octet-stream',
+                'content_encoding' => nil,
+                'content_language' => nil,
+                'content_disposition' => nil,
+                'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw==',
+                'cache_control' => nil,
+                'sequence_number' => 0,
+                'blob_type' => 'PageBlob',
+                'copy_id' => '0abcdc3b-4c3d-e277-97e0-0abca881f60c',
+                'copy_status' => 'success',
+                'copy_source' => 'https://testaccount.blob.core.windows.net/test_container/test_blob?snapshot=2016-02-04T08%3A35%3A55.3157696Z',
+                'copy_progress' => '4194304/4194304',
+                'copy_completion_time' => 'Thu, 04 Feb 2016 08:40:52 GMT',
+                'copy_status_description' => nil,
+                'accept_ranges' => 0
+              }
+            },
+            {
+              'name' => 'test_blob4',
+              'metadata' => {},
+              'properties' => {
+                'last_modified' => 'Tue, 04 Aug 2015 06:02:08 GMT',
+                'etag' => '0x8D29C92173526C8',
+                'lease_status' => 'unlocked',
+                'lease_state' => 'available',
+                'content_length' => 4_194_304,
+                'content_type' => 'application/octet-stream',
+                'content_encoding' => nil,
+                'content_language' => nil,
+                'content_disposition' => nil,
+                'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw==',
+                'cache_control' => nil,
+                'sequence_number' => 0,
+                'blob_type' => 'PageBlob',
+                'copy_id' => '0abcdc3b-4c3d-e277-97e0-0abca881f60c',
+                'copy_status' => 'success',
+                'copy_source' => 'https://testaccount.blob.core.windows.net/test_container/test_blob?snapshot=2016-02-04T08%3A35%3A55.3157696Z',
+                'copy_progress' => '4194304/4194304',
+                'copy_completion_time' => 'Thu, 04 Feb 2016 08:40:52 GMT',
+                'copy_status_description' => nil,
+                'accept_ranges' => 0
+              }
+            }
+          ]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def copy_blob(destination_container, destination_blob, source_container, source_blob, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "Copying blob: #{source_blob} from container #{source_container} to container #{destination_container} options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            copy_id, copy_status = @blob_client.copy_blob(destination_container, destination_blob, source_container, source_blob, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Copying blob: x-ms-copy-id: #{copy_id}, x-ms-copy-status: #{copy_status}"
+          [copy_id, copy_status]
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def copy_blob(*)
+          %w(abc123 pending)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def copy_blob_from_uri(destination_container, destination_blob, source_blob_uri, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "Copying blob: #{source_blob_uri} to container #{destination_container} options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            copy_id, copy_status = @blob_client.copy_blob_from_uri(destination_container, destination_blob, source_blob_uri, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          Fog::Logger.debug "Copying blob: x-ms-copy-id: #{copy_id}, x-ms-copy-status: #{copy_status}"
+          [copy_id, copy_status]
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def copy_blob_from_uri(*)
+          %w(abc123 pending)
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,108 @@
+module Fog
+  module Storage
+    class AzureRM
+      # This class provides the actual implementation for service calls.
+      class Real
+        def create_block_blob(container_name, blob_name, body, options = {})
+          options[:request_id] = SecureRandom.uuid
+          msg = "create_block_blob #{blob_name} to the container #{container_name}. options: #{options}"
+          Fog::Logger.debug msg
+
+          begin
+            if body.nil?
+              data = nil
+            elsif body.respond_to?(:read)
+              if body.respond_to?(:rewind)
+                begin
+                  body.rewind
+                rescue
+                  nil
+                end
+              end
+              data = body.read
+            else
+              data = Fog::Storage.parse_data(body)
+              options[:content_length] = data[:headers]['Content-Length']
+              options[:content_type] = data[:headers]['Content-Type']
+              data = data[:body]
+            end
+
+            raise ArgumentError.new('The maximum size for a block blob created via create_block_blob is 64 MB.') if !data.nil? && Fog::Storage.get_body_size(data) > 64 * 1024 * 1024
+            blob = @blob_client.create_block_blob(container_name, blob_name, data, options)
+          rescue Azure::Core::Http::HTTPError => ex
+            raise_azure_exception(ex, msg)
+          end
+
+          if data.nil?
+            Fog::Logger.debug "Create a block blob #{blob_name} successfully."
+          else
+            Fog::Logger.debug "Upload a block blob #{blob_name} successfully."
+          end
+          blob
+        end
+      end
+
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def create_block_blob(_container_name, _blob_name, body, _options = {})
+          Fog::Logger.debug 'Blob created successfully.'
+          if body.nil?
+            {
+              'name' => 'test_blob',
+              'metadata' => {},
+              'properties' => {
+                'last_modified' => 'Mon, 04 Jul 2016 02:50:20 GMT',
+                'etag' => '0x8D3A3B5F017F52D',
+                'lease_status' => nil,
+                'lease_state' => nil,
+                'content_length' => 0,
+                'content_type' => 'application/octet-stream',
+                'content_encoding' => nil,
+                'content_language' => nil,
+                'content_disposition' => nil,
+                'content_md5' => nil,
+                'cache_control' => nil,
+                'sequence_number' => 0,
+                'blob_type' => 'BlockBlob',
+                'copy_id' => nil,
+                'copy_status' => nil,
+                'copy_source' => nil,
+                'copy_progress' => nil,
+                'copy_completion_time' => nil,
+                'copy_status_description' => nil,
+                'accept_ranges' => 0
+              }
+            }
+          else
+            {
+              'name' => 'test_blob',
+              'metadata' => {},
+              'properties' => {
+                'last_modified' => 'Mon, 04 Jul 2016 02:50:20 GMT',
+                'etag' => '0x8D3A3B5F017F52D',
+                'lease_status' => 'unlocked',
+                'lease_state' => 'available',
+                'content_length' => 4_194_304,
+                'content_type' => 'application/octet-stream',
+                'content_encoding' => nil,
+                'content_language' => nil,
+                'content_disposition' => nil,
+                'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw==',
+                'cache_control' => nil,
+                'sequence_number' => 0,
+                'blob_type' => 'BlockBlob',
+                'copy_id' => '095adc3b-e277-4c3d-97e0-0abca881f60c',
+                'copy_status' => 'success',
+                'copy_source' => 'https://testaccount.blob.core.windows.net/test_container/test_blob?snapshot=2016-02-04T08%3A35%3A50.3256874Z',
+                'copy_progress' => '4194304/4194304',
+                'copy_completion_time' => 'Thu, 04 Feb 2016 08:35:52 GMT',
+                'copy_status_description' => nil,
+                'accept_ranges' => 0
+              }
+            }
+          end
+        end
+      end
+    end
+  end
+end