from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
from telegram.ext import CallbackContext, ConversationHandler, filters

from model.base import ServiceEnum
from model.genshinhelper import YuanShen, Genshin
from plugins.base import BasePlugins
from service import BaseService
from service.base import UserInfoData


class SignCommandData:
    user_info: UserInfoData = UserInfoData()
    chat_id: int = 0
    reply_to_message_id: int = 0


class Sign(BasePlugins):
    def __init__(self, service: BaseService):
        super().__init__(service)
        self._sign_y = YuanShen()
        self._sing_g = Genshin()

    CHECK_SERVER, COMMAND_RESULT = range(10400, 10402)

    async def _start_sign(self, uid: int, cookies: dict, service: ServiceEnum) -> str:
        if service == ServiceEnum.MIHOYO:
            sign_api = self._sign_y
        else:
            sign_api = self._sing_g
        sign_give = await sign_api.get_sign_give(cookies=cookies)
        if sign_give.error:
            return f"获取签到信息失败，API返回信息为 {sign_give.message}"
        is_sign = await sign_api.is_sign(uid, cookies=cookies)
        if is_sign.error:
            return f"获取签到状态失败，API返回信息为 {is_sign.message}"
        total_sign_day = is_sign.data["total_sign_day"]
        award_name = sign_give.data["awards"][total_sign_day]["name"]
        award_cnt = sign_give.data["awards"][total_sign_day]["cnt"]
        today = is_sign.data["today"]
        if not is_sign.data["is_sign"]:
            sign = await sign_api.sign(uid, cookies=cookies)
            if sign.code == 0:
                result = "OK"
            elif sign.code == -5003:
                result = "今天旅行者已经签到过了~"
            else:
                result = f"签到失败 返回错误代码为 {sign.code}"
        else:
            result = "今天旅行者已经签到过了~"
        message = f"###### {today} ######\n" \
                  f"UID: {uid}\n" \
                  f"今日奖励: {award_name} × {award_cnt}\n" \
                  f"签到结果: {result}"
        return message

    async def command_start(self, update: Update, context: CallbackContext) -> int:
        user = update.effective_user
        message = update.message
        sign_command_data: SignCommandData = context.chat_data.get("sign_command_data")
        if sign_command_data is None:
            sign_command_data = SignCommandData()
            context.chat_data["sign_command_data"] = sign_command_data
        user_info = await self.service.user_service_db.get_user_info(user.id)
        if user_info.service == ServiceEnum.NULL:
            message = "请选择你要签到的服务器"
            keyboard = [
                [
                    InlineKeyboardButton("miHoYo", callback_data="miHoYo"),
                    InlineKeyboardButton("HoYoLab", callback_data="HoYoLab")
                ]
            ]
            sign_command_data.user_info = user_info
            await update.message.reply_text(message, reply_markup=InlineKeyboardMarkup(keyboard))
            sign_command_data.chat_id = update.message.chat_id
            sign_command_data.reply_to_message_id = update.message.message_id
            return self.COMMAND_RESULT
        else:
            sign = await self._start_sign(user_info.mihoyo_game_uid, user_info.mihoyo_cookie, user_info.service)
            reply_message = await message.reply_text(sign)
            if filters.ChatType.GROUPS.filter(update.callback_query.message):
                self._add_delete_message_job(context, reply_message.chat_id, reply_message.message_id)
        return ConversationHandler.END

    async def command_result(self, update: Update, context: CallbackContext) -> int:
        sign_command_data: SignCommandData = context.chat_data["sign_command_data"]
        user_info = sign_command_data.user_info
        query = update.callback_query
        await query.answer()
        message = "签到失败"
        if query.data == "miHoYo":
            message = await self._start_sign(user_info.mihoyo_game_uid, user_info.mihoyo_cookie, ServiceEnum.MIHOYO)
        if query.data == "HoYoLab":
            message = await self._start_sign(user_info.hoyoverse_game_uid, user_info.hoyoverse_cookie,
                                             ServiceEnum.HOYOLAB)
        await query.edit_message_text(message)
        if filters.ChatType.GROUPS.filter(update.callback_query.message):
            self._add_delete_message_job(context, query.message.chat_id, query.message.message_id)
        return ConversationHandler.END