fix oauth login flow

1 files changed, 20 insertions, 2 deletions

diff --git a/bot.py b/bot.py
index 3b45906..79fba5d 100644
--- a/bot.py
+++ b/bot.py
@@ -56,8 +56,11 @@ from config import (
     LOCATION_CONFIRMATION,
     ADD_MEDIA,
     ADD_COMMENT,
-    BOT_TOKEN
+    BOT_TOKEN,
+    BOT_SCOPE
 )
+from mastodon import Mastodon
+
 
 # Enable logging
 logging.basicConfig(
@@ -85,7 +88,22 @@ class FediLoginCallbackContext(CallbackContext[ExtBot, dict, dict, dict]):
 
 
 async def process_oauth_login_callback(update: FediLoginCallbackUpdate, context: FediLoginCallbackContext) -> None:
-    text = "Login success, your OAuth code is: {}".format(update.code)
+    # query client_id and client_secret from in memory database
+    client_id = ""
+    client_secret = ""
+    home_instance = ""
+
+    mastodon_client = Mastodon(client_id=client_id, client_secret=client_secret, api_base_url=home_instance)
+    access_token = mastodon_client.log_in(
+        code=update.code,
+        redirect_uri="{}{}".format(BOT_DOMAIN, FEDI_LOGIN_CALLBACK_URL),
+        scopes=BOT_SCOPE
+    )
+
+    # TODO
+    # save access_token to database
+
+    text = "You have successfully logged in to your Mastodon account!"
     await context.bot.send_message(chat_id=update.state, text=text)