Update main.py

main.py

@@ -1,16 +1,17 @@
 import logging
+import time
+import os
 from pyrogram import Client, filters
 from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
 from pyrogram.types import *
 import akenoai as ak
-from config import TIKTOK_WEB API_ID, API_HASH, BOT_TOKEN
-import hashlib
+from config import API_KEY, API_ID, API_HASH, BOT_TOKEN
 
 logging.getLogger("pyrogram").setLevel(logging.WARNING)
 logging.basicConfig(level=logging.INFO)
 
 WELCOME_TEXT = """
-Halo {}
+Halo {name}
 Saya adalah bot untuk mengunduh video PornoHub di telegram.
 
 - Dilarang Keras spam
@@ -19,6 +20,14 @@ Saya adalah bot untuk mengunduh video PornoHub di telegram.
 Powered by @xtdevs
 """
 
+USER_TRACK = """
+• AkenoHUB
+
+UserID: {id}
+First Name: {first_name}
+Username: @{username}
+"""
+
 client = Client(
     "AkenoHub",
     api_id=API_ID,
@@ -26,7 +35,7 @@ client = Client(
     bot_token=BOT_TOKEN
 )
 
-api = ak.PornoHub(key="randigithub356")
+api = ak.PornoHub(key=API_KEY)
 
 @client.on_message(filters.command("start") & filters.private)
 async def welcome_start(client: Client, message: Message):
@@ -40,38 +49,86 @@ async def welcome_start(client: Client, message: Message):
             ]
         ]
     )
+    await client.send_message(
+        PRIVATE_LOGS,
+        USER_TRACK.format(
+            id=message.from_user.id if message.from_user else 0,
+            first_name=message.from_user.first_name,
+            username=message.from_user.username if message.from_user else None
+        )
+    )
     await message.reply_text(
-        WELCOME_TEXT.format(message.from_user.first_name),
+        WELCOME_TEXT.format(
+            name=message.from_user.first_name
+        ),
         reply_markup=keyboard
     )
 
-@client.on_message(filters.text & filters.private)
-async def tiktok_downloader(client: Client, message: Message):
-    if message.text:
-        query = message.text
-        response = await api.x_search(query=query)
-        link = response[0]
+@client.on_message(
+    filters.command(["hubsearch"])
+    & filters.private
+    & ~filters.forwarded
+)
+async def searchhub(client: Client, message: Message):
+    query = message.text.split(" ", 1)[1] if len(message.command) > 1 else None
+    if not query:
+        return await message.reply_text("please search for pornohub.")
+    response = await api.x_search(query=query)
+    link = response[0]
+    try:
         file_path, _, _, = await api.x_download(url=link, is_stream=True)
-        
-        callback_data = generate_callback_data(message.from_user.id, query)
-        keyboard = InlineKeyboardMarkup(
-            [
-                [
-                    InlineKeyboardButton(
-                        text="Audio Download",
-                        callback_data=callback_data
-                    )
-                ]
-            ]
+        upload_text = f"**⬆️ 𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 video ...**"
+        await pro.edit_text(upload_text)
+        await message.reply_video(
+            file_path,
+            caption=f"• Powered by {client.me.mention}",
+            thumb=thumb,
+            has_spoiler=True,
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text
+            )
+        )
+        await pro.delete()
+        os.remove(thumb)
+        os.remove(file_path)
+    except Exception as e:
+        await pro.edit_text(str(e))
+
+@client.on_message(
+    filters.command(["hubdl"])
+    & filters.private
+    & ~filters.forwarded
+)
+async def porno_download(client: Client, message: Message):
+    link = message.text.split(" ", 1)[1] if len(message.command) > 1 else None
+    if not link:
+        return await message.reply_text("please link for pornohub.")
+    if not link.startswith("https://www.xnxx.com/"):
+        return await message.reply_text("invalid link.")
+    try:
+        pro = await message.reply_text("Processing.....")
+        file_path, thumb, _ = await api.x_download(url=link, is_stream=True)
+        upload_text = f"**⬆️ 𝖴𝗉𝗅𝗈𝖺𝖽𝗂𝗇𝗀 video ...**"
+        await pro.edit_text(upload_text)
+        await message.reply_video(
+            file_path,
+            caption=f"• Powered by {client.me.mention}",
+            thumb=thumb,
+            has_spoiler=True,
+            progress=progress,
+            progress_args=(
+                pro,
+                time.time(),
+                upload_text
+            )
         )
-        try:
-            dll = await message.reply_text("Processing....")
-            await message.delete()
-            response = await Tiktok.download(tt, query)
-            await message.reply_video(response[0], reply_markup=keyboard)
-            await dll.delete()
-        except Exception as e:
-            await dll.delete()
-            await message.reply_text(f"Error: {str(e)}")
+        await pro.delete()
+        os.remove(thumb)
+        os.remove(file_path)
+    except Exception as e:
+        await pro.edit_text(str(e))
 
 client.run()