mirror of
https://github.com/Xtao-Labs/misskey2telegram.git
synced 2024-11-25 06:47:38 +00:00
83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
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)
|
||
host = misskey_bot.tg_user.host
|
||
text, button = gen_text(host, user), gen_button(host, 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)
|