import os
from aiogram import Router
from aiogram.filters import Command
from aiogram.fsm.state import State, StatesGroup
from aiogram.types import (BufferedInputFile, CallbackQuery,
                           InlineKeyboardButton, InlineKeyboardMarkup, Message)

from bot import bot
from database import add_referral, check_connection_exists, add_connection
from config import CHANNEL_URL, SUPPORT_CHAT_URL, BOT_VERSION  # Импорт BOT_VERSION

router = Router()

class FeedbackState(StatesGroup):
    waiting_for_feedback = State()

async def send_welcome_message(chat_id: int):
    welcome_text = (
        "<b>🎉 SoloNet — твой доступ в свободный интернет! 🌐✨</b>\n\n"
        "<b>Наши преимущества:</b>\n"
        "<blockquote>"
        "🚀 <b>Высокая скорость</b>\n"
        "🔄 <b>Стабильность</b>\n"
        "🌍 <b>Смена локаций</b>\n"
        "💬 <b>Отзывчивая поддержка</b>\n"
        "📱💻 <b>Для телефонов, компьютеров и планшетов</b>\n"
        "💰 <b>Реферальная программа: 25% от покупки</b>\n"
        "</blockquote>"
        "\n\n🎁 <b>1 день бесплатно!</b>\n\n" # Добавление версии бота
        "<i>Перейдите в профиль для продолжения 👇</i>"
    )

    image_path = os.path.join(os.path.dirname(__file__), 'pic.jpg')

    if not os.path.isfile(image_path):
        await bot.send_message(chat_id, "Файл изображения не найден.")
        return

    inline_keyboard = InlineKeyboardMarkup(inline_keyboard=[
        [InlineKeyboardButton(text='👤 Мой профиль', callback_data='view_profile')],
        [InlineKeyboardButton(text='🔒 О VPN', callback_data='about_vpn')],
        [InlineKeyboardButton(text='📞 Поддержка', url=SUPPORT_CHAT_URL)],  
        [InlineKeyboardButton(text='📢 Наш канал', url=CHANNEL_URL)]
    ])

    with open(image_path, 'rb') as image_from_buffer:
        await bot.send_photo(
            chat_id,
            BufferedInputFile(image_from_buffer.read(), filename="pic.jpg"),
            caption=welcome_text,
            parse_mode='HTML',
            reply_markup=inline_keyboard 
        )

@router.message(Command('start'))
async def start_command(message: Message):
    print(f"Received start command with text: {message.text}") 
    if 'referral_' in message.text:
        referrer_tg_id = int(message.text.split('referral_')[1])
        print(f"Referral ID: {referrer_tg_id}")
        if not await check_connection_exists(message.from_user.id):
            await add_connection(message.from_user.id)

            await add_referral(message.from_user.id, referrer_tg_id)
            
            await message.answer("Вас пригласил друг, добро пожаловать!")
        else:
            await message.answer("Вы уже зарегистрированы в системе!")
    await send_welcome_message(message.chat.id)

@router.callback_query(lambda c: c.data == 'about_vpn')
async def handle_about_vpn(callback_query: CallbackQuery):
    await callback_query.message.delete()
    info_message = (
        "*О VPN*\n\n"
        "<b>🚀 Высокоскоростные серверы</b>\n"
        "Мы используем высокоскоростные серверы в различных локациях для обеспечения стабильного и быстрого соединения.\n\n"
        
        "<b>🔐 Безопасность данных</b>\n"
        "Для защиты ваших данных мы применяем новейшие протоколы шифрования, которые гарантируют вашу конфиденциальность.\n\n"
        
        "<b>🔑 Ваш ключ — ваша безопасность!</b>\n"
        "Не передавайте своё шифрование сторонним лицам, чтобы избежать рисков.\n\n"
        
        f"<i>Версия бота: {BOT_VERSION}</i>"  # Добавление версии бота в информацию о VPN
    )

    button_back = InlineKeyboardButton(text='⬅️ Назад', callback_data='back_to_menu')
    inline_keyboard_back = InlineKeyboardMarkup(inline_keyboard=[[button_back]])

    await callback_query.message.answer(
        info_message,
        parse_mode='HTML',
        reply_markup=inline_keyboard_back
    )
    await callback_query.answer()

@router.callback_query(lambda c: c.data == 'back_to_menu')
async def handle_back_to_menu(callback_query: CallbackQuery):
    await callback_query.message.delete()
    await send_welcome_message(callback_query.from_user.id)
    await callback_query.answer()