2022-02-24 07:03:50 +00:00
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
|
|
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
|
|
from .. import Peer
|
|
|
|
|
|
|
|
from ..commands import GossipCommands, command_to_byte
|
|
|
|
import onionrplugins
|
|
|
|
|
|
|
|
|
|
|
|
def do_announce(peer_set):
|
|
|
|
"Announce with N peers of each identified transport"
|
|
|
|
def _announce(announce_peer: 'Peer', our_transport_address: str):
|
|
|
|
sock = announce_peer.get_socket()
|
|
|
|
sock.send(
|
|
|
|
command_to_byte(GossipCommands.ANNOUNCE) + our_transport_address)
|
|
|
|
if sock.dup
|
|
|
|
|
|
|
|
|
|
|
|
per_transport = 3
|
|
|
|
peer_types = {}
|
|
|
|
count_for_peer = 0
|
|
|
|
for peer in peer_set:
|
|
|
|
try:
|
|
|
|
count_for_peer = peer_types[peer.__name__]
|
|
|
|
except KeyError:
|
|
|
|
peer_types[peer.__name__] = 0
|
|
|
|
continue
|
|
|
|
|
|
|
|
if count_for_peer == per_transport:
|
|
|
|
continue
|
|
|
|
|
|
|
|
onionrplugins.events.event(
|
|
|
|
'get_our_transport',
|
|
|
|
data={'callback': _announce, 'peer': peer})
|
|
|
|
|
|
|
|
peer_types[peer.__name__] += 1
|