From b31c286a8b8442d48f9b0713a8cce41432c168d1 Mon Sep 17 00:00:00 2001
From: retoor <retoor@molodetz.nl>
Date: Wed, 9 Apr 2025 10:34:57 +0200
Subject: [PATCH] New video embedding

---
 src/snek/system/template.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/snek/system/template.py b/src/snek/system/template.py
index cff807b..7ee0d0f 100644
--- a/src/snek/system/template.py
+++ b/src/snek/system/template.py
@@ -91,10 +91,10 @@ def embed_youtube(text):
     for element in soup.find_all("a"):
         if (
             element.attrs["href"].startswith("https://www.you")
-            and "?v=" in element.attrs["href"]
+            and "v=" in element.attrs["href"] or "si=" in element.attrs["href"]
         ):
-            video_name = element.attrs["href"].split("?v=")[1].split("&")[0]
-            embed_template = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_name}" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>'
+            video_name = "?" + "?".join(element.attrs["href"].split("?")[1:])
+            embed_template = f'<br /><iframe width="560" height="315" src="https://www.youtube.com/embed/{video_name}" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>'
             element.replace_with(BeautifulSoup(embed_template, "html.parser"))
     return str(soup)