From f3c37c24987f193df792bb1fd5aaae25542dc090 Mon Sep 17 00:00:00 2001
From: MasterAcnolo <68693319+MasterAcnolo@users.noreply.github.com>
Date: Sun, 12 Apr 2026 18:13:38 +0200
Subject: [PATCH] Feat: Add createPlaylistFolders feature to organize downloads
 into dedicated folders with toggle settings

---
 app/ipcHandlers.js                       |  3 +-
 config/config.default.json               |  1 +
 public/index.html                        |  8 +++
 public/script/fetchinfo.js               | 15 +++++
 server/controller/download.controller.js |  1 +
 server/services/download.services.js     | 85 +++++++++++++++++++++++-
 6 files changed, 111 insertions(+), 2 deletions(-)

diff --git a/app/ipcHandlers.js b/app/ipcHandlers.js
index f4a3f4c..e2f8364 100644
--- a/app/ipcHandlers.js
+++ b/app/ipcHandlers.js
@@ -17,7 +17,8 @@ const FEATURE_WHITELIST = new Set([
   "verboseLogs",
   "autoDownloadPlaylist",
   "customCodec",
-  "theme"
+  "theme",
+  "createPlaylistFolders"
 ]);
 
 const configFolderPath = featuresPath;
diff --git a/config/config.default.json b/config/config.default.json
index 328a2b4..7b68c4c 100644
--- a/config/config.default.json
+++ b/config/config.default.json
@@ -7,6 +7,7 @@
   "addMetadata": true,
   "verboseLogs": false,
   "autoDownloadPlaylist": true,
+  "createPlaylistFolders": true,
   "customCodec": "h264",
   "logSystem": true,
   "outputTitleCheck": true,
diff --git a/public/index.html b/public/index.html
index 181659d..82becc3 100644
--- a/public/index.html
+++ b/public/index.html
@@ -88,6 +88,14 @@
                     </div>
                 </div>
 
+                <div class="setting-item">
+                    <input type="checkbox" data-key="createPlaylistFolders" id="createPlaylistFolders">
+                    <div class="setting-info">
+                    <span class="setting-title">Create Playlist Folders</span>
+                    <small>Create a dedicated folder for each playlist with its name.</small>
+                    </div>
+                </div>
+
                 <!-- <div class="setting-item">
                     <input type="checkbox" data-key="notifySystem" id="notifySystem">
                     <div class="setting-info">
