typhoeus 0.1.6 → 0.2.0

data/lib/typhoeus/request.rb

@@ -1,22 +1,87 @@
+require 'uri'
+
 module Typhoeus
   class Request
-    attr_accessor :method, :params, :body, :headers, :timeout, :user_agent, :response, :cache_timeout
     attr_reader   :url
-    
+    attr_writer   :headers
+    attr_accessor :method, :params, :body, :connect_timeout, :timeout,
+                  :user_agent, :response, :cache_timeout, :follow_location,
+                  :max_redirects, :proxy, :disable_ssl_peer_verification,
+                  :ssl_cert, :ssl_cert_type, :ssl_key, :ssl_key_type,
+                  :ssl_key_password, :ssl_cacert, :ssl_capath, :verbose,
+                  :username, :password, :auth_method, :user_agent
+
+    # Initialize a new Request
+    #
+    # Options:
+    # * +url+ : Endpoint (URL) of the request
+    # * +options+   : A hash containing options among :
+    # ** +:method+  : :get (default) / :post / :put
+    # ** +:params+  : params as a Hash
+    # ** +:body+
+    # ** +:timeout+ : timeout (ms)
+    # ** +:connect_timeout+ : connect timeout (ms)
+    # ** +:headers+  : headers as Hash
+    # ** +:user_agent+ : user agent (string)
+    # ** +:cache_timeout+ : cache timeout (ms)
+    # ** +:follow_location
+    # ** +:max_redirects
+    # ** +:proxy
+    # ** +:disable_ssl_peer_verification
+    # ** +:ssl_cert
+    # ** +:ssl_cert_type
+    # ** +:ssl_key
+    # ** +:ssl_key_type
+    # ** +:ssl_key_password
+    # ** +:ssl_cacert
+    # ** +:ssl_capath
+    # ** +:verbose
+    # ** +:username
+    # ** +:password
+    # ** +:auth_method
+    #
     def initialize(url, options = {})
       @method           = options[:method] || :get
       @params           = options[:params]
       @body             = options[:body]
       @timeout          = options[:timeout]
+      @connect_timeout  = options[:connect_timeout]
       @headers          = options[:headers] || {}
       @user_agent       = options[:user_agent] || Typhoeus::USER_AGENT
       @cache_timeout    = options[:cache_timeout]
-      @url              = @params ? "#{url}?#{params_string}" : url
+      @follow_location  = options[:follow_location]
+      @max_redirects    = options[:max_redirects]
+      @proxy            = options[:proxy]
+      @disable_ssl_peer_verification = options[:disable_ssl_peer_verification]
+      @ssl_cert         = options[:ssl_cert]
+      @ssl_cert_type    = options[:ssl_cert_type]
+      @ssl_key          = options[:ssl_key]
+      @ssl_key_type     = options[:ssl_key_type]
+      @ssl_key_password = options[:ssl_key_password]
+      @ssl_cacert       = options[:ssl_cacert]
+      @ssl_capath       = options[:ssl_capath]
+      @verbose          = options[:verbose]
+      @username         = options[:username]
+      @password         = options[:password]
+      @auth_method      = options[:auth_method]
+
+      if @method == :post
+        @url = url
+      else
+        @url = @params ? "#{url}?#{params_string}" : url
+      end
+
+      @parsed_uri = URI.parse(@url)
+
       @on_complete      = nil
       @after_complete   = nil
       @handled_response = nil
     end
-    
+
+    def localhost?
+      %(localhost 127.0.0.1 0.0.0.0).include?(@parsed_uri.host)
+    end
+
     def host
       slash_location = @url.index('/', 8)
       if slash_location
@@ -26,86 +91,108 @@ module Typhoeus
         return query_string_location ? @url.slice(0, query_string_location) : @url
       end
     end
-    
+
     def headers
       @headers["User-Agent"] = @user_agent
       @headers
     end
-    
+
     def params_string
-      params.keys.sort.collect do |k|
+      params.keys.sort { |a, b| a.to_s <=> b.to_s }.collect do |k|
         value = params[k]
         if value.is_a? Hash
