pgericson-handsoap 1.1.7

data/lib/handsoap/http.rb

@@ -0,0 +1,4 @@
+# -*- coding: utf-8 -*-
+require 'handsoap/http/request'
+require 'handsoap/http/response'
+require 'handsoap/http/drivers'

data/lib/handsoap/http/drivers.rb

@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+require 'handsoap/http/drivers/abstract_driver'
+require 'handsoap/http/drivers/curb_driver'
+require 'handsoap/http/drivers/event_machine_driver'
+require 'handsoap/http/drivers/http_client_driver'
+require 'handsoap/http/drivers/net_http_driver'
+require 'handsoap/http/drivers/mock_driver'
+
+module Handsoap
+  module Http
+    @@drivers = {
+      :curb => Drivers::CurbDriver,
+      :em => Drivers::EventMachineDriver,
+      :event_machine => Drivers::EventMachineDriver,
+      :httpclient => Drivers::HttpClientDriver,
+      :http_client => Drivers::HttpClientDriver,
+      :net_http => Drivers::NetHttpDriver,
+    }
+
+    def self.drivers
+      @@drivers
+    end
+  end
+end

data/lib/handsoap/http/drivers/abstract_driver.rb

@@ -0,0 +1,170 @@
+# -*- coding: utf-8 -*-
+
+module Handsoap
+  module Http
+    module Drivers
+      class AbstractDriver
+        def self.load!
+        end
+
+        def initialize
+          self.class.load!
+        end
+
+        # Parses a raw http response into a +Response+ or +Part+ object.
+        def parse_http_part(headers, body, status = nil, content_type = nil)
+          if headers.kind_of? String
+            headers = parse_headers(headers)
+          end
+          headers = headers.inject({}) {|collect,item| collect[item[0].downcase] = item[1]; collect }
+          if content_type.nil? && headers['content-type']
+            content_type = headers['content-type'].first
+          end
+          boundary = parse_multipart_boundary(content_type)
+          parts = if boundary
+            parse_multipart(boundary, body).map {|raw_part| parse_http_part(raw_part[:head], raw_part[:body]) }
+          end
+          if status.nil?
+            Handsoap::Http::Part.new(headers, body, parts)
+          else
+            Handsoap::Http::Response.new(status, headers, body, parts)
+          end
+        end
+
+        # Content-Type header string -> mime-boundary | nil
+        def parse_multipart_boundary(content_type)
+          if %r|\Amultipart.*boundary=\"?([^\";,]+)\"?|n.match(content_type)
+            $1.dup
+          end
+        end
+
+        # Parses a multipart http-response body into parts.
+        # +boundary+ is a string of the boundary token.
+        # +content_io+ is either a string or an IO. If it's an IO, then content_length must be specified.
+        # +content_length+ (optional) is an integer, specifying the length of +content_io+
+        #
+        # This code is lifted from cgi.rb
+        #
+        def parse_multipart(boundary, content_io, content_length = nil)
+          if content_io.kind_of? String
+            content_length = content_io.length
+            content_io = StringIO.new(content_io, 'r')
+          elsif !(content_io.kind_of? IO) || content_length.nil?
+            raise "Second argument must be String or IO with content_length"
+          end
+
+          boundary = "--" + boundary
+          quoted_boundary = Regexp.quote(boundary, "n")
+          buf = ""
+          bufsize = 10 * 1024
+          boundary_end = ""
+
+          # start multipart/form-data
+          content_io.binmode if defined? content_io.binmode
+          boundary_size = boundary.size + "\r\n".size
+          content_length -= boundary_size
+          status = content_io.read(boundary_size)
+          if nil == status
+            raise EOFError, "no content body"
+          elsif boundary + "\r\n" != status
+            raise EOFError, "bad content body"
+          end
+
+          parts = []
+
+          loop do
+            head = nil
+            if 10240 < content_length
+              require "tempfile"
+              body = Tempfile.new("Handsoap")
+            else
+              begin
+                require "stringio"
+                body = StringIO.new
+              rescue LoadError
+                require "tempfile"
+                body = Tempfile.new("Handsoap")
+              end
+            end
+            body.binmode if defined? body.binmode
+
+            until head and /#{quoted_boundary}(?:\r\n|--)/n.match(buf)
+
+              if (not head) and /\r\n\r\n/n.match(buf)
+                buf = buf.sub(/\A((?:.|\n)*?\r\n)\r\n/n) do
+                  head = $1.dup
+                  ""
+                end
+                next
+              end
+
+              if head and ( ("\r\n" + boundary + "\r\n").size < buf.size )
+                body.print buf[0 ... (buf.size - ("\r\n" + boundary + "\r\n").size)]
+                buf[0 ... (buf.size - ("\r\n" + boundary + "\r\n").size)] = ""
+              end
+
+              c = if bufsize < content_length
+                    content_io.read(bufsize)
+                  else
+                    content_io.read(content_length)
+                  end
+              if c.nil? || c.empty?
+                raise EOFError, "bad content body"
+              end
+              buf.concat(c)
+              content_length -= c.size
+            end
+
+            buf = buf.sub(/\A((?:.|\n)*?)(?:[\r\n]{1,2})?#{quoted_boundary}([\r\n]{1,2}|--)/n) do
+              body.print $1
+              if "--" == $2
+                content_length = -1
+              end
+              boundary_end = $2.dup
+              ""
+            end
+
+            body.rewind
+            parts << {:head => head, :body => body.read(body.size)}
+
+            break if buf.size == 0
+            break if content_length == -1
+          end
+          raise EOFError, "bad boundary end of body part" unless boundary_end =~ /--/
+          parts
+        end
+
+        # lifted from webrick/httputils.rb
+        def parse_headers(raw)
+          header = Hash.new([].freeze)
+          field = nil
+          tmp = raw.gsub(/^(\r\n)+|(\r\n)+$/, '')
+          (tmp.respond_to?(:lines) ? tmp.lines : tmp).each {|line|
+            case line
+            when /^([A-Za-z0-9!\#$%&'*+\-.^_`|~]+):\s*(.*?)\s*\z/om
+              field, value = $1, $2
+              field.downcase!
+              header[field] = [] unless header.has_key?(field)
+              header[field] << value
+            when /^\s+(.*?)\s*\z/om
+              value = $1
+              unless field
+                raise "bad header '#{line.inspect}'."
+              end
+              header[field][-1] << " " << value
+            else
+              raise "bad header '#{line.inspect}'."
+            end
+          }
+          header.each {|key, values|
+            values.each {|value|
+              value.strip!
+              value.gsub!(/\s+/, " ")
+            }
+          }
+          header
+        end
+      end
+    end
+  end
+end

data/lib/handsoap/http/drivers/curb_driver.rb

@@ -0,0 +1,65 @@
+# -*- coding: utf-8 -*-
+require 'handsoap/http/drivers/abstract_driver'
+
+module Handsoap
+  module Http
+    module Drivers
+      class CurbDriver < AbstractDriver
+        attr_accessor :enable_cookies
+
+        def initialize
+          @enable_cookies = false
+        end
+
+        def self.load!
+          require 'curb'
+        end
+
+        def get_curl(url)
+          if @curl
+            @curl.url = url
+          else
+            @curl                 = ::Curl::Easy.new(url)
+            @curl.timeout         = Handsoap.timeout
+            @curl.enable_cookies  = @enable_cookies
+            
+            if Handsoap.follow_redirects?
+              @curl.follow_location = true
+              @curl.max_redirects   = Handsoap.max_redirects
+            end
+          end
+          @curl
+        end
+        
+        private :get_curl
+
+        def send_http_request(request)
+          http_client = get_curl(request.url)
+          # Set credentials. The driver will negotiate the actual scheme
+          if request.username && request.password
+            http_client.userpwd = [request.username, ":", request.password].join
+          end
+          # pack headers
+          headers = request.headers.inject([]) do |arr, (k,v)|
+            arr + v.map {|x| "#{k}: #{x}" }
+          end
+          http_client.headers = headers
+          # I don't think put/delete is actually supported ..
+          case request.http_method
+          when :get
+            http_client.http_get
+          when :post
+            http_client.http_post(request.body)
+          when :put
+            http_client.http_put(request.body)
+          when :delete
+            http_client.http_delete
+          else
+            raise "Unsupported request method #{request.http_method}"
+          end
+          parse_http_part(http_client.header_str.gsub(/^HTTP.*\r\n/, ""), http_client.body_str, http_client.response_code, http_client.content_type)
+        end
+      end
+    end
+  end
+end

data/lib/handsoap/http/drivers/event_machine_driver.rb

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+
+module Handsoap
+  module Http
+    module Drivers
+      class EventMachineDriver < AbstractDriver
+        def self.load!
+          require 'eventmachine'
+          require 'em-http'
+        end
+
+        def send_http_request_async(request)
+          emr = EventMachine::HttpRequest.new(request.url)
+
+          if request.username && request.password
+            # TODO: Verify that this is actually supported?
+            request.headers['authorization'] = [request.username, request.password]
+          end
+
+          case request.http_method
+          when :get
+            emdef = emr.get(:head => request.headers)
+          when :post
+            emdef = emr.post(:head => request.headers, :body => request.body)
+          when :put
+            emdef = emr.put(:head => request.headers, :body => request.body)
+          when :delete
+            emdef = emr.delete
+          else
+            raise "Unsupported request method #{request.http_method}"
+          end
+
+          deferred = Handsoap::Deferred.new
+          emdef.callback do
+            http_response = parse_http_part(emdef.response_header, emdef.response, emdef.response_header.status)
+            deferred.trigger_callback http_response
+          end
+          emdef.errback do
+            deferred.trigger_errback emdef
+          end
+          deferred
+        end
+      end
+    end
+  end
+end

data/lib/handsoap/http/drivers/http_client_driver.rb

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+require 'handsoap/http/drivers/abstract_driver'
+
+module Handsoap
+  module Http
+    module Drivers
+      class HttpClientDriver < AbstractDriver
+        def self.load!
+          require 'httpclient'
+        end
+
+        def send_http_request(request)
+          http_client = HTTPClient.new
+          # Set credentials. The driver will negotiate the actual scheme
+          if request.username && request.password
+            domain = request.url.match(/^(http(s?):\/\/[^\/]+\/)/)[1]
+            http_client.set_auth(domain, request.username, request.password)
+          end
+          http_client.ssl_config.set_trust_ca(request.trust_ca_file) if request.trust_ca_file
+          http_client.ssl_config.set_client_cert_file(request.client_cert_file,request.client_cert_key_file) if request.client_cert_file and request.client_cert_key_file
+          # pack headers
+          headers = request.headers.inject([]) do |arr, (k,v)|
+            arr + v.map {|x| [k,x] }
+          end
+          response = http_client.request(request.http_method, request.url, nil, request.body, headers)
+          response_headers = response.header.all.inject({}) do |h, (k, v)|
+            k.downcase!
+            if h[k].nil?
+              h[k] = [v]
+            else
+              h[k] << v
+            end
+            h
+          end
+          parse_http_part(response_headers, response.content, response.status, response.contenttype)
+        end
+      end
+    end
+  end
+end

data/lib/handsoap/http/drivers/mock_driver.rb

@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+require 'handsoap/http/drivers/abstract_driver'
+
+module Handsoap
+  module Http
+    module Drivers
+      # A mock driver for your testing needs.
+      #
+      # To use it, create a new instance and assign to +Handsoap::Http.drivers+. Then configure +Handsoap::Service+ to use it:
+      #
+      #     Handsoap::Http.drivers[:mock] = Handsoap::Http::Drivers::MockDriver.new :status => 200, :headers => headers, :content => body
+      #     Handsoap.http_driver = :mock
+      #
+      # Remember that headers should use \r\n, rather than \n.
+      class MockDriver < AbstractDriver
+        attr_accessor :mock, :last_request, :is_loaded
+
+        def initialize(mock)
+          @mock = mock
+          @is_loaded = false
+        end
+
+        def load!
+          is_loaded = true
+        end
+
+        def new
+          self
+        end
+
+        def send_http_request(request)
+          @last_request = request
+          (mock.kind_of? Hash) ?
+            parse_http_part(mock[:headers], mock[:content], mock[:status], mock[:content_type]) : mock
+        end
+      end
+    end
+  end
+end

data/lib/handsoap/http/drivers/net_http_driver.rb

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+
+module Handsoap
+  module Http
+    module Drivers
+      class NetHttpDriver < AbstractDriver
+        def self.load!
+          require 'net/http'
+          require 'uri'
+        end
+
+        def send_http_request(request)
+          url = request.url
+          unless url.kind_of? ::URI::Generic
+            url = ::URI.parse(url)
+          end
+          ::URI::Generic.send(:public, :path_query) # hackety hack
+          path = url.path_query
+          http_request = case request.http_method
+                         when :get
+                           Net::HTTP::Get.new(path)
+                         when :post
+                           Net::HTTP::Post.new(path)
+                         when :put
+                           Net::HTTP::Put.new(path)
+                         when :delete
+                           Net::HTTP::Delete.new(path)
+                         else
+                           raise "Unsupported request method #{request.http_method}"
+                         end
+                         
+          http_client = Net::HTTP.new(url.host, url.port)
+          
+          #http_client.read_timeout = 120
+          http_client.read_timeout = Handsoap.timeout
+          
+          http_client.use_ssl = true if url.scheme == 'https'
+          
+          if request.username && request.password
+            # TODO: http://codesnippets.joyent.com/posts/show/1075
+            http_request.basic_auth request.username, request.password
+          end
+          request.headers.each do |k, values|
+            values.each do |v|
+              http_request.add_field(k, v)
+            end
+          end
+          http_request.body = request.body
+          # require 'stringio'
+          # debug_output = StringIO.new
+          # http_client.set_debug_output debug_output
+          http_response = http_client.start do |client|
+            client.request(http_request)
+          end
+          # puts debug_output.string
+          # hacky-wacky
+          def http_response.get_headers
+            @header.inject({}) do |h, (k, v)|
+              h[k.downcase] = v
+              h
+            end
+          end
+          # net/http only supports basic auth. We raise a warning if the server requires something else.
+          if http_response.code == 401 && http_response.get_headers['www-authenticate']
+            auth_type = http_response.get_headers['www-authenticate'].chomp.match(/\w+/)[0].downcase
+            if auth_type != "basic"
+              raise "Authentication type #{auth_type} is unsupported by net/http"
+            end
+          end
+          parse_http_part(http_response.get_headers, http_response.body, http_response.code)
+        end
+      end
+    end
+  end
+end