From e49aaa021646476b1499f7f612c13d5ba780462f Mon Sep 17 00:00:00 2001
From: Svallinn <41585298+Svallinn@users.noreply.github.com>
Date: Wed, 24 Mar 2021 05:15:06 +0000
Subject: [PATCH] Fix channel search API

---
 src/invidious/search.cr | 46 ++++++++++++++++++++++++++++-------------
 1 file changed, 32 insertions(+), 14 deletions(-)

diff --git a/src/invidious/search.cr b/src/invidious/search.cr
index cf8fd790..26161fab 100644
--- a/src/invidious/search.cr
+++ b/src/invidious/search.cr
@@ -231,20 +231,27 @@ end
 alias SearchItem = SearchVideo | SearchChannel | SearchPlaylist
 
 def channel_search(query, page, channel)
-  response = YT_POOL.client &.get("/channel/#{channel}?hl=en&gl=US")
-  response = YT_POOL.client &.get("/user/#{channel}?hl=en&gl=US") if response.headers["location"]?
-  response = YT_POOL.client &.get("/c/#{channel}?hl=en&gl=US") if response.headers["location"]?
+  response = YT_POOL.client &.get("/channel/#{channel}")
 
-  ucid = response.body.match(/\\"channelId\\":\\"(?<ucid>[^\\]+)\\"/).try &.["ucid"]?
+  if response.status_code == 404
+    response = YT_POOL.client &.get("/user/#{channel}")
+    response = YT_POOL.client &.get("/c/#{channel}") if response.status_code == 404
+    ucid = response.body.match(/HeaderRenderer":\{"channelId":"(?<ucid>[^\\"]+)"/).try &.["ucid"]?
+  else
+    ucid = channel
+  end
 
-  return 0, [] of SearchItem if !ucid
+  continuation = produce_channel_search_continuation(ucid, query, page)
+  response_json = request_youtube_api_browse(continuation)
 
-  url = produce_channel_search_url(ucid, query, page)
-  response = YT_POOL.client &.get(url)
-  initial_data = JSON.parse(response.body).as_a.find &.["response"]?
-  return 0, [] of SearchItem if !initial_data
-  author = initial_data["response"]?.try &.["metadata"]?.try &.["channelMetadataRenderer"]?.try &.["title"]?.try &.as_s
-  items = extract_items(initial_data.as_h, author, ucid)
+  result = JSON.parse(response_json.match(/"continuationItems": (?<items>\[.*\]),/m).try &.["items"] || "{}")
+  return 0, [] of SearchItem if result.size == 0
+
+  items = [] of SearchItem
+  result.as_a.select(&.as_h.has_key?("itemSectionRenderer")).each { |item|
+    extract_item(item["itemSectionRenderer"]["contents"].as_a[0])
+      .try { |t| items << t }
+  }
 
   return items.size, items
 end
@@ -361,17 +368,28 @@ def produce_search_params(page = 1, sort : String = "relevance", date : String =
   return params
 end
 
-def produce_channel_search_url(ucid, query, page)
+def produce_channel_search_continuation(ucid, query, page)
+  if page <= 1
+    idx = 0_i64
+  else
+    idx = 30_i64 * (page - 1)
+  end
+
   object = {
     "80226972:embedded" => {
       "2:string" => ucid,
       "3:base64" => {
         "2:string"  => "search",
+        "6:varint"  => 1_i64,
         "7:varint"  => 1_i64,
-        "15:string" => "#{page}",
+        "12:varint" => 1_i64,
+        "15:base64" => {
+          "3:varint" => idx,
+        },
         "23:varint" => 0_i64,
       },
       "11:string" => query,
+      "35:string" => "browse-feed#{ucid}search",
     },
   }
 
@@ -380,7 +398,7 @@ def produce_channel_search_url(ucid, query, page)
     .try { |i| Base64.urlsafe_encode(i) }
     .try { |i| URI.encode_www_form(i) }
 
-  return "/browse_ajax?continuation=#{continuation}&gl=US&hl=en"
+  return continuation
 end
 
 def process_search_query(query, page, user, region)