feat: now urls like youtu.be/ID is supported

nlif-m · Jun 23, 2023 · 77c3a18 · 77c3a18
1 parent 2b71454
commit 77c3a18
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/ytdlp/ytdlp.go b/ytdlp/ytdlp.go
@@ -80,6 +80,7 @@ func (yt *Ytdlp) GetVersion() (version string, err error) {
 }
 
 var youtubeVideoRegexp = regexp.MustCompile(`(https:\/\/|)(www\.|)youtube\.com\/watch\?v=.+`)
+var youtubeVideoRegexp2 = regexp.MustCompile(`(https:\/\/|)youtu\.be\/.+`)
 var youtubePlaylistRegexp = regexp.MustCompile(`(https:\/\/|)(www\.|)youtube\.com\/playlist\?list=.+`)
 var vkVideoRegexp = regexp.MustCompile(`(https:\/\/|)vk\.com\/video(s|).+`)
 
@@ -89,10 +90,16 @@ func (yt *Ytdlp) IsDownloadable(rawURL string) (ytType YtdlpURLType, URL string,
 	if URL != "" {
 		return YoutubeVideoType, URL, true
 	}
+	URL = youtubeVideoRegexp2.FindString(rawURL)
+	if URL != "" {
+		return YoutubeVideoType, URL, true
+	}
+
 	URL = youtubePlaylistRegexp.FindString(rawURL)
 	if URL != "" {
 		return YoutubePlaylistType, URL, true
 	}
+
 	URL = vkVideoRegexp.FindString(rawURL)
 	if URL != "" {
 		return VkVideoType, URL, true

diff --git a/ytdlp/ytdlp_test.go b/ytdlp/ytdlp_test.go
@@ -17,6 +17,8 @@ func TestIsDownloadable(t *testing.T) {
 		{"https://www.youtube.com/watch?v=123456789zxcvbnasdfqwew", YoutubeVideoType, true},
 		{"www.youtube.com/watch?v=123456789zxcvbnasdfqwew", YoutubeVideoType, true},
 		{"youtube.com/watch?v=123456789zxcvbnasdfqwew", YoutubeVideoType, true},
+		{"https://youtu.be/312u8fdjaf", YoutubeVideoType, true},
+		{"youtu.be/312u8fdjaf", YoutubeVideoType, true},
 
 		{"https://www.youtube.com/playlist?list=1234345sdfjlka", YoutubePlaylistType, true},
 		{"www.youtube.com/playlist?list=1234345sdfjlka", YoutubePlaylistType, true},