import traceback
from gossip.commands import GossipCommands, command_to_byte
from .peerset import gossip_peer_set

import logger


def connect_peer(peer):
    if peer in gossip_peer_set:
        return
    try:
        s = peer.get_socket(120)
    except Exception:
        logger.warn(f"Could not connect to {peer.transport_address}")
        logger.warn(traceback.format_exc())
    else:
        with s:
            s.sendall(command_to_byte(GossipCommands.PING))

            if s.recv(4).decode('utf-8') == 'PONG':
                gossip_peer_set.add(peer)
                logger.info(f"connected to {peer.transport_address}", terminal=True)