mirror of
https://gitea.invidious.io/iv-org/invidious-copy-2022-03-16.git
synced 2024-08-15 00:53:18 +00:00
parent
c6e086c6ff
commit
de00e86cd5
1 changed files with 44 additions and 18 deletions
|
@ -404,38 +404,33 @@ module YoutubeAPI
|
||||||
url = "#{endpoint}?key=#{client_config.api_key}"
|
url = "#{endpoint}?key=#{client_config.api_key}"
|
||||||
|
|
||||||
headers = HTTP::Headers{
|
headers = HTTP::Headers{
|
||||||
"Content-Type" => "application/json; charset=UTF-8",
|
"Content-Type" => "application/json; charset=UTF-8",
|
||||||
|
"Accept-Encoding" => "gzip, deflate",
|
||||||
}
|
}
|
||||||
|
|
||||||
# The normal HTTP client automatically applies accept-encoding: gzip,
|
|
||||||
# and decompresses. However, explicitly applying it will remove this functionality.
|
|
||||||
#
|
|
||||||
# https://github.com/crystal-lang/crystal/issues/11252#issuecomment-929594741
|
|
||||||
{% unless flag?(:disable_quic) %}
|
|
||||||
if CONFIG.use_quic
|
|
||||||
headers["Accept-Encoding"] = "gzip"
|
|
||||||
end
|
|
||||||
{% end %}
|
|
||||||
|
|
||||||
# Logging
|
# Logging
|
||||||
LOGGER.debug("YoutubeAPI: Using endpoint: \"#{endpoint}\"")
|
LOGGER.debug("YoutubeAPI: Using endpoint: \"#{endpoint}\"")
|
||||||
LOGGER.trace("YoutubeAPI: ClientConfig: #{client_config}")
|
LOGGER.trace("YoutubeAPI: ClientConfig: #{client_config}")
|
||||||
LOGGER.trace("YoutubeAPI: POST data: #{data}")
|
LOGGER.trace("YoutubeAPI: POST data: #{data}")
|
||||||
|
|
||||||
# Send the POST request
|
# Send the POST request
|
||||||
if client_config.proxy_region
|
if {{ !flag?(:disable_quic) }} && CONFIG.use_quic
|
||||||
response = YT_POOL.client(
|
# Using QUIC client
|
||||||
client_config.proxy_region,
|
response = YT_POOL.client(client_config.proxy_region,
|
||||||
&.post(url, headers: headers, body: data.to_json)
|
&.post(url, headers: headers, body: data.to_json)
|
||||||
)
|
)
|
||||||
|
body = response.body
|
||||||
else
|
else
|
||||||
response = YT_POOL.client &.post(
|
# Using HTTP client
|
||||||
url, headers: headers, body: data.to_json
|
body = YT_POOL.client(client_config.proxy_region) do |client|
|
||||||
)
|
client.post(url, headers: headers, body: data.to_json) do |response|
|
||||||
|
self._decompress(response.body_io, response.headers["Content-Encoding"]?)
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
# Convert result to Hash
|
# Convert result to Hash
|
||||||
initial_data = JSON.parse(response.body).as_h
|
initial_data = JSON.parse(body).as_h
|
||||||
|
|
||||||
# Error handling
|
# Error handling
|
||||||
if initial_data.has_key?("error")
|
if initial_data.has_key?("error")
|
||||||
|
@ -453,4 +448,35 @@ module YoutubeAPI
|
||||||
|
|
||||||
return initial_data
|
return initial_data
|
||||||
end
|
end
|
||||||
|
|
||||||
|
####################################################################
|
||||||
|
# _decompress(body_io, headers)
|
||||||
|
#
|
||||||
|
# Internal function that reads the Content-Encoding headers and
|
||||||
|
# decompresses the content accordingly.
|
||||||
|
#
|
||||||
|
# We decompress the body ourselves (when using HTTP::Client) because
|
||||||
|
# the auto-decompress feature is broken in the Crystal stdlib.
|
||||||
|
#
|
||||||
|
# Read more:
|
||||||
|
# - https://github.com/iv-org/invidious/issues/2612
|
||||||
|
# - https://github.com/crystal-lang/crystal/issues/11354
|
||||||
|
#
|
||||||
|
def _decompress(body_io : IO, encodings : String?) : String
|
||||||
|
if encodings
|
||||||
|
# Multiple encodings can be combined, and are listed in the order
|
||||||
|
# in which they were applied. E.g: "deflate, gzip" means that the
|
||||||
|
# content must be first "gunzipped", then "defated".
|
||||||
|
encodings.split(',').reverse.each do |enc|
|
||||||
|
case enc.strip(' ')
|
||||||
|
when "gzip"
|
||||||
|
body_io = Compress::Gzip::Reader.new(body_io, sync_close: true)
|
||||||
|
when "deflate"
|
||||||
|
body_io = Compress::Deflate::Reader.new(body_io, sync_close: true)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
return body_io.gets_to_end
|
||||||
|
end
|
||||||
end # End of module
|
end # End of module
|
||||||
|
|
Loading…
Reference in a new issue