azure 0.5.0 → 0.6.0

data/lib/azure/core/http/retry_policy.rb

@@ -1,68 +1,74 @@
-#-------------------------------------------------------------------------
-# Copyright (c) Microsoft. All rights reserved.
-#
-# 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_filter"
-
-module Azure
-  module Core
-    module Http
-    
-      # A HttpFilter implementation that handles retrying based on a 
-      # specific policy when HTTP layer errors occur
-      class RetryPolicy < HttpFilter
-    
-        def initialize(&block)
-          @block = block
-        end
-
-        attr_accessor :retry_data
-
-        # Overrides the base class implementation of call to implement 
-        # a retry loop that uses should_retry? to determine when to 
-        # break the loop
-        # 
-        # req   - HttpRequest. The HTTP request
-        # _next - HttpFilter. The next filter in the pipeline
-        def call(req, _next)
-          retry_data = {}
-          response = nil
-          begin
-            response = _next.call
-          end while should_retry?(response, retry_data)
-          response
-        end
-
-        # Determines if the HTTP request should continue retrying
-        # 
-        # response - HttpResponse. The response from the active request
-        # retry_data - Hash. Stores stateful retry data
-        #
-        # The retry_data is a Hash which can be used to store 
-        # stateful data about the request execution context (such as an 
-        # incrementing counter, timestamp, etc). The retry_data object 
-        # will be the same instance throughout the lifetime of the request.
-        #
-        # If an inline block was passed to the constructor, that block 
-        # will be used here and should return true to retry the job, or
-        # false to stop exit. If an inline block was not passed to the 
-        # constructor the method returns false.
-        #
-        # Alternatively, a subclass could override this method.
-        def should_try?(response, retry_data)
-          @block ? @block.call(response, retry_data) : false
-        end
-      end
-    end
-  end
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# 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_filter"
+
+module Azure
+  module Core
+    module Http
+    
+      # A HttpFilter implementation that handles retrying based on a 
+      # specific policy when HTTP layer errors occur
+      class RetryPolicy < HttpFilter
+    
+        def initialize(&block)
+          @block = block
+        end
+
+        attr_accessor :retry_data
+
+        # Overrides the base class implementation of call to implement 
+        # a retry loop that uses should_retry? to determine when to 
+        # break the loop
+        # 
+        # req   - HttpRequest. The HTTP request
+        # _next - HttpFilter. The next filter in the pipeline
+        def call(req, _next)
+          retry_data = {}
+          response = nil
+          begin
+            response = _next.call
+          rescue
+            retry_data[:error] = $!
+          end while should_retry?(response, retry_data)
+          if retry_data.has_key?(:error)
+            raise retry_data[:error]
+          else
+            response
+          end
+        end
+
+        # Determines if the HTTP request should continue retrying
+        # 
+        # response - HttpResponse. The response from the active request
+        # retry_data - Hash. Stores stateful retry data
+        #
+        # The retry_data is a Hash which can be used to store 
+        # stateful data about the request execution context (such as an 
+        # incrementing counter, timestamp, etc). The retry_data object 
+        # will be the same instance throughout the lifetime of the request.
+        #
+        # If an inline block was passed to the constructor, that block 
+        # will be used here and should return true to retry the job, or
+        # false to stop exit. If an inline block was not passed to the 
+        # constructor the method returns false.
+        #
+        # Alternatively, a subclass could override this method.
+        def should_retry?(response, retry_data)
+          @block ? @block.call(response, retry_data) : false
+        end
+      end
+    end
+  end
 end

data/lib/azure/core/http/signer_filter.rb

@@ -1,34 +1,34 @@
-#-------------------------------------------------------------------------
-# Copyright (c) Microsoft. All rights reserved.
-#
-# 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_filter"
-
-module Azure
-  module Core
-    module Http 
-      # A HttpFilter implementation that creates a authorization signature which is added to the request headers
-      class SignerFilter < HttpFilter
-        def initialize(signer)
-          @signer = signer
-        end
-
-        def call(req, _next)
-          signature = @signer.sign(req.method, req.uri, req.headers)
-          req.headers["Authorization"] = "#{@signer.name} #{signature}"
-          _next.call
-        end
-      end
-    end
-  end
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# 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_filter"
+
+module Azure
+  module Core
+    module Http 
+      # A HttpFilter implementation that creates a authorization signature which is added to the request headers
+      class SignerFilter < HttpFilter
+        def initialize(signer)
+          @signer = signer
+        end
+
+        def call(req, _next)
+          signature = @signer.sign(req.method, req.uri, req.headers)
+          req.headers["Authorization"] = "#{@signer.name} #{signature}"
+          _next.call
+        end
+      end
+    end
+  end
 end

