add support for passing cookies

Author: LumePart

docker-compose.yaml

@@ -8,6 +8,8 @@ services:
       - /path/to/musiclibrary/explo:/data/ # has to be in the same path you have your music system pointed to (it's recommended to put explo under a subfolder)
       # - /path/to/slskd/downloads:/slskd/ # if using slskd and MIGRATE_DOWNLOADS is set to true in .env
       # - $PLAYLIST_DIR:$PLAYLIST_DIR # for MPD. Both paths should be as defined in .env (e.g /my/playlists/:/my/playlists/)
+      
+      # - /path/to/cookies.txt:/opt/explo/cookies.txt # Path to optional cookies file (for yt-dlp)
     environment:
       - TZ=UTC # Change this to the timezone set in ListenBrainz (default is UTC)
 

sample.env

@@ -45,6 +45,8 @@ YOUTUBE_API_KEY=
 # FFMPEG_PATH=
 # Custom path to yt-dlp binary (default: defined in $PATH)
 # YTDLP_PATH=
+# Path to (optional) cookies file (default ./cookies.txt)
+# COOKIES_PATH=./cookies.txt
 # Comma-separated (without spaces) keywords to exclude from YouTube results (default: live,remix,instrumental,extended,clean,acapella)
 # FILTER_LIST=live,remix,instrumental,extended
 

src/config/config.go

@@ -87,6 +87,7 @@ type Youtube struct {
 	APIKey string `env:"YOUTUBE_API_KEY"`
 	FfmpegPath string `env:"FFMPEG_PATH"`
 	YtdlpPath string `env:"YTDLP_PATH"`
+	CookiesPath string `env:"COOKIES_PATH" env-default:"./cookies.txt"`
 	Filters Filters
 }
 

src/downloader/youtube.go

@@ -144,12 +144,16 @@ func getTopic(cfg cfg.Youtube, videos Videos, track models.Track) string { // ge
 }
 
 func getVideo(ctx context.Context, c Youtube, videoID string) (*goutubedl.DownloadResult, error) { // gets video stream using yt-dlp
-
 	if c.Cfg.YtdlpPath != "" {
 		goutubedl.Path = c.Cfg.YtdlpPath
 	}
 
-	result, err := goutubedl.New(ctx, videoID, goutubedl.Options{})
+	var opts goutubedl.Options
+	if _, err := os.Stat(c.Cfg.CookiesPath); err == nil {
+		opts.Cookies = c.Cfg.CookiesPath
+	}
+
+	result, err := goutubedl.New(ctx, videoID, opts)
 	if err != nil {
 		return nil, fmt.Errorf("could not create URL for video download (ID: %s): %s", videoID, err.Error())
 	}