url_fetcher 1.1.5 → 1.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 32494b5b29e118d6dffacb4b123e90aa1da50795
-  data.tar.gz: 19981639b3c1206540c1f584d3e34537e64d5170
+  metadata.gz: 8523decc234223ef23d0ad21e091b8e02b10697d
+  data.tar.gz: 54d317200ebbd7921eef0def888b0932a23a2f2f
 SHA512:
-  metadata.gz: 15b7108d62ece1765797169e2aa776c7fbbe0656b9ef693a1579dc7b8979e29d3b4375e0d56b9b1407d4a3fa9cd55cc66d138a173a192be7e7f8501aa04bb02a
-  data.tar.gz: 28a1b5adc3962cd66b1b5e4b39f6c96513f4ad7d01fc814777a1567023daab9ee8fdd16365041e7bb357a2e128a41347132e2e1fed7b33198a72af43becedd1b
+  metadata.gz: 82a6284bcd4a7096f8828b950431adb7940d2c9c2ad1b8e932ce16bf912ba4dac5fde62065c93ba4045028b97f8b8415f6fccb5de8fdebd0a6ebec56c26e5674
+  data.tar.gz: ad131782b83ac319e3819139c278f8f705c4c9a8f36705a1418aa494d54669521a2c64bde94dab23fa2990750504d3b4844d4e6714e55d2dbef494ad5c804e44

data/VERSION

@@ -1 +1 @@
-1.1.5
+1.1.6

data/lib/url_fetcher.rb

@@ -1,8 +1,8 @@
-require "url_fetcher/errors"
 require "net/http"
 require "openssl"
-require "open-uri"
+require "addressable/uri"
 require "tempfile"
+require "url_fetcher/errors"
 
 # This class will fetch the contents of a URL and store them in a Tempfile. The
 # results are exposed as a stream so you don't need to read potentialy huge responses
@@ -22,6 +22,7 @@ class UrlFetcher
   # * :max_size (10 megabytes)- The maximum size in bytes that should be fetched.
   # * :open_timeout (10) - Time in seconds to wait for a connection to be established.
   # * :read_timeout (20) - Time in seconds to wait for reading the HTTP response.
+  # * :headers ("Accept" => "*")- Hash of headers to send with request.
   def initialize(url, options = {}, &redirect_hook)
     @url = url
     @redirect_hook = redirect_hook
@@ -94,6 +95,17 @@ class UrlFetcher
       Net::HTTP::Get.new(uri.request_uri)
     end
 
+    if options[:headers]
+      options[:headers].each do |name, value|
+        name = name.to_s
+        values = Array(value)
+        request[name] = values[0].to_s
+        values[1, values.length].each do |val|          
+          request.add_field(name, val.to_s)
+        end
+      end
+    end
+    
     response = http.request(request) do |resp|
       unless resp.is_a?(Net::HTTPSuccess) || resp.is_a?(Net::HTTPRedirection)
         resp.value # Raises an appropriate HTTP error

data/spec/url_fetcher_spec.rb

@@ -43,6 +43,12 @@ describe UrlFetcher do
     expect(url_fetcher.header("content-length")).to eql("5")
     expect(url_fetcher.body.read).to eql("Hello")
   end
+  
+  it "should send custom headers" do
+    WebMock.stub_request(:get, "https://example.com/test").with(:headers => {"Accept" => ["image/jpeg", "image/gif"]}).to_return(:status => 200, :body => "Hello", :headers => {"Content-Length" => 5})
+    url_fetcher = UrlFetcher.new("https://example.com/test", :headers => {"Accept" => ["image/jpeg", "image/gif"]})
+    expect(url_fetcher).to be_success
+  end
 
   it "should honor redirects" do
     WebMock.stub_request(:get, "http://example.com/test1").to_return(:status => 301, :headers => {"Location" => "http://example.com/test2"})

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: url_fetcher
 version: !ruby/object:Gem::Version
-  version: 1.1.5
+  version: 1.1.6
 platform: ruby
 authors:
 - weheartit
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-11-18 00:00:00.000000000 Z
+date: 2016-12-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: addressable