from tgbot.handlers.send_button import show_request_msg
from tgbot.api import unmute_member, mute_member, delete_message
from tgbot.storage import Profile

def handle_join(msg):
    chat_id = str(msg['chat']['id'])
    from_id = str(msg['from']['id'])

    actor = Profile.get(from_id, msg)

    newcomer_id = str(msg['new_chat_member']['id'])
    if from_id == newcomer_id:
        if len(actor['parents']) == 0:
            # показываем сообщение с кнопкой "поручиться"
            request_msg_id = show_request_msg(msg)

            # до одобрения - мьют
            r = mute_member(chat_id, newcomer_id)
            print(r)

            # обновляем профиль присоединившегося
            actor['request_msg_id'] = f'{chat_id}:{request_msg_id}'
            Profile.save(actor)
        else:
            # за пользователя поручились ранее
            pass
    else:
        # пользователи приглашены другим участником
        print(f'{len(msg["new_chat_members"])} members were invited by {from_id}')    
        for m in msg['new_chat_members']:
            newcomer = Profile.get(m['id'])
            newcomer['parents'].append(from_id)
            Profile.save(newcomer)
            actor['children'].append(m['id'])
            r = unmute_member(chat_id, newcomer['id'])
            print(r)

        # обновляем профиль пригласившего
        Profile.save(actor)


def handle_left(msg):
    print(f'handling member leaving')
    member_id = msg["left_chat_member"]["id"]
    chat_id = msg['chat']['id']

    # профиль покидающего чат
    leaver = Profile.get(member_id)

    # удаление сообщения с кнопкой в этом чате
    if leaver['request_msg_id'].startswith(chat_id):
        chat_id, rmid = leaver['request_msg_id'].split(':')
        r = delete_message(chat_id, rmid)
        print(r)

    Profile.leaving(leaver)