azurex 0.6.5

data/lib/azure/base_management/location.rb

@@ -0,0 +1,27 @@
+#-------------------------------------------------------------------------
+# Copyright 2013 Microsoft Open Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require "azure/base_management/serialization"
+
+module Azure
+  module BaseManagement
+    class Location
+      attr_accessor :name, :available_services
+
+      def initialize
+        yield self if block_given?
+      end
+    end
+  end
+end

data/lib/azure/base_management/management_http_request.rb

@@ -0,0 +1,244 @@
+#-------------------------------------------------------------------------
+# Copyright 2013 Microsoft Open Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+require 'azure/core/http/http_response'
+require 'azure/core/http/http_request'
+include Azure::Core::Http
+
+# Represents an HTTP request that can perform synchronous queries to
+# an HTTP server, returning a HttpResponse
+module Azure
+  module BaseManagement
+    class ManagementHttpRequest < HttpRequest
+      attr_accessor :uri, :warn, :key, :cert
+
+      # Public: Creates the ManagementHttpRequest
+      #
+      # method  - Symbol. The HTTP method to use (:get, :post, :put, :del, etc...)
+      # path    - URI. The URI of the HTTP endpoint to query
+      # body    - IO or String. The request body (optional)
+      # key     - String. The request key
+      # cert    - String. The request certificate
+      def initialize(method, path, body = nil)
+        super
+        @warn = false
+        content_length = body ? body.bytesize.to_s : '0'
+        @headers = {
+          'x-ms-version' => '2014-05-01',
+          'Content-Type' => 'application/xml',
+          'Content-Length' => content_length
+        }
+        @uri = URI.parse(Azure.config.management_endpoint + Azure.config.subscription_id + path)
+        @key = Azure.config.http_private_key
+        @cert = Azure.config.http_certificate_key
+      end
+
+      # Public: Sends a request to HTTP server and returns a HttpResponse
+      #
+      # * +options+             - Hash. Optional parameters.
+      #
+      # ==== Options
+      #
+      # Accepted key/value pairs in options parameter are:
+      #
+      # * +:fire_and_forget+    - Boolean(optional). Default is false. If true, the client 
+      #                           does not wait until the request is completed
+      # * +:no_exit_on_failure+ - Boolean(optional). Default is false.
+      #
+      # Returns a Nokogiri::XML instance of HttpResponse body
+      def call(options={})
+        fire_and_forget = options[:fire_and_forget].nil? ? false : options[:fire_and_forget]
+        request = http_request_class.new(uri.request_uri, headers)
+        request.body = body if body
+        http = http_setup
+        # http.set_debug_output($stdout)
+        if fire_and_forget
+          response = validate_response(HttpResponse.new(http.request(request)), options)
+          Nokogiri::XML response.body unless response.nil?
+        else
+          response = wait_for_completion(HttpResponse.new(http.request(request)), options)
+          Nokogiri::XML response.body unless response.nil?
+        end
+      end
+
+      # Public: Wait for HTTP request completion.
+      #
+      # ==== Attributes
+      #
+      # * +response+            - Azure::Core::Http::HttpResponse. HttpResponse Response
+      # * +options+             - Hash. Optional parameters.
+      #
+      # ==== Options
+      #
+      # Accepted key/value pairs in options parameter are:
+      #
+      # * +:no_exit_on_failure+ - Boolean(optional). Default is false.
+      #
+      # Print Error or Success of HttpRequest
+      def wait_for_completion(response, options={})
+        no_exit_on_failure = options[:no_exit_on_failure].nil? ? false : options[:no_exit_on_failure]
+        ret_val = Nokogiri::XML response.body
+        if ret_val.at_css('Error Code') && ret_val.at_css('Error Code').content == 'AuthenticationFailed'
+          error_msg = ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+          return log_warn_or_exit(response, error_msg, no_exit_on_failure)
+        end
+        if response.status_code.to_i == 200 || response.status_code.to_i == 201
+          return response
+        elsif redirected? response
+          rebuild_request response
+        elsif response.status_code.to_i > 201 && response.status_code.to_i <= 299
+          check_completion(response.headers['x-ms-request-id'])
+        elsif warn && !response.success?
+          # Loggerx.warn ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+        elsif response.body
+          if ret_val.at_css('Error Code') && ret_val.at_css('Error Message')
+            error_msg = ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+            return log_warn_or_exit(response, error_msg, no_exit_on_failure)
+          else
+            Loggerx.exception_message "http error: #{response.status_code}"
+          end
+        else
+          Loggerx.exception_message "http error: #{response.status_code}"
+        end
+      end
+
+      # Public: Validate the Http response
+      #
+      # ==== Attributes
+      #
+      # * +response+            - Azure::Core::Http::HttpResponse. HttpResponse Response
+      # * +options+             - Hash. Optional parameters.
+      #
+      # ==== Options
+      #
+      # Accepted key/value pairs in options parameter are:
+      #
+      # * +:no_exit_on_failure+ - Boolean(optional). Default is false.
+      #
+      # Print Error or Success of HttpRequest
+      def validate_response(response, options={})
+        no_exit_on_failure = options[:no_exit_on_failure].nil? ? false : options[:no_exit_on_failure]
+        ret_val = Nokogiri::XML response.body
+        if ret_val.at_css('Error Code') && ret_val.at_css('Error Code').content == 'AuthenticationFailed'
+          error_msg = ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+          return log_warn_or_exit(response, error_msg, no_exit_on_failure)
+        end
+        status_code = response.status_code.to_i
+        if [200, 201, 202].include? status_code
+          return response
+        elsif response.body
+          if ret_val.at_css('Error Code') && ret_val.at_css('Error Message')
+            error_msg = ret_val.at_css('Error Code').content + ' : ' + ret_val.at_css('Error Message').content
+            return log_warn_or_exit(response, error_msg, no_exit_on_failure)
+          else
+            Loggerx.exception_message "http error: #{response.status_code}"
+          end
+        else
+          Loggerx.exception_message "http error: #{response.status_code}"
+        end
+      end
+
+      # Public: Gets the status of the specified operation and determines whether
+      # the operation has succeeded, failed, or is still in progress.
+      #
+      # ==== Attributes
+      #
+      # * +request_id+       - String.  x-ms-request-id response header of request
+      #
+      # See: http://msdn.microsoft.com/en-us/library/windowsazure/ee460783.aspx
+      #
+      # Print Error or Success of Operation.
+      def check_completion(request_id)
+        request_path = "/#{Azure.config.subscription_id}/operations/#{request_id}"
+        http = http_setup
+        headers['Content-Length'] = '0'
+        @method = :get
+        done = false
+        while not done
+          print '# '
+          request = http_request_class.new(request_path, headers)
+          response = HttpResponse.new(http.request(request))
+          ret_val = Nokogiri::XML response.body
+          status = xml_content(ret_val, 'Operation Status')
+          status_code = response.status_code.to_i
+          if status != 'InProgress'
+            done = true
+          end
+          if redirected? response
+            host_uri = URI.parse(response.headers['location'])
+            http = http_setup(host_uri)
+            done = false
+          end
+          if done
+            if status.downcase != 'succeeded'
+              error_code = xml_content(ret_val, 'Operation Error Code')
+              error_msg = xml_content(ret_val, 'Operation Error Message')
+              Loggerx.exception_message "#{error_code}: #{error_msg}"
+            else
+              Loggerx.success "#{status.downcase} (#{status_code})"
+            end
+            return
+          else
+            sleep(5)
+          end
+        end
+      end
+
+      def rebuild_request(response)
+        host_uri = URI.parse(response.headers['location'])
+        request = http_request_class.new(host_uri.request_uri, headers)
+        request.body = body if body
+        http = http_setup(host_uri)
+        wait_for_completion(HttpResponse.new(http.request(request)))
+      end
+
+      def redirected?(response)
+        (response.status_code.to_i == 307)
+      end
+
+      def http_setup(host_uri = nil)
+        @uri = host_uri if host_uri
+        http = nil
+        if ENV['HTTP_PROXY'] || ENV['HTTPS_PROXY']
+          if ENV['HTTP_PROXY']
+            proxy_uri = URI.parse(ENV['HTTP_PROXY'])
+          else
+            proxy_uri = URI.parse(ENV['HTTPS_PROXY'])
+          end
+          http = Net::HTTP::Proxy(proxy_uri.host, proxy_uri.port).new(uri.host, uri.port)
+        else
+          http = Net::HTTP.new(uri.host, uri.port)
+        end
+
+        if uri.scheme.downcase == 'https'
+          http.use_ssl = true
+          http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+          http.cert = cert
+          http.key = key
+        end
+        http
+      end
+
+      private
+        def log_warn_or_exit(response, error_msg, no_exit_on_failure)
+          if no_exit_on_failure
+            Loggerx.warn "Warning: #{error_msg}"
+            return response
+          else
+            Loggerx.error_with_exit error_msg
+          end
+        end
+    end
+  end
+end

