From 304c65d303439973977757ad6106759d9f0ba9f5 Mon Sep 17 00:00:00 2001
From: Aevann1 <randomname42029@gmail.com>
Date: Thu, 6 Oct 2022 06:54:05 +0200
Subject: [PATCH] convert webm to mp4

---
 files/helpers/media.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/files/helpers/media.py b/files/helpers/media.py
index 5a5af305f..6cd35125d 100644
--- a/files/helpers/media.py
+++ b/files/helpers/media.py
@@ -47,6 +47,12 @@ def process_audio(file):
 	return f'{SITE_FULL}{name}'
 
 
+def webm_to_mp4(old, new):
+	subprocess.run(["ffmpeg", "-y", "-loglevel", "warning", "-i", old, "-map_metadata", "-1", new, "-threads:v", "1"], check=True, stderr=subprocess.STDOUT)
+	os.remove(old)
+	requests.post(f'https://api.cloudflare.com/client/v4/zones/{CF_ZONE}/purge_cache', headers=CF_HEADERS, 
+		data=f'{{"files": ["{SITE_FULL}{new}"]}}', timeout=5)
+
 def process_video(file):
 	old = f'/videos/{time.time()}'.replace('.','')
 	file.save(old)
@@ -61,8 +67,14 @@ def process_video(file):
 		extension = 'mp4'
 	new = old + '.' + extension
 
-	subprocess.run(["ffmpeg", "-y", "-loglevel", "warning", "-i", old, "-map_metadata", "-1", "-c:v", "copy", "-c:a", "copy", new], check=True)
-	os.remove(old)
+	if extension == 'webm':
+		new = new.replace('.webm', '.mp4')
+		file.save(new)
+		gevent.spawn(webm_to_mp4, old, new)
+	else:
+		subprocess.run(["ffmpeg", "-y", "-loglevel", "warning", "-i", old, "-map_metadata", "-1", "-c:v", "copy", "-c:a", "copy", new], check=True)
+		os.remove(old)
+
 	return f'{SITE_FULL}{new}'