fog-azure-rm 0.0.0

data/lib/fog/azurerm/requests/compute/get_virtual_machine.rb

@@ -0,0 +1,24 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def get_virtual_machine(resource_group, name)
+          begin
+            promise = @compute_mgmt_client.virtual_machines.get(resource_group, name)
+            response = promise.value!
+            response.body
+          rescue MsRestAzure::AzureOperationError => e
+            msg = "Exception getting Virtual Machine #{name} from Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def get_virtual_machine(resource_group, name)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/list_availability_sets.rb

@@ -0,0 +1,24 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def list_availability_sets(resource_group)
+          begin
+            promise = @compute_mgmt_client.availability_sets.list(resource_group)
+            response = promise.value!
+            response.body.value
+          rescue MsRestAzure::AzureOperationError => e
+            msg = "Exception listing availability sets in Resource Group #{resource_group}. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_availability_sets(resource_group)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/list_available_sizes_for_virtual_machine.rb

@@ -0,0 +1,24 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def list_available_sizes_for_virtual_machine(resource_group, name)
+          begin
+            response = @compute_mgmt_client.virtual_machines.list_available_sizes(resource_group, name)
+            result = response.value!
+            result.body.value
+          rescue MsRestAzure::AzureOperationError => e
+            msg = "Error listing Sizes for Virtual Machine #{name} in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_available_sizes_for_virtual_machine(resource_group, name)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/list_virtual_machines.rb

@@ -0,0 +1,64 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def list_virtual_machines(resource_group)
+          begin
+            response = @compute_mgmt_client.virtual_machines.list(resource_group)
+            result = response.value!
+            Azure::ARM::Compute::Models::VirtualMachineListResult.serialize_object(result.body)['value']
+          rescue MsRestAzure::AzureOperationError => e
+            msg = "Error listing Virtual Machines in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def list_virtual_machines(resource_group)
+          [
+            {
+              'id' => "/subscriptions/########-####-####-####-############/resourceGroups/#{resource_group}/providers/Microsoft.Compute/virtualMachines/fog-test-server",
+              'name' => 'fog-test-server',
+              'location' => 'West US',
+              'properties' => {
+                'hardwareProfile' => {
+                    'vmSize' => 'Basic_A0'
+                },
+                'storageProfile' => {
+                  'imageReference' => {
+                    'publisher' => 'Canonical',
+                    'offer' => 'UbuntuServer',
+                    'sku' => '14.04.2-LTS',
+                    'version' => 'latest'
+                  },
+                  'osDisk' => {
+                    'name' => 'fog-test-server_os_disk',
+                    'vhd' => {
+                      'uri' => 'http://storageAccount.blob.core.windows.net/vhds/fog-test-server_os_disk.vhd'
+                    }
+                  }
+                },
+                'osProfile' => {
+                  'computerName' => 'fog-test-server',
+                  'adminUsername' => 'shaffan',
+                  'linuxConfiguration' => {
+                      'disablePasswordAuthentication' => false
+                  }
+                },
+                'networkProfile' => {
+                  'networkInterfaces' => [
+                    {
+                        'id' => "/subscriptions/########-####-####-####-############/resourceGroups/#{resource_group}/providers/Microsoft.Network/networkInterfaces/fogtestnetworkinterface"
+                    }
+                  ]
+                }
+              }
+            }
+          ]
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/power_off_virtual_machine.rb

@@ -0,0 +1,26 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def power_off_virtual_machine(resource_group, name)
+          Fog::Logger.debug "Powering off Virtual Machine #{name} in Resource Group #{resource_group}."
+          begin
+            promise = @compute_mgmt_client.virtual_machines.power_off(resource_group, name)
+            result = promise.value!
+            Fog::Logger.debug "Virtual Machine #{name} Powered off Successfully."
+            result
+          rescue  MsRestAzure::AzureOperationError => e
+            msg = "Error Powering off Virtual Machine '#{name}' in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def power_off_virtual_machine(resource_group, name)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/redeploy_virtual_machine.rb

