VeritableValor/invidious
1
0
Fork 0
mirror of https://github.com/iv-org/invidious.git synced 2026-07-27 16:06:47 -05:00
Code Issues Packages Projects Releases Wiki Activity

chore: improve invidious companion proxying

Browse Source
This commit is contained in:
Fijxu 2026-07-14 02:41:11 -04:00
parent c84acc10ad
commit c6dfc0017d
No known key found for this signature in database
GPG Key ID: 32C1DDF333EDA6A4
1 changed files with 20 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
src/invidious/routes/companion.cr
View File

@ -1,60 +1,43 @@
module Invidious::Routes::Companion
# GET /companion
def self.get_companion(env)
url = env.request.path
if env.request.query
url += "?#{env.request.query}"
end
begin
COMPANION_POOL.client do |wrapper|
wrapper.client.get(url, env.request.headers) do |resp|
return self.proxy_companion(env, resp)
end
end
rescue ex
end
url = self.make_url
self.proxy_companion(env, "GET", url)
end
# POST /companion
def self.post_companion(env)
url = env.request.path
if env.request.query
url += "?#{env.request.query}"
end
begin
COMPANION_POOL.client do |wrapper|
wrapper.client.post(url, env.request.headers, env.request.body) do |resp|
return self.proxy_companion(env, resp)
end
end
rescue ex
end
url = self.make_url
self.proxy_companion(env, "POST", url)
end
# OPTIONS /companion
def self.options_companion(env)
url = self.make_url
self.proxy_companion(env, "OPTIONS", url)
end
private def make_url(env)
url = env.request.path
if env.request.query
url += "?#{env.request.query}"
end
end
private def self.proxy_companion(env, method, url)
begin
COMPANION_POOL.client do |wrapper|
wrapper.client.options(url, env.request.headers) do |resp|
return self.proxy_companion(env, resp)
end
end
rescue ex
end
end
private def self.proxy_companion(env, response)
env.response.status_code = response.status_code
response.headers.each do |key, value|
wrapper.client.exec(method, url, env.request.headers, (env.request.body if method == "POST")) do |resp|
env.response.status_code = resp.status_code
resp.headers.each do |key, value|
env.response.headers[key] = value
end
return IO.copy response.body_io, env.response
env.response.headers["Via"] = "1.1 Invidious"
return IO.copy resp.body_io, env.response
end
end
rescue ex
return error_json(502, "Couldn't proxy request to Invidious Companion.")
end
end
end