main.py

import os
import yt_dlp
from telegram import Update, ReplyKeyboardMarkup
from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, filters, ContextTypes
from telegram.constants import ParseMode
from urllib.parse import urlparse

BOT_TOKEN = ""

# 📝 Сообщения по языкам
MESSAGES = {
    "ru": {
        "greeting": "<b>Привет!</b> 👋\nЯ помогу тебе скачать любое видео (до 10 минут).\nПросто пришли ссылку, и через пару секунд оно появится прямо в чате 🎬",
        "choose_lang": "🌐 Выберите язык:",
        "lang_set": "✅ Язык установлен: Русский",
        "video_too_long": "⚠️ Видео длиннее 10 минут, я не могу его отправить.",
        "download_start": "Принял ссылку, начинаю загрузку... ⏳",
        "send_error": "⚠️ Видео скачалось, но не удалось отправить его в чат.\nВозможные причины: файл слишком большой или формат не поддерживается Telegram.",
        "download_fail": "Упс! 😅 Не удалось скачать видео.\nПричина: <code>{}</code>",
        "cookie_error": "⚠️ Стоит обновить куки файлы.\nНапиши админу, чтобы он не расслаблялся и обновил их 😅",
        "invalid_url": "❌ Это не похоже на ссылку 🤔 Попробуй отправить корректный URL.",
        "menu_start": "🎬 Старт",
        "menu_language": "🌐 Выбор языка",
        "lang_buttons": ["Русский", "Украинский", "English"],
        "back": "⬅ Назад"
    },
    "ua": {
        "greeting": "<b>Привіт!</b> 👋\nЯ допоможу тобі завантажити будь-яке відео (до 10 хвилин).\nПросто надішли посилання, і через кілька секунд воно з’явиться в чаті 🎬",
        "choose_lang": "🌐 Оберіть мову:",
        "lang_set": "✅ Мова встановлена: Українська",
        "video_too_long": "⚠️ Відео довше 10 хвилин, я не можу його надіслати.",
        "download_start": "Прийняв посилання, починаю завантаження... ⏳",
        "send_error": "⚠️ Відео завантажено, але не вдалося надіслати його в чат.\nМожливі причини: файл занадто великий або формат не підтримується Telegram.",
        "download_fail": "Упс! 😅 Не вдалося завантажити відео.\nПричина: <code>{}</code>",
        "cookie_error": "⚠️ Потрібно оновити кукі файли.\nНапиши адміну, щоб він не лінувався і оновив їх 😅",
        "invalid_url": "❌ Це не схоже на посилання 🤔 Спробуй надіслати коректний URL.",
        "menu_start": "🎬 Почати",
        "menu_language": "🌐 Вибір мови",
        "lang_buttons": ["Російська", "Українська", "Англійська"],
        "back": "⬅ Назад"
    },
    "en": {
        "greeting": "<b>Hello!</b> 👋\nI can help you download any video (up to 10 minutes).\nJust send the link, and in a few seconds it will appear in chat 🎬",
        "choose_lang": "🌐 Choose a language:",
        "lang_set": "✅ Language set: English",
        "video_too_long": "⚠️ Video is longer than 10 minutes, I can't send it.",
        "download_start": "Received the link, starting download... ⏳",
        "send_error": "⚠️ Video downloaded, but couldn't send it in the chat.\nPossible reasons: file too large or format not supported by Telegram.",
        "download_fail": "Oops! 😅 Failed to download video.\nReason: <code>{}</code>",
        "cookie_error": "⚠️ Cookies file needs to be updated.\nContact the admin so he finally refreshes them 😅",
        "invalid_url": "❌ This doesn’t look like a valid link 🤔 Try sending a correct URL.",
        "menu_start": "🎬 Start",
        "menu_language": "🌐 Language",
        "lang_buttons": ["Russian", "Ukrainian", "English"],
        "back": "⬅ Back"
    }
}

# ✏️ Текст подписи под видео по языкам (с сердечком)
THANK_CAPTION = {
    "ru": "Спасибо, что пользуетесь — @LinkerGrab_bot ❤️",
    "ua": "Дякуємо, що користуєтесь — @LinkerGrab_bot ❤️",
    "en": "Thank you for using — @LinkerGrab_bot ❤️",
}

# 📥 Проверка, является ли строка ссылкой
def is_valid_url(text: str) -> bool:
    try:
        result = urlparse(text)
        return all([result.scheme, result.netloc])
    except Exception:
        return False

# 📥 Функция создания клавиатуры
def get_main_menu(lang):
    return ReplyKeyboardMarkup(
        keyboard=[[MESSAGES[lang]["menu_start"], MESSAGES[lang]["menu_language"]]],
        resize_keyboard=True
    )

