some fixes

wbb/modules/inline.py

@@ -380,7 +380,8 @@ async def inline_query_handler(client, query):
                 )
                 return
             tex = query.query.split(None, 1)[1].strip()
-            answerss = await yt_music_func(answers, tex)
+            user_id = query.from_user.id
+            answerss = await yt_music_func(answers, tex, user_id)
             await client.answer_inline_query(
                 query.id,
                 results=answerss,

wbb/modules/music.py

@@ -31,14 +31,14 @@ import ffmpeg
 from random import randint
 from hurry.filesize import size as format_size
 from pyrogram import filters
-from wbb import app, arq
+from wbb import app, arq, SUDOERS
 from wbb.core.decorators.errors import capture_err
 from wbb.utils.pastebin import paste
 from wbb.utils.functions import file_size_from_url
 
 
 __MODULE__ = "Music"
-__HELP__ = """/ytmusic [link] To Download Music From Various Websites Including Youtube.
+__HELP__ = """/ytmusic [link] To Download Music From Various Websites Including Youtube. [SUDOERS]
 /saavn [query] To Download Music From Saavn.
 /deezer [query] To Download Music From Deezer.
 /lyrics [query] To Get Lyrics Of A Song."""
@@ -70,7 +70,7 @@ async def download_youtube_audio(url: str, m = 0):
     return [title, performer, duration, audio_file]
 
 
-@app.on_message(filters.command("ytmusic") & ~filters.edited)
+@app.on_message(filters.command("ytmusic") & ~filters.edited & filters.user(SUDOERS))
 @capture_err
 async def music(_, message):
     global is_downloading

wbb/utils/inlinefuncs.py

@@ -829,7 +829,16 @@ async def nsfw_scan_func(answers, url: str):
     return answers
 
 
-async def yt_music_func(answers, url):
+async def yt_music_func(answers, url, user_id):
+    if user_id not in SUDOERS:
+        msg = "**ERROR**\n__THIS FEATURE IS ONLY FOR SUDO USERS__"
+        answers.append(
+            InlineQueryResultArticle(
+                title="ERROR",
+                description="THIS FEATURE IS ONLY FOR SUDO USERS",
+                input_message_content=InputTextMessageContent(msg)
+            ))
+        return answers
     title, performer, duration, audio_file = await download_youtube_audio(url)
     m = await app.send_audio(MESSAGE_DUMP_CHAT, audio_file)
     answers.append(