-          value.keys.collect {|sk| Rack::Utils.escape("#{k}[#{sk}]") + "=" + Rack::Utils.escape(value[sk].to_s)}
+          value.keys.collect {|sk| Typhoeus::Utils.escape("#{k}[#{sk}]") + "=" + Typhoeus::Utils.escape(value[sk].to_s)}
         elsif value.is_a? Array
-          key = Rack::Utils.escape(k.to_s)
-          value.collect { |v| "#{key}=#{Rack::Utils.escape(v.to_s)}" }.join('&')
+          key = Typhoeus::Utils.escape(k.to_s)
+          value.collect { |v| "#{key}[]=#{Typhoeus::Utils.escape(v.to_s)}" }.join('&')
         else
-          "#{Rack::Utils.escape(k.to_s)}=#{Rack::Utils.escape(params[k].to_s)}"
+          "#{Typhoeus::Utils.escape(k.to_s)}=#{Typhoeus::Utils.escape(params[k].to_s)}"
         end
       end.flatten.join("&")
     end
-    
+
     def on_complete(&block)
       @on_complete = block
     end
-    
+
     def on_complete=(proc)
       @on_complete = proc
     end
-    
+
     def after_complete(&block)
       @after_complete = block
     end
-    
+
     def after_complete=(proc)
       @after_complete = proc
     end
-    
+
     def call_handlers
       if @on_complete
         @handled_response = @on_complete.call(response)
         call_after_complete
       end
     end
-    
+
     def call_after_complete
-       @after_complete.call(@handled_response) if @after_complete
+      @after_complete.call(@handled_response) if @after_complete
     end
-    
+
     def handled_response=(val)
       @handled_response = val
     end
-    
+
     def handled_response
       @handled_response || response
     end
-    
+
+    def inspect
+      result = ":method => #{self.method.inspect},\n" <<
+               "\t:url => #{URI.parse(self.url).to_s}"
+      if self.body and !self.body.empty?
+        result << ",\n\t:body => #{self.body.inspect}"
+      end
+
+      if self.params and !self.params.empty?
+        result << ",\n\t:params => #{self.params.inspect}"
+      end
+
+      if self.headers and !self.headers.empty?
+        result << ",\n\t:headers => #{self.headers.inspect}"
+      end
+
+      result
+    end
+
     def cache_key
       Digest::SHA1.hexdigest(url)
     end
-    
+
     def self.run(url, params)
       r = new(url, params)
       Typhoeus::Hydra.hydra.queue r
       Typhoeus::Hydra.hydra.run
       r.response
     end
-    
-    def self.get(url, params)
+
+    def self.get(url, params = {})
       run(url, params.merge(:method => :get))
     end
-    
-    def self.post(url, params)
+
+    def self.post(url, params = {})
       run(url, params.merge(:method => :post))
     end
-    
-    def self.put(url, params)
+
+    def self.put(url, params = {})
       run(url, params.merge(:method => :put))
     end
-    
-    def self.delete(url, params)
+
+    def self.delete(url, params = {})
       run(url, params.merge(:method => :delete))
     end
+
+    def self.head(url, params = {})
+      run(url, params.merge(:method => :head))
+    end
   end
-end
+end

data/lib/typhoeus/response.rb

@@ -1,19 +1,76 @@
 module Typhoeus
   class Response
-    attr_accessor :request
+    attr_accessor :request, :mock
     attr_reader :code, :headers, :body, :time,
                 :requested_url, :requested_remote_method,
-                :requested_http_method, :start_time
-    
+                :requested_http_method, :start_time,
+                :effective_url
+    attr_writer :headers_hash
+
     def initialize(params = {})
       @code                  = params[:code]
-      @headers               = params[:headers]
+      @status_message        = params[:status_message]
+      @http_version          = params[:http_version]
+      @headers               = params[:headers] || ''
       @body                  = params[:body]
       @time                  = params[:time]
       @requested_url         = params[:requested_url]
       @requested_http_method = params[:requested_http_method]
       @start_time            = params[:start_time]
       @request               = params[:request]
+      @effective_url         = params[:effective_url]
+      @mock                  = params[:mock] || false  # default
+      @headers_hash          = NormalizedHeaderHash.new(params[:headers_hash]) if params[:headers_hash]
     end