def get_language_menu(lang):
    buttons = MESSAGES[lang]["lang_buttons"]
    return ReplyKeyboardMarkup(
        keyboard=[buttons, [MESSAGES[lang]["back"]]],
        resize_keyboard=True
    )

# 📥 Приветствие
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
    lang = context.user_data.get("lang", "ru")
    context.user_data["choosing_lang"] = False
    await update.message.reply_text(
        MESSAGES[lang]["greeting"],
        parse_mode=ParseMode.HTML,
        reply_markup=get_main_menu(lang)
    )

# 📥 Обработка меню и языков
async def menu_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
    text = update.message.text
    lang = context.user_data.get("lang", "ru")

    if text == MESSAGES[lang]["menu_start"]:
        await start(update, context)
    elif text == MESSAGES[lang]["menu_language"]:
        context.user_data["choosing_lang"] = True
        await update.message.reply_text(MESSAGES[lang]["choose_lang"], reply_markup=get_language_menu(lang))
    elif text in sum([MESSAGES[l]["lang_buttons"] for l in ["ru","ua","en"]], []):
        lang_map = {
            "Русский": "ru", "Украинский": "ua", "English": "en",
            "Російська": "ru", "Українська": "ua", "Англійська": "en",
            "Russian": "ru", "Ukrainian": "ua", "English": "en"
        }
        context.user_data["lang"] = lang_map[text]
        new_lang = context.user_data["lang"]
        context.user_data["choosing_lang"] = False
        await update.message.reply_text(MESSAGES[new_lang]["lang_set"], reply_markup=get_main_menu(new_lang))
    elif text in [MESSAGES[l]["back"] for l in ["ru","ua","en"]]:
        context.user_data["choosing_lang"] = False
        await update.message.reply_text("Меню:", reply_markup=get_main_menu(lang))
    else:
        if context.user_data.get("choosing_lang", False):
            await update.message.reply_text("❌ Такой язык не поддерживается.\nВыберите из списка кнопок.")

# 📥 Функция загрузки и отправки видео
async def download_video(update: Update, context: ContextTypes.DEFAULT_TYPE):
    lang = context.user_data.get("lang", "ru")
    url = update.message.text.strip()

    if not is_valid_url(url):
        await update.message.reply_text(MESSAGES[lang]["invalid_url"])
        return

    output_path = "downloads"
    os.makedirs(output_path, exist_ok=True)

    try:
        await update.message.reply_text(MESSAGES[lang]["download_start"])

        ydl_opts = {
            "outtmpl": f"{output_path}/%(title)s.%(ext)s",
            "format": "mp4/best",
            "noplaylist": True,
            "cookiefile": "cookies.txt"
        }

        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
            info = ydl.extract_info(url, download=True)

            duration = info.get("duration")
            if duration and duration > 600:
                await update.message.reply_text(MESSAGES[lang]["video_too_long"])
                return

            filename = ydl.prepare_filename(info)

        # 👇 Фиксированный текст благодарности с сердечком
        caption_text = THANK_CAPTION.get(lang, "Спасибо, что пользуетесь — @LinkerGrab_bot ❤️")

        try:
            await update.message.reply_video(
                video=open(filename, "rb"),
                caption=caption_text
            )
        except Exception:
            await update.message.reply_text(MESSAGES[lang]["send_error"])

        if os.path.exists(filename):
            os.remove(filename)

    except Exception as e:
        error_text = str(e).lower()
        if "cookies" in error_text or "login required" in error_text:
            msg = MESSAGES[lang]["cookie_error"]
        else:
            msg = MESSAGES[lang]["download_fail"].format(e)

        await update.message.reply_text(msg, parse_mode=ParseMode.HTML)

# 📥 Универсальный обработчик
async def message_handler(update: Update, context: ContextTypes.DEFAULT_TYPE):
    text = update.message.text
    lang = context.user_data.get("lang", "ru")

    menu_texts = [MESSAGES[l]["menu_start"] for l in ["ru","ua","en"]] + \
                 [MESSAGES[l]["menu_language"] for l in ["ru","ua","en"]] + \
                 sum([MESSAGES[l]["lang_buttons"] for l in ["ru","ua","en"]], []) + \
                 [MESSAGES[l]["back"] for l in ["ru","ua","en"]]

    if text in menu_texts or context.user_data.get("choosing_lang", False):
        await menu_handler(update, context)
    else:
        await download_video(update, context)

# 🚀 Основной запуск
def main():
    app = ApplicationBuilder().token(BOT_TOKEN).build()

    app.add_handler(CommandHandler("start", start))
    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, message_handler))

    print("✅ Бот запущен и ждёт ссылок...")
    app.run_polling()

if __name__ == "__main__":
    main()