From 830d6fae8e09937b6c0a0dfffc58f46d6be91fd9 Mon Sep 17 00:00:00 2001 From: Omar Roth Date: Sun, 15 Apr 2018 22:57:17 -0500 Subject: [PATCH] Add support for using local links in manifest --- src/invidious.cr | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/src/invidious.cr b/src/invidious.cr index fb405ffd..7e950317 100644 --- a/src/invidious.cr +++ b/src/invidious.cr @@ -540,6 +540,7 @@ get "/api/manifest/dash/id/:id" do |env| env.response.headers.add("Access-Control-Allow-Origin", "*") env.response.content_type = "application/dash+xml" + local = env.params.query["local"]?.try &.== "true" id = env.params.url["id"] yt_client = get_client(youtube_pool) @@ -560,12 +561,18 @@ get "/api/manifest/dash/id/:id" do |env| halt env, status_code: 403 end - signature = false - if adaptive_fmts[0]? && adaptive_fmts[0]["s"]? - signature = true + if local + adaptive_fmts.each do |fmt| + if Kemal.config.ssl + scheme = "https://" + end + scheme ||= "http://" + + fmt["url"] = scheme + env.request.headers["Host"] + URI.parse(fmt["url"]).full_path + end end - if signature + if adaptive_fmts[0]? && adaptive_fmts[0]["s"]? adaptive_fmts.each do |fmt| fmt["url"] += "&signature=" + decrypt_signature(fmt["s"]) end