# Ultroid - UserBot
# Copyright (C) 2020 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.

import re

from telethon.errors.rpcerrorlist import (
    ChatWriteForbiddenError,
    MediaEmptyError,
    PeerIdInvalidError,
)
from telethon.utils import get_display_name

from . import *

# taglogger


@ultroid_bot.on(
    events.NewMessage(
        incoming=True,
        func=lambda e: (e.mentioned),
    ),
)
async def all_messages_catcher(e):
    if udB.get("TAG_LOG"):
        try:
            NEEDTOLOG = int(udB.get("TAG_LOG"))
        except Exception:
            return LOGS.info("you given Wrong Grp/Channel ID in TAG_LOG.")
        x = e.sender
        if x.bot or x.verified:
            return
        y = e.chat
        where_n = get_display_name(y)
        who_n = get_display_name(x)
        where_l = f"https://t.me/c/{y.id}/{e.id}"
        send = await ultroid_bot.get_messages(e.chat_id, ids=e.id)
        try:
            if x.username:
                who_l = f"https://t.me/{x.username}"
                await asst.send_message(
                    NEEDTOLOG,
                    send,
                    buttons=[
                        [Button.url(who_n, who_l)],
                        [Button.url(where_n, where_l)],
                    ],
                )
            else:
                await asst.send_message(
                    NEEDTOLOG,
                    send,
                    buttons=[
                        [Button.inline(who_n, data=f"who{x.id}")],
                        [Button.url(where_n, where_l)],
                    ],
                )
        except MediaEmptyError:
            if x.username:
                who_l = f"https://t.me/{x.username}"
                await asst.send_message(
                    NEEDTOLOG,
                    "`Unsupported Media`",
                    buttons=[
                        [Button.url(who_n, who_l)],
                        [Button.url(where_n, where_l)],
                    ],
                )
            else:
                await asst.send_message(
                    NEEDTOLOG,
                    "`Unsupported Media`",
                    buttons=[
                        [Button.inline(who_n, data=f"who{x.id}")],
                        [Button.url(where_n, where_l)],
                    ],
                )
        except PeerIdInvalidError:
            await ultroid_bot.send_message(
                int(udB.get("LOG_CHANNEL")),
                "The Chat Id You Set In Tag Logger Is Wrong , Please Correct It",
            )
        except ChatWriteForbiddenError:
            await ultroid_bot.send_message(NEEDTOLOG, "Please Give Your Assistant Bot")
        except Exception as er:
            LOGS.info(str(er))
    else:
        return


@callback(re.compile("who(.*)"))
async def _(e):
    wah = e.pattern_match.group(1).decode("UTF-8")
    y = await ultroid_bot.get_entity(int(wah))
    who = f"[{get_display_name(y)}](tg://user?id={y.id})"
    x = await e.reply(f"Mention By user : {who}")
    await asyncio.sleep(6)
    await x.delete()


# log for assistant
@asst.on(events.ChatAction)
async def when_asst_added_to_chat(event):
    if event.user_added:
        user = await event.get_user()
        chat = (await event.get_chat()).title
        tmp = event.added_by
        if user.is_self:
            buttons = Button.inline("Leave Chat", data=f"leave_ch_{event.chat_id}|bot")
            return await asst.send_message(
                int(udB.get("LOG_CHANNEL")),
                f"#ADD_LOG\n\n[{tmp.first_name}](tg://user?id={tmp.id}) added [{user.first_name}](tg://user?id={user.id}) to {chat}.",
                buttons=buttons,
            )


# log for user's new joins


@ultroid.on(events.ChatAction)
async def when_ultd_added_to_chat(event):
    if event.user_added:
        user = await event.get_user()
        chat = (await event.get_chat()).title
        tmp = event.added_by
        if user.is_self:
            buttons = Button.inline("Leave Chat", data=f"leave_ch_{event.chat_id}|user")
            return await asst.send_message(
                int(udB.get("LOG_CHANNEL")),
                f"#ADD_LOG\n\n[{tmp.first_name}](tg://user?id={tmp.id}) just added [{user.first_name}](tg://user?id={user.id}) to {chat}.",
                buttons=buttons,
            )
    elif event.user_joined:
        user = await event.get_user()
        chat = (await event.get_chat()).title
        if user.is_self:
            buttons = Button.inline("Leave Chat", data=f"leave_ch_{event.chat_id}|user")
            return await asst.send_message(
                int(udB.get("LOG_CHANNEL")),
                f"#JOIN_LOG\n\n[{user.first_name}](tg://user?id={user.id}) just joined {chat}.",
                buttons=buttons,
            )


@callback(
    re.compile(
        "leave_ch_(.*)",
    ),
)
@owner
async def leave_ch_at(event):
    cht = event.data_match.group(1).decode("UTF-8")
    ch_id, client = cht.split("|")
    if client == "bot":
        name = (await asst.get_entity(int(ch_id))).title
        await asst.delete_dialog(int(ch_id))
    elif client == "user":
        name = (await ultroid_bot.get_entity(int(ch_id))).title
        await ultroid_bot.delete_dialog(int(ch_id))
    await event.edit(f"Left `{name}`")