[remark-wiki-link][xs]: add support youtu.be links

datopian · Sep 26, 2023 · 71716ab · 71716ab
1 parent 06d3977
commit 71716ab
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/.changeset/spicy-terms-worry.md 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
@@ -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",