import os

from tgbot.rest import delete_message, register_webhook, send_message, ban_member, unban_member
from sanic import Sanic
from sanic.response import json, text


app = Sanic()
WEBHOOK = os.environ.get('VERCEL_URL')
CHAT_ID = os.environ.get('CHAT_ID').replace("-", "-100")
WELCOME_MSG = os.environ.get('WELCOME_MSG') or 'Welcome! Press the wright button'

BUTTON_OK = os.environ.get('BUTTON_OK') or 'Ok'
BUTTON_NO = os.environ.get('BUTTON_NO') or 'No'

newcomers = {}


@app.route('/', methods=["GET", "PUT", "OPTIONS"])
async def register(req):
    r = register_webhook(WEBHOOK)
    print('\n\t\t\tWEBHOOK REGISTERED\n')
    return json(r.json())


@app.post('/')
async def handle(req):
    print(req)
    try:
        update = req.json
        print(update)
        msg = update.get('message', update.get('my_chat_member'))
        if msg['chat']['id'] == CHAT_ID:
            chat_id = msg['chat']['id']
            member_id = msg['user']['id']
            if msg['chat']['type'] == 'new_chat_member':
                print(f'new member {member_id}')
                newcomers[member_id] = 'newcomer'
                reply_markup = {
                    "inline_keyboard": [
                        [
                            {"text": BUTTON_NO, "callback_data": BUTTON_NO},
                            {"text": BUTTON_OK, "callback_data": BUTTON_OK}
                        ]
                    ]
                }
                welcome_msg_id = send_message(
                    chat_id,
                    WELCOME_MSG,
                    reply_to=msg['message_id'],
                    reply_markup=reply_markup
                )
                newcomers[member_id] = 'newcomer' + welcome_msg_id
            elif msg['chat']['type'] == 'text':
                data = newcomers[member_id]
                if data:
                    if data.startswith('newcomer'):
                        if 'text' in msg:
                            answer = msg['text']
                            if BUTTON_OK.lower() in answer.lower():
                                del newcomers[member_id]
                            else:
                                delete_message(msg['message_id'])
                        else:
                            callback_data = update['callback_query']['data']  
                            if callback_data == BUTTON_NO:
                                ban_member(member_id, chat_id)
                            elif callback_data == BUTTON_OK:
                                del newcomers[member_id]
                                delete_message(CHAT_ID, data.replace('newcomer', ''))
    except Exception:
        pass
    return text('ok')