@@ -0,0 +1,26 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def redeploy_virtual_machine(resource_group, name)
+          Fog::Logger.debug "Redeploying Virtual Machine #{name} in Resource Group #{resource_group}."
+          begin
+            promise = @compute_mgmt_client.virtual_machines.redeploy(resource_group, name)
+            result = promise.value!
+            Fog::Logger.debug "Virtual Machine #{name} Redeployed Successfully."
+            result
+          rescue  MsRestAzure::AzureOperationError => e
+            msg = "Error Redeploying Virtual Machine '#{name}' in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def redeploy_virtual_machine(resource_group, name)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/restart_virtual_machine.rb

@@ -0,0 +1,26 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def restart_virtual_machine(resource_group, name)
+          Fog::Logger.debug "Restarting Virtual Machine #{name} in Resource Group #{resource_group}."
+          begin
+            promise = @compute_mgmt_client.virtual_machines.restart(resource_group, name)
+            result = promise.value!
+            Fog::Logger.debug "Virtual Machine #{name} Restarted Successfully."
+            result
+          rescue  MsRestAzure::AzureOperationError => e
+            msg = "Error Restarting Virtual Machine '#{name}' in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def restart_virtual_machine(resource_group, name)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/compute/start_virtual_machine.rb

@@ -0,0 +1,26 @@
+module Fog
+  module Compute
+    class AzureRM
+      # This class provides the actual implemention for service calls.
+      class Real
+        def start_virtual_machine(resource_group, name)
+          Fog::Logger.debug "Starting Virtual Machine #{name} in Resource Group #{resource_group}."
+          begin
+            promise = @compute_mgmt_client.virtual_machines.start(resource_group, name)
+            result = promise.value!
+            Fog::Logger.debug "Virtual Machine #{name} started Successfully."
+            result
+          rescue  MsRestAzure::AzureOperationError => e
+            msg = "Error starting Virtual Machine '#{name}' in Resource Group '#{resource_group}'. #{e.body['error']['message']}"
+            raise msg
+          end
+        end
+      end
+      # This class provides the mock implementation for unit tests.
+      class Mock
+        def start_virtual_machine(resource_group, name)
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/check_for_zone.rb

@@ -0,0 +1,39 @@
+module Fog
+  module DNS
+    class AzureRM
+      class Real
+        def check_for_zone(resource_group, name)
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{resource_group}/providers/Microsoft.Network/dnsZones/#{name}?api-version=2015-05-04-preview"
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            dns_response = RestClient.get(
+                resource_url,
+                accept: 'application/json',
+                content_type: 'application/json',
+                authorization: token)
+            dns_hash = JSON.parse(dns_response)
+            if dns_hash.key?('id') && !dns_hash['id'].nil?
+              true
+            else
+              false
+            end
+          rescue RestClient::Exception => e
+            if e.http_code == 404
+              false
+            else
+              body = JSON.parse(e.http_body)
+              msg = "Exception checking if the zone exists: #{body['code']}, #{body['message']}"
+              fail msg
+            end
+          end
+        end
+      end
+
+      class Mock
+        def check_for_zone(resource_group, name)
+
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/create_record_set.rb

@@ -0,0 +1,61 @@
+module Fog
+  module DNS
+    class AzureRM
+      class Real
+        def create_record_set(dns_resource_group, zone_name, record_set_name, records, record_type, ttl)
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{dns_resource_group}/providers/Microsoft.Network/dnsZones/#{zone_name}/#{record_type}/#{record_set_name}?api-version=2015-05-04-preview"
+          Fog::Logger.debug "Creating/Updating RecordSet #{record_set_name} of type '#{record_type}' in zone #{zone_name}"
+
+          case record_type
+            when 'A'
+              a_type_records_array = Array.new
+              records.each do |ip|
+                a_type_records_array.push({'ipv4Address' => ip})
+              end
+              body = {
+                  :location => 'global',
+                  :tags => '',
+                  :properties => {
+                      :TTL => ttl,
+                      :ARecords => a_type_records_array
+                  }
+              }
+            when 'CNAME'
+              body = {
+                  :location => 'global',
+                  :tags => '',
+                  :properties => {
+                      :TTL => ttl,
+                      :CNAMERecord => {
+                          'cname' => records.first # because cname only has 1 value and we know the object is an array passed in.
+                      }
+                  }
+              }
+          end
+
+          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
+            )
+            Fog::Logger.debug "RecordSet #{record_set_name} Created/Updated Successfully!"
+          rescue Exception => e
+            Fog::Logger.warning "Exception setting #{record_type} records for the record set: #{record_set_name}"
+            msg = "AzureDns::RecordSet - Exception is: #{e.message}"
+            raise msg
+          end
+        end
+      end
+
+      class Mock
+        def create_record_set(dns_resource_group, zone_name, record_set_name, records, record_type, ttl)
+
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/create_zone.rb

