import asyncio
import sys
import logging
sys.path.append("./")
import websockets
from websockets import serve, broadcast

CONNECTIONS = set()
stop = asyncio.Future()
global wsServer

logging.basicConfig(
    format="%(asctime)s %(message)s",
    level=logging.WARN,
)

async def connectHandler(websocket):
    CONNECTIONS.add(websocket)
    #print("clientConnected")
    try:
        await websocket.wait_closed()
    finally:
        CONNECTIONS.remove(websocket)
        #print("clientgone")

def message_all(message):
    broadcast(CONNECTIONS, message)
    #print(message)
async def serveWebSocket():
   wsServer = await serve(connectHandler, "", 1337)
        #await asyncio.Future()  # run forever
async def closeWebSocket():
    wsServer.close()