from bot.Handler import Handler
from db.services.ChatService import ChatService
from bot import SET_EMAIL_STEP
from bot.DaMsg import DaMsg
from bot.msg.SetEmailMsg import SetEmailMsg


class SetRoleHandler(Handler):

    @staticmethod
    async def set_partner(update, context):
        chat = ChatService.get_by_chat_id(update.effective_chat.id)
        ChatService.edit(chat, "status", 20)

        msg = SetEmailMsg(update.effective_message)

        await context.bot.send_message(chat_id=update.effective_chat.id, text=msg.get_msg())

        return SET_EMAIL_STEP

    @staticmethod
    async def set_developer(update, context):
        chat = ChatService.get_by_chat_id(update.effective_chat.id)
        ChatService.edit(chat, "status", 10)

        msg = SetEmailMsg(update.effective_message)

        await context.bot.send_message(chat_id=update.effective_chat.id, text=msg.get_msg())

        return SET_EMAIL_STEP

    @staticmethod
    async def set_email(update, context):
        chat = ChatService.get_by_chat_id(update.effective_chat.id)
        msg = DaMsg(update.effective_message)
        ChatService.edit(chat, "email", msg.text)

        await context.bot.send_message(chat_id=update.effective_chat.id, text="Спасибо)")

        return SET_EMAIL_STEP