- bot: support pleroma instances

- status_update() in mastodon.py currently doesn't support content_type, thus it still fallbacks to text/plain on pleroma instances when adding comments

1 files changed, 35 insertions, 12 deletions

diff --git a/callback.py b/callback.py
index a1b4b2d..ecce2bf 100644
--- a/callback.py
+++ b/callback.py
@@ -7,8 +7,7 @@ from telegram.ext import CallbackContext
 
 from command import *
 from dbstore.peewee_store import get_poi_by_fsq_id
-from foursquare.poi import OSM_ENDPOINT
-from foursquare.poi import query_poi
+from foursquare.poi import query_poi, query_poi_by_fsq_id, OSM_ENDPOINT
 from config import BOT_SCOPE, ENCRYPT_KEY
 from dbstore.peewee_store import User, db, TOOT_VISIBILITY_PRIVATE, TOOT_VISIBILITY_PUBLIC, TOOT_VISIBILITY_UNLISTED
 import uuid
@@ -24,7 +23,10 @@ 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=decrypt(user.access_key, ENCRYPT_KEY), api_base_url=user.home_instance)
+            feature_set = "pleroma" if user.home_instance_type == "pleroma" else "mainline"
+
+            return Mastodon(access_token=decrypt(user.access_key, ENCRYPT_KEY),
+                            api_base_url=user.home_instance, feature_set=feature_set, version_check_mode='none', )
 
 
 def generate_toot_text(poi_name, poi_locality, poi_region, poi_lat, poi_lon):
@@ -94,11 +96,12 @@ async def callback_generate_fedi_login_url(update: Update, context: ContextTypes
             u.save()
 
     oauth_url = m.auth_request_url(redirect_uris="{}{}".format(BOT_DOMAIN, FEDI_LOGIN_CALLBACK_URL),
-                                                 scopes=BOT_SCOPE,
-                                                 state=state)
+                                   scopes=BOT_SCOPE,
+                                   state=state)
 
     msg = await update.message.reply_text(PROMPT_FEDI_LOGIN,
-                                          reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("Login", url=oauth_url)]]),
+                                          reply_markup=InlineKeyboardMarkup(
+                                              [[InlineKeyboardButton("Login", url=oauth_url)]]),
                                           parse_mode=ParseMode.MARKDOWN)
 
     context.user_data[PROMPT_FEDI_LOGIN] = msg.message_id
@@ -112,9 +115,18 @@ 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_poi_by_fsq_id(context.user_data.get("fsq_id"))
+        poi = query_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 = get_mastodon_client(update.effective_user.id).status_post(content, visibility=TOOT_VISIBILITY_PRIVATE, media_ids=[])
+
+        # TODO
+        # mastodon.py status_update() currently does not support content_type parameter
+        with db.connection_context():
+            u = User.get(User.telegram_user_id == update.effective_user.id)
+            content_type = "text/markdown" if u.home_instance_type == "pleroma" else "text/plain"
+        status = get_mastodon_client(update.effective_user.id).status_post(content,
+                                                                           visibility=TOOT_VISIBILITY_PRIVATE,
+                                                                           content_type=content_type,
+                                                                           media_ids=[])
 
         context.user_data[KEY_TOOT_STATUS_ID] = status["id"]
         context.user_data[KEY_TOOT_STATUS_CONTENT] = content
@@ -158,7 +170,8 @@ async def callback_location_keyword_search(update: Update, context: ContextTypes
     await context.bot.delete_message(update.effective_chat.id, context.user_data.get(PROMPT_LOCATION_KEYWORD))
     key = update.effective_message.text
 
-    keyboard = await _process_location_search(key, context.user_data.get("latitude"), context.user_data.get("longitude"))
+    keyboard = await _process_location_search(key, context.user_data.get("latitude"),
+                                              context.user_data.get("longitude"))
 
     if len(keyboard) == 0:
         msg = await update.message.reply_text(PROMPT_WAIT_LOCATION_CONFIRMATION_NO_NEARBY_POI)
@@ -188,9 +201,17 @@ async def _process_location_selection(context: ContextTypes.DEFAULT_TYPE) -> int
         poi_name = poi["name"]
         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"))
+        content = generate_toot_text(poi_name, "", "", context.user_data.get("latitude"),
+                                     context.user_data.get("longitude"))
 
-    status = get_mastodon_client(context.user_data["user_id"]).status_post(content, visibility=TOOT_VISIBILITY_PRIVATE, media_ids=[])
+    with db.connection_context():
+        u = User.get(User.telegram_user_id == context.user_data["user_id"])
+        content_type = "text/markdown" if u.home_instance_type == "pleroma" else "text/plain"
+
+    status = get_mastodon_client(context.user_data["user_id"]).status_post(content,
+                                                                           visibility=TOOT_VISIBILITY_PRIVATE,
+                                                                           content_type=content_type,
+                                                                           media_ids=[])
 
     context.user_data[KEY_TOOT_STATUS_ID] = status["id"]
     context.user_data[KEY_TOOT_STATUS_CONTENT] = content
@@ -238,11 +259,13 @@ async def _process_comment(context: ContextTypes.DEFAULT_TYPE) -> int:
 
 async def callback_add_comment(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
     context.user_data["chat_id"] = update.effective_chat.id
+    context.user_data["user_id"] = update.effective_user.id
     await context.bot.delete_message(update.effective_chat.id, context.user_data.get(PROMPT_ADD_COMMENT))
 
     comment = update.effective_message.text
     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))
+                                                                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)
 
     return await _process_comment(context)