From d33eebb7bb625d1e7ac5d03ddec01633943bb2b6 Mon Sep 17 00:00:00 2001
From: Aevann <randomname42029@gmail.com>
Date: Sat, 2 Mar 2024 22:03:21 +0200
Subject: [PATCH] sync nonlive youtube orgy and make it end automatically

---
 files/classes/orgy.py     | 5 +++++
 files/routes/admin.py     | 7 +++++++
 files/templates/orgy.html | 2 +-
 requirements.txt          | 1 +
 4 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/files/classes/orgy.py b/files/classes/orgy.py
index 80bee5d2f..1ba4b5eae 100644
--- a/files/classes/orgy.py
+++ b/files/classes/orgy.py
@@ -36,6 +36,11 @@ class Orgy(Base):
 			t += 303
 		return t
 
+	@property
+	@lazy
+	def seconds_since_starts(self):
+		return int(time.time() - self.start_utc)
+
 def get_running_orgy(v):
 	if not (v and v.allowed_in_chat): return None
 
diff --git a/files/routes/admin.py b/files/routes/admin.py
index e745a788c..1e1d7b04d 100644
--- a/files/routes/admin.py
+++ b/files/routes/admin.py
@@ -3,6 +3,7 @@ from math import floor
 import os
 import ffmpeg
 import random
+import isodate
 
 from sqlalchemy.orm import load_only
 
@@ -1946,6 +1947,12 @@ def schedule_orgy(v):
 	if bare_youtube_regex.match(normalized_link):
 		orgy_type = 'youtube'
 		data, _ = get_youtube_id_and_t(normalized_link)
+		if YOUTUBE_KEY != DEFAULT_CONFIG_VALUE:
+			req = requests.get(f"https://www.googleapis.com/youtube/v3/videos?id={data}&key={YOUTUBE_KEY}&part=contentDetails", headers=HEADERS, timeout=5).json()
+			duration = req['items'][0]['contentDetails']['duration']
+			if duration != 'P0D':
+				duration = isodate.parse_duration(duration).total_seconds()
+				end_utc = int(start_utc + duration)
 	elif rumble_regex.match(normalized_link):
 		orgy_type = 'rumble'
 		data = normalized_link
diff --git a/files/templates/orgy.html b/files/templates/orgy.html
index c0dff1f6d..1fe1efb25 100644
--- a/files/templates/orgy.html
+++ b/files/templates/orgy.html
@@ -17,7 +17,7 @@
 			<div>
 				<p id="orgy-file-container" class="resizable">
 					{% if orgy.type == 'youtube' %}
-						<iframe style="border: none" sandbox="allow-scripts allow-same-origin allow-popups allow-presentation allow-popups-to-escape-sandbox" title="Play" allow="accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture" allowfullscreen src="https://cdpn.io/pen/debug/NWeVNRj?v={{orgy.data}}&autoplay=1&modestbranding=1"></iframe>
+						<iframe style="border: none" sandbox="allow-scripts allow-same-origin allow-popups allow-presentation allow-popups-to-escape-sandbox" title="Play" allow="accelerometer; autoplay; encrypted-media; gyroscope; picture-in-picture" allowfullscreen src="https://cdpn.io/pen/debug/NWeVNRj?v={{orgy.data}}&autoplay=1&modestbranding=1{% if orgy.end_utc %}&start={{orgy.seconds_since_starts}}{% endif %}"></iframe>
 					{% elif orgy.type == 'rumble' %}
 						<iframe src="{{orgy.data}}" frameborder="0" allowfullscreen></iframe>
 					{% elif orgy.type == 'twitch' %}
diff --git a/requirements.txt b/requirements.txt
index df49441fb..a18e01e29 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -15,6 +15,7 @@ greenlet
 gunicorn
 imagehash
 inotify
+isodate
 lxml
 mistletoe
 matplotlib