Merge pull request #46 from omarroth/add-channel-rss

Add channel RSS

3 files changed, 104 insertions, 6 deletions

diff --git a/src/invidious.cr b/src/invidious.cr
index 250f7624..9f971482 100644
--- a/src/invidious.cr
+++ b/src/invidious.cr
@@ -1126,10 +1126,7 @@ get "/api/v1/channels/:ucid/videos" do |env|
   videos = JSON.build do |json|
     json.array do
       document.xpath_nodes(%q(//li[contains(@class, "feed-item-container")])).each do |item|
-        anchor = item.xpath_node(%q(.//h3[contains(@class,"yt-lockup-title")]/a))
-        if !anchor
-          halt env, status_code: 403
-        end
+        anchor = item.xpath_node(%q(.//h3[contains(@class,"yt-lockup-title")]/a)).not_nil!
         title = anchor.content.strip
         video_id = anchor["href"].lchop("/watch?v=")
 
@@ -1786,6 +1783,96 @@ get "/feed/subscriptions" do |env|
   end
 end
 
+get "/feed/channel/:ucid" do |env|
+  ucid = env.params.url["ucid"]
+
+  url = produce_videos_url(ucid)
+  client = make_client(YT_URL)
+  response = client.get(url)
+
+  channel = get_channel(ucid, client, PG_DB, pull_all_videos: false)
+
+  json = JSON.parse(response.body)
+  content_html = json["content_html"].as_s
+  if content_html.empty?
+    halt env, status_code: 403
+  end
+  document = XML.parse_html(content_html)
+
+  if Kemal.config.ssl || CONFIG.https_only
+    scheme = "https://"
+  else
+    scheme = "http://"
+  end
+  host = env.request.headers["Host"]
+  path = env.request.path
+
+  feed = XML.build(indent: "  ", encoding: "UTF-8") do |xml|
+    xml.element("feed", "xmlns:yt": "http://www.youtube.com/xml/schemas/2015",
+      "xmlns:media": "http://search.yahoo.com/mrss/", xmlns: "http://www.w3.org/2005/Atom") do
+      xml.element("link", rel: "self", href: "#{scheme}#{host}#{path}")
+      xml.element("id") { xml.text "yt:channel:#{ucid}" }
+      xml.element("yt:channelId") { xml.text ucid }
+      xml.element("title") { xml.text channel.author }
+      xml.element("link", rel: "alternate", href: "#{scheme}#{host}/channel/#{ucid}")
+
+      xml.element("author") do
+        xml.element("name") { xml.text channel.author }
+        xml.element("uri") { xml.text "#{scheme}#{host}/channel/#{ucid}" }
+      end
+
+      document.xpath_nodes(%q(//li[contains(@class, "feed-item-container")])).each do |item|
+        anchor = item.xpath_node(%q(.//h3[contains(@class,"yt-lockup-title")]/a)).not_nil!
+        title = anchor.content.strip
+        video_id = anchor["href"].lchop("/watch?v=")
+
+        view_count = item.xpath_node(%q(.//div[@class="yt-lockup-meta"]/ul/li[2])).not_nil!
+        view_count = view_count.content.rchop(" views").delete(",").to_i
+
+        descriptionHtml = item.xpath_node(%q(.//div[contains(@class, "yt-lockup-description")])).not_nil!.to_s
+        description = descriptionHtml.gsub("<br>", "\n")
+        description = description.gsub("<br/>", "\n")
+        description = XML.parse_html(description).content.strip("\n ")
+
+        published = item.xpath_node(%q(.//div[@class="yt-lockup-meta"]/ul/li[1]))
+        if !published
+          next
+        end
+        published = published.content
+        published = decode_date(published)
+
+        xml.element("entry") do
+          xml.element("id") { xml.text "yt:video:#{video_id}" }
+          xml.element("yt:videoId") { xml.text video_id }
+          xml.element("yt:channelId") { xml.text ucid }
+          xml.element("title") { xml.text title }
+          xml.element("link", rel: "alternate", href: "#{scheme}#{host}/watch?v=#{video_id}")
+
+          xml.element("author") do
+            xml.element("name") { xml.text channel.author }
+            xml.element("uri") { xml.text "#{scheme}#{host}/channel/#{ucid}" }
+          end
+
+          xml.element("published") { xml.text published.to_s("%Y-%m-%dT%H:%M:%S%:z") }
+
+          xml.element("media:group") do
+            xml.element("media:title") { xml.text title }
+            xml.element("media:thumbnail", url: "https://i.ytimg.com/vi/#{video_id}/hqdefault.jpg", width: "480", height: "360")
+            xml.element("media:description") { xml.text description }
+          end
+
+          xml.element("media:community") do
+            xml.element("media:statistics", views: view_count)
+          end
+        end
+      end
+    end
+  end
+
+  env.response.content_type = "text/xml"
+  feed
+end
+
 get "/feed/private" do |env|
   token = env.params.query["token"]?
 
diff --git a/src/invidious/helpers.cr b/src/invidious/helpers.cr
index 676b3585..a091d1e2 100644
--- a/src/invidious/helpers.cr
+++ b/src/invidious/helpers.cr
@@ -950,7 +950,7 @@ def produce_playlist_url(ucid, index)
   return url
 end
 
-def produce_videos_url(ucid, page)
+def produce_videos_url(ucid, page = 1)
   page = "#{page}"
 
   meta = "\x12\x06videos \x00\x30\x02\x38\x01\x60\x01\x6a\x00\x7a"
diff --git a/src/invidious/views/channel.ecr b/src/invidious/views/channel.ecr
index 084ad10d..886ecc93 100644
--- a/src/invidious/views/channel.ecr
+++ b/src/invidious/views/channel.ecr
@@ -2,7 +2,17 @@
 <title><%= author %> - Invidious</title>
 <% end %>
 
-<h1><%= author %></h1>
+<div class="pure-g" style="padding:1em;">
+    <div class="pure-u-2-3">
+        <h3><%= author %></h3>
+    </div>
+    <div class="pure-u-1-3" style="text-align:right;">
+        <h3>
+            <a href="/feed/channel/<%= ucid %>"><i class="fas fa-rss-square"></i></a>
+        </h3>
+    </div>
+</div>
+
 <% if user %>
     <% if subscriptions.includes? ucid %>
     <p>
@@ -24,6 +34,7 @@
         </a>
     </p>
 <% end %>
+
 <% videos.each_slice(4) do |slice| %>
 <div class="pure-g">
     <% slice.each do |video| %>