From df8526545383f4def3605fb61551edbd851c18c7 Mon Sep 17 00:00:00 2001
From: Samantaz Fox <coding@samantaz.fr>
Date: Sat, 6 May 2023 20:20:27 +0200
Subject: [PATCH] Comments: Move link utility functions to own file + module

---
 src/invidious/comments.cr             | 73 -------------------------
 src/invidious/comments/links_util.cr  | 76 +++++++++++++++++++++++++++
 src/invidious/routes/api/v1/videos.cr |  4 +-
 src/invidious/routes/watch.cr         |  8 +--
 4 files changed, 82 insertions(+), 79 deletions(-)
 create mode 100644 src/invidious/comments/links_util.cr

diff --git a/src/invidious/comments.cr b/src/invidious/comments.cr
index 6a3aa4c2..3c7e2bb4 100644
--- a/src/invidious/comments.cr
+++ b/src/invidious/comments.cr
@@ -1,76 +1,3 @@
-def replace_links(html)
-  # Check if the document is empty
-  # Prevents edge-case bug with Reddit comments, see issue #3115
-  if html.nil? || html.empty?
-    return html
-  end
-
-  html = XML.parse_html(html)
-
-  html.xpath_nodes(%q(//a)).each do |anchor|
-    url = URI.parse(anchor["href"])
-
-    if url.host.nil? || url.host.not_nil!.ends_with?("youtube.com") || url.host.not_nil!.ends_with?("youtu.be")
-      if url.host.try &.ends_with? "youtu.be"
-        url = "/watch?v=#{url.path.lstrip('/')}#{url.query_params}"
-      else
-        if url.path == "/redirect"
-          params = HTTP::Params.parse(url.query.not_nil!)
-          anchor["href"] = params["q"]?
-        else
-          anchor["href"] = url.request_target
-        end
-      end
-    elsif url.to_s == "#"
-      begin
-        length_seconds = decode_length_seconds(anchor.content)
-      rescue ex
-        length_seconds = decode_time(anchor.content)
-      end
-
-      if length_seconds > 0
-        anchor["href"] = "javascript:void(0)"
-        anchor["onclick"] = "player.currentTime(#{length_seconds})"
-      else
-        anchor["href"] = url.request_target
-      end
-    end
-  end
-
-  html = html.xpath_node(%q(//body)).not_nil!
-  if node = html.xpath_node(%q(./p))
-    html = node
-  end
-
-  return html.to_xml(options: XML::SaveOptions::NO_DECL)
-end
-
-def fill_links(html, scheme, host)
-  # Check if the document is empty
-  # Prevents edge-case bug with Reddit comments, see issue #3115
-  if html.nil? || html.empty?
-    return html
-  end
-
-  html = XML.parse_html(html)
-
-  html.xpath_nodes("//a").each do |match|
-    url = URI.parse(match["href"])
-    # Reddit links don't have host
-    if !url.host && !match["href"].starts_with?("javascript") && !url.to_s.ends_with? "#"
-      url.scheme = scheme
-      url.host = host
-      match["href"] = url
-    end
-  end
-
-  if host == "www.youtube.com"
-    html = html.xpath_node(%q(//body/p)).not_nil!
-  end
-
-  return html.to_xml(options: XML::SaveOptions::NO_DECL)
-end
-
 def text_to_parsed_content(text : String) : JSON::Any
   nodes = [] of JSON::Any
   # For each line convert line to array of nodes
diff --git a/src/invidious/comments/links_util.cr b/src/invidious/comments/links_util.cr
new file mode 100644
index 00000000..f89b86d3
--- /dev/null
+++ b/src/invidious/comments/links_util.cr
@@ -0,0 +1,76 @@
+module Invidious::Comments
+  extend self
+
+  def replace_links(html)
+    # Check if the document is empty
+    # Prevents edge-case bug with Reddit comments, see issue #3115
+    if html.nil? || html.empty?
+      return html
+    end
+
+    html = XML.parse_html(html)
+
+    html.xpath_nodes(%q(//a)).each do |anchor|
+      url = URI.parse(anchor["href"])
+
+      if url.host.nil? || url.host.not_nil!.ends_with?("youtube.com") || url.host.not_nil!.ends_with?("youtu.be")
+        if url.host.try &.ends_with? "youtu.be"
+          url = "/watch?v=#{url.path.lstrip('/')}#{url.query_params}"
+        else
+          if url.path == "/redirect"
+            params = HTTP::Params.parse(url.query.not_nil!)
+            anchor["href"] = params["q"]?
+          else
+            anchor["href"] = url.request_target
+          end
+        end
+      elsif url.to_s == "#"
+        begin
+          length_seconds = decode_length_seconds(anchor.content)
+        rescue ex
+          length_seconds = decode_time(anchor.content)
+        end
+
+        if length_seconds > 0
+          anchor["href"] = "javascript:void(0)"
+          anchor["onclick"] = "player.currentTime(#{length_seconds})"
+        else
+          anchor["href"] = url.request_target
+        end
+      end
+    end
+
+    html = html.xpath_node(%q(//body)).not_nil!
+    if node = html.xpath_node(%q(./p))
+      html = node
+    end
+
+    return html.to_xml(options: XML::SaveOptions::NO_DECL)
+  end
+
+  def fill_links(html, scheme, host)
+    # Check if the document is empty
+    # Prevents edge-case bug with Reddit comments, see issue #3115
+    if html.nil? || html.empty?
+      return html
+    end
+
+    html = XML.parse_html(html)
+
+    html.xpath_nodes("//a").each do |match|
+      url = URI.parse(match["href"])
+      # Reddit links don't have host
+      if !url.host && !match["href"].starts_with?("javascript") && !url.to_s.ends_with? "#"
+        url.scheme = scheme
+        url.host = host
+        match["href"] = url
+      end
+    end
+
+    if host == "www.youtube.com"
+      html = html.xpath_node(%q(//body/p)).not_nil!
+    end
+
+    return html.to_xml(options: XML::SaveOptions::NO_DECL)
+  end
+end
diff --git a/src/invidious/routes/api/v1/videos.cr b/src/invidious/routes/api/v1/videos.cr
index 6feaaef4..af4fc806 100644
--- a/src/invidious/routes/api/v1/videos.cr
+++ b/src/invidious/routes/api/v1/videos.cr
@@ -362,8 +362,8 @@ module Invidious::Routes::API::V1::Videos
         return reddit_thread.to_json
       else
         content_html = Frontend::Comments.template_reddit(comments, locale)
-        content_html = fill_links(content_html, "https", "www.reddit.com")
-        content_html = replace_links(content_html)
+        content_html = Comments.fill_links(content_html, "https", "www.reddit.com")
+        content_html = Comments.replace_links(content_html)
         response = {
           "title"       => reddit_thread.title,
           "permalink"   => reddit_thread.permalink,
diff --git a/src/invidious/routes/watch.cr b/src/invidious/routes/watch.cr
index 6b441a48..e5cf3716 100644
--- a/src/invidious/routes/watch.cr
+++ b/src/invidious/routes/watch.cr
@@ -101,8 +101,8 @@ module Invidious::Routes::Watch
               comments, reddit_thread = Comments.fetch_reddit(id)
               comment_html = Frontend::Comments.template_reddit(comments, locale)
 
-              comment_html = fill_links(comment_html, "https", "www.reddit.com")
-              comment_html = replace_links(comment_html)
+              comment_html = Comments.fill_links(comment_html, "https", "www.reddit.com")
+              comment_html = Comments.replace_links(comment_html)
             end
           end
         elsif source == "reddit"
@@ -110,8 +110,8 @@ module Invidious::Routes::Watch
             comments, reddit_thread = Comments.fetch_reddit(id)
             comment_html = Frontend::Comments.template_reddit(comments, locale)
 
-            comment_html = fill_links(comment_html, "https", "www.reddit.com")
-            comment_html = replace_links(comment_html)
+            comment_html = Comments.fill_links(comment_html, "https", "www.reddit.com")
+            comment_html = Comments.replace_links(comment_html)
           rescue ex
             if preferences.comments[1] == "youtube"
               comment_html = JSON.parse(Comments.fetch_youtube(id, nil, "html", locale, preferences.thin_mode, region))["contentHtml"]