data/lib/azure/core/service.rb

@@ -1,49 +1,63 @@
-#-------------------------------------------------------------------------
-# Copyright (c) Microsoft. All rights reserved.
-#
-# 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_request'
-
-module Azure
-  module Core
-    # A base class for Service implementations
-    class Service
-
-      # Create a new instance of the Service
-      # 
-      # host            - String. The hostname. (optional, Default empty)
-      def initialize(host='')
-        @host = host
-      end
-
-      attr_accessor :host
-
-      def call(method, uri, body=nil, headers=nil)
-        request = Core::Http::HttpRequest.new(method, uri, body)
-        request.headers.merge!(headers) if headers
-
-        request.headers['connection'] = 'keep-alive' if request.respond_to? :headers
-
-        yield request if block_given?
-
-        request.call
-      end
-
-      def generate_uri(path='', query={})
-        uri = URI.parse(File.join(host, path))
-        uri.query = URI.encode_www_form(query) unless query == nil or query.empty?
-        uri
-      end
-    end
-  end
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# 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_request'
+
+module Azure
+  module Core
+    # A base class for Service implementations
+    class Service
+
+      # Create a new instance of the Service
+      # 
+      # host            - String. The hostname. (optional, Default empty)
+      def initialize(host='')
+        @host = host
+      end
+
+      attr_accessor :host
+
+      def call(method, uri, body=nil, headers=nil)
+        if headers && !body.nil?
+          if headers['Content-Encoding'].nil?
+            headers['Content-Encoding'] = body.encoding.to_s
+          else 
+            body.force_encoding(headers['Content-Encoding']) 
+          end
+        end
+
+        request = Core::Http::HttpRequest.new(method, uri, body)
+        request.headers.merge!(headers) if headers
+
+        request.headers['connection'] = 'keep-alive' if request.respond_to? :headers
+
+        yield request if block_given?
+
+        response = request.call
+
+        if !response.nil? && !response.body.nil? && response.headers['content-encoding']
+          response.body.force_encoding(response.headers['content-encoding']) 
+        end
+
+        response
+      end
+
+      def generate_uri(path='', query={})
+        uri = URI.parse(File.join(host, path))
+        uri.query = URI.encode_www_form(query) unless query == nil or query.empty?
+        uri
+      end
+    end
+  end
 end

data/lib/azure/core/signed_service.rb

@@ -1,43 +1,43 @@
-#-------------------------------------------------------------------------
-# Copyright (c) Microsoft. All rights reserved.
-#
-# 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/filtered_service'
-require 'azure/core/http/signer_filter'
-require 'azure/core/auth/shared_key'
-
-module Azure
-  module Core
-    # A base class for Service implementations
-    class SignedService < FilteredService
-
-      # Create a new instance of the SignedService
-      #
-      # signer        - Azure::Core::Auth::Signer. An implementation of Signer used for signing requests. (optional, Default=Azure::Core::Auth::SharedKey.new)
-      # account_name  - String. The account name (optional, Default=Azure.config.storage_account_name)  
-      def initialize(signer=Core::Auth::SharedKey.new, account_name=Azure.config.storage_account_name)
-        super()
-        @account_name = account_name
-        @signer = signer
-        filters.unshift Core::Http::SignerFilter.new(signer) if signer
-      end
-
-      attr_accessor :account_name
-      attr_accessor :signer
-
-      def call(method, uri, body=nil, headers=nil)
-        super(method, uri, body, headers)
-      end
-    end
-  end
+#-------------------------------------------------------------------------
+# # Copyright (c) Microsoft and contributors. All rights reserved.
+#
+# 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/filtered_service'
+require 'azure/core/http/signer_filter'
+require 'azure/core/auth/shared_key'
+
+module Azure
+  module Core
+    # A base class for Service implementations
+    class SignedService < FilteredService
+
+      # Create a new instance of the SignedService
+      #
+      # signer        - Azure::Core::Auth::Signer. An implementation of Signer used for signing requests. (optional, Default=Azure::Core::Auth::SharedKey.new)
+      # account_name  - String. The account name (optional, Default=Azure.config.storage_account_name)  
+      def initialize(signer=Core::Auth::SharedKey.new, account_name=Azure.config.storage_account_name)
+        super()
+        @account_name = account_name
+        @signer = signer
+        filters.unshift Core::Http::SignerFilter.new(signer) if signer
+      end
+
+      attr_accessor :account_name
+      attr_accessor :signer
+
+      def call(method, uri, body=nil, headers=nil)
+        super(method, uri, body, headers)
+      end
+    end
+  end
 end

