diff --git a/program/music.py b/program/music.py index f0afd6f..f4a87f7 100644 --- a/program/music.py +++ b/program/music.py @@ -2,15 +2,17 @@ # Commit Start Date 20/10/2021 # Finished On 28/10/2021 -import asyncio import re +import asyncio -from config import BOT_USERNAME, GROUP_SUPPORT, IMG_1, IMG_2, UPDATES_CHANNEL -from driver.filters import command, other_filters -from driver.queues import QUEUE, add_to_queue -from driver.veez import call_py from pyrogram import Client +from driver.veez import call_py +from program import BOT_ID, USERBOT_ID +from driver.queues import QUEUE, add_to_queue +from driver.filters import command, other_filters +from pyrogram.errors import UserAlreadyParticipant, UserNotParticipant from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message +from config import BOT_USERNAME, GROUP_SUPPORT, IMG_1, IMG_2, UPDATES_CHANNEL, ASSISTANT_NAME from pytgcalls import StreamType from pytgcalls.types.input_stream import AudioPiped from youtubesearchpython import VideosSearch @@ -64,7 +66,53 @@ async def play(_, m: Message): ] ] ) - + + chat_title = m.chat.title + a = await _.get_chat_member(m.chat.id, BOT_ID) + if a.status != "administrator": + await m.reply_text(f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**") + return + if not a.can_manage_voice_chats: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Manage voice chat__") + return + if not a.can_delete_messages: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Delete messages__") + return + if not a.can_invite_users: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Add users__") + return + if not a.can_restrict_members: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Ban users__") + return + try: + b = await _.get_chat_member(m.chat.id, USERBOT_ID) + if b.status == "kicked": + await m.reply_text(f"@{ASSISTANT_NAME} **is banned in group** {chat_title}\n\n» **unban the userbot first if you want to use this bot.**") + return + except UserNotParticipant: + if m.chat.username: + try: + await call_py.join_chat(f"{m.chat.username}") + except Exception as e: + await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") + return + else: + try: + pope = await _.export_chat_invite_link(m.chat.id) + pepo = await _.revoke_chat_invite_link(m.chat.id, pope) + await call_py.join_chat(pepo.invite_link) + except UserAlreadyParticipant: + pass + except Exception as e: + return await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") replied = m.reply_to_message chat_id = m.chat.id if replied: @@ -215,7 +263,53 @@ async def stream(_, m: Message): ] ] ) - + + chat_title = m.chat.title + a = await _.get_chat_member(m.chat.id, BOT_ID) + if a.status != "administrator": + await m.reply_text(f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**") + return + if not a.can_manage_voice_chats: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Manage voice chat__") + return + if not a.can_delete_messages: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Delete messages__") + return + if not a.can_invite_users: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Add users__") + return + if not a.can_restrict_members: + await m.reply_text( + "missing required permission:" + + "\n\n» ❌ __Ban users__") + return + try: + b = await _.get_chat_member(m.chat.id, USERBOT_ID) + if b.status == "kicked": + await m.reply_text(f"@{ASSISTANT_NAME} **is banned in group** {chat_title}\n\n» **unban the userbot first if you want to use this bot.**") + return + except UserNotParticipant: + if m.chat.username: + try: + await call_py.join_chat(f"{m.chat.username}") + except Exception as e: + await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") + return + else: + try: + pope = await _.export_chat_invite_link(m.chat.id) + pepo = await _.revoke_chat_invite_link(m.chat.id, pope) + await call_py.join_chat(pepo.invite_link) + except UserAlreadyParticipant: + pass + except Exception as e: + return await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}") chat_id = m.chat.id if len(m.command) < 2: await m.reply("» give me a live-link/m3u8 url/youtube link to stream.")