From 0759290306a90e5ecd224cf2726e97ddc3840ee0 Mon Sep 17 00:00:00 2001
From: Yuzu <yuzu@kirameki.cafe>
Date: Sun, 30 Jan 2022 21:35:19 +0700
Subject: [PATCH] Add YouTube Cookies

---
 src/providers/DiscordMusicPlayer.ts | 8 ++++++++
 src/providers/Environment.ts        | 4 +++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/providers/DiscordMusicPlayer.ts b/src/providers/DiscordMusicPlayer.ts
index 0d59e23..95a9701 100644
--- a/src/providers/DiscordMusicPlayer.ts
+++ b/src/providers/DiscordMusicPlayer.ts
@@ -4,9 +4,17 @@ import { AudioPlayer, VoiceConnection, createAudioPlayer, joinVoiceChannel, crea
 import { EventEmitter } from "stream";
 
 import DiscordProvider from "./Discord";
+import Environment from "./Environment";
 
 export type ValidTracks = YouTubeVideo;
 
+if (Environment.get().YOUTUBE_COOKIE_BASE64) {
+    playdl.setToken({
+        youtube: {
+            cookie: Buffer.from(Environment.get().YOUTUBE_COOKIE_BASE64, 'base64').toString()
+        }
+    })
+}
 export class Queue {
     public track: (ValidTracks)[] = [];
 }
diff --git a/src/providers/Environment.ts b/src/providers/Environment.ts
index edfee12..a9e1c11 100644
--- a/src/providers/Environment.ts
+++ b/src/providers/Environment.ts
@@ -41,6 +41,7 @@ class Environment {
         const PRIVATE_BOT = process.env.PRIVATE_BOT;
 
         const OSU_API_KEY = process.env.OSU_API_KEY;
+        const YOUTUBE_COOKIE_BASE64 = process.env.YOUTUBE_COOKIE_BASE64;
       
         return {
             NODE_ENV,
@@ -49,7 +50,8 @@ class Environment {
             DEVELOPER_IDS,
             PRIVATE_BOT,
 
-            OSU_API_KEY
+            OSU_API_KEY,
+            YOUTUBE_COOKIE_BASE64
         };
     }