@@ -0,0 +1,47 @@
+module Fog
+  module DNS
+    class AzureRM
+      class Real
+        def create_zone(dns_resource_group, zone_name)
+          Fog::Logger.debug "Creating Zone #{zone_name} ..."
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{dns_resource_group}/providers/Microsoft.Network/dnsZones/#{zone_name}?api-version=2015-05-04-preview"
+          
+          body = {
+              location: 'global',
+              tags: {},
+              properties: {}
+          }
+
+          begin
+            token = Fog::Credentials::AzureRM.get_token(@tenant_id, @client_id, @client_secret)
+            dns_response = RestClient.put(
+                resource_url,
+                body.to_json,
+                accept: 'application/json',
+                content_type: 'application/json',
+                authorization: token)
+            response_hash = JSON.parse(dns_response)
+            Fog::Logger.debug "Zone #{zone_name} created successfully."
+            response_hash
+          rescue RestClient::Exception => e
+            body = JSON.parse(e.http_body)
+            if body.key?('error')
+              body = body['error']
+              msg = "Exception creating zone: #{body['code']}, #{body['message']}"
+            else
+              msg = "Exception creating zone: #{body['code']}, #{body['message']}"
+            end
+
+            raise msg
+          end
+        end
+      end
+
+      class Mock
+        def create_zone(dns_resource_group, zone_name)
+
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/delete_record_set.rb

@@ -0,0 +1,38 @@
+module Fog
+  module DNS
+    class AzureRM
+      class Real
+        def delete_record_set(record_set_name, dns_resource_group, zone_name, record_type)
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{dns_resource_group}/providers/Microsoft.Network/dnsZones/#{zone_name}/#{record_type}/#{record_set_name}?api-version=2015-05-04-preview"
+          Fog::Logger.debug "Deleting RecordSet #{record_set_name} of type '#{record_type}' in zone #{zone_name}"
+
+          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
+            )
+            Fog::Logger.debug "RecordSet #{record_set_name} Deleted Successfully!"
+          rescue RestClient::Exception => e
+            body = JSON.parse(e.http_body)
+            if body.key?('error')
+              body = body['error']
+              msg = "Exception deleting zone: #{body['code']}, #{body['message']}"
+            else
+              msg = "Exception deleting zone: #{body['code']}, #{body['message']}"
+            end
+            raise msg
+          end
+        end
+      end
+
+      class Mock
+        def delete_record_set(record_set_name, dns_resource_group, zone_name, record_type)
+
+        end
+      end
+    end
+  end
+end

data/lib/fog/azurerm/requests/dns/delete_zone.rb

@@ -0,0 +1,38 @@
+module Fog
+  module DNS
+    class AzureRM
+      class Real
+
+        def delete_zone(zone_name, dns_resource_group)
+          Fog::Logger.debug "Deleting Zone #{zone_name} ..."
+          resource_url = "#{AZURE_RESOURCE}/subscriptions/#{@subscription_id}/resourceGroups/#{dns_resource_group}/providers/Microsoft.Network/dnsZones/#{zone_name}?api-version=2015-05-04-preview"
+          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)
+            Fog::Logger.debug "Zone #{zone_name} deleted successfully."
+            true
+          rescue RestClient::Exception => e
+            body = JSON.parse(e.http_body)
+            if body.key?('error')
+              body = body['error']
+              msg = "Exception deleting zone: #{body['code']}, #{body['message']}"
+            else
+              msg = "Exception deleting zone: #{body['code']}, #{body['message']}"
+            end
+            raise msg
+          end
+        end
+      end
+
+      class Mock
+        def delete_zone(zone_name, dns_resource_group)
+
+        end
+      end
+    end
+  end
+end