import struct

def read(socket, callback):
    buff=""
    size=0
    while True:
        buff+=socket.read(1024)
        if len(buff)>=size + 4:
            if size>0:
                callback(buff[:size].decode('utf-8'))
            size,=struct.unpack("!L", buff[size:size+4])
            buff=buff[size+4:]

def listen_for_messages():
    while True:
        message = s.recv(1024).decode()
        print(message)
        # bot(message)

import socket
from threading import Thread
from rich import print

print('\n'*100)

# server's IP address
SERVER_HOST = "0.0.0.0"
SERVER_PORT = 5002 # port we want to use
separator_token = "<SEP>" # we will use this to separate the client name & message

# initialize list/set of all connected client's sockets
client_sockets = set()
# create a TCP socket
s = socket.socket()
# make the port as reusable port
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
# bind the socket to the address we specified

def listen_for_messages(socket=s, callback=print):
    buff=b""
    size=0
    while True:
        buff+=(socket.recv(1024))
        if size == 0:
            size,=struct.unpack("!L", buff[:4])
            buff=buff[4:]
        if len(buff)>=size:
            callback(buff[:size].decode('utf-8'))
            buff=buff[size:]
            size=0\