data/lib/azure/base_management/serialization.rb

@@ -0,0 +1,129 @@
+#-------------------------------------------------------------------------
+# Copyright 2013 Microsoft Open Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+
+module Azure
+  module BaseManagement
+    module Serialization
+      def self.locations_from_xml(locationXML)
+        location_objs = []
+        xml = locationXML.css('Locations Location')
+        xml.each do |meta_node|
+          loc = Location.new
+          loc.name = xml_content(meta_node, 'Name')
+          available_services = meta_node.css('AvailableServices').children
+          loc.available_services = available_services.to_ary.join(', ')
+          location_objs << loc
+        end
+        location_objs
+      end
+
+      def self.affinity_group_to_xml(name, location, label, options = {})
+        builder = Nokogiri::XML::Builder.new do |xml|
+          xml.CreateAffinityGroup(
+            'xmlns' => 'http://schemas.microsoft.com/windowsazure'
+          ) do
+            xml.Name name
+            xml.Label Base64.encode64(label).strip
+            xml.Description options[:description]
+            xml.Location location
+          end
+        end
+        builder.doc.to_xml
+      end
+
+      def self.affinity_groups_from_xml(affinity_xml)
+        affinity_groups = []
+        affinity_group_services_xml = affinity_xml.css(
+          'AffinityGroups  AffinityGroup'
+        )
+
+        affinity_group_services_xml.each do |ag_xml|
+          affinity_group = AffinityGroup.new
+          affinity_group.name = xml_content(ag_xml, 'Name')
+          affinity_group.label = Base64.decode64(xml_content(ag_xml, 'Label'))
+          affinity_group.description = xml_content(ag_xml, 'Description')
+          affinity_group.location = xml_content(ag_xml, 'Location')
+
+          capabilities = ag_xml.css('Capabilities Capability')
+          affinity_group.capability = capabilities.map { |x|  x.content }
+          affinity_groups << affinity_group
+        end
+        affinity_groups.compact
+      end
+
+      def self.affinity_group_from_xml(affinity_xml)
+        hosted_services_xml = affinity_xml.css(
+          'AffinityGroup HostedServices HostedService'
+        )
+
+        storage_services_xml = affinity_xml.css(
+          'AffinityGroup StorageServices StorageService'
+        )
+
+        capability_xml = affinity_xml.css(
+          'AffinityGroup Capabilities Capability'
+        )
+
+        AffinityGroup.new do |affinity_group|
+          affinity_group.name = xml_content(
+            affinity_xml,
+            'AffinityGroup Name'
+          )
+          affinity_group.label = Base64.decode64(
+            xml_content(
+              affinity_xml,
+              'AffinityGroup Label'
+            )
+          )
+          affinity_group.description = xml_content(
+            affinity_xml,
+            'AffinityGroup Description'
+          )
+          affinity_group.location = xml_content(
+            affinity_xml,
+            'AffinityGroup Location'
+          )
+          affinity_group.hosted_services = []
+          hosted_services_xml.each do |hosted_service_xml|
+            affinity_group.hosted_services << {
+              url: xml_content(hosted_service_xml, 'Url'),
+              service_name: xml_content(hosted_service_xml, 'ServiceName')
+            }
+          end
+          affinity_group.storage_services = []
+          storage_services_xml.each do |storage_service_xml|
+            affinity_group.storage_services << {
+              url: xml_content(storage_service_xml, 'Url'),
+              service_name: xml_content(storage_service_xml, 'ServiceName')
+            }
+          end
+          affinity_group.capability = capability_xml.map { |x| x.content }
+        end
+      end
+
+      def self.resource_to_xml(label, options = {})
+        builder = Nokogiri::XML::Builder.new do |xml|
+          xml.UpdateAffinityGroup(
+            'xmlns' => 'http://schemas.microsoft.com/windowsazure'
+          ) do
+            xml.Label Base64.encode64(label).strip
+            xml.Description options[:description] if options[:description]
+          end
+        end
+        builder.doc.to_xml
+      end
+    end
+  end
+end

