Minor fix (#76)

* Minor fix

* Forgot tag in gd_search.py

bot/helper/ext_utils/help_messages.py

@@ -270,7 +270,7 @@
 """
 
 thumbnail_layout = """
-Thumbnail Layout: -tl
+<b>Thumbnail Layout</b>: -tl
 
 /cmd link -tl 3x3 (widthxheight) 3 photos in row and 3 photos in column
 """
@@ -282,6 +282,17 @@
 /cmd link -med (Leech as media)
 """
 
+metadata = """
+<b>Metadata</b>: -md
+/cmd link -md text
+It will add text in your video metadata. (MKV & MP4 supports only)
+
+
+<b>Metadata Attachment</b>: -mda
+/cmd link -mda tg-message-link(doc or photo) or any direct link
+It will embed thumb in your video. (MKV & MP4 supports only)
+"""
+
 YT_HELP_DICT = {
     "main": yt,
     "ʀᴇɴᴀᴍᴇ\nꜰɪʟᴇ": f"{new_name}\nNote: Don't add file extension",
@@ -303,6 +314,7 @@
     "ʜʏʙʀɪᴅ\nʟᴇᴇᴄʜ": mixed_leech,
     "ᴛʜᴜᴍʙ\nʟᴀʏᴏᴜᴛ": thumbnail_layout,
     "ʟᴇᴇᴄʜ\nᴛʏᴘᴇ": leech_as,
+    "ᴍᴇᴛᴀᴅᴀᴛᴀ\nᴀᴛᴛᴀᴄʜ": metadata,
 }
 
 MIRROR_HELP_DICT = {
@@ -332,6 +344,7 @@
     "ʜʏʙʀɪᴅ\nʟᴇᴇᴄʜ": mixed_leech,
     "ᴛʜᴜᴍʙ\nʟᴀʏᴏᴜᴛ": thumbnail_layout,
     "ʟᴇᴇᴄʜ\nᴛʏᴘᴇ": leech_as,
+    "ᴍᴇᴛᴀᴅᴀᴛᴀ\nᴀᴛᴛᴀᴄʜ": metadata,
 }
 
 CLONE_HELP_DICT = {

bot/helper/ext_utils/media_utils.py

@@ -35,6 +35,8 @@
     ARCH_EXT, 
     get_mime_type
 )
+from .links_utils import is_telegram_link
+from ..telegram_helper.message_utils import get_tg_link_message
 
 
 async def convert_video(listener, video_file, ext, retry=False):
@@ -890,18 +892,12 @@ async def create_sample_video(listener, video_file, sample_duration, part_durati
         return False
 
 
-SUPPORTED_VIDEO_EXTENSIONS = {
-    ".mp4",
-    ".mkv"
-}
-
-
 async def edit_video_metadata(listener, dir):
 
     data = listener.metadata
     dir_path = Path(dir)
 
-    if dir_path.suffix.lower() not in SUPPORTED_VIDEO_EXTENSIONS:
+    if not dir_path.lower().endswith(("mkv", "mp4")):
         return dir
 
     file_name = dir_path.name
@@ -1074,12 +1070,19 @@ async def add_attachment(listener, dir):
     data = listener.m_attachment
     dir_path = Path(dir)
 
-    if dir_path.suffix.lower() not in SUPPORTED_VIDEO_EXTENSIONS:
+    if not dir_path.lower().endswith(("mkv", "mp4")):
         return dir
 
     file_name = dir_path.name
     work_path = dir_path.with_suffix(".temp.mkv")
 
+    if data:
+        if is_telegram_link(data):
+            msg = (await get_tg_link_message(data))[0]
+            data = (
+                await create_thumb(msg) if msg.photo or msg.document else ""
+            )
+
     data_ext = data.split(".")[-1].lower()
     if not (mime_type := MIME_TYPES.get(data_ext)):
         LOGGER.error(f"Unsupported attachment type: {data_ext}")

bot/modules/gd_search.py

@@ -20,13 +20,15 @@
     get_telegraph_list
 )
 from ..helper.ext_utils.status_utils import get_readable_time
+from ..helper.ext_utils.token_manager import checking_access
 from ..helper.task_utils.gdrive_utils.search import GoogleDriveSearch
 from ..helper.telegram_helper.bot_commands import BotCommands
 from ..helper.telegram_helper.button_build import ButtonMaker
 from ..helper.telegram_helper.filters import CustomFilters
 from ..helper.telegram_helper.message_utils import (
     anno_checker,
     auto_delete_message,
+    is_admin,
     send_message,
     edit_message
 )
@@ -224,6 +226,14 @@ async def gdrive_search(_, message):
         )
         return
     user_id = from_user.id
+    if not await is_admin(message, user_id):
+        msg, btn = await checking_access(user_id)
+        if msg is not None:
+            msg += f"\n\n<b>cc</b>: {message.from_user.mention}"
+            msg += f"\n<b>Thank You</b>"
+            gdmsg = await send_message(message, msg, btn.build_menu(1))
+            await auto_delete_message(message, gdmsg)
+            return
     buttons = await list_buttons(user_id)
     gmsg = await send_message(
         message,