Merge pull request #134 from internetarchive/telegram-url

fix: recursion present in Telegram URLs
internetarchive · Aug 20, 2024 · be08a01 · be08a01
2 parents 764b35c + 4c2c5ef
commit be08a01
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/internal/pkg/crawl/sitespecific/telegram/telegram.go b/internal/pkg/crawl/sitespecific/telegram/telegram.go
@@ -6,7 +6,7 @@ import (
 )
 
 func IsTelegramEmbedURL(url string) bool {
-	return strings.Contains(url, "/t.me/") && strings.Contains(url, "?embed=1")
+	return strings.Contains(url, "/t.me/") && strings.Contains(url, "embed=1")
 }
 
 func IsTelegramURL(url string) bool {
@@ -15,9 +15,8 @@ func IsTelegramURL(url string) bool {
 
 func TransformURL(URL *url.URL) {
 	// Add embed=1 to the URL, without changing the original URL
-	if len(URL.RawQuery) > 0 {
-		URL.RawQuery += "&embed=1&mode=tme"
-	} else {
-		URL.RawQuery = "embed=1&mode=tme"
-	}
+	q := URL.Query()
+	q.Add("embed", "1")
+	q.Add("mode", "tme")
+	URL.RawQuery = q.Encode()
 }