Download Speed

server/controller/download.controller.js

@@ -5,6 +5,7 @@ const { buildYtDlpArgs } = require("../helpers/buildArgs");
 const { notifyDownloadFinished } = require("../helpers/notify");
 
 const listeners = [];
+const speedListeners = [];
 
 async function downloadController(req, res) {
   try {
@@ -18,7 +19,7 @@ async function downloadController(req, res) {
     if (!options.url || !isValidUrl(options.url)) return res.status(400).send("❌ URL invalide !");
     if (options.outputFolder && !isSafePath(options.outputFolder)) return res.status(400).send("❌ Chemin de sauvegarde non autorisé.");
 
-    const filePath = await fetchDownload(options, listeners);
+    const filePath = await fetchDownload(options, listeners, speedListeners);
     notifyDownloadFinished(filePath);
     res.send("✅ Téléchargement terminé !");
     
@@ -44,4 +45,21 @@ function progressController(req, res) {
   });
 }
 
-module.exports = { downloadController, progressController };
+function speedController(req, res) {
+  res.set({
+    'Content-Type': 'text/event-stream',
+    'Cache-Control': 'no-cache',
+    'Connection': 'keep-alive',
+  });
+
+  const sendSpeed = speed => res.write(`data: ${speed}\n\n`);
+  speedListeners.push(sendSpeed);
+
+  req.on('close', () => {
+    speedListeners.splice(speedListeners.indexOf(sendSpeed), 1);
+    res.end();
+  });
+}
+
+
+module.exports = { downloadController, progressController, speedController};

server/routes/download.route.js

@@ -1,10 +1,11 @@
 const express = require("express");
 const router = express.Router();
-const { downloadController, progressController } = require("../controller/download.controller");
+const { downloadController, progressController, speedController } = require("../controller/download.controller");
 
 router.post("/", downloadController);
 
 // SSE pour la progression du téléchargement
 router.get("/progress", progressController);
+router.get("/speed", speedController);
 
-module.exports = router;
+module.exports = router;

server/services/download.services.js

@@ -5,7 +5,7 @@ const fs = require("fs");
 const { logger } = require("../logger");
 const { buildYtDlpArgs } = require("../helpers/buildArgs");
 
-function fetchDownload(options, listeners) {
+function fetchDownload(options, listeners, speedListeners) {
 
   return new Promise((resolve, reject) => {
     const outputFolder = options.outputFolder || path.join(process.env.USERPROFILE, "Downloads", "Freedom Loader");
@@ -34,12 +34,24 @@ function fetchDownload(options, listeners) {
         if (line.startsWith("[download] Destination:")) listeners.forEach(fn => fn("reset"));
         const match = line.match(/\[download\]\s+(\d+\.\d+)%/);
         if (match) listeners.forEach(fn => fn(parseFloat(match[1])));
+
+        if (line.includes("MiB/s") || line.includes("KiB/s")) {
+            const match = line.match(/\d+(\.\d+)?[KM]?iB\/s/);
+            if (match) {
+                const speed = match[0];
+                speedListeners.forEach(fn => fn(speed));
+            }
+        }
+
+
       });
     });
 
     child.stderr.on("data", data => {
       data.toString().split("\n").forEach(line => line.trim() && logger.error(`[yt-dlp stderr] ${line}`));
     });
+
+    
   });
 }