MissKatyPyro/misskaty/plugins/ubot_plugin.py

"""
 * @author        yasir <yasiramunandar@gmail.com>
 * @date          2022-12-01 09:12:27
 * @lastModified  2022-12-01 09:32:31
 * @projectName   MissKatyPyro
 * Copyright @YasirPedia All rights reserved
"""
# Code in this plugin to learn basic userbot in pyrogram
import os
from datetime import datetime

from pyrogram import enums, filters
from pyrogram.raw import functions
from pyrogram.types import ChatEventFilter, InlineKeyboardButton, InlineKeyboardMarkup, Message

from misskaty import app, user

f = filters.chat([])


@user.on_message(f)
async def auto_read(_, message: Message):
    await user.read_history(message.chat.id)
    await message.continue_propagation()


@user.on_message(filters.command("autoscroll", "!") & filters.me)
async def add_keep(_, message: Message):
    if message.chat.id in f:
        f.remove(message.chat.id)
        await message.edit("Autoscroll dimatikan")
    else:
        f.add(message.chat.id)
        await message.edit("Autoscroll diaktifkan, semua chat akan otomatis terbaca")


# @user.on_deleted_messages(filters.chat([-1001455886928, -1001255283935]))
async def del_msg(client, message):
    async for a in user.get_chat_event_log(message[0].chat.id, limit=1, filters=ChatEventFilter(deleted_messages=True)):
        try:
            ustat = (await user.get_chat_member(message[0].chat.id, a.deleted_message.from_user.id)).status
        except:
            ustat = enums.ChatMemberStatus.MEMBER
        if ustat in [enums.ChatMemberStatus.ADMINISTRATOR, enums.ChatMemberStatus.OWNER] or a.deleted_message.from_user.is_bot:
            return
        if a.user.id == a.deleted_message.from_user.id:
            if a.deleted_message.text:
                await app.send_message(
                    a.deleted_message.chat.id,
                    f"#DELETED_MESSAGE\n\n<a href='tg://user?id={a.deleted_message.from_user.id}'>{a.deleted_message.from_user.first_name}</a> menghapus pesannya 🧐.\n<b>Pesan:</b> {a.deleted_message.text}",
                )
            elif a.deleted_message.video:
                await app.send_message(
                    a.deleted_message.chat.id,
                    f"#DELETED_MESSAGE\n\n<a href='tg://user?id={a.deleted_message.from_user.id}'>{a.deleted_message.from_user.first_name}</a> menghapus pesannya 🧐.\n<b>Nama file:</b> {a.deleted_message.video.file_name}",
                )


# @user.on_edited_message(filters.text & filters.chat(-1001455886928))
async def edit_msg(client, message):
    try:
        ustat = (await user.get_chat_member(message.chat.id, message.from_user.id)).status
    except:
        ustat = enums.ChatMemberStatus.MEMBER
    if message.from_user.is_bot or ustat in [
        enums.ChatMemberStatus.ADMINISTRATOR,
        enums.ChatMemberStatus.OWNER,
    ]:
        return
    async for a in user.get_chat_event_log(message.chat.id, limit=1, filters=ChatEventFilter(edited_messages=True)):
        if a.old_message.text.startswith(("/mirror", "/leech", "/unzipmirror", "/unzipleech")):
            await app.send_message(
                message.chat.id,
                f"#EDITED_MESSAGE\n\n<a href='tg://user?id={a.user.id}'>{a.user.first_name}</a> mengedit pesannya 🧐.\n<b>Pesan:</b> {a.old_message.text}",
            )


@user.on_message(filters.private & ~filters.bot & ~filters.me & filters.text)
async def message_pm(client, message):
    await app.send_message(
        617426792,
        f"Ada pesan baru dari {message.from_user.mention}\n\n<b>Pesan: </b>{message.text}",
    )


@user.on_message(~filters.bot & filters.group & filters.mentioned)
async def mentioned(client, message):
    if message.sender_chat:
        return
    cid = message.chat.id
    pesan = message.text or message.caption
    await app.send_message(
        617426792,
        f"{message.from_user.mention} mention kamu di {message.chat.title}\n\n<b>Pesan:</b> {pesan}",
        reply_markup=InlineKeyboardMarkup(
            [
                [
                    InlineKeyboardButton(
                        text="💬 Lihat Pesan",
                        url=f"https://t.me/c/{str(cid)[4:]}/{message.id}",
                    )
                ]
            ]
        ),
    )


@user.on_message(filters.command("joindate", "!") & filters.me)
async def join_date(app, message: Message):
    members = []
    async for m in app.iter_chat_members(message.chat.id):
        members.append(
            (
                m.user.first_name,
                m.joined_date or (await app.get_messages(message.chat.id, 1)).date,
            )
        )
    members.sort(key=lambda member: member[1])

    with open("joined_date.txt", "w", encoding="utf8") as f:
        f.write("Join Date      First Name\n")
        for member in members:
            f.write(str(datetime.fromtimestamp(member[1]).strftime("%y-%m-%d %H:%M")) + f" {member[0]}\n")

    await user.send_document(message.chat.id, "joined_date.txt")
    os.remove("joined_date.txt")


@user.on_message(filters.command("memberstats", "!") & filters.me)
async def memberstats(client, message):
    people = {}
    total = await user.get_chat_members_count(message.chat.id)
    async for msg in user.iter_history(message.chat.id, limit=1000):
        if msg.from_user and not msg.from_user.is_bot:
            people[msg.from_user.id] = msg.from_user.first_name
    await message.edit(f"{round(len(people) / total)}%")


@user.on_message(filters.command("recent_action", "!") & filters.me)
async def recent_act(client, message):
    full_log = await user.invoke(
        functions.channels.GetAdminLog(
            channel=await user.resolve_peer(message.chat.id),
            q="",
            max_id=0,
            min_id=0,
            limit=0,
        )
    )
    with open(f"recent_actions_{message.chat.id}.txt", "w", encoding="utf8") as log_file:
        log_file.write(str(full_log))
    await message.reply_document(f"recent_actions_{message.chat.id}.txt")


@user.on_message(filters.command(["screenshot"], prefixes="!"))
async def take_a_screenshot(client, message):
    await message.delete()
    await user.invoke(
        functions.messages.SendScreenshotNotification(
            peer=await user.resolve_peer(message.chat.id),
            reply_to_msg_id=0,
            random_id=app.rnd_id(),
        )
    )