From 6a100c8cb16c36f567439523004dde9bae41213d Mon Sep 17 00:00:00 2001
From: TC <th-ch@users.noreply.github.com>
Date: Fri, 8 Jan 2021 22:29:25 +0100
Subject: [PATCH] Allow custom audio extensions in downloader

---
 plugins/downloader/youtube-dl.js | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/plugins/downloader/youtube-dl.js b/plugins/downloader/youtube-dl.js
index 890659f3..e2f57575 100644
--- a/plugins/downloader/youtube-dl.js
+++ b/plugins/downloader/youtube-dl.js
@@ -73,6 +73,7 @@ const toMP3 = async (
 	options
 ) => {
 	const safeVideoName = randomBytes(32).toString("hex");
+	const extension = options.extension || "mp3";
 
 	try {
 		if (!ffmpeg.isLoaded()) {
@@ -88,14 +89,16 @@ const toMP3 = async (
 			"-i",
 			safeVideoName,
 			...(options.ffmpegArgs || []),
-			safeVideoName + ".mp3"
+			safeVideoName + "." + extension
 		);
 
 		const folder = options.downloadFolder || downloadsFolder();
-		const filename = filenamify(videoName + ".mp3", { replacement: "_" });
+		const filename = filenamify(videoName + "." + extension, {
+			replacement: "_",
+		});
 		writeFileSync(
 			join(folder, filename),
-			ffmpeg.FS("readFile", safeVideoName + ".mp3")
+			ffmpeg.FS("readFile", safeVideoName + "." + extension)
 		);
 
 		reinit();