From 81ace95c13f7cc9a9fa8e786040d5b7a21041873 Mon Sep 17 00:00:00 2001
From: craftingmod <alyac1234@gmail.com>
Date: Sun, 18 Apr 2021 04:10:12 +0900
Subject: [PATCH] [youtube] Use youtube instead of youtube music if premium
 isn't presented.

---
 youtube_dl/extractor/youtube.py | 42 ++++++++++++++++++++++-----------
 1 file changed, 28 insertions(+), 14 deletions(-)

diff --git a/youtube_dl/extractor/youtube.py b/youtube_dl/extractor/youtube.py
index 1c9213ffd..a31d351dc 100644
--- a/youtube_dl/extractor/youtube.py
+++ b/youtube_dl/extractor/youtube.py
@@ -1483,6 +1483,18 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
         webpage = self._download_webpage(
             webpage_url + '&bpctr=9999999999&has_verified=1', video_id, fatal=False)
 
+        # define early for use
+        def get_text(x):
+            if not x:
+                return
+            text = x.get('simpleText')
+            if text and isinstance(text, compat_str):
+                return text
+            runs = x.get('runs')
+            if not isinstance(runs, list):
+                return
+            return ''.join([r['text'] for r in runs if isinstance(r.get('text'), compat_str)])
+
         # check is music
         is_music = re.match(r'^https?://music\.youtube\.com/.+', url)
 
@@ -1506,9 +1518,22 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
                 lambda x: x['player_response'][0],
                 compat_str) or '{}', video_id)
             if pr:
-                self.to_screen("Parsed youtube music info!")
-                player_response = pr
-        elif webpage:
+                # check cookie works
+                playability_status = pr.get('playabilityStatus') or {}
+                pemr = try_get(
+                    playability_status,
+                    lambda x: x['errorScreen']['playerErrorMessageRenderer'],
+                    dict) or {}
+                reason = get_text(pemr.get('reason')) or playability_status.get('reason')
+                if reason != None and reason.find("This video is only available to Music Premium members") >= 0:
+                    # accept only music premium 
+                    self.report_warning(reason.replace("video", "URL") + ". Using youtube mode instead of yt-music mode.", video_id)
+                else:
+                    self.to_screen("Using yt-music.")
+                    player_response = pr
+
+
+        if player_response == None and webpage:
             player_response = self._extract_yt_initial_variable(
                 webpage, self._YT_INITIAL_PLAYER_RESPONSE_RE,
                 video_id, 'initial player response')
@@ -1539,17 +1564,6 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
             return self.url_result(
                 trailer_video_id, self.ie_key(), trailer_video_id)
 
-        def get_text(x):
-            if not x:
-                return
-            text = x.get('simpleText')
-            if text and isinstance(text, compat_str):
-                return text
-            runs = x.get('runs')
-            if not isinstance(runs, list):
-                return
-            return ''.join([r['text'] for r in runs if isinstance(r.get('text'), compat_str)])
-
         search_meta = (
             lambda x: self._html_search_meta(x, webpage, default=None)) \
             if webpage else lambda x: None