feat(yts): cookies support (#117)

Co-authored-by: Rushaway <rushaway@hotmail.fr>

Author: Dolly132

README.md

@@ -67,6 +67,75 @@ You can find more information about available commands with
 torchlight --help
 ```
 
+## YouTube Cookies
+
+In some environments (VPS, dedicated servers, VPN/WARP containers), YouTube search may fail due to rate limits or bot protection.  
+Providing a valid YouTube cookie file can resolve this issue.
+
+Configure the cookie file in:
+
+`Sounds > CookieFile`
+
+inside your JSON configuration.
+
+---
+
+### How to Get YouTube Cookies
+
+1. Install a browser extension that exports cookies in **Netscape format (`cookies.txt`)**
+   - Chrome: **Get cookies.txt LOCALLY**
+   - Firefox: **cookies.txt**
+
+2. Open https://youtube.com and make sure you are logged in.
+
+3. Use the extension to export cookies for `youtube.com`.
+
+4. Save the file as:
+
+   ```
+   cookies.txt
+   ```
+
+5. Upload it to your server (example):
+
+   ```
+   /app/config/cookies.txt
+   ```
+
+6. Set the path in your configuration:
+
+   ```json
+   {
+     "YoutubeSearch": {
+      "parameters": {
+          "proxy": "",
+          "keywords_banned": [],
+          "cookies": "config/cookies.txt"
+      }
+     }
+   }
+   ```
+
+7. Restart the container after updating the configuration.
+
+8. It's preferred to use **Cloudflare WARP** (recommended for VPS environments), you can run it using:  
+   https://github.com/cmj2002/warp-docker  
+
+   Then configure Torchlight to use the WARP container's socks5 proxy:
+
+   ```yaml
+   "proxy": "socks5://0.0.0.0:PORT"
+   ```
+
+   This routes all requests through WARP, which helps prevent YouTube blocking server IP addresses.
+---
+
+### Notes
+
+- Keep your cookie file private (it contains session authentication data).
+- If YouTube search stops working, export fresh cookies.
+- Do **not** share your cookie file publicly.
+
 ## Docker
 ```
 version: '3.7'

config/config.json

@@ -289,7 +289,8 @@
 					"porn",
 					"algorithm",
 					"loud"
-				]
+				],
+				"cookies": "config/cookies.txt"
 			},
 			"description": "Search YouTube or play a YouTube link (e.g. !yt never gonna give you up)."
 		},

src/torchlight/Commands.py

@@ -911,10 +911,21 @@ async def _func(self, message: list[str], player: Player) -> int:
 
         real_time = get_url_real_time(url=input_url)
 
-        proxy = command_config.get("parameters", {}).get("proxy", "")
+        proxy: str = command_config.get("parameters", {}).get("proxy", "")
+        cookies: str = command_config.get("parameters", {}).get("cookies", "")
+        if not cookies:
+            self.logger.warning("Parameters/Cookies is empty, please consider adding your own cookies file")
+
+        if cookies:
+            if not os.path.isfile(cookies):
+                self.logger.warning(f"Cookies file not found: {cookies}, ignoring")
+                cookies = ""
+            elif os.path.getsize(cookies) == 0:
+                self.logger.warning(f"Cookies file is empty: {cookies}, ignoring")
+                cookies = ""
 
         try:
-            info = get_url_youtube_info(url=input_url, proxy=proxy)
+            info = get_url_youtube_info(url=input_url, proxy=proxy, cookies=cookies)
         except Exception as exc:
             self.logger.error(f"Failed to extract youtube info from: {input_url}")
             self.logger.error(exc)
@@ -925,9 +936,9 @@ async def _func(self, message: list[str], player: Player) -> int:
             return 1
 
         if "title" not in info and "url" in info:
-            info = get_url_youtube_info(url=info["url"], proxy=proxy)
+            info = get_url_youtube_info(url=info["url"], proxy=proxy, cookies=cookies)
         if info["extractor_key"] == "YoutubeSearch":
-            info = get_first_valid_entry(entries=info["entries"], proxy=proxy)
+            info = get_first_valid_entry(entries=info["entries"], proxy=proxy, cookies=cookies)
 
         title = info["title"]
         url = get_audio_format(info=info)

src/torchlight/URLInfo.py

@@ -102,7 +102,7 @@ def get_url_real_time(url: str) -> int:
 
 
 # @profile
-def get_url_youtube_info(url: str, proxy: str = "") -> dict:
+def get_url_youtube_info(url: str, proxy: str = "", cookies: str = "") -> dict:
     # https://github.com/ytdl-org/youtube-dl/blob/3e4cedf9e8cd3157df2457df7274d0c842421945/youtube_dl/YoutubeDL.py#L137-L312
     # https://github.com/yt-dlp/yt-dlp/blob/master/yt_dlp/YoutubeDL.py#L192
     ydl_opts = {
@@ -115,19 +115,24 @@ def get_url_youtube_info(url: str, proxy: str = "") -> dict:
         "simulate": True,
         "keepvideo": False,
     }
+
     if proxy:
         ydl_opts["proxy"] = proxy
+
+    if cookies:
+        ydl_opts["cookiefile"] = cookies
+
     ydl = yt_dlp.YoutubeDL(ydl_opts)
     ydl.add_default_info_extractors()
     return ydl.extract_info(url, download=False)
 
 
 # @profile
-def get_first_valid_entry(entries: list[Any], proxy: str = "") -> dict[str, Any]:
+def get_first_valid_entry(entries: list[Any], proxy: str = "", cookies: str = "") -> dict[str, Any]:
     for entry in entries:
         input_url = f"https://youtube.com/watch?v={entry['id']}"
         try:
-            info = get_url_youtube_info(url=input_url, proxy=proxy)
+            info = get_url_youtube_info(url=input_url, proxy=proxy, cookies=cookies)
             return info
         except yt_dlp.utils.DownloadError:
             logger.warning(f"Error trying to download <{input_url}>")

src/torchlight/__init__.py

@@ -1 +1 @@
-__version__ = "1.7.0"
+__version__ = "1.8.0"