| | |
| |
|
| | import math |
| | from pyrogram import filters, Client |
| | from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, Message |
| | from pyrogram.enums.parse_mode import ParseMode |
| | from WebStreamer import __version__ |
| | from WebStreamer.bot import StreamBot |
| | from WebStreamer.server.exceptions import FIleNotFound |
| | from WebStreamer.utils.bot_utils import is_user_accepted_tos, validate_user |
| | from WebStreamer.vars import Var |
| | from WebStreamer.utils.database import Database |
| | from WebStreamer.utils.Translation import Language, BUTTON |
| |
|
| | db = Database(Var.DATABASE_URL, Var.SESSION_NAME) |
| |
|
| | @StreamBot.on_message(filters.command('start') & filters.private) |
| | async def start(_: Client, message: Message): |
| | lang = Language(message) |
| | if not await validate_user(message, lang): |
| | return |
| | |
| | |
| | await message.reply_text( |
| | text=lang.START_TEXT.format(message.from_user.mention), |
| | parse_mode=ParseMode.HTML, |
| | disable_web_page_preview=True, |
| | reply_markup=BUTTON.START_BUTTONS |
| | ) |
| |
|
| | @StreamBot.on_message(filters.command("about") & filters.private) |
| | async def about(_: Client, message: Message): |
| | lang = Language(message) |
| | if not await validate_user(message, lang): |
| | return |
| | await message.reply_text( |
| | text=lang.ABOUT_TEXT.format(__version__), |
| | disable_web_page_preview=True, |
| | reply_markup=BUTTON.ABOUT_BUTTONS |
| | ) |
| |
|
| |
|
| | @StreamBot.on_message((filters.command('help')) & filters.private) |
| | async def help_handler(_: Client, message: Message): |
| | lang = Language(message) |
| | if not await validate_user(message, lang): |
| | return |
| | await message.reply_text( |
| | text=lang.HELP_TEXT.format(Var.UPDATES_CHANNEL), |
| | parse_mode=ParseMode.HTML, |
| | disable_web_page_preview=True, |
| | reply_markup=BUTTON.HELP_BUTTONS |
| | ) |
| |
|
| | |
| |
|
| | @StreamBot.on_message(filters.command('myfiles') & filters.private) |
| | async def my_files(_: Client, message: Message): |
| | if not await validate_user(message): |
| | return |
| | user_files, total_files=await db.find_files(message.from_user.id, [1,10]) |
| |
|
| | file_list=[] |
| | async for x in user_files: |
| | file_list.append([InlineKeyboardButton(x["file_name"], callback_data=f"myfile_{x['_id']}_{1}")]) |
| | if total_files > 10: |
| | file_list.append( |
| | [ |
| | InlineKeyboardButton("<<", callback_data="N/A"), |
| | InlineKeyboardButton(f"1/{math.ceil(total_files/10)}", callback_data="N/A"), |
| | InlineKeyboardButton(">>", callback_data="userfiles_2") |
| | ] |
| | ) |
| | if not file_list: |
| | file_list.append([InlineKeyboardButton("Empty", callback_data="N/A")]) |
| | await message.reply_photo(photo=Var.IMAGE_FILEID, |
| | caption="Total files: {total_files}", |
| | reply_markup=InlineKeyboardMarkup(file_list)) |
| |
|
| | @StreamBot.on_message(filters.command('tos') & filters.private) |
| | async def tos_handler(_: Client, message: Message): |
| | if not Var.TOS: |
| | await message.reply_text("This bot does not have any terms of service.") |
| | return |
| | if await is_user_accepted_tos(message): |
| | await message.reply_text( |
| | Var.TOS, |
| | reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("✅ I accepted the TOS", callback_data="N/A")]]) |
| | ) |
| |
|
| | @StreamBot.on_message(filters.command('info') & filters.private) |
| | async def info_handler(_: Client, message: Message): |
| | lang = Language(message) |
| | if not await validate_user(message, lang): |
| | return |
| | i_cmd=message.text.split() |
| | if (message.from_user.id == Var.OWNER_ID) and (len(i_cmd) > 1): |
| | message.from_user.id=int(i_cmd[1]) |
| | user = await db.get_user(message.from_user.id) |
| | files=await db.total_files(message.from_user.id) |
| | links="N/A" |
| | if (user.get("Plan") == "Free") and (Var.LINK_LIMIT): |
| | links=Var.LINK_LIMIT-files |
| | await message.reply_text(lang.INFO_TEXT.format(message.from_user.id, user.get("Plan"), files, links)) |
| |
|
| | @StreamBot.on_message(filters.command('getfile') & filters.private & filters.user(Var.OWNER_ID)) |
| | async def getfile(_: Client, message: Message): |
| | if not await validate_user(message): |
| | return |
| | usr_cmd=message.text.split()[1:] |
| | if len(usr_cmd) < 1: |
| | return await message.reply_text("Invalid Format\nUsage: `/getfile _id`") |
| | for x in usr_cmd: |
| | try: |
| | myfile = await db.get_file(x) |
| | await message.reply_cached_media(myfile['file_id']) |
| | except FIleNotFound: |
| | await message.reply_text(f"{x} :File Not Found") |
| |
|