data/lib/azure/base_management/sql_management_http_request.rb

@@ -0,0 +1,45 @@
+#-------------------------------------------------------------------------
+# Copyright 2013 Microsoft Open Technologies, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#--------------------------------------------------------------------------
+# Represents an HTTP request that can perform synchronous queries to
+# an HTTP server, returning a HttpResponse
+module Azure
+  module BaseManagement
+    # This class is used for communicating with the Management certificate authentication API endpoint
+    class SqlManagementHttpRequest < ManagementHttpRequest
+      attr_accessor :uri
+      # Public: Creates the ManagementHttpRequest
+      #
+      # method  - Symbol. The HTTP method to use (:get, :post, :put, :del, etc...)
+      # path    - URI. The URI of the HTTP endpoint to query
+      # body    - IO or String. The request body (optional)
+      def initialize(method, path, body = nil)
+        if sql_endpoint?
+          super(method, path, body)
+          @headers['x-ms-version'] = '1.0'
+          @uri = URI.parse(Azure.config.sql_database_management_endpoint + Azure.config.subscription_id + path)
+        else
+          path = "/services/sqlservers#{path}"
+          super(method, path, body)
+        end
+      end
+
+      private
+
+      def sql_endpoint?
+        Azure.config.sql_database_authentication_mode == :sql_server
+      end
+    end
+  end
+end