lib/patron/request.rb in patron-0.4.0 vs lib/patron/request.rb in patron-0.4.1

@@ -38,13 +38,49 @@
       @timeout = 0
       @connect_timeout = 0
       @max_redirects = -1
     end
 
-    attr_accessor :url, :username, :password, :upload_data, :file_name, :proxy
+    attr_accessor :url, :username, :password, :file_name, :proxy, :auth_type
     attr_reader :action, :timeout, :connect_timeout, :max_redirects, :headers
+    attr_reader :auth_type
+    
+      # Set the type of authentication to use for this request. 
+      # 
+      # @param [String, Symbol] type - The type of authentication to use for this request, can be one of
+      #   :basic, :digest, or :any
+      #
+      # @example
+      #   sess.username = "foo"
+      #   sess.password = "sekrit"
+      #   sess.auth_type = :digest
+      def auth_type=(type=:basic)
+        @auth_type = case type
+        when :basic, "basic"
+          Request::AuthBasic
+        when :digest, "digest"
+          Request::AuthDigest
+        when :any, "any"
+          Request::AuthAny
+        else
+          raise "#{type.inspect} is an unknown authentication type"
+        end
+      end
 
+    def upload_data=(data)
+      @upload_data = case data
+      when Hash:
+        hash_to_string(data)
+      else
+        data
+      end
+    end
+    
+    def upload_data
+      @upload_data
+    end
+
     def action=(new_action)
       if !VALID_ACTIONS.include?(new_action)
         raise ArgumentError, "Action must be one of #{VALID_ACTIONS.join(', ')}"
       end
 
@@ -90,7 +126,22 @@
     def credentials
       return nil if username.nil? || password.nil?
       "#{username}:#{password}"
     end
 
+    private
+    
+    # serialize hash for Rails-style params
+    def hash_to_string(hash)
+      pairs = []
+      recursive = Proc.new do |h, prefix| 
+        h.each_pair do |k,v|
+          key = prefix == '' ? k : "#{prefix}[#{k}]"
+          v.is_a?(Hash) ? recursive.call(v, key) : pairs << "#{key}=#{v}"
+        end
+      end
+      recursive.call(hash, '')
+      return pairs.join('&')
+    end
+    
   end
 end