Fix host in redirect

6 years ago · eee973fe86
parent 61769c6f9c
commit eee973fe86
1 changed files with 4 additions and 2 deletions
--- a/src/invidious.cr
+++ b/src/invidious.cr
@ -4240,10 +4240,11 @@ get "/videoplayback" do |env|

  if response.headers["Location"]?
    url = URI.parse(response.headers["Location"])
+    host = url.host
    env.response.headers["Access-Control-Allow-Origin"] = "*"

    url = url.full_path
-    url += "&host=#{host.lchop("https://")}"
+    url += "&host=#{host}"

    if region
      url += "&region=#{region}"
@ -4267,10 +4268,11 @@ get "/videoplayback" do |env|

    if response.headers["Location"]?
      url = URI.parse(response.headers["Location"])
+      host = url.host
      env.response.headers["Access-Control-Allow-Origin"] = "*"

      url = url.full_path
-      url += "&host=#{host.lchop("https://")}"
+      url += "&host=#{host}"

      if region
        url += "&region=#{region}"