+
+    # Returns true if this is a mock response.
+    def mock?
+      @mock
+    end
+
+    def headers_hash
+      @headers_hash ||= begin
+        headers.split("\n").map {|o| o.strip}.inject(Typhoeus::NormalizedHeaderHash.new) do |hash, o|
+          if o.empty? || o =~ /^HTTP\/[\d\.]+/
+            hash
+          else
+            i = o.index(":") || o.size
+            key = o.slice(0, i)
+            value = o.slice(i + 1, o.size)
+            value = value.strip unless value.nil?
+            if hash.has_key? key
+              hash[key] = [hash[key], value].flatten
+            else
+              hash[key] = value
+            end
+
+            hash
+          end
+        end
+      end
+    end
+
+    def status_message
+      # http://rubular.com/r/eAr1oVYsVa
+      @status_message ||= first_header_line ? first_header_line[/\d{3} (.*)$/, 1].chomp : nil
+    end
+
+    def http_version
+      @http_version ||= first_header_line ? first_header_line[/HTTP\/(\S+)/, 1] : nil
+    end
+
+    def success?
+      @code >= 200 && @code < 300
+    end
+
+    def modified?
+      @code != 304
+    end
+
+    private
+
+      def first_header_line
+        @first_header_line ||= headers.split("\n").first
+      end
   end
 end

data/lib/typhoeus/service.rb

@@ -0,0 +1,20 @@
+module Typhoeus
+  class Service
+    def initialize(host, port)
+      @host = host
+      @port = port
+    end
+    
+    def get(resource, params)
+    end
+    
+    def put(resource, params)
+    end
+    
+    def post(resource, params)
+    end
+    
+    def delete(resource, params)
+    end
+  end
+end

data/lib/typhoeus/utils.rb

@@ -0,0 +1,24 @@
+module Typhoeus
+  module Utils
+    # Taken from Rack::Utils, 1.2.1 to remove Rack dependency.
+    def escape(s)
+      s.to_s.gsub(/([^ a-zA-Z0-9_.-]+)/u) {
+        '%'+$1.unpack('H2'*bytesize($1)).join('%').upcase
+      }.tr(' ', '+')
+    end
+    module_function :escape
+
+    # Return the bytesize of String; uses String#size under Ruby 1.8 and
+    # String#bytesize under 1.9.
+    if ''.respond_to?(:bytesize)
+      def bytesize(string)
+        string.bytesize
+      end
+    else
+      def bytesize(string)
+        string.size
+      end
+    end
+    module_function :bytesize
+  end
+end

data/lib/typhoeus.rb

@@ -1,7 +1,9 @@
 $LOAD_PATH.unshift(File.dirname(__FILE__)) unless $LOAD_PATH.include?(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.dirname(__FILE__) + "/../ext")
 
-require 'rack/utils'
 require 'digest/sha2'
+require 'typhoeus/utils'
+require 'typhoeus/normalized_header_hash'
 require 'typhoeus/easy'
 require 'typhoeus/multi'
 require 'typhoeus/native'
@@ -12,25 +14,26 @@ require 'typhoeus/remote_proxy_object'
 require 'typhoeus/response'
 require 'typhoeus/request'
 require 'typhoeus/hydra'
+require 'typhoeus/hydra_mock'
 
 module Typhoeus
-  VERSION = "0.1.6"
-  
+  VERSION = File.read(File.dirname(__FILE__) + "/../VERSION").chomp
+
   def self.easy_object_pool
     @easy_objects ||= []
   end
-  
+
   def self.init_easy_object_pool
     20.times do
       easy_object_pool << Typhoeus::Easy.new
     end
   end
-  
+
   def self.release_easy_object(easy)
     easy.reset
     easy_object_pool << easy
   end
-  
+
   def self.get_easy_object
     if easy_object_pool.empty?
       Typhoeus::Easy.new
@@ -38,12 +41,12 @@ module Typhoeus
       easy_object_pool.pop
     end
   end
-  
+
   def self.add_easy_request(easy_object)
     Thread.current[:curl_multi] ||= Typhoeus::Multi.new
     Thread.current[:curl_multi].add(easy_object)
   end
-  
+
   def self.perform_easy_requests
     multi = Thread.current[:curl_multi]
     start_time = Time.now

