fix: YouTube sanitizer should also clean youtu.be domain (#507)

svenjacobs · Nov 24, 2024 · 1336218 · 1336218
1 parent c61d66d
commit 1336218
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/...src/main/kotlin/com/svenjacobs/app/leon/core/domain/sanitizer/youtube/YoutubeSanitizer.kt b/...src/main/kotlin/com/svenjacobs/app/leon/core/domain/sanitizer/youtube/YoutubeSanitizer.kt
@@ -38,5 +38,5 @@ class YoutubeSanitizer :
 	)
 
 	override fun matchesDomain(input: String) =
-		input.matchesDomainRegex("(?:m(?:usic)?\\.)?youtube\\.com")
+		input.matchesDomainRegex("(?:(?:(?:m(?:usic)?\\.)?youtube\\.com)|youtu\\.be)")
 }
diff --git a/...test/kotlin/com/svenjacobs/app/leon/core/domain/sanitizer/youtube/YoutubeSanitizerTest.kt b/...test/kotlin/com/svenjacobs/app/leon/core/domain/sanitizer/youtube/YoutubeSanitizerTest.kt
@@ -33,7 +33,7 @@ class YoutubeSanitizerTest :
 						"https://m.youtube.com/watch?v=CvFH_6DNRCY&pp=ygUHZGVidXNzeQ%3D%3D",
 					) shouldBe "https://m.youtube.com/watch?v=CvFH_6DNRCY"
 				}
-				
+
 				"remove all parameters except \"t\" from video URLs" {
 					sanitizer(
 						"https://m.youtube.com/watch?v=CvFH_6DNRCY&t=125",
@@ -59,6 +59,12 @@ class YoutubeSanitizerTest :
 					) shouldBe
 						"https://music.youtube.com/watch?v=KGFkMD2zotU"
 				}
+
+				"remove parameters from youtu.be domain" {
+					sanitizer(
+						"https://youtu.be/RvRhUHTV_8k?si=OXYBmGTMXib1jlA2?si=UA0UVnrmPbK612Lu",
+					) shouldBe "https://youtu.be/RvRhUHTV_8k"
+				}
 			}
 
 			"matchesDomain" should {
@@ -74,6 +80,10 @@ class YoutubeSanitizerTest :
 				"match music.youtube.com domain" {
 					sanitizer.matchesDomain("https://music.youtube.com/") shouldBe true
 				}
+
+				"match youtu.be domain" {
+					sanitizer.matchesDomain("https://youtu.be/") shouldBe true
+				}
 			}
 		},
 	)