import os, logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
from telegram.error import BadRequest
from telegram import Update, User, InlineKeyboardMarkup, InlineKeyboardButton
from telegram.ext import Updater, CallbackContext, CallbackQueryHandler, CommandHandler, MessageHandler, PicklePersistence
from dotenv import load_dotenv
load_dotenv()
from Gardening import Plant, get_plant_info

def reply(update: Update, context: CallbackContext, text: str = "", markup: str = ""):
    return context.bot.send_message(chat_id=update.effective_chat.id, text=text, reply_markup=markup, parse_mode='Markdown')

def get_plant(context: CallbackContext, user_id: int):
        try:
            plant = context.bot_data[user_id]["plant"]
        except KeyError:
            return None
        
        plant.update()
        return plant

def start_handler(update: Update, context: CallbackContext):
    
    plant = get_plant(context, update.effective_user.id)
    new = False
    
    if plant is None:
        context.bot_data[update.effective_user.id] = { "plant" : Plant(update.effective_user.id) }
        new = True

    if plant.dead or plant.stage == 5:
        plant.start_over()
        new = True
    
    if new:
        return reply(update, context, "Hai piantato un nuovo seme! Adesso usa /water per innaffiarlo.")
    
    return reply(update, context, "La tua pianta non è ancora pronta per andarsene!")

def water(context: CallbackContext, user_id: int):
    plant = get_plant(context, user_id)
    
    if plant is None:
        return "Non hai nessuna pianta da innaffiare! Usa /start per piantarne una."

    if plant.dead:
        return "La pianta è morta..."
    
    plant.water()
    return "Pianta innaffiata."

def show(context: CallbackContext, user_id: int):
    plant = get_plant(context, user_id)
    
    if plant is None:
        return "Non hai nessuna pianta da mostrare! Usa /start per piantarne una."

    return get_plant_info(plant)

def water_handler(update: Update, context: CallbackContext):
    answer = water(context, update.effective_user.id)
    return reply(update, context, answer)

def show_handler(update: Update, context: CallbackContext):
    user_id = update.effective_user.id
    answer = show(context, user_id)
    
    markup = InlineKeyboardMarkup([[InlineKeyboardButton(text="Innaffia 🚰", callback_data=f"water {user_id}")]])
    reply(update, context, answer, markup)

def keyboard_handler(update: Update, context: CallbackContext):
    query = update.callback_query
    data = query.data
    
    if data.startswith("water"):
        user_id = int(data.split(" ")[1])
        answer = water(context, user_id)
        if user_id != update.effective_user.id:
            reply(update, context, f"{update.effective_user.full_name} ha innaffiato la pianta di qualcuno!")
        return query.answer(answer)
    
    return query.answer("Questo tasto non fa nulla.")

def main():
    
    updater = Updater(token=os.getenv("token"),
                      persistence=PicklePersistence(filename='bot-data.pkl',
                                                    store_user_data=False,
                                                    store_bot_data=True,
                                                    store_callback_data=False,
                                                    store_chat_data=False
                                                    ))
    
    dispatcher = updater.dispatcher
    
    
    # commands
    dispatcher.add_handler(CommandHandler('start', start_handler))
    dispatcher.add_handler(CommandHandler('water', water_handler))
    dispatcher.add_handler(CommandHandler('show', show_handler))
    
    dispatcher.add_handler(CallbackQueryHandler(callback=keyboard_handler))
    
    updater.start_polling()
    print(os.getenv("bot_name"))
    updater.idle()

if __name__ == "__main__":
    main()