data/profilers/valgrind.rb

@@ -0,0 +1,24 @@
+#!/usr/bin/env ruby
+# go to ext/typhoeus and run ruby extconf.rb && make before running
+# this.
+
+$LOAD_PATH.unshift(File.dirname(__FILE__) + "/../ext")
+require File.dirname(__FILE__) + "/../lib/typhoeus"
+
+klass = Class.new { include Typhoeus }
+
+loops = ENV['LOOPS'].to_i
+url = ARGV.first || (raise "requires URL!")
+
+loops.times do |i|
+  puts "On loop #{i}" if i % 10 == 0
+  results = []
+  5.times do
+    results << klass.get(url)
+  end
+ 
+  # fire requests
+  results[0].code
+end
+
+puts "Ran #{loops} loops on #{url}!"

data/spec/fixtures/result_set.xml

@@ -0,0 +1,60 @@
+<result_set>
+  <ttl>20</ttl>
+  <result>
+    <id>1</id>
+    <name>hello</name>
+    <description>
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+    </description>
+  </result>
+  <result>
+    <id>2</id>
+    <name>hello</name>
+    <description>
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+    </description>
+  </result>
+  <result>
+    <id>3</id>
+    <name>hello</name>
+    <description>
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+      this is a long description for a text field of some kind.
+    </description>
+  </result>
+</result_set>

data/spec/servers/app.rb

@@ -1,6 +1,8 @@
+#!/usr/bin/env ruby
 require 'rubygems'
 require 'sinatra'
 require 'json'
+require 'zlib'
 
 @@fail_count = 0
 get '/fail/:number' do
@@ -16,11 +18,56 @@ get '/fail_forever' do
   error 500, "oh noes!"
 end
 
+get '/redirect' do
+  redirect '/'
+end
+
+get '/bad_redirect' do
+  redirect '/bad_redirect'
+end
+
+get '/auth_basic/:username/:password' do
+  @auth ||=  Rack::Auth::Basic::Request.new(request.env)
+  # Check that we've got a basic auth, and that it's credentials match the ones
+  # provided in the request
+  if @auth.provided? && @auth.basic? && @auth.credentials == [ params[:username], params[:password] ]
+    # auth is valid - confirm it
+    true
+  else
+    # invalid auth - request the authentication
+    response['WWW-Authenticate'] = %(Basic realm="Testing HTTP Auth")
+    throw(:halt, [401, "Not authorized\n"])
+  end
+end
+
+get '/auth_ntlm' do
+  # we're just checking for the existence if NTLM auth header here. It's validation
+  # is too troublesome and really doesn't bother is much, it's up to libcurl to make
+  # it valid
+  response['WWW-Authenticate'] = 'NTLM'
+  is_ntlm_auth = /^NTLM/ =~ request.env['HTTP_AUTHORIZATION']
+  true if is_ntlm_auth
+  throw(:halt, [401, "Not authorized\n"]) if !is_ntlm_auth
+end
+
+get '/gzipped' do
+  req_env = request.env.to_json
+  z = Zlib::Deflate.new
+  gzipped_env = z.deflate(req_env, Zlib::FINISH)
+  z.close
+  response['Content-Encoding'] = 'gzip'
+  gzipped_env
+end
+
 get '/**' do
   sleep params["delay"].to_i if params.has_key?("delay")
   request.env.merge!(:body => request.body.read).to_json
 end
 
+head '/**' do
+  sleep params["delay"].to_i if params.has_key?("delay")
+end
+
 put '/**' do
   puts request.inspect
   request.env.merge!(:body => request.body.read).to_json
@@ -34,4 +81,4 @@ end
 delete '/**' do
   puts request.inspect
   request.env.merge!(:body => request.body.read).to_json
-end
+end

data/spec/spec_helper.rb

@@ -8,4 +8,4 @@ begin require "redgreen" unless ENV['TM_CURRENT_LINE']; rescue LoadError; end
 path = File.expand_path(File.dirname(__FILE__) + "/../lib/")
 $LOAD_PATH.unshift(path) unless $LOAD_PATH.include?(path)
 
-require "lib/typhoeus"
+require path + '/typhoeus'