8125d8e796
* Add a `Danbooru.config.http_headers` config option. * Refactor various places to use this option instead of setting the user agent manually.
27 lines
615 B
Ruby
27 lines
615 B
Ruby
class ImageProxy
|
|
def self.needs_proxy?(url)
|
|
fake_referer_for(url).present?
|
|
end
|
|
|
|
def self.fake_referer_for(url)
|
|
Sources::Site.new(url).strategy.try(:fake_referer)
|
|
end
|
|
|
|
def self.get_image(url)
|
|
if url.blank?
|
|
raise "Must specify url"
|
|
end
|
|
|
|
if !needs_proxy?(url)
|
|
raise "Proxy not allowed for this site"
|
|
end
|
|
|
|
response = HTTParty.get(url, Danbooru.config.httparty_options.deep_merge(headers: {"Referer" => fake_referer_for(url)}))
|
|
if response.success?
|
|
return response
|
|
else
|
|
raise "HTTP error code: #{response.code} #{response.message}"
|
|
end
|
|
end
|
|
end
|