From 62b6de6db2f6cd27484fbe048abf3b89ee72203f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafa=C5=82=20Dzi=C4=99giel?= <rafostar.github@gmail.com>
Date: Sat, 17 Apr 2021 16:14:21 +0200
Subject: [PATCH] YT: support live HLS videos

---
 src/youtube.js | 67 +++++++++++++++++++++++++++++++++++++-------------
 1 file changed, 50 insertions(+), 17 deletions(-)

diff --git a/src/youtube.js b/src/youtube.js
index 8d11439b..63ea0e9d 100644
--- a/src/youtube.js
+++ b/src/youtube.js
@@ -319,23 +319,28 @@ var YouTubeClient = GObject.registerClass({
             codec: 'h264',
             type: settings.get_string('yt-quality-type'),
         };
-        const dashInfo = await this.getDashInfoAsync(info, itagOpts).catch(debug);
 
-        if(dashInfo) {
-            debug('parsed video info to dash info');
-            const dash = Dash.generateDash(dashInfo);
+        uri = this.getHLSUri(info, itagOpts);
 
-            if(dash) {
-                debug('got dash data');
+        if(!uri) {
+            const dashInfo = await this.getDashInfoAsync(info, itagOpts).catch(debug);
 
-                const dashFile = await FileOps.saveFilePromise(
-                    'tmp', null, 'clapper.mpd', dash
-                ).catch(debug);
+            if(dashInfo) {
+                debug('parsed video info to dash info');
+                const dash = Dash.generateDash(dashInfo);
 
-                if(dashFile)
-                    uri = dashFile.get_uri();
+                if(dash) {
+                    debug('got dash data');
 
-                debug('got dash file');
+                    const dashFile = await FileOps.saveFilePromise(
+                        'tmp', null, 'clapper.mpd', dash
+                    ).catch(debug);
+
+                    if(dashFile)
+                        uri = dashFile.get_uri();
+
+                    debug('got dash file');
+                }
             }
         }
 
@@ -437,6 +442,27 @@ var YouTubeClient = GObject.registerClass({
         };
     }
 
+    getHLSUri(info, itagOpts)
+    {
+        const isLive = info.videoDetails.isLiveContent;
+        debug(`video is live: ${isLive}`);
+
+        /* YouTube only uses HLS for live content */
+        if(!isLive)
+            return null;
+
+        const hlsUri = info.streamingData.hlsManifestUrl;
+        if(!hlsUri) {
+            debug(new Error('no HLS manifest URL'));
+            return null;
+        }
+
+        /* TODO: download manifest and select best resolution
+         * for monitor when adaptive streaming is disabled */
+
+        return hlsUri;
+    }
+
     getBestCombinedUri(info, itagOpts)
     {
         debug(`obtaining best combined URL for resolution: ${itagOpts.width}x${itagOpts.height}`);
@@ -708,14 +734,21 @@ var YouTubeClient = GObject.registerClass({
 
     _getIsCipher(data)
     {
-        /* Check only first best combined,
-         * AFAIK there are no videos without it */
-        if(data.formats[0].url)
+        const stream = (data.formats.length)
+            ? data.formats[0]
+            : data.adaptiveFormats[0];
+
+        if(!stream) {
+            debug(new Error('no streams'));
+            return false;
+        }
+
+        if(stream.url)
             return false;
 
         if(
-            data.formats[0].signatureCipher
-            || data.formats[0].cipher
+            stream.signatureCipher
+            || stream.cipher
         )
             return true;