import logging import re from pyrogram import Client, filters from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton from pyrogram.types import * from RyuzakiLib import Tiktok from config import API_ID, API_HASH, BOT_TOKEN import hashlib logging.getLogger("pyrogram").setLevel(logging.WARNING) logging.basicConfig(level=logging.INFO) TIKTOK_WEB = "https://www.tikwm.com" WELCOME_TEXT = """ Halo {} Saya adalah bot untuk mengunduh video tiktok di telegram. Saya dapat mengunduh video dengan tanda air atau tanpa tanda air dan mengunduh audio dari url. Kirimkan saja saya url tiktok. """ client = Client( "TTK-BOT", api_id=API_ID, api_hash=API_HASH, bot_token=BOT_TOKEN ) link_storage = {} def generate_callback_data(user_id, query): identifier = hashlib.md5(query.encode()).hexdigest() callback_data = f"audiodownload_{user_id}_{identifier}" link_storage[callback_data] = query return callback_data @client.on_message(filters.command("start") & filters.private) async def welcome_start(client: Client, message: Message): keyboard = InlineKeyboardMarkup( [ [ InlineKeyboardButton( text="📢 Saluran Bot", url="https://t.me/RendyProjects" ) ] ] ) await message.reply_text( WELCOME_TEXT.format(message.from_user.first_name), reply_markup=keyboard ) @client.on_callback_query(filters.regex("^audiodownload_")) async def callback_button(client: Client, cb: CallbackQuery): try: data = cb.data user_id = cb.from_user.id query = link_storage.get(data) if query: response = await Tiktok.download(TIKTOK_WEB, query) await client.send_audio(user_id, response[1]) await cb.answer("Audio sent successfully!") else: await cb.answer("Invalid or expired link.", show_alert=True) except Exception as e: await cb.answer(f"Error: {str(e)}", show_alert=True) def is_tiktok_url(url): pattern = r"(https?)://(vt|www)\.tiktok\.com/(\w+)" match = re.search(pattern, url) return bool(match) @client.on_message(filters.text & filters.private) async def tiktok_downloader(client: Client, message: Message): if message.text: query_url = message.text if not is_tiktok_url(query_url): return await message.reply_text("Invalid link") callback_data = generate_callback_data(message.from_user.id, query_url) keyboard = InlineKeyboardMarkup( [ [ InlineKeyboardButton( text="Audio Download", callback_data=callback_data ) ] ] ) try: dll = await message.reply_text("Processing....") await message.delete() response = await Tiktok.download(TIKTOK_WEB, query_url) 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)}") client.run()