feat(env): add YTDL_COOKIE_PATH for YouTube cookie support in content retrieval

Author: qlaffont

.gitignore

@@ -76,3 +76,5 @@ logs
 docs
 data
 sqlite.db*
+
+ytdl.json

README.md

@@ -62,6 +62,7 @@ docker run -p 3000:3000 qlaffont-livechatccb \
 -e DEFAULT_DURATION='5' \ # <-- Durée par défaut si le contenu n'est pas vidéo ou audio
 -e HIDE_COMMANDS_DISABLED='false' \ # <-- Si vous souhaitez désactiver les commandes masquées, vous pouvez modifier la valeur de 'false' à 'true'
 -e API_URL='API-URL-TO-REPLACE' # <-- Remplacer par l'adresse où l'utilisateur se connectera (Ex: https://livechat.domainname.com)
+-e YTDL_COOKIE_PATH='YTDL-COOKIE-PATH-TO-REPLACE' # <-- Remplacer par le chemin du fichier cookies.json (voir https://github.com/distubejs/ytdl-core#cookies-support)
 ```
 
 OU
@@ -147,6 +148,7 @@ docker run -p 3000:3000 qlaffont-livechatccb \
 -e DEFAULT_DURATION='5' \ # <-- Default duration if content is not video or audio
 -e HIDE_COMMANDS_DISABLED='false' \ # <-- If you want to disable hided commands, you can change the value from 'false' to 'true'
 -e API_URL='API-URL-TO-REPLACE' \ # <--Replace with the endpoint where user will connect (Ex: https://livechat.domainname.com)
+-e YTDL_COOKIE_PATH='YTDL-COOKIE-PATH-TO-REPLACE' # <-- Replace with the path of the cookies.json file (see https://github.com/distubejs/ytdl-core#cookies-support)
 -e I18N='en'
 ```
 

src/services/content-utils.ts

@@ -1,3 +1,4 @@
+import fs from 'fs';
 import fetch from 'node-fetch';
 import { getVideoDurationInSeconds } from 'get-video-duration';
 import { fileTypeFromBuffer } from 'file-type';
@@ -49,7 +50,13 @@ export const getContentInformationsFromUrl = async (url: string) => {
 
   //if it is a youtube video, get the duration from the url
   if (url.includes('youtube.com') || url.includes('youtu.be')) {
-    const info = await ytdl.getInfo(url);
+    let agent;
+
+    //if file exist on root of app, use it
+    if (fs.existsSync(env.YTDL_COOKIE_PATH)) {
+      agent = ytdl.createAgent(JSON.parse(fs.readFileSync(env.YTDL_COOKIE_PATH, 'utf8')));
+    }
+    const info = await ytdl.getInfo(url, { agent });
     mediaDuration = info.videoDetails.lengthSeconds;
     mediaIsShort = info.videoDetails.isCrawlable;
     contentType = 'video/mp4';

src/services/env.ts

@@ -27,6 +27,8 @@ export const env = createEnv({
       .string()
       .default('5')
       .transform((s) => parseInt(s)),
+
+    YTDL_COOKIE_PATH: z.string(),
   },
   runtimeEnv: process.env,
 });