diff --git a/public/script/fetchinfo.js b/public/script/fetchinfo.js
index 1fc1f75..e992282 100644
--- a/public/script/fetchinfo.js
+++ b/public/script/fetchinfo.js
@@ -33,6 +33,17 @@ async function init() {
     const urlInput = document.getElementById("UrlInput");
     const infoDiv = document.getElementById("videoInfo");
     const loaderBox = document.getElementById("loaderBox");
+    const form = document.getElementById("downloadForm");
+
+    // Input caché pour stocker le titre de la playlist
+    let playlistTitleInput = document.getElementById("playlistTitle");
+    if (!playlistTitleInput) {
+      playlistTitleInput = document.createElement("input");
+      playlistTitleInput.type = "hidden";
+      playlistTitleInput.name = "playlistTitle";
+      playlistTitleInput.id = "playlistTitle";
+      form.appendChild(playlistTitleInput);
+    }
 
     let lastFetchedUrl = "";
 
@@ -47,6 +58,7 @@ async function init() {
       if (!url || url.length < 2) {
         infoDiv.innerHTML = "";
         infoDiv.classList.remove("visible", "playlist-mode");
+        playlistTitleInput.value = "";
         lastFetchedUrl = "";
         return;
       }
@@ -70,12 +82,14 @@ async function init() {
         `;
         infoDiv.classList.add("visible");
         infoDiv.classList.remove("playlist-mode");
+        playlistTitleInput.value = "";
         loaderBox.style.display = "none";
         return;
       }
 
       // ---------- PLAYLIST ----------
       if (data.type === "playlist") {
+        playlistTitleInput.value = data.title;
         infoDiv.classList.add("playlist-mode");
         infoDiv.innerHTML = `
           <div class="playlist-header">
@@ -194,6 +208,7 @@ async function init() {
       }
 
       infoDiv.classList.remove("playlist-mode");
+      playlistTitleInput.value = "";
 
       // ---------- VIDEO NORMALE ----------
       const durationStr = data.duration
diff --git a/server/controller/download.controller.js b/server/controller/download.controller.js
index dcc2dc2..b87ab33 100644
--- a/server/controller/download.controller.js
+++ b/server/controller/download.controller.js
@@ -14,6 +14,7 @@ async function downloadController(req, res) {
       audioOnly: req.body.audioOnly === "1",
       quality: req.body.quality || "best",
       outputFolder: req.body.savePath || undefined,
+      playlistTitle: req.body.playlistTitle || undefined,
     };
 
     if (!options.url || !isValidUrl(options.url)) return res.status(400).send("Invalid URL !");
diff --git a/server/services/download.services.js b/server/services/download.services.js
index a258282..b36e6f1 100644
--- a/server/services/download.services.js
+++ b/server/services/download.services.js
@@ -6,13 +6,80 @@ const { buildYtDlpArgs } = require("../helpers/buildArgs.helpers");
 const notify = require("../helpers/notify.helpers");
 const path = require("path");
 const { isSafePath } = require("../helpers/validation.helpers");
+const { configFeatures } = require("../../config.js");
 
 let currentDownloadProcess = null;
 
+// Détecte si une URL est une playlist
+function isPlaylistUrl(url) {
+  return url.includes("?list=") || url.includes("&list=");
+}
+
+function createPlaylistFolder(basePath, playlistTitle) {
+  const sanitizedTitle = playlistTitle
+    .replace(/[<>:"|?*]/g, '') 
+    .replace(/\s+/g, ' ') 
+    .trim()
+    .substring(0, 200);
+
+  const playlistPath = path.join(basePath, sanitizedTitle);
+  
+  try {
+    if (!fs.existsSync(playlistPath)) {
+      fs.mkdirSync(playlistPath, { recursive: true });
+      logger.info(`Playlist folder created: ${playlistPath}`);
+      return playlistPath;
+    }
+    
+    let counter = 1;
+    let newPath;
+    
+    while (counter <= 1000) {
+      newPath = path.join(basePath, `${sanitizedTitle} ${counter}`);
+      if (!fs.existsSync(newPath)) {
+        fs.mkdirSync(newPath, { recursive: true });
+        logger.info(`Playlist folder created with increment: ${newPath}`);
+        return newPath;
+      }
+      counter++;
+    }
+    
+    logger.error(`Could not find available playlist folder after 1000 attempts`);
+    throw new Error("Unable to create playlist folder");
+    
+  } catch (err) {
+    logger.warn(`Failed to create playlist folder with title "${sanitizedTitle}": ${err.message}`);
+    
+    // Fallback: create folder "Untitled Playlist X"
+    let counter = 1;
+    let fallbackPath;
+    
+    while (counter <= 1000) {
+      fallbackPath = path.join(basePath, `Untitled Playlist ${counter}`);
+      if (!fs.existsSync(fallbackPath)) {
+        try {
+          fs.mkdirSync(fallbackPath, { recursive: true });
+          logger.info(`Fallback playlist folder created: ${fallbackPath}`);
+          return fallbackPath;
+        } catch (mkErr) {
+          counter++;
+          continue;
+        }
+      }
+      counter++;
+    }
+    
+    logger.error(`Could not create playlist folder after 1000 attempts`);
+    throw new Error("Unable to create playlist folder");
+  }
+}
+
 function fetchDownload(options, listeners, speedListeners, stageListeners) {
 
   return new Promise((resolve, reject) => {
-    const outputFolder = options.outputFolder || defaultDownloadFolder;
+    logger.info(`CONFIG createPlaylistFolders: ${configFeatures.createPlaylistFolders}`);
+    
+    let outputFolder = options.outputFolder || defaultDownloadFolder;
     
     // Normalize path and validate it's safe (within Users folder)
     let safeOutputFolder = path.resolve(outputFolder);
@@ -31,6 +98,22 @@ function fetchDownload(options, listeners, speedListeners, stageListeners) {
       return reject(new Error(`Unable to create download folder: ${err.message}`));
     }
 
+    // Détecte si c'est une playlist et crée un dossier approprié
+    const isPlaylist = options.playlistTitle || isPlaylistUrl(options.url);
+    
+    if (isPlaylist && configFeatures.createPlaylistFolders) {
+      try {
+        const playlistName = options.playlistTitle || "Untitled Playlist";
+        safeOutputFolder = createPlaylistFolder(safeOutputFolder, playlistName);
+        logger.info(`Playlist detected, using folder: ${safeOutputFolder}`);
+      } catch (err) {
+        logger.error(`Failed to create playlist folder: ${err.message}`);
+        return reject(err);
+      }
+    } else if (isPlaylist && !configFeatures.createPlaylistFolders) {
+      logger.info(`Playlist detected but createPlaylistFolders is disabled, using base folder`);
+    }
+
     const args = buildYtDlpArgs({ ...options, outputFolder: safeOutputFolder });
     logger.info(`[yt-dlp args] ${args.join(" ")}`);