From 823603650f7f45e7cdc27f0beb7d6e346efb6ec5 Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@protonmail.com>
Date: Tue, 3 Dec 2019 19:14:11 -0500
Subject: [PATCH] Add support for /sorry/index CAPTCHA

---
 src/invidious/helpers/jobs.cr | 54 +++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)

diff --git a/src/invidious/helpers/jobs.cr b/src/invidious/helpers/jobs.cr
index 5838b5b35..f368d6df3 100644
--- a/src/invidious/helpers/jobs.cr
+++ b/src/invidious/helpers/jobs.cr
@@ -290,6 +290,60 @@ def bypass_captcha(captcha_key, logger)
         response = YT_POOL.client &.post("/das_captcha", headers, form: inputs)
 
         yield response.cookies.select { |cookie| cookie.name != "PREF" }
+      elsif response.headers["Location"]?.try &.includes?("/sorry/index")
+        location = response.headers["Location"].try { |u| URI.parse(u) }
+        client = QUIC::Client.new(location.host.not_nil!)
+        response = client.get(location.full_path)
+
+        html = XML.parse_html(response.body)
+        form = html.xpath_node(%(//form[@action="index"])).not_nil!
+        site_key = form.xpath_node(%(.//div[@class="g-recaptcha"])).try &.["data-sitekey"]
+
+        inputs = {} of String => String
+        form.xpath_nodes(%(.//input[@name])).map do |node|
+          inputs[node["name"]] = node["value"]
+        end
+
+        response = JSON.parse(HTTP::Client.post("https://api.anti-captcha.com/createTask", body: {
+          "clientKey" => CONFIG.captcha_key,
+          "task"      => {
+            "type"       => "NoCaptchaTaskProxyless",
+            "websiteURL" => location.to_s,
+            "websiteKey" => site_key,
+          },
+        }.to_json).body)
+
+        if response["error"]?
+          raise response["error"].as_s
+        end
+
+        task_id = response["taskId"].as_i
+
+        loop do
+          sleep 10.seconds
+
+          response = JSON.parse(HTTP::Client.post("https://api.anti-captcha.com/getTaskResult", body: {
+            "clientKey" => CONFIG.captcha_key,
+            "taskId"    => task_id,
+          }.to_json).body)
+
+          if response["status"]?.try &.== "ready"
+            break
+          elsif response["errorId"]?.try &.as_i != 0
+            raise response["errorDescription"].as_s
+          end
+        end
+
+        inputs["g-recaptcha-response"] = response["solution"]["gRecaptchaResponse"].as_s
+        client.close
+        client = QUIC::Client.new("www.google.com")
+        response = client.post(location.full_path, form: inputs)
+        headers = HTTP::Headers{
+          "Cookie" => URI.parse(response.headers["location"]).query_params["google_abuse"].split(";")[0],
+        }
+        cookies = HTTP::Cookies.from_headers(headers)
+
+        yield cookies
       end
     rescue ex
       logger.puts("Exception: #{ex.message}")