From b84ca6d84fc7635b42e518dec66fc00cc205633f Mon Sep 17 00:00:00 2001
From: Alek Ratzloff <alekratz@gmail.com>
Date: Fri, 29 Jul 2022 19:59:10 -0700
Subject: [PATCH] Add Youtube embeds

This will embed a youtube video inline on the page. Hopefully it works!

Signed-off-by: Alek Ratzloff <alekratz@gmail.com>
---
 board/static/board/post.js      | 21 +++++++++++++++++++++
 board/static/board/style.css    |  3 +++
 board/templatetags/post_body.py | 23 ++++++++++++++++++++---
 tests/test_replybuilder.py      | 12 ++++++++++++
 4 files changed, 56 insertions(+), 3 deletions(-)

diff --git a/board/static/board/post.js b/board/static/board/post.js
index 53228cb..4fea7bd 100644
--- a/board/static/board/post.js
+++ b/board/static/board/post.js
@@ -345,6 +345,27 @@ $(window).on("load", () => {
             $(e.target).attr("src", thumbUrl);
         }
     });
+
+    $(".youtube_embed_link").on("click", (e) => {
+        e.preventDefault();
+        let fullEnt = $(e.target).parents(".youtube_embed");
+        let frame = fullEnt.find(".youtube_embed_frame");
+        if (frame.length) {
+            frame.remove();
+        } else {
+            let hash = $(e.target).attr("data-yt-hash");
+            let embed = $("<span>")
+                .attr("class", "youtube_embed_frame")
+                .append("<br/>")
+                .append(
+                    $("<iframe>")
+                        .attr("src", "https://www.youtube.com/embed/" + hash)
+                        .attr("frameborder", "0")
+                        .attr("allow", "accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture")
+                );
+            fullEnt.find(".youtube_embed_container").append(embed);
+        }
+    });
 });
 
 // Load event that actually does stuff
diff --git a/board/static/board/style.css b/board/static/board/style.css
index 96162e9..86ce0eb 100644
--- a/board/static/board/style.css
+++ b/board/static/board/style.css
@@ -169,6 +169,9 @@ th {
     right: 0px;
 }
 
+/* Youtube embeds */
+
+
 /* Spoilers */
 
 .spoiler {
diff --git a/board/templatetags/post_body.py b/board/templatetags/post_body.py
index aa7b551..6691767 100644
--- a/board/templatetags/post_body.py
+++ b/board/templatetags/post_body.py
@@ -1,4 +1,5 @@
 import re
+from typing import Match, Pattern
 from django import template
 from board.models import Post
 
@@ -6,6 +7,9 @@ from board.models import Post
 REPLY_START_RE = re.compile(r"^&gt;&gt;\d+")
 REPLY_RE = re.compile(r"&gt;&gt;\d+")
 QUOTE_RE = re.compile(r"&gt;.+$")
+YOUTUBE_RE = re.compile(
+    r"https?://(((www\.)?youtube\.com/watch\?v=)|(youtu\.be/(watch\?v=)?))(?P<yt_hash>[a-zA-Z0-9_\-]{11})"
+)
 register = template.Library()
 
 
@@ -19,9 +23,12 @@ def htmlspecialchars(t: str):
 
 
 class ReplyBuilder:
-    def __init__(self, text: str, parse_spoilers: bool = False):
+    def __init__(
+        self, text: str, parse_spoilers: bool = False, parse_youtube: bool = True
+    ):
         self.text = text
         self.parse_spoilers = parse_spoilers
+        self.parse_youtube = parse_youtube
         self.index = 0
         self.final = ""
 
@@ -46,8 +53,12 @@ class ReplyBuilder:
     def adv(self, n=1):
         self.index += n
 
-    def matches(self, prefix: str) -> bool:
-        return self.remain[: len(prefix)] == prefix
+    def matches(self, prefix: str | Pattern) -> bool | Match:
+        if isinstance(prefix, str):
+            return self.remain.startswith(prefix)
+        else:
+            mat = prefix.match(self.remain)
+            return mat or False
 
     def build(self) -> str:
         while self.c:
@@ -60,6 +71,12 @@ class ReplyBuilder:
                 self.adv(9)
             elif self.parse_spoilers and self.matches("[spoiler]"):
                 self.do_spoiler()
+            elif self.parse_youtube and (mat := self.matches(YOUTUBE_RE)):
+                assert isinstance(mat, Match)
+                url = mat[0]
+                yt_hash = mat["yt_hash"]
+                self.final += f'<span class="youtube_embed"><span class="youtube_embed_container">{url} [<a class="youtube_embed_link" data-yt-hash="{yt_hash}" href="{url}">Embed</a>]</span></span>'
+                self.adv(len(url))
             else:
                 self.final += self.c
                 self.adv()
diff --git a/tests/test_replybuilder.py b/tests/test_replybuilder.py
index 9266e75..f42d38c 100644
--- a/tests/test_replybuilder.py
+++ b/tests/test_replybuilder.py
@@ -103,3 +103,15 @@ what happened to the 1 2 3 4 5 6 7 8 9
         for input, output in expected:
             rb = ReplyBuilder(input.strip(), parse_spoilers=True)
             self.assertEquals(rb.build(), output)
+
+    def test_youtube_embed(self):
+        expected = [
+            (
+                "https://www.youtube.com/watch?v=O_3_-UrhZH0",
+                '<span class="youtube_embed"><span class="youtube_embed_container">https://www.youtube.com/watch?v=O_3_-UrhZH0 [<a class="youtube_embed_link" data-yt-hash="O_3_-UrhZH0" href="https://www.youtube.com/watch?v=O_3_-UrhZH0">Embed</a>]</span></span>',
+            ),
+        ]
+
+        for input, output in expected:
+            rb = ReplyBuilder(input.strip())
+            self.assertEquals(rb.build(), output)