Automatically retrieve Apple Music media API key

1 files changed, 14 insertions, 0 deletions

diff --git a/code/bot.py b/code/bot.py
index aafeb0d..2864e85 100644
--- a/code/bot.py
+++ b/code/bot.py
@@ -6,6 +6,7 @@ import openai
 
 import utils.config as config
 from utils.command_tree import Tree
+from utils.media_api_key import get_media_api_key
 
 
 class MyBot(commands.Bot):
@@ -19,6 +20,7 @@ class MyBot(commands.Bot):
 
     async def setup_hook(self):
         get_access_token.start()
+        refresh_media_api_key.start()
         config.LOG.info("Loading cogs...")
         for ext in os.listdir("./code/cogs"):
             if ext.endswith(".py"):
@@ -59,6 +61,18 @@ async def get_access_token():
     bot.spotify_headers = {"Authorization": f"Bearer {access_token}"}
 
 
+@tasks.loop(hours=24)
+async def refresh_media_api_key():
+    media_api_key = get_media_api_key()
+    if media_api_key is not None:
+        bot.apple_headers = {
+            "Authorization": f"Bearer {media_api_key}",
+            "Origin": "https://apple.com",
+        }
+    else:
+        bot.apple_headers = None
+
+
 if __name__ == "__main__":
     config.load_config()
     bot.run(config.TOKEN)