Misc fixes on yt downloader

commands/youtube.js

@@ -15,9 +15,9 @@ function getYtDlpPath() {
   const platform = os.platform();
   if (platform === 'linux') {
     return 'yt-dlp';
-  }l
+  }
   return ytDlpPaths[platform] || 'yt-dlp';
-};
+}
 
 async function downloadFromYoutube(command, args) {
   return new Promise((resolve, reject) => {
@@ -29,7 +29,7 @@ async function downloadFromYoutube(command, args) {
       }
     });
   });
-};
+}
 
 async function getApproxSize(command, videoUrl) {
   const args = [videoUrl, '--compat-opt', 'manifest-filesize-approx', '-O', 'filesize_approx'];
@@ -50,11 +50,19 @@ async function getApproxSize(command, videoUrl) {
       }
     });
   });
-};
+}
+
+function timeoutPromise(timeout) {
+  return new Promise((_, reject) => {
+    setTimeout(() => {
+      reject(new Error('Timeout: Check took too long'));
+    }, timeout);
+  });
+}
 
 module.exports = (bot) => {
   bot.command(['yt', 'ytdl'], spamwatchMiddleware, async (ctx) => {
-    const strings = getStrings(ctx.from.language_code);
+    const Strings = getStrings(ctx.from.language_code);
     const ytDlpPath = getYtDlpPath();
     const userId = ctx.from.id;
     const videoUrl = ctx.message.text.split(' ').slice(1).join(' ');
@@ -63,20 +71,28 @@ module.exports = (bot) => {
     const cmdArgs = "--max-filesize 2G --no-playlist --merge-output-format mp4 -o";
     const dlpCommand = ytDlpPath;
 
-    const downloadingMessage = await ctx.reply(strings.ytDownloading, {
+    const downloadingMessage = await ctx.reply(Strings.ytCheckingSize, {
       parse_mode: 'Markdown',
       reply_to_message_id: ctx.message.message_id,
     });
 
+    if (fs.existsSync(ytDlpPath)) {
       try {
-      const approxSizeInMB = await getApproxSize(ytDlpPath, videoUrl);
-      let videoFormat = "";
+        const approxSizeInMB = await Promise.race([
+          getApproxSize(ytDlpPath, videoUrl),
+          timeoutPromise(5000)
+        ]);
 
-      if (approxSizeInMB >= 50) {
-        videoFormat = `-f best`;
-      } else {
-        videoFormat = "-f bestvideo+bestaudio";
-      }
+        let videoFormat = approxSizeInMB >= 50 ? `-f best` : "-f bestvideo+bestaudio";
+
+        await ctx.telegram.editMessageText(
+          ctx.chat.id,
+          downloadingMessage.message_id,
+          null,
+          Strings.ytDownloading, {
+          parse_mode: 'Markdown',
+          reply_to_message_id: ctx.message.message_id,
+        });
 
         const dlpArgs = [videoUrl, videoFormat, ...cmdArgs.split(' '), mp4File];
         await downloadFromYoutube(dlpCommand, dlpArgs);
@@ -85,13 +101,13 @@ module.exports = (bot) => {
           ctx.chat.id,
           downloadingMessage.message_id,
           null,
-        strings.ytUploading, {
+          Strings.ytUploading, {
           parse_mode: 'Markdown',
           reply_to_message_id: ctx.message.message_id,
         });
 
         if (fs.existsSync(mp4File)) {
-        const message = strings.ytUploadDesc
+          const message = Strings.ytUploadDesc
             .replace("{userId}", userId)
             .replace("{userName}", ctx.from.first_name);
 
@@ -102,18 +118,20 @@ module.exports = (bot) => {
               parse_mode: 'Markdown',
             });
 
+            if (approxSizeInMB >= 50) {
               await ctx.telegram.editMessageText(
                 ctx.chat.id,
                 downloadingMessage.message_id,
                 null,
-            strings.ytUploadLimit2, {
+                Strings.ytUploadLimit2, {
                 parse_mode: 'Markdown',
                 reply_to_message_id: ctx.message.message_id,
               });
+            }
 
             fs.unlinkSync(mp4File);
           } catch (error) {
-          await ctx.reply(`\`${error}\``, {
+            await ctx.reply(`\`${error.message}\``, {
               parse_mode: 'Markdown',
               reply_to_message_id: ctx.message.message_id,
             });
@@ -123,15 +141,17 @@ module.exports = (bot) => {
         }
       } catch (error) {
         fs.unlinkSync(mp4File);
-      let errStatus = "";
+        let errStatus;
 
-      if (error == "Error: 413: Request Entity Too Large") {
+        if (error.message === "Error: 413: Request Entity Too Large") {
           errStatus = Strings.ytUploadLimit;
         } else {
-        errStatus = error.error ? error.error.message : 'Unknown error';
+          errStatus = error.message === 'Timeout: Check took too long'
+            ? 'The check for video size timed out.'
+            : error.error ? error.error.message : 'Unknown error';
         }
 
-      const message = strings.ytDownloadErr
+        const message = Strings.ytDownloadErr
           .replace("{err}", errStatus)
           .replace("{userName}", ctx.from.first_name);
 
@@ -140,5 +160,15 @@ module.exports = (bot) => {
           reply_to_message_id: ctx.message.message_id,
         });
       }
+    } else {
+      await ctx.telegram.editMessageText(
+        ctx.chat.id,
+        downloadingMessage.message_id,
+        null,
+        Strings.ytFileErr, {
+        parse_mode: 'Markdown',
+        reply_to_message_id: ctx.message.message_id,
+      });
+    };
   });
 };

locales/english.json

@@ -61,6 +61,8 @@
   "maInvalidModule": "Please provide a valid module ID from The Mod Archive.\nExample: `/modarchive 81574`",
   "maDownloadError": "Error downloading the file. Check the module ID and try again.",
   "ytDownloading": "*Downloading video...*",
+  "ytFileErr": "*It seems that the yt-dlp executable does not exist on our server...\n\nIn that case, the problem is on our end! Please wait until we have noticed and solved the problem.*",
+  "ytCheckingSize": "*Checking if the video exceeds the 50MB limit...*",
   "ytUploading": "*Uploading video...*",
   "ytUploadDesc": "*[{userName}](tg://user?id={userId}), there is your downloaded video.*",
   "ytDownloadErr": "*Error during YT video download:*\n\n`{err}`",

locales/portuguese.json

@@ -61,6 +61,8 @@
   "maInvalidModule": "Por favor, forneça um ID de módulo válido do The Mod Archive.\nExemplo: `/modarchive 81574`",
   "maDownloadError": "Erro ao baixar o arquivo. Verifique o ID do módulo e tente novamente.",
   "ytDownloading": "*Baixando vídeo...*",
+  "ytCheckingSize": "Verificando se o vídeo excede o limite de 50 MB...",
+  "ytFileErr": "*Parece que o executável do yt-dlp não existe no nosso servidor...\n\nNesse caso, o problema está no nosso lado! Aguarde até que tenhamos notado e resolvido o problema.*",
   "ytUploading": "*Enviando video...*",
   "ytUploadDesc": "*[{userName}](tg://user?id={userId}), aqui está o seu vídeo baixado.*",
   "ytDownloadErr": "*Erro durante o download do vídeo do YT:*\n\n`{err}`",