import asyncio
import json
import logging
from websockets.asyncio.server import serve
from websockets.asyncio.server import broadcast

addr = "localhost"
port = 3636

logging.basicConfig(level=logging.INFO)

ulist = {}
ulist_connection_ids = {}
clients = []

class util:
    def error(code, listener):
        return json.dumps({
            "error": True,
            "code": code,
            "form": "helium-util",
            "listener": listener
        })
    
    def field_check(expects, gets):
        for i in expects:
            if i not in gets:
                return "malformedJson"
            if type(gets[i]) in [str, dict, list]:
                if len(gets[i]) not in expects[i]:
                    return "lengthInvalid"
        return True

async def handler(websocket):
    clients.append(websocket)
    async for message in websocket:
        try:
            r = json.loads(message)
        except:
            await websocket.send(util.error("malformedJson", None))
            continue
        if "listener" not in r:
            r["listener"] = None
        listener = r["listener"]
        if "command" not in r:
            await websocket.send(util.error("malformedJson", listener))
            continue
        if r["command"] == "register":
            fc = util.field_check({"username": range(1,21), "password": range(8,256), "invite_code": range(16,17)}, r)
            if fc != True:
                await websocket.send(util.error(fc, listener))
                continue
        else:
            await websocket.send(util.error("malformedJson", listener))
    if websocket in clients:
        clients.append(websocket)

async def main():
    async with serve(handler, addr, port) as server:
        await server.serve_forever()

asyncio.run(main())