geetarista-httparty 0.4.5

data/lib/httparty.rb

@@ -0,0 +1,206 @@
+$:.unshift(File.dirname(__FILE__))
+
+require 'net/http'
+require 'net/https'
+require 'httparty/module_inheritable_attributes'
+require 'rubygems'
+gem 'crack'
+require 'crack'
+
+require 'httparty/cookie_hash'
+
+module HTTParty
+  
+  AllowedFormats = {
+    'text/xml'               => :xml,
+    'application/xml'        => :xml,
+    'application/json'       => :json,
+    'text/json'              => :json,
+    'application/javascript' => :json,
+    'text/javascript'        => :json,
+    'text/html'              => :html,
+    'application/x-yaml'     => :yaml,
+    'text/yaml'              => :yaml,
+    'text/plain'             => :plain
+  } unless defined?(AllowedFormats)
+  
+  def self.included(base)
+    base.extend ClassMethods
+    base.send :include, HTTParty::ModuleInheritableAttributes
+    base.send(:mattr_inheritable, :default_options)
+    base.send(:mattr_inheritable, :default_cookies)
+    base.instance_variable_set("@default_options", {})
+    base.instance_variable_set("@default_cookies", CookieHash.new)
+  end
+  
+  module ClassMethods
+    # Allows setting http proxy information to be used
+    #
+    #   class Foo
+    #     include HTTParty
+    #     http_proxy 'http://foo.com', 80
+    #   end
+    def http_proxy(addr=nil, port = nil)
+      default_options[:http_proxyaddr] = addr
+      default_options[:http_proxyport] = port
+    end
+    
+    # Allows setting a base uri to be used for each request.
+    # Will normalize uri to include http, etc.
+    #
+    #   class Foo
+    #     include HTTParty
+    #     base_uri 'twitter.com'
+    #   end
+    def base_uri(uri=nil)
+      return default_options[:base_uri] unless uri
+      default_options[:base_uri] = HTTParty.normalize_base_uri(uri)
+    end
+    
+    # Allows setting basic authentication username and password.
+    #
+    #   class Foo
+    #     include HTTParty
+    #     basic_auth 'username', 'password'
+    #   end
+    def basic_auth(u, p)
+      default_options[:basic_auth] = {:username => u, :password => p}
+    end
+    
+    # Allows setting default parameters to be appended to each request.
+    # Great for api keys and such.
+    #
+    #   class Foo
+    #     include HTTParty
+    #     default_params :api_key => 'secret', :another => 'foo'
+    #   end
+    def default_params(h={})
+      raise ArgumentError, 'Default params must be a hash' unless h.is_a?(Hash)
+      default_options[:default_params] ||= {}
+      default_options[:default_params].merge!(h)
+    end
+    
+    # Allows setting a base uri to be used for each request.
+    #
+    #   class Foo
+    #     include HTTParty
+    #     headers 'Accept' => 'text/html'
+    #   end
+    def headers(h={})
+      raise ArgumentError, 'Headers must be a hash' unless h.is_a?(Hash)
+      default_options[:headers] ||= {}
+      default_options[:headers].merge!(h)
+    end
+
+    def cookies(h={})
+      raise ArgumentError, 'Cookies must be a hash' unless h.is_a?(Hash)
+      default_cookies.add_cookies(h)
+    end
+    
+    # Allows setting the format with which to parse.
+    # Must be one of the allowed formats ie: json, xml
+    #
+    #   class Foo
+    #     include HTTParty
+    #     format :json
+    #   end
+    def format(f)
+      raise UnsupportedFormat, "Must be one of: #{AllowedFormats.values.map { |v| v.to_s }.uniq.sort.join(', ')}" unless AllowedFormats.value?(f)
+      default_options[:format] = f
+    end
+    
+    # Allows making a get request to a url.
+    #
+    #   class Foo
+    #     include HTTParty
+    #   end
+    #   
+    #   # Simple get with full url
+    #   Foo.get('http://foo.com/resource.json')
+    #   
+    #   # Simple get with full url and query parameters
+    #   # ie: http://foo.com/resource.json?limit=10
+    #   Foo.get('http://foo.com/resource.json', :query => {:limit => 10})
+    def get(path, options={})
+      perform_request Net::HTTP::Get, path, options
+    end
+    
+    # Allows making a post request to a url.
+    #
+    #   class Foo
+    #     include HTTParty
+    #   end
+    #   
+    #   # Simple post with full url and setting the body
+    #   Foo.post('http://foo.com/resources', :body => {:bar => 'baz'})
+    #
+    #   # Simple post with full url using :query option, 
+    #   # which gets set as form data on the request.
+    #   Foo.post('http://foo.com/resources', :query => {:bar => 'baz'})
+    def post(path, options={})
+      perform_request Net::HTTP::Post, path, options
+    end
+
+    def put(path, options={})
+      perform_request Net::HTTP::Put, path, options
+    end
+
+    def delete(path, options={})
+      perform_request Net::HTTP::Delete, path, options
+    end
+    
+    def default_options #:nodoc:
+      @default_options
+    end
+
+    private
+      def perform_request(http_method, path, options) #:nodoc:
+        process_cookies(options)
+        Request.new(http_method, path, default_options.dup.merge(options)).perform
+      end
+
+      def process_cookies(options) #:nodoc:
+        return unless options[:cookies] || default_cookies
+        options[:headers] ||= {}
+        options[:headers]["cookie"] = cookies.merge(options[:cookies] || {}).to_cookie_string
+
+        options.delete(:cookies)
+      end
+  end
+
+  def self.normalize_base_uri(url) #:nodoc:
+    normalized_url = url.dup
+    use_ssl = (normalized_url =~ /^https/) || normalized_url.include?(':443')
+    ends_with_slash = normalized_url =~ /\/$/
+    
+    normalized_url.chop! if ends_with_slash
+    normalized_url.gsub!(/^https?:\/\//i, '')
+    
+    "http#{'s' if use_ssl}://#{normalized_url}"
+  end
+  
+  class Basement #:nodoc:
+    include HTTParty
+  end
+  
+  def self.get(*args)
+    Basement.get(*args)
+  end
+  
+  def self.post(*args)
+    Basement.post(*args)
+  end
+
+  def self.put(*args)
+    Basement.put(*args)
+  end
+
+  def self.delete(*args)
+    Basement.delete(*args)
+  end
+end
+
+require 'httparty/core_extensions'
+require 'httparty/exceptions'
+require 'httparty/request'
+require 'httparty/response'

data/lib/httparty/cookie_hash.rb

@@ -0,0 +1,9 @@
+class HTTParty::CookieHash < Hash #:nodoc:
+  def add_cookies(hash)
+    merge!(hash)
+  end
+
+  def to_cookie_string
+    collect { |k, v| "#{k}=#{v}" }.join("; ")
+  end
+end

data/lib/httparty/core_extensions.rb

@@ -0,0 +1,29 @@
+if RUBY_VERSION.to_f == 1.8
+  class BlankSlate #:nodoc:
+    instance_methods.each { |m| undef_method m unless m =~ /^__|instance_eval|object_id/ }
+  end
+else
+  class BlankSlate < BasicObject; end
+end
+ 
+# 1.8.6 has mistyping of transitive in if statement
+require "rexml/document"
+module REXML #:nodoc:
+  class Document < Element #:nodoc:
+    def write( output=$stdout, indent=-1, transitive=false, ie_hack=false )
+      if xml_decl.encoding != "UTF-8" && !output.kind_of?(Output)
+        output = Output.new( output, xml_decl.encoding )
+      end
+      formatter = if indent > -1
+          if transitive
+            REXML::Formatters::Transitive.new( indent, ie_hack )
+          else
+            REXML::Formatters::Pretty.new( indent, ie_hack )
+          end
+        else
+          REXML::Formatters::Default.new( ie_hack )
+        end
+      formatter.write( self, output )
+    end
+  end
+end

data/lib/httparty/exceptions.rb

@@ -0,0 +1,7 @@
+module HTTParty
+  # Exception raised when you attempt to set a non-existant format
+  class UnsupportedFormat < StandardError; end
+  
+  # Exception that is raised when request has redirected too many times
+  class RedirectionTooDeep < StandardError; end
+end

data/lib/httparty/module_inheritable_attributes.rb

@@ -0,0 +1,25 @@
+module HTTParty
+  module ModuleInheritableAttributes #:nodoc:
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods #:nodoc:
+      def mattr_inheritable(*args)
+        @mattr_inheritable_attrs ||= [:mattr_inheritable_attrs]
+        @mattr_inheritable_attrs += args
+        args.each do |arg|
+          module_eval %(class << self; attr_accessor :#{arg} end)
+        end
+        @mattr_inheritable_attrs
+      end
+
+      def inherited(subclass)
+        @mattr_inheritable_attrs.each do |inheritable_attribute|
+          instance_var = "@#{inheritable_attribute}" 
+          subclass.instance_variable_set(instance_var, instance_variable_get(instance_var))
+        end
+      end
+    end
+  end
+end

data/lib/httparty/request.rb

@@ -0,0 +1,141 @@
+require 'uri'
+
+module HTTParty
+  class Request #:nodoc:
+    SupportedHTTPMethods = [Net::HTTP::Get, Net::HTTP::Post, Net::HTTP::Put, Net::HTTP::Delete]
+    
+    attr_accessor :http_method, :path, :options
+    
+    def initialize(http_method, path, o={})
+      self.http_method = http_method
+      self.path = path
+      self.options = {
+        :limit => o.delete(:no_follow) ? 0 : 5, 
+        :default_params => {},
+      }.merge(o)
+    end
+
+    def path=(uri)
+      @path = URI.parse(uri)
+    end
+    
+    def uri
+      new_uri = path.relative? ? URI.parse("#{options[:base_uri]}#{path}") : path
+      
+      # avoid double query string on redirects [#12]
+      unless @redirect
+        new_uri.query = query_string(new_uri)
+      end
+      
+      new_uri
+    end
+    
+    def format
+      options[:format]
+    end
+    
+    def perform
+      validate
+      setup_raw_request
+      handle_response(get_response)
+    end
+
+    private
+      def http
+        http = Net::HTTP.new(uri.host, uri.port, options[:http_proxyaddr], options[:http_proxyport])
+        http.use_ssl = (uri.port == 443)
+        http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+        http
+      end
+
+      def body
+        options[:body].is_a?(Hash) ? options[:body].to_params : options[:body]
+      end
+      
+      def username
+        options[:basic_auth][:username]
+      end
+      
+      def password
+        options[:basic_auth][:password]
+      end
+
+      def setup_raw_request
+        @raw_request = http_method.new(uri.request_uri)
+        @raw_request.body = body if body
+        @raw_request.initialize_http_header(options[:headers])
+        @raw_request.basic_auth(username, password) if options[:basic_auth]
+      end
+
+      def perform_actual_request
+        http.request(@raw_request)
+      end
+
+      def get_response
+        response = perform_actual_request
+        options[:format] ||= format_from_mimetype(response['content-type'])
+        response
+      end
+      
+      def query_string(uri)
+        query_string_parts = []
+        query_string_parts << uri.query unless uri.query.nil?
+
+        if options[:query].is_a?(Hash)
+          query_string_parts << options[:default_params].merge(options[:query]).to_params
+        else
+          query_string_parts << options[:default_params].to_params unless options[:default_params].nil?
+          query_string_parts << options[:query] unless options[:query].nil?
+        end
+        
+        query_string_parts.size > 0 ? query_string_parts.join('&') : nil
+      end
+      
+      # Raises exception Net::XXX (http error code) if an http error occured
+      def handle_response(response)
+        case response
+          when Net::HTTPRedirection
+            options[:limit] -= 1
+            self.path = response['location']
+            @redirect = true
+            perform
+          else
+            parsed_response = parse_response(response.body)
+            Response.new(parsed_response, response.body, response.code, response.message, response.to_hash)
+          end
+      end
+      
+      def parse_response(body)
+        return nil if body.nil? or body.empty?
+        case format
+          when :xml
+            Crack::XML.parse(body)
+          when :json
+            Crack::JSON.parse(body)
+          when :yaml
+            YAML::load(body)
+          else
+            body
+          end
+      end
+  
+      # Uses the HTTP Content-Type header to determine the format of the response
+      # It compares the MIME type returned to the types stored in the AllowedFormats hash
+      def format_from_mimetype(mimetype)
+        return nil if mimetype.nil?
+        AllowedFormats.each { |k, v| return v if mimetype.include?(k) }
+      end
+      
+      def validate
+        raise HTTParty::RedirectionTooDeep, 'HTTP redirects too deep' if options[:limit].to_i <= 0
+        raise ArgumentError, 'only get, post, put and delete methods are supported' unless SupportedHTTPMethods.include?(http_method)
+        raise ArgumentError, ':headers must be a hash' if options[:headers] && !options[:headers].is_a?(Hash)
+        raise ArgumentError, ':basic_auth must be a hash' if options[:basic_auth] && !options[:basic_auth].is_a?(Hash)
+        raise ArgumentError, ':query must be hash if using HTTP Post' if post? && !options[:query].nil? && !options[:query].is_a?(Hash)
+      end
+      
+      def post?
+        Net::HTTP::Post == http_method
+      end
+  end
+end

data/lib/httparty/response.rb

@@ -0,0 +1,18 @@
+module HTTParty
+  class Response < BlankSlate #:nodoc:
+    attr_accessor :body, :code, :message, :headers
+    attr_reader :delegate
+
+    def initialize(delegate, body, code, message, headers={})
+      @delegate = delegate
+      @body = body
+      @code = code.to_i
+      @message = message
+      @headers = headers
+    end
+
+    def method_missing(name, *args, &block)
+      @delegate.send(name, *args, &block)
+    end
+  end
+end

data/lib/httparty/version.rb

@@ -0,0 +1,3 @@
+module HTTParty #:nodoc:
+  Version = '0.4.4'
+end