From f085006f1da0ac431caf391e721f0ee140d1bfb8 Mon Sep 17 00:00:00 2001
From: RblSb <msrblsb@gmail.com>
Date: Sat, 15 Feb 2025 06:16:13 +0300
Subject: Youtube cache cookies support

closes #59
---
 src/server/Cache.hx | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'src/server/Cache.hx')

diff --git a/src/server/Cache.hx b/src/server/Cache.hx
index cfd5d88..88899a6 100644
--- a/src/server/Cache.hx
+++ b/src/server/Cache.hx
@@ -1,5 +1,6 @@
 package server;
 
+import haxe.Json;
 import haxe.io.Path;
 import js.lib.Promise;
 import js.node.Buffer;
@@ -7,6 +8,7 @@ import js.node.ChildProcess;
 import js.node.Fs.Fs;
 import js.node.stream.Readable;
 import sys.FileSystem;
+import sys.io.File;
 import utils.YoutubeUtils;
 
 class Cache {
@@ -119,7 +121,14 @@ class Cache {
 				data: outName
 			}
 		});
-		final promise:Promise<YouTubeVideoInfo> = ytdl.getInfo(url);
+		var agent:Any = null;
+		final cookiesPath = '${main.userDir}/cookies.json';
+		if (FileSystem.exists(cookiesPath)) {
+			agent = ytdl.createAgent(Json.parse(File.getContent(cookiesPath)));
+		}
+		final promise:Promise<YouTubeVideoInfo> = ytdl.getInfo(url, {
+			agent: agent,
+		});
 		promise.then(info -> {
 			trace('Get info with ${info.formats.length} formats');
 			final audioFormat:YoutubeVideoFormat = try {
@@ -140,6 +149,7 @@ class Cache {
 
 			final dlVideo:Readable<Dynamic> = ytdl(url, {
 				format: videoFormat,
+				agent: agent,
 			});
 			dlVideo.pipe(Fs.createWriteStream('$cacheDir/$inVideoName'));
 			dlVideo.on("error", err -> {
@@ -150,6 +160,7 @@ class Cache {
 
 			final dlAudio:Readable<Dynamic> = ytdl(url, {
 				format: audioFormat,
+				agent: agent,
 			});
 			dlAudio.pipe(Fs.createWriteStream('$cacheDir/$inAudioName'));
 			dlAudio.on("error", err -> {
-- 
cgit v1.2.3