misskey2telegram/modules/react.py

from mipac.errors import NoSuchNoteError, AlreadyReactedError
from pyrogram import Client, filters
from pyrogram.types import CallbackQuery

from misskey_init import get_misskey_bot
from models.filters import timeline_filter


# react:note_id:reaction
@Client.on_callback_query(filters.regex(r"^react:(\w+):(\w+)$") & timeline_filter)
async def renote_callback(_: Client, callback_query: CallbackQuery):
    note_id = callback_query.matches[0].group(1)
    match callback_query.matches[0].group(2):
        case "love":
            reaction = "❤️"
        case _:
            reaction = "❤️"
    try:
        misskey_bot = get_misskey_bot(callback_query.from_user.id)
        await misskey_bot.core.api.note.reaction.action.add(
            reaction=reaction,
            note_id=note_id,
        )
    except NoSuchNoteError:
        await callback_query.answer("该嘟文不存在", show_alert=True)
    except AlreadyReactedError:
        try:
            misskey_bot = get_misskey_bot(callback_query.from_user.id)
            await misskey_bot.core.api.note.reaction.action.remove(
                note_id=note_id,
            )
            await callback_query.answer("取消表态成功", show_alert=True)
        except Exception as e:
            if callback_query.message:
                await callback_query.message.reply(f"取消表态失败：{e}", quote=True)
            await callback_query.answer("取消表态失败", show_alert=True)
        return
    except Exception as e:
        if callback_query.message:
            await callback_query.message.reply(f"表态失败：{e}", quote=True)
        await callback_query.answer("表态失败", show_alert=True)
    else:
        await callback_query.answer("表态成功", show_alert=True)