midou/invidious
1
0
Fork 1
mirror of https://github.com/iv-org/invidious.git synced 2024-11-26 16:52:12 +05:30
Code Issues Packages Projects Releases Wiki Activity

Remove useless proc usage in images.cr

Browse Source
This commit is contained in:
syeopite 2024-04-25 13:28:58 -07:00
parent 003c6f81dc
commit 75b68618ab
No known key found for this signature in database
GPG Key ID: A73C186DA3955A1A
1 changed files with 3 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/invidious/routes/images.cr
View File

@ -41,14 +41,10 @@ module Invidious::Routes::Images
end end
end end
# A callable proc to be used inside #proxy_image
callable_proc = ->(env : HTTP::Server::Context) {
env.response.headers["Connection"] = "close"
}
begin begin
get_ytimg_pool(authority).client &.get(url, headers) do |resp| get_ytimg_pool(authority).client &.get(url, headers) do |resp|
return self.proxy_image(env, resp, callable_proc: callable_proc) env.response.headers["Connection"] = "close"
return self.proxy_image(env, resp)
end end
rescue ex rescue ex
end end
@ -138,7 +134,7 @@ module Invidious::Routes::Images
end end
end end
private def self.proxy_image(env, response, callable_proc = nil) private def self.proxy_image(env, response)
env.response.status_code = response.status_code env.response.status_code = response.status_code
response.headers.each do |key, value| response.headers.each do |key, value|
if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase) if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase)
@ -148,10 +144,6 @@ module Invidious::Routes::Images
env.response.headers["Access-Control-Allow-Origin"] = "*" env.response.headers["Access-Control-Allow-Origin"] = "*"
if callable_proc
callable_proc.call(env)
end
if response.status_code >= 300 if response.status_code >= 300
return env.response.headers.delete("Transfer-Encoding") return env.response.headers.delete("Transfer-Encoding")
end end