Contents

require 'net/http/persistent'

module LinkThumbnailer

  class Fetcher

    attr_accessor :url

    def fetch(url, redirect_count = 0)
      if redirect_count > LinkThumbnailer.configuration.redirect_limit
        raise ArgumentError, "too many redirects (#{redirect_count})"
      end

      self.url = url.is_a?(URI) ? url : URI(url)

      if self.url.is_a?(URI::HTTP)
        http                        = Net::HTTP::Persistent.new('linkthumbnailer')
        http.headers['User-Agent']  = LinkThumbnailer.configuration.user_agent
        http.verify_mode            = OpenSSL::SSL::VERIFY_NONE unless LinkThumbnailer.configuration.verify_ssl
        http.open_timeout           = LinkThumbnailer.configuration.http_timeout
        resp                        = http.request(self.url)
        case resp
        when Net::HTTPSuccess     then resp.body
        when Net::HTTPRedirection then fetch(resp['location'], redirect_count + 1)
        else resp.error!
        end
      end
    end

  end

end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
link_thumbnailer-1.0.8	lib/link_thumbnailer/fetcher.rb
link_thumbnailer-1.0.7	lib/link_thumbnailer/fetcher.rb
link_thumbnailer-1.0.6	lib/link_thumbnailer/fetcher.rb