""" Pagermaid message plugin. """ from pyrogram.enums import ChatType from pyrogram.errors import Forbidden, FloodWait from pyrogram.raw.functions.messages import ForwardMessages from pagermaid import log from pagermaid.config import Config from pagermaid.listener import listener from pagermaid.utils import lang from pagermaid.enums import Client, Message @listener(is_plugin=False, outgoing=True, command="id", description=lang("id_des")) async def userid(message: Message): """Query the UserID of the sender of the message you replied to.""" reply = message.reply_to_message text = f"Message ID: `{str(message.id)}" + "`\n\n" text += "**Chat**\nid:`" + str(message.chat.id) + "`\n" msg_from = message.chat if msg_from.type == ChatType.PRIVATE: try: text += f"first_name: `{msg_from.first_name}" + "`\n" except TypeError: text += "**死号**\n" if msg_from.last_name: text += f"last_name: `{msg_from.last_name}" + "`\n" if msg_from.username: text += f"username: @{msg_from.username}" + "\n" if msg_from.type in [ChatType.SUPERGROUP, ChatType.CHANNEL]: text += f"title: `{msg_from.title}" + "`\n" try: if msg_from.username: text += f"username: @{msg_from.username}" + "\n" except AttributeError: return await message.edit(lang("leave_not_group")) text += f"protected: `{str(msg_from.has_protected_content)}" + "`\n" if reply: text += "\n" + lang("id_hint") + "\nMessage ID: `" + str(reply.id) + "`" try: text += "\n\n**User**\nid: `" + str(reply.from_user.id) + "`" if reply.from_user.is_bot: text += f"\nis_bot: {lang('id_is_bot_yes')}" try: text += "\nfirst_name: `" + reply.from_user.first_name + "`" except TypeError: text += f"\n**{lang('id_da')}**" if reply.from_user.last_name: text += "\nlast_name: `" + reply.from_user.last_name + "`" if reply.from_user.username: text += "\nusername: @" + reply.from_user.username if reply.from_user.dc_id: text += "\ndc: `" + str(reply.from_user.dc_id) + "`" except AttributeError: pass try: text += "\n\n**Chat**\nid: `" + str(reply.sender_chat.id) + "`" text += "\ntitle: `" + reply.sender_chat.title + "`" if reply.sender_chat.username: text += "\nusername: @" + reply.sender_chat.username except AttributeError: pass if reply.forward_from_chat: text += ( "\n\n**Forward From Channel**\n" "id: `" + str(reply.forward_from_chat.id) + "`\ntitle: `" + reply.forward_from_chat.title + "`" ) if reply.forward_from_chat.username: text += "\nusername: @" + reply.forward_from_chat.username if reply.forward_from_message_id: text += "\nmessage_id: `" + str(reply.forward_from_message_id) + "`" if reply.forward_sender_name: text += "\npost_author: `" + reply.forward_sender_name + "`" elif reply.forward_from: text += ( "\n\n**Forward From User**\nid: `" + str(reply.forward_from.id) + "`" ) try: if reply.forward_from.is_bot: text += f"\nis_bot: {lang('id_is_bot_yes')}" try: text += "\nfirst_name: `" + reply.forward_from.first_name + "`" except TypeError: text += f"\n**{lang('id_da')}**" if reply.forward_from.last_name: text += "\nlast_name: `" + reply.forward_from.last_name + "`" if reply.forward_from.username: text += "\nusername: @" + reply.forward_from.username if reply.forward_from.dc_id: text += "\ndc: `" + str(reply.forward_from.dc_id) + "`" except AttributeError: pass elif reply.forward_sender_name: text += ( "\n\n**Forward From User**\nsender_name: `" + str(reply.forward_sender_name) + "`" ) await message.edit(text) @listener( is_plugin=False, outgoing=True, command="uslog", description=lang("uslog_des"), parameters="", ) async def uslog(message: Message): """Forwards a message into log group""" if Config.LOG: if message.reply_to_message: reply_msg = message.reply_to_message await reply_msg.forward(Config.LOG_ID) elif message.arguments: await log(message.arguments) else: return await message.edit(lang("arg_error")) await message.edit(lang("uslog_success")) else: await message.edit(lang("uslog_log_disable")) @listener( is_plugin=False, outgoing=True, command="log", description=lang("log_des"), parameters="", ) async def logging(message: Message): """Forwards a message into log group""" if Config.LOG: if message.reply_to_message: reply_msg = message.reply_to_message await reply_msg.forward(Config.LOG_ID) elif message.arguments: await log(message.arguments) else: return await message.edit(lang("arg_error")) await message.safe_delete() else: await message.edit(lang("uslog_log_disable")) @listener( is_plugin=False, outgoing=True, command="re", description=lang("re_des"), parameters=lang("re_parameters"), ) async def re(bot: Client, message: Message): """Forwards a message into this group""" if reply := message.reply_to_message: if message.arguments == "": num = 1 else: try: num = int(message.arguments) if num > 100: await message.edit(lang("re_too_big")) except Exception: return await message.edit(lang("re_arg_error")) await message.safe_delete() for _ in range(num): try: if not message.chat.has_protected_content: await forward_msg(bot, message.reply_to_message) else: await reply.copy( reply.chat.id, reply_to_message_id=message.reply_to_top_message_id, ) except (Forbidden, FloodWait, Exception): return else: await message.edit(lang("not_reply")) async def forward_msg(bot: Client, message: Message): message_ids = [message.id] await bot.invoke( ForwardMessages( to_peer=await bot.resolve_peer(message.chat.id), from_peer=await bot.resolve_peer(message.chat.id), id=message_ids, silent=None, random_id=[bot.rnd_id() for _ in message_ids], schedule_date=None, noforwards=None, top_msg_id=message.reply_to_top_message_id, ) )