diff --git a/server.py b/server.py index 3c0f9fb..60f5bc4 100644 --- a/server.py +++ b/server.py @@ -70,7 +70,7 @@ def fmt(msg: str, name: str = "", action: bool = False) -> str: return f" {name}{' '*(20-len(name))}: {msg}" -async def handle_client(reader, writer): +async def handleClient(reader, writer): try: global G writer.write(b"Please identify yourself. Nick limit is 20 chars.\n") @@ -150,19 +150,28 @@ async def handle_client(reader, writer): G.serverLinks += 1 G.servers[sName] = [] msgIndex = 0 - writer.write("I Awaiting client listing.\n") + writer.write("I am awaiting your client listing.\n") while 1: - client = raw((await reader.read(967)).decode("utf8")) + client = raw((await reader.read(1024)).decode("utf8")) if client == f"END OF CLIENT LISTING FROM {sName}": break if client in G.servers[sName] or client in G.clientsConnected: writer.write(b"K Client rejected: Already exists\n") await writer.drain() continue - writer.write(b"I Added client.\n") + writer.write(b"I added that client.\n") await writer.drain() G.msgs.append(log(f"{client} has connected from {sName}")) G.servers[sName].append(client) + writer.write(f"{G.remoteID}\n".encode("utf8")) + await writer.drain() + await reader.read() + for client in G.clientsConnected: + writer.write(f"{client}\n".encode("utf8")) + resp = raw((await reader.read(1024)).decode("utf8")) + if resp.startswith("K"): + G.killList[client] = True + writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}\n".encode("utf8")) G.clientsConnected.extend(G.servers[sName]) while 1: try: @@ -184,7 +193,7 @@ async def handle_client(reader, writer): G.clientsConnected.append(cName) writer.write(b"I Mmm... Pineapples\n") else: - writer.write(b"K Nick Collision") + writer.write(f"K {cName}".encode("utf8")) case "-": cName = buffer[2:] G.msgs.append(log(f"{cName} has disconnected from {sName}")) @@ -251,18 +260,69 @@ async def connectServer(hostname: str, port: int): for client in G.clientsConnected: writer.write(f"{client}\n".encode("utf8")) await writer.drain() - response = await reader.read(1024) - if response.startswith("K"): + resp = raw((await reader.read(1024)).decode("utf8")) + if resp.startswith("K"): G.killList[client] = True - writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}".encode("utf8")) + writer.write(f"END OF CLIENT LISTING FROM {G.remoteID}\n".encode("utf8")) await writer.drain() + rID = raw((await reader.read(1024)).decode("utf8")) + G.serverLinks += 1 + G.servers.append(rID) + writer.write(b"I recieved your remote ID, now awaiting client listing\n") # recieve client list from the other server - # copy the handling code from the S2S section above + while 1: + client = raw((await reader.read(1024)).decode("utf8")) + if client == f"END OF CLIENT LISTING FROM {rID}": + break + if client in G.servers[rID] or client in G.clientsConnected: + writer.write(b"K Client rejected: Already exists\n") + await writer.drain() + continue + writer.write(b"I added that client.\n") + await writer.drain() + G.msgs.append(log(f"{client} has connected from {rID}")) + G.servers[rID].append(client) + G.clientsConnected.extend(G.servers[rID]) + while 1: + try: + rawMsg = await asyncio.wait_for(reader.read(967), 0.1) + buffer = raw(rawMsg.decode("utf8")) + match buffer[0]: + case "S": + G.msgs.extend([log(buffer[2:])]) + writer.write(b"I Mmm... Blueberries\n") + case "I": + pass + case "+": + cName = buffer[2:] + if cName not in G.clientsConnected: + G.msgs.append(log(f"{cName} has connected from {rID}")) + G.servers[rID].append(cName) + G.clientsConnected.append(cName) + writer.write(b"I Mmm... Pineapples\n") + else: + writer.write(f"K {cName}\n".encode("utf8")) + case "-": + cName = buffer[2:] + G.msgs.append(log(f"{cName} has disconnected from {sName}")) + G.servers[rID].remove(cName) + G.clientsConnected.remove(cName) + case "M": + cName = buffer[2:].split("|", 1)[0] + message = buffer[2:].split("|", 1)[1] + G.msgs.append(log(f" {cName}: {message}")) + writer.write( + b"I Get these damn heretic ghost clients out of my store so i can buy my cult candles in peace." + ) + case _: + pass + except Exception: + pass async def runServer(port: int): global G - server = await asyncio.start_server(handle_client, "0.0.0.0", port) + server = await asyncio.start_server(handleClient, "0.0.0.0", port) log(f"Listening on port {port}...") G.msgs.append(log("Server startup")) links = [] @@ -304,4 +364,4 @@ def interruptCatch(s, f): signal.signal(signal.SIGINT, interruptCatch) -asyncio.run(run_server(port)) +asyncio.run(runServer(port))