fog-azure-rm 0.1.0 → 0.1.1

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

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

@@ -4,13 +4,16 @@ module Fog
       # This class provides the actual implementation for service calls.
       class Real
         def create_container(name, options = {})
-          msg = "Creating container: #{name}."
+          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
@@ -20,16 +23,15 @@ module Fog
       class Mock
         def create_container(*)
           {
-            'name' => 'testcontainer1',
-            'properties' =>
-              {
-                'last_modified' => 'Mon, 04 Jul 2016 02:50:20 GMT',
-                'etag' => '0x8D3A3B5F017F52D',
-                'lease_status' => 'unlocked',
-                'lease_state' => 'available'
-              },
+            'name' => 'test_container',
             'public_access_level' => nil,
-            'metadata' => {}
+            'metadata' => {},
+            'properties' => {
+              'last_modified' => 'Mon, 04 Jul 2016 02:50:20 GMT',
+              'etag' => '0x8D3A3B5F017F52D',
+              'lease_status' => 'unlocked',
+              'lease_state' => 'available'
+            }
           }
         end
       end

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

@@ -1,34 +1,60 @@
 module Fog
   module Storage
     class AzureRM
-      # This class provides the actual implemention for service calls.
+      # This class provides the actual implementation for service calls.
       class Real
-        def create_disk(blob_name, options = {})
-          msg = "Creating disk(blob) #{blob_name}"
+        # 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
-          ::File.open('out.txt', 'w') { |f| f.write('Sample File') }
-          path = ::File.expand_path(::File.dirname('out.txt')) + '/' + 'out.txt'
+
+          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
-            disk = upload_block_blob_from_file('vhds', "#{blob_name}.vhd", path.to_s, options)
-          rescue Azure::Core::Http::HTTPError => e
-            raise_azure_exception(e, msg)
+            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
-          disk
+
+          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(Blob) created successfully.'
-          {
-            'name' => 'test_blob',
-            'properties' =>
-              {
-                'last_modified' => 'Thu, 28 Jul 2016 06:53:05 GMT',
-                'etag' => '0x8D3B6B3D353FFCA',
-                'content_md5' => 'tXAohIyxuu/t94Lp/ujeRw=='
-              }
-          }
+          Fog::Logger.debug 'Disk created successfully.'
+          true
         end
       end
     end

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

@@ -27,7 +27,7 @@ module Fog
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Recovery Vault #{name} created/updated successfully"
-          JSON.parse(response)
+          Fog::JSON.decode(response)
         end
       end
 
@@ -46,7 +46,7 @@ module Fog
               "name": "standard"
             }
           }'
-          JSON.parse(recovery_vault)
+          Fog::JSON.decode(recovery_vault)
         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

@@ -14,8 +14,8 @@ module Fog
             storage_account = @storage_mgmt_client.storage_accounts.create(storage_account_hash[:resource_group],
                                                                            storage_account_hash[:name],
                                                                            storage_account_params)
-          rescue MsRestAzure::AzureOperationError => e
-            raise_azure_exception(e, msg)
+          rescue MsRestAzure::AzureOperationError => ex
+            raise_azure_exception(ex, msg)
           end
           Fog::Logger.debug 'Storage Account created successfully.'
           storage_account

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

@@ -1,17 +1,22 @@
 module Fog
   module Storage
     class AzureRM
-      # This class provides the actual implemention for service calls.
+      # This class provides the actual implementation for service calls.
       class Real
         def delete_blob(container_name, blob_name, options = {})
-          Fog::Logger.debug "Deleting blob: #{blob_name} in container #{container_name}."
+          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)
-            Fog::Logger.debug "Blob #{blob_name} deleted successfully."
-            true
           rescue Azure::Core::Http::HTTPError => ex
-            raise "Exception in deleting the blob #{blob_name}: #{ex.inspect}"
+            return true if ex.message.include?('(404)')
+            raise_azure_exception(ex, msg)
           end
+
+          Fog::Logger.debug "Blob #{blob_name} deleted successfully."
+          true
         end
       end
 

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

@@ -3,14 +3,18 @@ module Fog
     class AzureRM
       # This class provides the actual implementation for service calls.
       class Real
-        def delete_container(name, options = {})
-          msg = "Deleting container: #{name}."
+        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

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

@@ -3,28 +3,31 @@ module Fog
     class AzureRM
       # This class provides the actual implementation for service calls.
       class Real
-        def delete_disk(blob_name)
-          msg = "Deleting Disk: #{blob_name}."
+        # 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
-          begin
-            result = delete_blob('vhds', "#{blob_name}.vhd")
-          rescue Azure::Core::Http::HTTPError => e
-            raise_azure_exception(e, msg)
-          end
-          if result.nil?
-            Fog::Logger.debug "Successfully deleted Disk: #{blob_name}."
-            true
-          else
-            Fog::Logger.debug 'Error deleting Disk.'
-            false
-          end
+
+          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 'Deleting Disk: test_blob.'
-          Fog::Logger.debug 'Successfully deleted Disk: test_blob.'
+          Fog::Logger.debug 'Successfully deleted Disk'
           true
         end
       end

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

@@ -20,7 +20,7 @@ module Fog
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Successfully retrieved backup jobs for Recovery Vault #{rv_name}"
-          JSON.parse(response)['value']
+          Fog::JSON.decode(response)['value']
         end
       end
 
@@ -48,7 +48,7 @@ module Fog
               }
             }]
           }'
-          JSON.parse(body)['value']
+          Fog::JSON.decode(body)['value']
         end
       end
     end

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

@@ -20,7 +20,7 @@ module Fog
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Successfully retrieved backup container from Recovery Vault #{rv_name}"
-          JSON.parse(response)['value']
+          Fog::JSON.decode(response)['value']
         end
       end
 
@@ -45,7 +45,7 @@ module Fog
               }
             }]
           }'
-          JSON.parse(body)['value']
+          Fog::JSON.decode(body)['value']
         end
       end
     end

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

@@ -20,7 +20,7 @@ module Fog
             raise_azure_exception(e, msg)
           end
           Fog::Logger.debug "Successfully retrieved backup item from Recovery Vault #{rv_name}"
-          JSON.parse(response)['value']
+          Fog::JSON.decode(response)['value']
         end
       end
 
@@ -50,7 +50,7 @@ module Fog
               }
             }]
           }'
-          JSON.parse(body)['value']
+          Fog::JSON.decode(body)['value']
         end
       end
     end