aws-sdk 1.0.0

data/lib/aws/http/builtin_handler.rb

@@ -0,0 +1,69 @@
+# Copyright 2011 Amazon.com, Inc. or its affiliates. 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. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file 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 'net/http'
+require 'net/https'
+
+module AWS
+  module Http
+    
+    # @private
+    class BuiltinHandler
+
+      def handle(request, response)
+
+        http = Net::HTTP.new(request.host, request.use_ssl? ? 443 : 80)
+
+        # http://www.ruby-lang.org/en/news/2007/10/04/net-https-vulnerability/
+        if request.use_ssl?
+          http.use_ssl = true
+          http.verify_mode = OpenSSL::SSL::VERIFY_PEER
+          store = OpenSSL::X509::Store.new
+          store.set_default_paths
+          http.cert_store = store
+        end
+
+        http.start do 
+
+          # Net::HTTP adds this header for us when the body is
+          # provided, but it messes up signing
+          headers = { 'content-type' => '' }
+
+          # Net::HTTP calls strip on each header value, this causes errors
+          # when the values are numbers (like Content-Length)
+          request.headers.each_pair do |key,value|
+            headers[key] = value.to_s
+          end
+
+          http_request_class = case request.http_method
+          when 'HEAD'   then Net::HTTP::Head
+          when 'GET'    then Net::HTTP::Get
+          when 'PUT'    then Net::HTTP::Put
+          when 'POST'   then Net::HTTP::Post
+          when 'DELETE' then Net::HTTP::Delete
+          else raise "unsupported http request method: #{request.http_method}"
+          end
+
+          http_request = http_request_class.new(request.uri, headers)
+          http.request(http_request, request.body) do |http_response|
+            response.body = http_response.body
+            response.status = http_response.code.to_i
+            response.headers = http_response.to_hash
+          end
+
+        end
+      end
+
+    end
+  end
+end

data/lib/aws/http/curb_handler.rb

@@ -0,0 +1,123 @@
+# Copyright 2011 Amazon.com, Inc. or its affiliates. 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. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file 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 'thread'
+
+module AWS
+  module Http
+
+    # @private
+    class CurbHandler
+
+      def initialize
+        @q = []
+        @sem = Mutex.new
+        @multi = Curl::Multi.new
+
+        start_processor
+      end
+
+      def handle request, response
+        raise "unsupport http reqest method: #{request.http_method}" unless
+          ['GET', 'HEAD', 'PUT', 'POST', 'DELETE'].include? request.http_method
+        @sem.synchronize do
+          @q << [request, response, Thread.current]
+          @processor.wakeup
+        end
+        Thread.stop
+        nil
+      end
+
+      # fills the Curl::Multi handle with the given array of queue
+      # items, calling make_easy_handle on each one first
+      private
+      def fill_multi(items)
+        items.each do |item|
+          c = make_easy_handle(*item)
+          @multi.add(c)
+        end
+      end
+
+      # starts a background thread that waits for new items and
+      # sends them through the Curl::Multi handle
+      private
+      def start_processor
+        @processor = Thread.new do
+          loop do
+            items = nil
+            @sem.synchronize do
+              items = @q.slice!(0..-1)
+            end
+            unless items.empty?
+              fill_multi(items)
+              @multi.perform do
+                # curl is idle, so process more items if we can get them
+                # without blocking
+                if !@q.empty? && @sem.try_lock
+                  begin
+                    fill_multi(@q.slice!(0..-1))
+                  ensure
+                    @sem.unlock
+                  end
+                end
+              end
+            end
+
+            # wait for a new item to show up before continuing
+            Thread.stop if @q.empty?
+          end
+        end
+      end
+
+      private
+      def make_easy_handle request, response, thread = nil
+
+        url = request.use_ssl? ? 
+          "https://#{request.host}:443#{request.uri}" :
+          "http://#{request.host}#{request.uri}"
+        puts url
+
+        curl = Curl::Easy.new(url)
+        curl.headers = request.headers
+
+        curl.on_header {|header_data|
+          name, value = header_data.strip.split(/:\s+/, 2)
+          response.headers[name] = value
+          header_data.length
+        }
+
+        case request.http_method
+        when 'GET'
+          # ....
+        when 'HEAD'
+          curl.head = true
+        when 'PUT'
+          curl.put_data = request.body
+        when 'POST'
+          curl.post_body = request.body
+        when 'DELETE'
+          curl.delete = true
+        end
+
+        curl.on_complete do
+          response.body = curl.body_str
+          response.status = curl.response_code
+          thread.run if thread
+        end
+
+        curl
+      end
+
+    end
+  end
+end

data/lib/aws/http/handler.rb

