new update

bot/config.py

@@ -8,7 +8,7 @@ class Telegram:
     BOT_USERNAME = env.get("TELEGRAM_BOT_USERNAME", "xxxx_streambot")
     BOT_TOKEN = env.get("TELEGRAM_BOT_TOKEN", "1234567:xyz")
     CHANNEL_ID = int(env.get("TELEGRAM_CHANNEL_ID", -1001684936508))
-    SECRET_CODE_LENGTH = int(env.get("SECRET_CODE_LENGTH", 24))
+    SECRET_CODE_LENGTH = int(env.get("SECRET_CODE_LENGTH", 10))
 
 class Server:
     BASE_URL = env.get("BASE_URL", "https://kawaiimizo-maakichu.hf.space")

bot/modules/static.py

@@ -65,6 +65,18 @@ FileLinksText = \
 `%(dl_link)s`
 """
 
+FileLinkText = """
+<i><u>𝗬𝗼𝘂𝗿 𝗟𝗶𝗻𝗸 𝗚𝗲𝗻𝗲𝗿𝗮𝘁𝗲𝗱 !</u></i> 🎉
+
+📂 **Fɪʟᴇ ɴᴀᴍᴇ:** `{file_name}`
+
+📦 **Fɪʟᴇ ꜱɪᴢᴇ:** `{file_size}`
+
+📥 **Dᴏᴡɴʟᴏᴀᴅ:** __{dl_link}__
+
+🚸 **Nᴏᴛᴇ:** **LINK WON'T EXPIRE TILL YOU REVOKE**
+"""
+
 MediaLinksText = \
 """
 **Download Link:**
@@ -73,6 +85,20 @@ MediaLinksText = \
 `%(stream_link)s`
 """
 
+MediaLinkText = """
+<i><u>𝗬𝗼𝘂𝗿 𝗟𝗶𝗻𝗸 𝗚𝗲𝗻𝗲𝗿𝗮𝘁𝗲𝗱 !</u></i> 🎉
+
+📂 **Fɪʟᴇ ɴᴀᴍᴇ:** __{file_name}__
+
+📦 **Fɪʟᴇ ꜱɪᴢᴇ:** `{file_size}`
+
+📥 **Dᴏᴡɴʟᴏᴀᴅ:** __{dl_link}__
+
+🖥 **WATCH:** __{stream_link}__
+
+🚸 **Nᴏᴛᴇ:** **LINK WON'T EXPIRE TILL YOU REVOKE**
+"""
+
 InvalidQueryText = \
 """
 Query data mismatched.

bot/plugins/files.py

@@ -5,6 +5,7 @@ from bot import TelegramBot
 from bot.config import Telegram, Server
 from bot.modules.decorators import verify_user
 from bot.modules.static import *
+from bot.utils import humanbytes
 
 @TelegramBot.on_message(
     filters.private
@@ -21,17 +22,38 @@ from bot.modules.static import *
 async def handle_user_file(_, msg: Message):
     sender_id = msg.from_user.id
     secret_code = token_hex(Telegram.SECRET_CODE_LENGTH)
+
+    # Determine the file type and extract file name and size dynamically
+    file_type = None
+    if msg.document:
+        file_type = "document"
+    elif msg.video:
+        file_type = "video"
+    elif msg.video_note:
+        file_type = "video_note"
+    elif msg.audio:
+        file_type = "audio"
+    elif msg.voice:
+        file_type = "voice"
+    elif msg.photo:
+        file_type = "photo"
+
+    # Use getattr to dynamically access file_name and file_size
+    file_name = getattr(msg, file_type).file_name if hasattr(getattr(msg, file_type), 'file_name') else file_type.title()
+    file_size = getattr(getattr(msg, file_type), 'file_size', 0)
     file = await msg.copy(
         chat_id=Telegram.CHANNEL_ID,
         caption=f'||{secret_code}/{sender_id}||'
     )
     file_id = file.id
     dl_link = f'{Server.BASE_URL}/dl/{file_id}?code={secret_code}'
+    stream_link = f'{Server.BASE_URL}/stream/{file_id}?code={secret_code}'
+    formatted_message = MediaLinkText.format(file_name=file_name, file_size=humanbytes(file_size), dl_link=dl_link, stream_link=stream_link)
+    file_message = FileLinkText.format(file_name=file_name, file_size=humanbytes(file_size), dl_link=dl_link)
 
     if (msg.document and 'video' in msg.document.mime_type) or msg.video:
-        stream_link = f'{Server.BASE_URL}/stream/{file_id}?code={secret_code}'
         await msg.reply(
-            text=MediaLinksText % {'dl_link': dl_link, 'stream_link': stream_link},
+            text=formatted_message,
             quote=True,
             reply_markup=InlineKeyboardMarkup(
                 [
@@ -47,7 +69,7 @@ async def handle_user_file(_, msg: Message):
         )
     else:
         await msg.reply(
-            text=FileLinksText % {'dl_link': dl_link},
+            text=file_message,
             quote=True,
             reply_markup=InlineKeyboardMarkup(
                 [

bot/server/main.py

@@ -37,7 +37,7 @@ async def health():
     })
 
 
-@bp.route('/dl/<int:file_id>')
+@bp.route('/dl/<int:file_id>', methods=['GET', 'POST', 'HEAD'])
 async def transmit_file(file_id):
     file = await get_message(file_id) or abort(404)
     code = request.args.get('code') or abort(401)
@@ -102,7 +102,7 @@ async def transmit_file(file_id):
 
     return Response(file_stream(), headers=headers, status=status_code)
 
-@bp.route('/stream/<int:file_id>')
+@bp.route('/stream/<int:file_id>', methods=['GET', 'POST', 'HEAD'])
 async def stream_file(file_id):
     code = request.args.get('code') or abort(401)
     return await render_template('player.html', mediaLink=f'{Server.BASE_URL}/dl/{file_id}?code={code}')

bot/utils/__init__.py

@@ -1 +1,2 @@
-from .time_format import get_readable_time, get_system_info
+from .time_format import get_readable_time, get_system_info
+from .util import humanbytes

bot/utils/util.py

@@ -0,0 +1,13 @@
+
+def humanbytes(size):
+    if not size:
+        return "0 B"
+    for unit in ["", "K", "M", "G", "T"]:
+        if size < 1024:
+            break
+        size /= 1024
+    if isinstance(size, int):
+        size = f"{size} {unit}B"
+    elif isinstance(size, float):
+        size = f"{size:.2f} {unit}B"
+    return size