import config
import logger

from getsocks import get_socks
from torpeer import TorPeer


def on_announce_rec(api, data=None):
    socks_address, socks_port = get_socks()[0]

    announced: str = data['address']
    try:
        announced = announced.decode('utf-8')
    except AttributeError:
        pass
    announced = announced.strip()

    if announced.removesuffix('.onion') == config.get(
            'tor.transport_address', '').removesuffix('.onion'):
        logger.warn(
            "Received announcement for our own node, which shouldn't happen")
        return

    if not announced.endswith('.onion'):
        announced += '.onion'

    logger.info(f"Peer {announced} announced to us.", terminal=True)

    data['callback'](TorPeer(socks_address, socks_port, announced))