From 71716ab01875f04f51bc23ba551c7a6a9a24a335 Mon Sep 17 00:00:00 2001
From: Ola Rubaj <52197250+olayway@users.noreply.github.com>
Date: Tue, 26 Sep 2023 21:23:39 +0200
Subject: [PATCH] [remark-wiki-link][xs]: add support youtu.be links

---
 .changeset/spicy-terms-worry.md               |  5 +++++
 packages/remark-embed/src/lib/remark-embed.ts | 21 ++++++++++++++++---
 2 files changed, 23 insertions(+), 3 deletions(-)
 create mode 100644 .changeset/spicy-terms-worry.md

diff --git a/.changeset/spicy-terms-worry.md b/.changeset/spicy-terms-worry.md
new file mode 100644
index 000000000..0c05aa68d
--- /dev/null
+++ b/.changeset/spicy-terms-worry.md
@@ -0,0 +1,5 @@
+---
+'@portaljs/remark-embed': patch
+---
+
+Add support for shortened yt links (https://youtu.be).
diff --git a/packages/remark-embed/src/lib/remark-embed.ts b/packages/remark-embed/src/lib/remark-embed.ts
index 141602906..6d99f727a 100644
--- a/packages/remark-embed/src/lib/remark-embed.ts
+++ b/packages/remark-embed/src/lib/remark-embed.ts
@@ -1,14 +1,29 @@
 import { visit } from "unist-util-visit";
 
+// https://youtu.be/vDuh7vDgGIg
+
+// TODO write tests!
 function transformer(tree) {
   visit(tree, "paragraph", (node) => {
     visit(node, "text", (textNode) => {
       if (
-        textNode.value.includes("https://www.youtube.com") &&
+        (textNode.value.includes("https://www.youtube.com") ||
+          textNode.value.includes("https://youtu.be")
+        ) &&
         !textNode.value.includes("\n")
       ) {
-        const urlSplit = textNode.value.split(/[=&]+/);
-        const iframeUrl = `https://www.youtube.com/embed/${urlSplit[1]}`;
+        let videoId: string;
+
+        if (textNode.value.includes("https://youtu.be")) {
+          // Extract video ID for short YT URLs
+          videoId = textNode.value.split("/").pop();
+        } else {
+          // Extract video ID for full YT URLs
+          const urlSplit = textNode.value.split(/[=&]+/);
+          videoId = urlSplit[1];
+        }
+
+        const iframeUrl = `https://www.youtube.com/embed/${videoId}`;
         Object.assign(node, {
           ...node,
           type: "element",