Adds Token Refresh Route

Signed-off-by: Hassan Abouelela <[email protected]>

1 files changed, 8 insertions, 3 deletions

diff --git a/backend/discord.py b/backend/discord.py
index d6310b7..9cdd2c4 100644
--- a/backend/discord.py
+++ b/backend/discord.py
@@ -8,16 +8,21 @@ from backend.constants import (
 API_BASE_URL = "https://discord.com/api/v8"
 
 
-async def fetch_bearer_token(access_code: str) -> dict:
+async def fetch_bearer_token(code: str, *, refresh: bool) -> dict:
     async with httpx.AsyncClient() as client:
         data = {
             "client_id": OAUTH2_CLIENT_ID,
             "client_secret": OAUTH2_CLIENT_SECRET,
-            "grant_type": "authorization_code",
-            "code": access_code,
             "redirect_uri": OAUTH2_REDIRECT_URI
         }
 
+        if refresh:
+            data["grant_type"] = "refresh_token"
+            data["refresh_token"] = code
+        else:
+            data["grant_type"] = "authorization_code"
+            data["code"] = code
+
         r = await client.post(f"{API_BASE_URL}/oauth2/token", headers={
             "Content-Type": "application/x-www-form-urlencoded"
         }, data=data)