From 0c8dff162d2431e087b28f334827dbe3a65a233e Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Wed, 27 Feb 2019 15:15:24 -0600
Subject: [PATCH] Fix embed extractor for age-gated videos

---
 src/invidious/videos.cr | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/invidious/videos.cr b/src/invidious/videos.cr
index c6ee72327..1de591b32 100644
--- a/src/invidious/videos.cr
+++ b/src/invidious/videos.cr
@@ -636,7 +636,10 @@ def fetch_video(id, proxies, region)
 
   # Try to pull streams from embed URL
   if info["reason"]?
-    embed_info = HTTP::Params.parse(client.get("/get_video_info?video_id=#{id}&ps=default&eurl=&gl=US&hl=en&disable_polymer=1").body)
+    embed_page = client.get("/embed/#{id}").body
+    sts = embed_page.match(/"sts"\s*:\s*(?<sts>\d+)/).try &.["sts"]?
+    sts ||= ""
+    embed_info = HTTP::Params.parse(client.get("/get_video_info?video_id=#{id}&eurl=https://youtube.googleapis.com/v/#{id}&gl=US&hl=en&disable_polymer=1&sts=#{sts}").body)
 
     if !embed_info["reason"]?
       embed_info.each do |key, value|