From 11ff40bcd68f4941fa40017f0bb6f5a8826effa1 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Mon, 11 Mar 2019 12:55:05 -0500
Subject: [PATCH] Fix paths for 'local=true&raw=1'

---
 src/invidious.cr | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/invidious.cr b/src/invidious.cr
index 998b9291..c89c6771 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -405,8 +405,8 @@ get "/watch" do |env|
   adaptive_fmts = video.adaptive_fmts(decrypt_function)
 
   if params[:local]
-    fmt_stream.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).query.not_nil! }
-    adaptive_fmts.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).query.not_nil! }
+    fmt_stream.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).full_path }
+    adaptive_fmts.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).full_path }
   end
 
   video_streams = video.video_streams(adaptive_fmts)
@@ -511,8 +511,8 @@ get "/embed/:id" do |env|
   adaptive_fmts = video.adaptive_fmts(decrypt_function)
 
   if params[:local]
-    fmt_stream.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).query.not_nil! }
-    adaptive_fmts.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).query.not_nil! }
+    fmt_stream.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).full_path }
+    adaptive_fmts.each { |fmt| fmt["url"] = URI.parse(fmt["url"]).full_path }
   end
 
   video_streams = video.video_streams(adaptive_fmts)