From f26ad00155eb0258e667b18b04c03a7cabc60c6d Mon Sep 17 00:00:00 2001
From: Omar Roth <omarroth@hotmail.com>
Date: Tue, 19 Feb 2019 17:05:27 -0600
Subject: [PATCH] Add /api/v1/channels/playlists/:ucid

---
 src/invidious.cr | 133 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 133 insertions(+)

diff --git a/src/invidious.cr b/src/invidious.cr
index 587929ff..ead41755 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -3179,6 +3179,139 @@ end
   end
 end
 
+["/api/v1/channels/:ucid/playlists", "/api/v1/channels/playlists/:ucid"].each do |route|
+  get route do |env|
+    locale = LOCALES[env.get("locale").as(String)]?
+
+    env.response.content_type = "application/json"
+
+    ucid = env.params.url["ucid"]
+    continuation = env.params.query["continuation"]?
+    sort_by = env.params.query["sort"]?.try &.downcase
+    sort_by ||= env.params.query["sort_by"]?.try &.downcase
+    sort_by ||= "last"
+
+    begin
+      author, ucid, auto_generated = get_about_info(ucid, locale)
+    rescue ex
+      error_message = ex.message
+      halt env, status_code: 500, response: error_message
+    end
+
+    client = make_client(YT_URL)
+
+    if continuation
+      url = produce_channel_playlists_url(ucid, continuation, sort_by, auto_generated)
+
+      response = client.get(url)
+      json = JSON.parse(response.body)
+
+      if json["load_more_widget_html"].as_s.empty?
+        response = {
+          "playlists"    => [] of String,
+          "continuation" => nil,
+        }
+
+        if env.params.query["pretty"]? && env.params.query["pretty"] == "1"
+          response = response.to_pretty_json
+        else
+          response = response.to_json
+        end
+
+        halt env, status_code: 200, response: response
+      end
+
+      continuation = XML.parse_html(json["load_more_widget_html"].as_s)
+      continuation = continuation.xpath_node(%q(//button[@data-uix-load-more-href]))
+      if continuation
+        continuation = extract_channel_playlists_cursor(continuation["data-uix-load-more-href"], auto_generated)
+      end
+
+      html = XML.parse_html(json["content_html"].as_s)
+      nodeset = html.xpath_nodes(%q(//li[contains(@class, "feed-item-container")]))
+    else
+      url = "/channel/#{ucid}/playlists?disable_polymer=1&flow=list"
+
+      if auto_generated
+        url += "&view=50"
+      else
+        url += "&view=1"
+      end
+
+      case sort_by
+      when "last", "last_added"
+        #
+      when "oldest", "oldest_created"
+        url += "&sort=da"
+      when "newest", "newest_created"
+        url += "&sort=dd"
+      end
+
+      response = client.get(url)
+      html = XML.parse_html(response.body)
+
+      continuation = html.xpath_node(%q(//button[@data-uix-load-more-href]))
+      if continuation
+        continuation = extract_channel_playlists_cursor(continuation["data-uix-load-more-href"], auto_generated)
+      end
+
+      nodeset = html.xpath_nodes(%q(//ul[@id="browse-items-primary"]/li[contains(@class, "feed-item-container")]))
+    end
+
+    if auto_generated
+      items = extract_shelf_items(nodeset, ucid, author)
+    else
+      items = extract_items(nodeset, ucid, author)
+    end
+
+    response = JSON.build do |json|
+      json.object do
+        json.field "playlists" do
+          json.array do
+            items.each do |item|
+              json.object do
+                if item.is_a?(SearchPlaylist)
+                  json.field "title", item.title
+                  json.field "playlistId", item.id
+
+                  json.field "author", item.author
+                  json.field "authorId", item.ucid
+                  json.field "authorUrl", "/channel/#{item.ucid}"
+
+                  json.field "videoCount", item.video_count
+                  json.field "videos" do
+                    json.array do
+                      item.videos.each do |video|
+                        json.object do
+                          json.field "title", video.title
+                          json.field "videoId", video.id
+                          json.field "lengthSeconds", video.length_seconds
+
+                          json.field "videoThumbnails" do
+                            generate_thumbnails(json, video.id)
+                          end
+                        end
+                      end
+                    end
+                  end
+                end
+              end
+            end
+          end
+        end
+
+        json.field "continuation", continuation
+      end
+    end
+
+    if env.params.query["pretty"]? && env.params.query["pretty"] == "1"
+      JSON.parse(response).to_pretty_json
+    else
+      response
+    end
+  end
+end
+
 get "/api/v1/channels/search/:ucid" do |env|
   locale = LOCALES[env.get("locale").as(String)]?