import socket

def receive_messages(sock: socket.socket):
    while True:
        try:
            message = sock.recv(4096).decode()
            if not message:
                print("\nConnexion fermée.")
                sock.close()
                break
            if message.startswith("QUIT:"):
                print(f"\n{message.split(':', 1)[1]} s'est déconnecté.")
                sock.close()
                break
            print(f"\n{message}")
            print("Vous > ", end="", flush=True)
        except:
            break

def send_messages(sock: socket.socket, username: str):
    while True:
        try:
            message = input("Vous > ")
            if message.upper() == 'Q' \
            or message.upper() == 'QUIT':
                sock.sendall(f"QUIT:{username}".encode())
                break
            if message:
                complete_message = f"{username} > {message}"
                sock.sendall(complete_message.encode())
        except:
            break