@@ -0,0 +1,77 @@
+# Copyright 2011 Amazon.com, Inc. or its affiliates. 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. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file 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 'aws/meta_utils'
+
+module AWS
+  module Http
+    
+    # @private
+    class Handler
+
+      attr_reader :base
+
+      def initialize(base, &block)
+        @base = base
+        if base.respond_to?(:handle)
+
+          unless block.arity == 2
+            raise ArgumentError, 'passed block must accept 2 arguments'
+          end
+          MetaUtils.extend_method(self, :handle, &block)
+
+        elsif base.respond_to?(:handle_async)
+
+          unless block.arity == 3
+            raise ArgumentError, 'passed block must accept 3 arguments'
+          end
+
+          MetaUtils.extend_method(self, :handle_async) do |req, resp, handle|
+            @base.handle_async(req, resp, handle)
+          end
+          MetaUtils.extend(self) do
+            define_method(:handle) do |req, resp|
+              raise "attempted to call #handle on an async handler"
+            end
+            define_method(:handle_async, &block)
+          end
+
+        else
+          raise ArgumentError, 'base must respond to #handle or #handle_async'
+        end
+      end
+
+      def handle(request, http_response)
+        @base.handle(request, http_response)
+      end
+
+      def handle_async(request, http_response, handle)
+        Thread.new do
+          begin
+            self.handle(request, http_response)
+          rescue => e
+            handle.signal_failure
+          else
+            handle.signal_success
+          end
+        end
+      end
+
+      def sleep_with_callback seconds, &block
+        Kernel.sleep(seconds)
+        yield
+      end
+
+    end
+  end
+end

data/lib/aws/http/httparty_handler.rb

@@ -0,0 +1,61 @@
+# Copyright 2011 Amazon.com, Inc. or its affiliates. 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. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file 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 'httparty'
+
+module AWS
+  module Http
+
+    # @private
+    class HTTPartyHandler
+
+      include HTTParty
+
+      class NoOpParser < HTTParty::Parser
+        SupportedFormats = {}
+      end
+
+      def handle(request, response)
+
+        url = request.use_ssl? ?
+          "https://#{request.host}:443#{request.uri}" :
+          "http://#{request.host}#{request.uri}"
+
+        # get, post, put, delete, head
+        method = request.http_method.downcase
+
+        # Net::HTTP adds this header for us when the body is
+        # provided, but it messes up signing
+        headers = { 'content-type' => '' }
+
+        # headers must have string values (net http calls .strip on them)
+        request.headers.each_pair do |key,value|
+          headers[key] = value.to_s
+        end
+
+        begin
+          http_response = self.class.send(method, url,
+                                          :headers => headers,
+                                          :body => request.body,
+                                          :parser => NoOpParser)
+        rescue Timeout::Error => e
+          response.timeout = true
+        else
+          response.body = http_response.body
+          response.status = http_response.code.to_i
+          response.headers = http_response.to_hash
+        end
+      end
+    end
+  end
+end

data/lib/aws/http/request.rb

@@ -0,0 +1,136 @@
+# Copyright 2011 Amazon.com, Inc. or its affiliates. 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. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file 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 'stringio'
+require 'aws/http/request_param'
+
+module AWS
+  module Http
+
+    # Base class for all service reqeusts.
+    # @private
+    class Request
+
+      # Returns a new empty http request object.
+      def initialize
+        @host = nil
+        @http_method = 'POST'
+        @path = '/'
+        @headers = CaseInsensitiveHash.new
+        @params = []
+        @use_ssl = true
+      end
+
+      # @return [String] hostname of the request
+      attr_accessor :host
+
+      # @return [CaseInsensitiveHash] request headers
+      attr_reader :headers
+
+      # @return [Array] An array of request params, each param responds to
+      #   #name and #value.
+      attr_reader :params
+
+      # @return [String] GET, PUT POST, HEAD or DELETE, defaults to POST
+      attr_accessor :http_method
+
+      # @return [String] path of the request URI, defaults to /
+      attr_reader :path
+
+      # @return [String] the AWS access key ID used to authorize the
+      #   request
+      attr_accessor :access_key_id
+
+      # @param [Boolean] ssl If the request should be sent over ssl or not.
+      def use_ssl= use_ssl
+        @use_ssl = use_ssl
+      end
+
+      # @return [Boolean] If this request should be sent over ssl or not.
+      def use_ssl?
+        @use_ssl
+      end
+
+      # Adds a request param.
+      #
+      # @overload add_param(param_name, param_value = nil)
+      #   Add a param (name/value)
+      #   @param [String] param_name
+      #   @param [String] param_value Leave blank for sub resources
+      #
+      # @overload add_param(param_obj)
+      #   Add a param (object)
+      #   @param [Param] param_obj
+      #
+      def add_param name_or_param, value = nil
+        if name_or_param.kind_of?(Param)
+          @params << name_or_param
+        else
+          @params << Param.new(name_or_param, value)
+        end
+      end
+
+      # @private
+      def get_param param_name
+        @params.detect{|p| p.name == param_name } ||
+          raise("undefined param #{param_name}")
+      end
+
+      # @return [String] the request uri
+      def uri
+        querystring ? "#{path}?#{querystring}" : path
+      end
+
+      # @return [String] Returns the request params url encoded, or nil if 
+      #   this request has no params.
+      def url_encoded_params
+        if @params.empty?
+          nil
+        else
+          @params.sort.collect{|p| p.encoded }.join('&')
+        end
+      end
+
+      # @return [String, nil] Returns the requesty querystring.
+      def querystring
+        nil
+      end
+
+      # @return [String, nil] Returns the request body.
+      def body
+        url_encoded_params
+      end
+
+      # @private
+      class CaseInsensitiveHash < Hash
+        
+        def []= key, value
+          super(key.to_s.downcase, value)
+        end
+
+        def [] key
+          super(key.to_s.downcase)
+        end
+
+        def has_key?(key)
+          super(key.to_s.downcase)
+        end
+        alias_method :key?, :has_key?
+        alias_method :include?, :has_key?
+        alias_method :member?, :has_key?
+
+      end
+
+    end
+  end
+end