Adding custom parsers. One failing spec right now.

lib/httparty.rb

@@ -110,6 +110,16 @@ module HTTParty
       default_options[:format] = f
     end
     
+    # Allows setting a custom parser for the response.
+    #
+    #   class Foo
+    #     include HTTParty
+    #     parser Proc.new {|data| ...}
+    #   end
+    def parser(customer_parser)
+      default_options[:parser] = customer_parser
+    end
+    
     # Allows making a get request to a url.
     #
     #   class Foo

lib/httparty/parsers.rb

@@ -0,0 +1,31 @@
+module HTTParty
+  module Xml
+    def self.parse(body)
+      Crack::XML.parse(body)
+    end
+  end
+  
+  module Json
+    def self.parse(body)
+      Crack::JSON.parse(body)
+    end
+  end
+  
+  module Yaml
+    def self.parse(str)
+      ::YAML.load(str)
+    end
+  end
+  
+  module Html
+    def self.parse(str)
+      str
+    end
+  end
+  
+  module Text
+    def self.parse(str)
+      str
+    end
+  end
+end

lib/httparty/request.rb

@@ -107,20 +107,29 @@ module HTTParty
           end
       end
       
+      # HTTParty.const_get((self.format.to_s || 'text').capitalize)
       def parse_response(body)
         return nil if body.nil? or body.empty?
-        case format
+        if options[:parser].blank?
-          when :xml
+          case format
-            Crack::XML.parse(body)
+            when :xml
-          when :json
+              Crack::XML.parse(body)
-            Crack::JSON.parse(body)
+            when :json
-          when :yaml
+              Crack::JSON.parse(body)
-            YAML::load(body)
+            when :yaml
+              YAML::load(body)
+            else
+              body
+            end
+        else
+          if options[:parser].is_a?(Proc)
+            options[:parser].call(body)
           else
             body
           end
+        end
       end
-      
+            
       def capture_cookies(response)
         return unless response['Set-Cookie']
         cookies_hash = HTTParty::CookieHash.new()

spec/httparty_spec.rb

@@ -1,5 +1,11 @@
 require File.expand_path(File.join(File.dirname(__FILE__), 'spec_helper'))
 
+class CustomParser
+  def self.parse(body)
+    return {:sexy => true}
+  end
+end
+
 describe HTTParty do
   before(:each) do
     @klass = Class.new
@@ -174,6 +180,22 @@ describe HTTParty do
     end
   end
   
+  describe "parser" do
+    before(:each) do
+      @klass.parser Proc.new{ |data| CustomParser.parse(data) }
+    end
+    
+    it "should set parser options" do
+      @klass.default_options[:parser].class.should == Proc
+    end
+
+    it "should be able parse response with custom parser" do
+      stub_http_response_with 'twitter.xml'
+      custom_parsed_response = @klass.get('http://twitter.com/statuses/public_timeline.xml')
+      custom_parsed_response[:sexy].should == true
+    end
+  end
+  
   describe "format" do
     it "should allow xml" do
       @klass.format :xml