# Intended to extend the Net::HTTP response object
# and adds support for decoding gzip and deflate encoded pages
#
# Author: Jason Stirk <http://griffin.oobleyboo.com>
# Home: http://griffin.oobleyboo.com/projects/http_encoding_helper
# Created: 5 September 2007
# Last Updated: 23 November 2007
#
# Usage:
#
# require 'net/http'
# require 'http_encoding_helper'
# headers={'Accept-Encoding' => 'gzip, deflate' }
# http = Net::HTTP.new('griffin.oobleyboo.com', 80)
# http.start do |h|
#   request = Net::HTTP::Get.new('/', headers)
#   response = http.request(request)
#   content=response.plain_body     # Method from our library
#   puts "Transferred: #{response.body.length} bytes"
#   puts "Compression: #{response['content-encoding']}"
#   puts "Extracted: #{response.plain_body.length} bytes"
# end
#

require 'zlib'
require 'stringio'

class Net::HTTPResponse
  # Return the uncompressed content
  def plain_body
    encoding = self['Content-Encoding']
    @_content ||= nil

    return @_content if @_content

    if encoding
      case encoding
      when 'gzip'
        i = Zlib::GzipReader.new(StringIO.new(body))
        @_content = i.read
      when 'deflate'
        i = Zlib::Inflate.new
        @_content = i.inflate(body)
      else
        raise "Unknown encoding - #{encoding}"
      end
    else
      @_content = body
    end

    @_content
  end
end
