from aiogram import Bot, Dispatcher, types
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
from aiogram.utils import executor
from aiogram.dispatcher.filters import Text
from pomodoro_bot.user_manager import user_manager
from pomodoro_bot.timer_manager import TimerManager
import os

API_TOKEN = os.getenv("BOT_TOKEN")

bot = Bot(token=API_TOKEN)
dp = Dispatcher(bot)
timer_manager = TimerManager(bot)

menu_kb = InlineKeyboardMarkup(row_width=2)
menu_kb.add(
    InlineKeyboardButton("▶️ Pomodoro", callback_data="pomodoro"),
    InlineKeyboardButton("☕ Short Break", callback_data="shortbreak"),
    InlineKeyboardButton("😌 Long Break", callback_data="longbreak"),
    InlineKeyboardButton("⏹ Stop", callback_data="stop")
)

@dp.message_handler(commands=['start', 'help'])
async def send_welcome(message: Message):
    await message.reply("Привет! Я твой личный Pomodoro-бот. Выбирай действие:", reply_markup=menu_kb)

@dp.callback_query_handler(Text(equals=["pomodoro", "shortbreak", "longbreak", "stop"]))
async def handle_callback(call: types.CallbackQuery):
    if call.data == "pomodoro":
        await timer_manager.start_timer(call.from_user.id, 25*60, call.message.chat.id, 'Pomodoro')
        await call.message.answer("Начался 25-минутный Pomodoro! 🔥")
    elif call.data == "shortbreak":
        await timer_manager.start_timer(call.from_user.id, 5*60, call.message.chat.id, 'Short Break')
        await call.message.answer("Начался 5-минутный перерыв ☕")
    elif call.data == "longbreak":
        await timer_manager.start_timer(call.from_user.id, 15*60, call.message.chat.id, 'Long Break')
        await call.message.answer("Начался длинный перерыв 😌")
    elif call.data == "stop":
        await timer_manager.stop_timer(call.from_user.id)
        await call.message.answer("Таймер остановлен ⏹")
    await call.answer()

from pomodoro_bot.redis_client import init_redis

async def on_startup(dispatcher):
    await init_redis()

def run_bot():
    executor.start_polling(dp, skip_updates=True, on_startup=on_startup)