import discord
import asyncio

client = discord.Client()
server = None
lobbyChannel = None

def init():
    print("Starting the bot")
    global client
    client.run("XXX")

@client.event
async def on_ready():
    global client, server, lobbyChannel

    print('Logged in as')
    print(client.user.name)
    print(client.user.id)
    print('------')

    lobbyChannel = client.get_channel("234378707853246464") #client.get_channel("234378707853246464")
    server = client.get_server("234365566608080896")

    client.loop.create_task(handle_lobby())

@client.event
async def on_member_join(member):
    await client.send_message(lobbyChannel, "Witaj {}! Napisz tutaj swój typ osobowości, a zostanie Ci przydzielona ranga.".format(member.mention))

@asyncio.coroutine
async def handle_lobby():
    valid_types = [ "ENFJ", "ENFP", "ENTJ", "ENTP", "ESFJ", "ESFP", "ESTJ", "ESTP", "INFJ", "INFP", "INTJ", "INTP", "ISFJ", "ISFP", "ISTJ", "ISTP" ]
    type_roles = { }

    await client.wait_until_ready()

    for role in server.roles:
        if role.name.upper() in valid_types:
            type_roles[role.name.upper()] = role

    # await client.send_message(lobbyChannel, "Bot został uruchomiony!")

    while not client.is_closed:
        msg = await client.wait_for_message(channel=lobbyChannel)

        if msg.author == client.user:
            continue

        for type_str in valid_types:
            if type_str in msg.content.upper():
                await client.remove_roles(msg.author, *type_roles.values())
                await client.add_roles(msg.author, type_roles[type_str])

                await client.send_message(lobbyChannel, msg.author.mention + ", od teraz twój typ to **" + type_str + "**")

                break


init()