7868e5045e
3cdf67920 changed it so that Danbooru::Http follows redirects by
default. This broke some things in the Nico Seiga strategy, so disable
following redirects in the Nico Seiga API client for now.
Also change it so that Danbooru::Http follows redirects after a POST
request (by setting `strict: false`). Nico Seiga needs this because it
sends a redirect after we POST the login form.
92 lines
2.3 KiB
Ruby
92 lines
2.3 KiB
Ruby
module Danbooru
|
|
class Http
|
|
DEFAULT_TIMEOUT = 3
|
|
MAX_REDIRECTS = 5
|
|
|
|
attr_writer :cache, :http
|
|
|
|
class << self
|
|
delegate :get, :put, :post, :delete, :cache, :follow, :timeout, :auth, :basic_auth, :headers, to: :new
|
|
end
|
|
|
|
def get(url, **options)
|
|
request(:get, url, **options)
|
|
end
|
|
|
|
def put(url, **options)
|
|
request(:get, url, **options)
|
|
end
|
|
|
|
def post(url, **options)
|
|
request(:post, url, **options)
|
|
end
|
|
|
|
def delete(url, **options)
|
|
request(:delete, url, **options)
|
|
end
|
|
|
|
def cache(expiry)
|
|
dup.tap { |o| o.cache = expiry.to_i }
|
|
end
|
|
|
|
def follow(*args)
|
|
dup.tap { |o| o.http = o.http.follow(*args) }
|
|
end
|
|
|
|
def timeout(*args)
|
|
dup.tap { |o| o.http = o.http.timeout(*args) }
|
|
end
|
|
|
|
def auth(*args)
|
|
dup.tap { |o| o.http = o.http.auth(*args) }
|
|
end
|
|
|
|
def basic_auth(*args)
|
|
dup.tap { |o| o.http = o.http.basic_auth(*args) }
|
|
end
|
|
|
|
def headers(*args)
|
|
dup.tap { |o| o.http = o.http.headers(*args) }
|
|
end
|
|
|
|
protected
|
|
|
|
def request(method, url, **options)
|
|
if @cache.present?
|
|
cached_request(method, url, **options)
|
|
else
|
|
raw_request(method, url, **options)
|
|
end
|
|
rescue HTTP::Redirector::TooManyRedirectsError
|
|
::HTTP::Response.new(status: 598, body: "", version: "1.1")
|
|
rescue HTTP::TimeoutError
|
|
# return a synthetic http error on connection timeouts
|
|
::HTTP::Response.new(status: 599, body: "", version: "1.1")
|
|
end
|
|
|
|
def cached_request(method, url, **options)
|
|
key = Cache.hash({ method: method, url: url, headers: http.default_options.headers.to_h, **options }.to_json)
|
|
|
|
cached_response = Cache.get(key, @cache) do
|
|
response = raw_request(method, url, **options)
|
|
{ status: response.status, body: response.to_s, headers: response.headers.to_h, version: "1.1" }
|
|
end
|
|
|
|
::HTTP::Response.new(**cached_response)
|
|
end
|
|
|
|
def raw_request(method, url, **options)
|
|
http.send(method, url, **options)
|
|
end
|
|
|
|
def http
|
|
@http ||= ::HTTP.
|
|
follow(strict: false, max_hops: MAX_REDIRECTS).
|
|
timeout(DEFAULT_TIMEOUT).
|
|
use(:auto_inflate).
|
|
headers(Danbooru.config.http_headers).
|
|
headers("Accept-Encoding" => "gzip")
|
|
end
|
|
end
|
|
end
|