import os
import random
from aiogram import Bot, Dispatcher, types, F
from aiogram.filters import Command
from aiogram.types import ReplyKeyboardMarkup, KeyboardButton
from fastapi import FastAPI
from contextlib import asynccontextmanager

# Загрузка токена из переменных окружения
BOT_TOKEN = os.getenv("BOT_TOKEN")
if not BOT_TOKEN:
    raise ValueError("Не задан токен бота. Укажите его в переменной окружения BOT_TOKEN")

# Инициализация бота и диспетчера
bot = Bot(token=BOT_TOKEN)
dp = Dispatcher()

# Создаем клавиатуру
keyboard = ReplyKeyboardMarkup(
    keyboard=[
        [KeyboardButton(text="Завтрак"), KeyboardButton(text="Обед")],
        [KeyboardButton(text="Ужин"), KeyboardButton(text="Прочее")]
    ],
    resize_keyboard=True,
    input_field_placeholder="Выберите тип блюда"
)

# Функция для получения случайного блюда из файла
def get_random_dish(meal_type: str) -> str:
    filename = f"{meal_type.lower()}.txt"
    try:
        with open(filename, 'r', encoding='utf-8') as file:
            dishes = file.read().splitlines()
            if dishes:
                return random.choice(dishes)
            return f"Файл {filename} пуст"
    except FileNotFoundError:
        return f"Файл с блюдами для {meal_type} не найден"

# Обработчик команды /start
@dp.message(Command("start"))
async def start_handler(message: types.Message):
    await message.answer(
        "Привет! Я помогу выбрать блюдо. Выбери тип:",
        reply_markup=keyboard
    )

# Обработчик кнопок
@dp.message(F.text.in_(["Завтрак", "Обед", "Ужин", "Прочее"]))
async def meal_handler(message: types.Message):
    meal_type = message.text
    dish = get_random_dish(meal_type)
    await message.answer(f"📌 Для {meal_type}а рекомендую:\n\n<b>{dish}</b>", parse_mode="HTML")

# Инициализация FastAPI
@asynccontextmanager
async def lifespan(app: FastAPI):
    await bot.delete_webhook()
    await bot.set_webhook(f"{BASE_URL}/webhook")
    yield
    await bot.session.close()

app = FastAPI(lifespan=lifespan)

@app.post("/webhook")
async def webhook(update: dict):
    telegram_update = types.Update(**update)
    await dp.feed_update(bot=bot, update=telegram_update)

@app.get("/")
async def index():
    return {"status": "Bot is running"}

# Конфигурация (замените на ваш публичный URL)
BASE_URL = "https://bubu.frps.botforge.biz"