Skip to content

Commit

Permalink
Remove useless proc usage in images.cr
Browse files Browse the repository at this point in the history
  • Loading branch information
syeopite committed Apr 25, 2024
1 parent 7ab2d1e commit 16146bc
Showing 1 changed file with 3 additions and 11 deletions.
14 changes: 3 additions & 11 deletions src/invidious/routes/images.cr
Expand Up @@ -41,14 +41,10 @@ module Invidious::Routes::Images
end
end

# A callable proc to be used inside #proxy_image
callable_proc = ->(env : HTTP::Server::Context) {
env.response.headers["Connection"] = "close"
}

begin
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
rescue ex
end
Expand Down Expand Up @@ -138,7 +134,7 @@ module Invidious::Routes::Images
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
response.headers.each do |key, value|
if !RESPONSE_HEADERS_BLACKLIST.includes?(key.downcase)
Expand All @@ -148,10 +144,6 @@ module Invidious::Routes::Images

env.response.headers["Access-Control-Allow-Origin"] = "*"

if callable_proc
callable_proc.call(env)
end

if response.status_code >= 300
return env.response.headers.delete("Transfer-Encoding")
end
Expand Down

0 comments on commit 16146bc

Please sign in to comment.