data/lib/azure/core/utility.rb

@@ -0,0 +1,190 @@
+#-------------------------------------------------------------------------
+# 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 'ipaddr'
+
+if RUBY_VERSION.to_f < 2.0
+  begin
+    require 'Win32/Console/ANSI' if RUBY_PLATFORM =~ /win32|mingw32/
+  rescue LoadError
+    puts 'WARNING: Output will look weird on Windows unless'\
+         ' you install the "win32console" gem.'
+  end
+end
+
+module Azure
+  module Error
+    # Azure Error
+    class Error <  Azure::Core::Error
+      attr_reader :description
+      attr_reader :status_code
+      attr_reader :type
+
+      def initialize(type, status, description)
+        @type = type
+        @status_code = status
+        @description = description
+        super("#{type} (#{status_code}): #{description}")
+      end
+    end
+  end
+
+  module Core
+    module Utility
+      def random_string(str = 'azure', no_of_char = 5)
+        str + (0...no_of_char).map { ('a'..'z').to_a[rand(26)] }.join
+      end
+
+      def xml_content(xml, key, default = '')
+        content = default
+        node = xml.at_css(key)
+        content = node.text if node
+        content
+      end
+
+      def locate_file(name)
+        if File.exist? name
+          name
+        elsif File.exist?(File.join(ENV['HOME'], name))
+          File.join(ENV['HOME'], name)
+        else
+          Loggerx.error_with_exit "Unable to find #{name} file  "
+        end
+      end
+
+      def export_der(cert, key, pass = nil, name = nil)
+        pkcs12 = OpenSSL::PKCS12.create(pass, name, key, cert)
+        Base64.encode64(pkcs12.to_der)
+      rescue Exception => e
+        puts e.message
+        abort
+      end
+
+      def export_fingerprint(certificate)
+        Digest::SHA1.hexdigest(certificate.to_der)
+      end
+
+      def enable_winrm?(winrm_transport)
+        (!winrm_transport.nil? && (winrm_transport.select { |x| x.downcase == 'http' || x.downcase == 'https' }.size > 0))
+      end
+    end
+
+    # Logger
+    module Logger
+      class << self
+        def info(msg)
+          puts msg.bold.white
+        end
+
+        def error_with_exit(msg)
+          puts  msg.bold.red
+          raise msg.bold.red
+        end
+
+        def warn(msg)
+          puts msg.yellow
+          msg
+        end
+
+        def error(msg)
+          puts msg.bold.red
+          msg
+        end
+
+        def exception_message(msg)
+          print msg.bold.red
+          raise msg.bold.red
+        end
+
+        def success(msg)
+          msg_with_new_line = msg + "\n"
+          print msg_with_new_line.green
+        end
+      end
+    end
+  end
+end
+
+class String
+  { reset:  0,
+    bold:  1,
+    dark:  2,
+    underline:  4,
+    blink:  5,
+    orange:  6,
+    negative:  7,
+    black: 30,
+    red: 31,
+    green: 32,
+    yellow: 33,
+    blue: 34,
+    magenta: 35,
+    cyan: 36,
+    white: 37,
+  }.each do |key, value|
+    define_method key do
+      "\e[#{value}m" + self + "\e[0m"
+    end
+  end
+end
+
+# Code validate private/public IP acceptable ranges.
+class IPAddr
+  PRIVATE_RANGES = [
+    IPAddr.new('10.0.0.0/8'),
+    IPAddr.new('172.16.0.0/12'),
+    IPAddr.new('192.168.0.0/16')
+  ]
+
+  def private?
+    return false unless self.ipv4?
+    PRIVATE_RANGES.each do |ipr|
+      return true if ipr.include?(self)
+    end
+    false
+  end
+
+  def public?
+    !private?
+  end
+
+  class << self
+    def validate_ip_and_prefix(ip, cidr)
+      if cidr.to_s.empty?
+        raise "Cidr is missing for IP '#{ip}'."
+      elsif valid?(ip)
+        raise "Ip address '#{ip}' is invalid."
+      elsif !IPAddr.new(ip).private?
+        raise "Ip Address #{ip} must be private."
+      end
+    end
+
+    def validate_address_space(ip)
+      if ip.split('/').size != 2
+        raise "Cidr is invalid for IP #{ip}."
+      elsif valid?(ip)
+        raise "Address space '#{ip}' is invalid."
+      end
+    end
+
+    def address_prefix(ip, cidr)
+      ip + '/' + cidr.to_s
+    end
+
+    def valid?(ip)
+      (IPAddr.new(ip) rescue nil).nil?
+    end
+  end
+end