from .template import *

import datetime

FIRST = 1

class Weather(BotFunc):
    """Shows a weatherforecast for a given location"""
    def __init__(self, api, db):
        """initialize api and persistence"""
        super().__init__(db)
        self.api = api
        self.city = ""


    def create_handler(self):
        """returns the handlers with button-logic"""
        conv_handler = ConversationHandler(
            entry_points=[CommandHandler('weather', self.entry_point)],
            states={
                FIRST: [
                    CallbackQueryHandler(self.choose_city, pattern="^city-"),
                    CallbackQueryHandler(self.choose_time, pattern="^time-"),
                ]
            },
            fallbacks=[CommandHandler('weather', self.entry_point)],
        )

        return conv_handler


    def entry_point(self, update: Update, context: CallbackContext) -> None:
        super().entry_point(update, context)
        """Reacts the call of the command. Prints the first buttons"""
        keyboard = [
            [
                InlineKeyboardButton("Zürich", callback_data="city-zurich"),
                InlineKeyboardButton("Freiburg", callback_data="city-freiburg"),
                InlineKeyboardButton("Mulhouse", callback_data="city-mulhouse"),
            ]
        ]
        reply_markup = InlineKeyboardMarkup(keyboard)
        if update.message:
            update.message.reply_text("Which city?", reply_markup=reply_markup)
        else:
            update.callback_query.edit_message_text("Which city", reply_markup=reply_markup)
        return FIRST


    def choose_city(self, update: Update, context: CallbackContext) -> None:
        """Prompt same text & keyboard as `start` does but not as new message"""
        # Get CallbackQuery from Update
        query = update.callback_query
        data = query.data
        self.city = data.replace("city-","")
        query.answer()
        keyboard = [
            [
                InlineKeyboardButton("Now", callback_data="time-now"),
                InlineKeyboardButton("Tomorrow", callback_data="time-tomorrow"),
                InlineKeyboardButton("7 days", callback_data="time-7"),
            ]
        ]
        reply_markup = InlineKeyboardMarkup(keyboard)
        query.edit_message_text(
            text = "Which time?", reply_markup=reply_markup
        )
        return FIRST


    def choose_time(self, update: Update, context: CallbackContext) -> None:
        """Show new choice of buttons"""
        query = update.callback_query
        query.answer()
        forecast_time = query.data.replace("time-","")
        weather = self.get_weather(self.city, forecast_time)
        query.edit_message_text(
            text = "Weather: \n\n" + weather,
            parse_mode = ParseMode.HTML
        )
        super().log_activity(read = True, execute = True, send = True)
        return ConversationHandler.END


    def get_weather(self, city, forecast_time) -> None:
        """get the weather that matches the given params"""
        locations = {"freiburg": [47.9990, 7.8421], "zurich": [47.3769, 8.5417], "mulhouse": [47.7508, 7.3359]}

        city = locations[city]
        
        categories = {"Clouds": "☁", "Rain": "🌧", "Thunderstorm": "🌩", "Drizzle": ":droplet:", "Snow": "❄", "Clear": "☀", "Mist": "🌫", "Smoke": "Smoke", "Haze": "Haze", "Dust": "Dust", "Fog": "Fog", "Sand": "Sand", "Dust": "Dust", "Ash": "Ash", "Squall": "Squall", "Tornado": "Tornado",}
        days = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
        today = datetime.datetime.today().weekday()
        weather = self.api.show_weather(city)
        message = ""
        if forecast_time == "now" or forecast_time == "7":
            now = weather.pop(0)
            message += "<b>Now:</b> " + categories[now["short"]] + "\n"
            message += "🌡" + str(now["temps"][0]) + "°\n\n"
            tod = weather.pop(0)
            message += "<b>" + "Today" + ":</b> " + categories[tod["short"]] + "\n"
            message += "🌡 ❄ " + str(tod["temps"][0]) + "° , 🌡 🔥 " + str(tod["temps"][1]) + "°\n\n"

        if forecast_time == "tomorrow" or forecast_time == "7":
            if forecast_time == "tomorrow": # previous statement was not executed: tomorrow is at weather[2]
                tom = weather.pop(2)
            else:
                tom = weather.pop(0)
            message += "<b>" + "Tomorrow" + ":</b> " + categories[tom["short"]] + "\n"
            message += "🌡 ❄ " + str(tom["temps"][0]) + "° , 🌡 🔥 " + str(tom["temps"][1]) + "°\n\n"

        if forecast_time == "7":
            for i, day in enumerate(weather):
                message += "<b>" + days[(today + i + 2) % 7] + ":</b> " + categories[day["short"]] + "\n"
                message += "🌡 ❄ " + str(day["temps"][0]) + "° , 🌡 🔥 " + str(day["temps"][1]) + "°\n\n"

        return message