youtube: detect login required video

iawia002 · Dec 13, 2019 · b975c0f · b975c0f
1 parent 3675c72
commit b975c0f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/extractors/defs.go b/extractors/defs.go
@@ -5,3 +5,4 @@ import (
 )
 
 var ErrURLParseFailed = errors.New("url parse failed")
+var ErrLoginRequired = errors.New("login required")
diff --git a/extractors/youtube/youtube.go b/extractors/youtube/youtube.go
@@ -99,6 +99,10 @@ func youtubeDownload(uri string) downloader.Data {
 	}
 	ytplayer := utils.MatchOneOf(html, `;ytplayer\.config\s*=\s*({.+?});`)
 	if ytplayer == nil || len(ytplayer) < 2 {
+		if strings.Contains(html, "LOGIN_REQUIRED") ||
+			strings.Contains(html, "Sign in to confirm your age") {
+			return downloader.EmptyData(uri, extractors.ErrLoginRequired)
+		}
 		return downloader.EmptyData(uri, extractors.ErrURLParseFailed)
 	}