From c281ceabda1e483bb531685579a4cfa78e29d738 Mon Sep 17 00:00:00 2001
From: girst <tobi@isticktoit.net>
Date: Fri, 29 Apr 2022 20:27:51 +0200
Subject: [PATCH] allow selecting muxed video quality

listing all urls in the <video> element doesn't work (in firefox at
least), so only load highest resolution one by default. we can then just
override that choice when a &itag=XX is present.
---
 app/static/style.css                |  2 +-
 app/youtube/__init__.py             |  4 ++++
 app/youtube/templates/watch.html.j2 | 12 +++++++++---
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/app/static/style.css b/app/static/style.css
index 85a6af2..18a8f50 100644
--- a/app/static/style.css
+++ b/app/static/style.css
@@ -313,7 +313,7 @@ off-screen, but only on browsers that support nesting calc/max: */
 .header-left:not(:first-of-type) {
   margin-left: 1em;
 }
-.header-left.bold {
+.bold {
   font-weight: bold;
 }
 .header-left.large {
diff --git a/app/youtube/__init__.py b/app/youtube/__init__.py
index aee084f..d303079 100644
--- a/app/youtube/__init__.py
+++ b/app/youtube/__init__.py
@@ -81,8 +81,12 @@ def watch():
 
     video_id = request.args.get('v')
     show = request.args.get("show")
+    itag = request.args.get("itag")
     video_url, stream_map, metadata, error, errdetails = get_video_info(video_id, metaOnly=(show == 'meta'))
 
+    if itag:  # if itag is specified and available, force its use even if not highest resolution
+        video_url = next((e['url'] for e in stream_map['muxed'] if e['itag'] == int(itag)), video_url)
+
     proxy_on = getattr(g, 'proxy_on', None)
     proxy_msg = {
         False: "Log in to use proxy.", # proxy for logged-in users only
diff --git a/app/youtube/templates/watch.html.j2 b/app/youtube/templates/watch.html.j2
index ce2b082..da588ef 100644
--- a/app/youtube/templates/watch.html.j2
+++ b/app/youtube/templates/watch.html.j2
@@ -12,9 +12,7 @@
 <div class="aspect-ratio main-video" style="--aspect-ratio:{{ aspectr }}">
 <video controls poster="{{ poster }}">
 	{% set offset = "#t="~request.args.t|timeoffset if request.args.t else "" %}
-	{% for v in stream_map.muxed|sort(attribute='height', reverse=True) %}
-	<source src="{{v.url}}{{offset}}">
-	{% endfor %}
+	<source src="{{video_url}}{{offset}}">
 	{% set cc_default = False %}
 	{% for cc in subtitles %}
 	<track label="{{ cc.name }}" kind="subtitles" srclang="{{ cc.code }}" src="{{ url_for('youtube.timedtext') }}?{{ cc.query }}" {{ 'default' if cc_default and not loop.counter }}>
@@ -93,6 +91,14 @@ var sha256=function a(b){function c(a,b){return a>>>b|a<<32-b}for(var d,e,f=Math
 	<dd>{{ '{0:,}'.format(views | int)|replace(",","'") }}
 	<dt>Visibility
 	<dd>{{ 'unlisted' if unlisted else 'public' }}
+	<dt>Resolution
+	{% for v in stream_map.muxed|selectattr("qualityLabel", "ne", "144p")|sort(attribute='height', reverse=True) %}
+		{% set active = v.url == video_url %}
+		{% set itag_url = url_for('.watch', v=video_id, t=request.args.t, itag=v.itag) %}
+		<dd><a href="{{ itag_url }}" class="{{'bold' if active else ''}}">{{v.qualityLabel}}</a>
+	{% else %}
+		<dd>Adaptive {# livestreams have no muxed formats#}
+	{% endfor %}
 	</dl>
 </details>
 <script>
-- 
2.39.3