misskey2telegram/modules/search_user.py

import contextlib
from mipac.errors import (
    InternalErrorError,
    AlreadyFollowingError,
    FolloweeIsYourselfError,
)
from pyrogram import Client, filters
from pyrogram.types import Message, CallbackQuery

from defs.search_user import search_user, gen_text, gen_button
from misskey_init import get_misskey_bot
from models.filters import notice_filter


# @xxx
@Client.on_message(filters.incoming & notice_filter & filters.regex(r"^@(.*)$"))
async def search_user_command(_: Client, message: Message):
    """
    搜索用户
    """
    username = message.matches[0].group(1)
    path = username.strip().split("@")
    username = path[0]
    host = path[1] if len(path) > 1 else None
    try:
        misskey_bot = get_misskey_bot(message.from_user.id)
        user = await search_user(misskey_bot, username, host)
        if not user:
            return await message.reply("没有找到用户", quote=True)
        text, button = gen_text(user), gen_button(user)
        if user.avatar_url:
            try:
                await message.reply_photo(
                    user.avatar_url, caption=text, reply_markup=button, quote=True
                )
            except Exception:
                await message.reply(
                    text, reply_markup=button, quote=True, disable_web_page_preview=True
                )
        else:
            await message.reply(
                text, reply_markup=button, quote=True, disable_web_page_preview=True
            )
    except Exception as e:
        return await message.reply(f"搜索用户失败：{e}", quote=True)


# follow:xxx
@Client.on_callback_query(filters.regex(r"^follow:(.*)$") & notice_filter)
async def follow_user_callback(_: Client, callback_query: CallbackQuery):
    """
    关注/取消关注用户
    """
    user_id = callback_query.matches[0].group(1)
    button = callback_query.message.reply_markup
    follow = True
    try:
        misskey_bot = get_misskey_bot(callback_query.from_user.id)
        await misskey_bot.core.api.follow.action.add(user_id)
        await callback_query.answer("关注成功", show_alert=True)
    except InternalErrorError:
        await callback_query.answer("关注申请未批准，请等待对方同意", show_alert=True)
    except AlreadyFollowingError:
        try:
            misskey_bot = get_misskey_bot(callback_query.from_user.id)
            await misskey_bot.core.api.follow.action.remove(user_id)
            await callback_query.answer("取消关注成功", show_alert=True)
            follow = False
        except Exception as e:
            await callback_query.answer("取消关注失败", show_alert=True)
            await callback_query.message.reply(f"取消关注失败：{e}", quote=True)
    except FolloweeIsYourselfError:
        await callback_query.answer("不能关注自己", show_alert=True)
    except Exception as e:
        await callback_query.answer("关注失败", show_alert=True)
        await callback_query.message.reply(f"关注失败：{e}", quote=True)
    if button:
        with contextlib.suppress(Exception):
            button.inline_keyboard[1][0].text = "➖" if follow else "➕"
            await callback_query.message.edit_reply_markup(button)