fix: YT sanitizer should clean YT music domains (#380)

svenjacobs · Mar 16, 2024 · 891a4b0 · 891a4b0
1 parent 9466443
commit 891a4b0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
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
@@ -37,5 +37,6 @@ class YoutubeSanitizer :
 		name = context.getString(R.string.sanitizer_youtube_name),
 	)
 
-	override fun matchesDomain(input: String) = input.matchesDomainRegex("(m\\.)?youtube\\.com")
+	override fun matchesDomain(input: String) =
+		input.matchesDomainRegex("(?:m(?:usic)?\\.)?youtube\\.com")
 }
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
@@ -46,6 +46,13 @@ class YoutubeSanitizerTest :
 					) shouldBe
 						"https://youtube.com/playlist?list=PLkqz3S84Tw-QYEdfTLBzxJ1FAprtqeEpJ"
 				}
+
+				"remove all parameters except \"v\" from YouTube Music URLs" {
+					sanitizer(
+						"https://music.youtube.com/watch?v=KGFkMD2zotU&si=JrZ7QzX4VeMrfzp8",
+					) shouldBe
+						"https://music.youtube.com/watch?v=KGFkMD2zotU"
+				}
 			}
 
 			"matchesDomain" should {
@@ -57,6 +64,10 @@ class YoutubeSanitizerTest :
 				"match m.youtube.com domain" {
 					sanitizer.matchesDomain("https://m.youtube.com/") shouldBe true
 				}
+
+				"match music.youtube.com domain" {
+					sanitizer.matchesDomain("https://music.youtube.com/") shouldBe true
+				}
 			}
 		},
 	)