Allow storyboard queries with QUIC disabled

This commit is contained in:
syeopite 2021-10-08 06:03:33 -07:00
parent c3747c2d49
commit 6b8450558d
No known key found for this signature in database
GPG key ID: 6FA616E5A5294A82

View file

@ -78,7 +78,9 @@ module Invidious::Routes::Images
headers = HTTP::Headers.new headers = HTTP::Headers.new
headers[":authority"] = "#{authority}.ytimg.com" {% unless flag?(:disable_quic) %}
headers[":authority"] = "#{authority}.ytimg.com"
{% end %}
REQUEST_HEADERS_WHITELIST.each do |header| REQUEST_HEADERS_WHITELIST.each do |header|
if env.request.headers[header]? if env.request.headers[header]?
@ -86,25 +88,41 @@ module Invidious::Routes::Images
end end
end end
request_proc = ->(response : HTTP::Client::Response) {
env.response.status_code = response.status_code
response.headers.each do |key, value|
if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase)
env.response.headers[key] = value
end
end
env.response.headers["Connection"] = "close"
env.response.headers["Access-Control-Allow-Origin"] = "*"
if response.status_code >= 300
return env.response.headers.delete("Transfer-Encoding")
end
proxy_file(response, env)
}
begin begin
YT_POOL.client &.get(url, headers) do |response| {% unless flag?(:disable_quic) %}
env.response.status_code = response.status_code if CONFIG.use_quic
response.headers.each do |key, value| YT_POOL.client &.get(url, headers) do |resp|
if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase) return request_proc.call(resp)
env.response.headers[key] = value end
else
HTTP::Client.get("#{authority}.ytimg.com#{url}") do |resp|
return request_proc.call(resp)
end end
end end
{% else %}
env.response.headers["Connection"] = "close" # This can likely be optimized into a (small) pool sometime in the future.
env.response.headers["Access-Control-Allow-Origin"] = "*" HTTP::Client.get("#{authority}.ytimg.com#{url}") do |resp|
return request_proc.call(resp)
if response.status_code >= 300
env.response.headers.delete("Transfer-Encoding")
break
end end
{% end %}
proxy_file(response, env)
end
rescue ex rescue ex
end end
end end
@ -113,34 +131,60 @@ module Invidious::Routes::Images
def self.s_p_image(env) def self.s_p_image(env)
id = env.params.url["id"] id = env.params.url["id"]
name = env.params.url["name"] name = env.params.url["name"]
url = env.request.resource url = env.request.resource
headers = HTTP::Headers{":authority" => "i9.ytimg.com"} headers = (
{% unless flag?(:disable_quic) %}
if CONFIG.use_quic
HTTP::Headers{":authority" => "i9.ytimg.com"}
else
HTTP::Headers.new
end
{% else %}
HTTP::Headers.new
{% end %}
)
REQUEST_HEADERS_WHITELIST.each do |header| REQUEST_HEADERS_WHITELIST.each do |header|
if env.request.headers[header]? if env.request.headers[header]?
headers[header] = env.request.headers[header] headers[header] = env.request.headers[header]
end end
end end
request_proc = ->(response : HTTP::Client::Response) {
env.response.status_code = response.status_code
response.headers.each do |key, value|
if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase)
env.response.headers[key] = value
end
end
env.response.headers["Access-Control-Allow-Origin"] = "*"
if response.status_code >= 300 && response.status_code != 404
return env.response.headers.delete("Transfer-Encoding")
end
proxy_file(response, env)
}
begin begin
YT_POOL.client &.get(url, headers) do |response| {% unless flag?(:disable_quic) %}
env.response.status_code = response.status_code if CONFIG.use_quic
response.headers.each do |key, value| YT_POOL.client &.get(url, headers) do |resp|
if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase) return request_proc.call(resp)
env.response.headers[key] = value end
else
HTTP::Client.get("i9.ytimg.com#{url}") do |resp|
return request_proc.call(resp)
end end
end end
{% else %}
env.response.headers["Access-Control-Allow-Origin"] = "*" # This can likely be optimized into a (small) pool sometime in the future.
HTTP::Client.get("i9.ytimg.com#{url}") do |resp|
if response.status_code >= 300 && response.status_code != 404 return request_proc.call(resp)
env.response.headers.delete("Transfer-Encoding")
break
end end
{% end %}
proxy_file(response, env)
end
rescue ex rescue ex
end end
end end