server.py

from typing import Tuple, List
import socket
import threading
import queue

PORT: int = 1234
SERVER_NAME: str = "localhost"
ADDRESS: Tuple[str, int] = (SERVER_NAME, PORT)
DATA_FORMAT: str = "utf-8"

server: socket.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server.bind(ADDRESS)
messages = queue.Queue()
clients: List = []
num_clients: int = -1


def receive():
    while True:
        try:
            message, client_address = server.recvfrom(1024)
            messages.put((message, client_address))
        except:
            pass


def broadcast():
    while True:
        while True:
            message, client_address = messages.get()
            print(f"message: {message.decode(DATA_FORMAT)}")
            print(f"client_address: {client_address}")
            if message.decode(DATA_FORMAT).startswith("SIGNUP_TAG:"):
                name = message.decode(DATA_FORMAT)[message.decode(DATA_FORMAT).index(":")+1:]
                server.sendto(f"{name} joined!".encode(DATA_FORMAT), client_address)
                clients.append(client_address)
            if len(clients) == num_clients:
                print(f"there are {num_clients} clients")
                break

        # send to client 1
        # send to client 2
        # print("send connection information to both clients")
        # client1_info: str = f"INFO:0, {clients[1][0]}, {clients[1][1]}"
        # client2_info: str = f"INFO:1, {clients[0][0]}, {clients[0][1]}"
        # server.sendto(client1_info.encode(DATA_FORMAT), clients[0])
        # server.sendto(client2_info.encode(DATA_FORMAT), clients[1])
        # print("server finished its jobs, terminate")

        for i in range(len(clients)):
            connection_information: str = f"INFO:{i}"
            for j in range(len(clients)):
                if i != j:
                    server.sendto(f"{connection_information}, {clients[j][0]}, {clients[j][1]}".encode(DATA_FORMAT), clients[i])
        
                


t1 = threading.Thread(target=receive)
t2 = threading.Thread(target=broadcast)
t1.start()
t2.start()
num_clients = int(input("Enter the number of clients in the counter: "))
print("server is starting ...")