1 files changed, 37 insertions, 14 deletions

diff --git a/callback.py b/callback.py
index c7cb60c..4fe0540 100644
--- a/callback.py
+++ b/callback.py
@@ -7,13 +7,23 @@ from telegram.error import BadRequest
 from telegram.ext import CallbackContext
 
 from command import *
-from dbstore.dbm_store import get_loc
+from dbstore.peewee_store import get_poi_by_fsq_id
 from foursquare.poi import OSM_ENDPOINT
 from foursquare.poi import query_poi
-# from toot import mastodon_client
 from config import BOT_SCOPE
+from dbstore.peewee_store import User, db
+import uuid
 
-mastodon_client = None
+
+def generate_uuid():
+    return str(uuid.uuid4())
+
+
+def get_mastodon_client(user_id: int):
+    with db.connection_context():
+        user = User.get(User.telegram_user_id == user_id)
+        if user.home_instance and user.access_key:
+            return Mastodon(access_token=user.access_key, api_base_url=user.home_instance)
 
 
 def generate_toot_text(poi_name, poi_locality, poi_region, poi_lat, poi_lon):
@@ -41,6 +51,7 @@ async def get_img_file_bytes(telegram_media_file):
 async def process_media_group(context: CallbackContext):
     context.job.data = cast(List[MsgDict], context.job.data)
 
+    mastodon_client = get_mastodon_client(context.user_data["user_id"])
     media_id = []
     chat_id = context.job.data[0].get("chat_id")
     for media_dict in context.job.data:
@@ -61,9 +72,6 @@ async def process_media_group(context: CallbackContext):
 
 
 async def callback_generate_fedi_login_url(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
-    # generate fedi OAuth login url
-    global mastodon_client
-
     home_instance = update.effective_message.text
     client_id, client_secret = Mastodon.create_app(
         "Checkin.bot",
@@ -71,13 +79,25 @@ async def callback_generate_fedi_login_url(update: Update, context: ContextTypes
         redirect_uris="{}{}".format(BOT_DOMAIN, FEDI_LOGIN_CALLBACK_URL),
         api_base_url=home_instance,
     )
+    print("client_id: {}".format(client_id))
+    print("client_secret: {}".format(client_secret))
+
     m = Mastodon(client_id=client_id, client_secret=client_secret, api_base_url=home_instance)
 
-    # TODO
-    # generate random string as OAuth state
+    user_id = update.effective_user.id
+    state = generate_uuid()
+
+    db.connect()
+    u = User.get_or_none(telegram_user_id=user_id)
+    if u is None:
+        u = User.create(telegram_user_id=user_id, access_key="", home_instance=home_instance,
+                        client_id=client_id, client_secret=client_secret, state=state)
+        u.save()
+    db.close()
+
     oauth_url = m.auth_request_url(redirect_uris="{}{}".format(BOT_DOMAIN, FEDI_LOGIN_CALLBACK_URL),
                                                  scopes=BOT_SCOPE,
-                                                 state=update.effective_user.id)
+                                                 state=state)
 
     await update.message.reply_text(PROMPT_FEDI_LOGIN,
                                     reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("Login", url=oauth_url)]]),
@@ -93,9 +113,9 @@ async def callback_location_sharing(update: Update, context: ContextTypes.DEFAUL
         context.user_data["latitude"] = update.message.venue.location.latitude
         context.user_data["longitude"] = update.message.venue.location.longitude
 
-        poi = get_loc(context.user_data.get("fsq_id"))
+        poi = get_poi_by_fsq_id(context.user_data.get("fsq_id"))
         content = generate_toot_text(poi["name"], poi["locality"], poi["region"], poi["latitude"], poi["longitude"])
-        status = mastodon_client.status_post(content, visibility=DEFAULT_TOOT_VISIBILITY, media_ids=[])
+        status = get_mastodon_client(update.effective_user.id).status_post(content, visibility=DEFAULT_TOOT_VISIBILITY, media_ids=[])
 
         context.user_data[KEY_TOOT_STATUS_ID] = status["id"]
         context.user_data[KEY_TOOT_STATUS_CONTENT] = content
@@ -165,12 +185,12 @@ async def callback_skip_location_keyword_search(update: Update, context: Context
 async def _process_location_selection(context: ContextTypes.DEFAULT_TYPE) -> int:
     poi_name = context.user_data.get("poi_name")
     if context.user_data.get("fsq_id") is not None:
-        poi = get_loc(context.user_data.get("fsq_id"))
+        poi = get_poi_by_fsq_id(context.user_data.get("fsq_id"))
         content = generate_toot_text(poi["name"], poi["locality"], poi["region"], poi["latitude"], poi["longitude"])
     else:
         content = generate_toot_text(poi_name, "", "", context.user_data.get("latitude"), context.user_data.get("longitude"))
 
-    status = mastodon_client.status_post(content, visibility=DEFAULT_TOOT_VISIBILITY, media_ids=[])
+    status = get_mastodon_client(context.user_data["user_id"]).status_post(content, visibility=DEFAULT_TOOT_VISIBILITY, media_ids=[])
 
     context.user_data[KEY_TOOT_STATUS_ID] = status["id"]
     context.user_data[KEY_TOOT_STATUS_CONTENT] = content
@@ -192,6 +212,7 @@ async def callback_location_confirmation(update: Update, context: ContextTypes.D
     query = update.callback_query
     await query.answer()
     context.user_data["fsq_id"] = query.data
+    context.user_data["user_id"] = update.effective_user.id
     await query.delete_message()
 
     context.user_data["chat_id"] = update.effective_chat.id
@@ -202,6 +223,7 @@ async def callback_location_confirmation(update: Update, context: ContextTypes.D
 async def callback_manual_location(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
     context.user_data["poi_name"] = update.effective_message.text
     context.user_data["chat_id"] = update.effective_chat.id
+    context.user_data["user_id"] = update.effective_user.id
 
     return await _process_location_selection(context)
 
@@ -219,7 +241,7 @@ async def callback_add_comment(update: Update, context: ContextTypes.DEFAULT_TYP
     await context.bot.delete_message(update.effective_chat.id, context.user_data.get(PROMPT_ADD_COMMENT))
 
     comment = update.effective_message.text
-    mastodon_client.status_update(id=context.user_data.get(KEY_TOOT_STATUS_ID),
+    get_mastodon_client(update.effective_user.id).status_update(id=context.user_data.get(KEY_TOOT_STATUS_ID),
                                   status=f"{comment} " + context.user_data.get(KEY_TOOT_STATUS_CONTENT))
     context.user_data[KEY_TOOT_STATUS_CONTENT] = f"{comment} " + context.user_data.get(KEY_TOOT_STATUS_CONTENT)
 
@@ -243,6 +265,7 @@ async def callback_add_media(update: Update, context: CallbackContext):
         if "not found" in str(e.message):
             pass
 
+    mastodon_client = get_mastodon_client(update.effective_user.id)
     status_id = context.user_data.get(KEY_TOOT_STATUS_ID)
     status_content = context.user_data.get(KEY_TOOT_STATUS_CONTENT)