diff --git a/libp2p/__init__.py b/libp2p/__init__.py index 2e1cf88..38a285a 100644 --- a/libp2p/__init__.py +++ b/libp2p/__init__.py @@ -67,16 +67,18 @@ def initialize_default_swarm( if not id_opt: id_opt = generate_id() + # TODO parse transport_opt to determine transport transport_opt = transport_opt or ["/ip4/127.0.0.1/tcp/8001"] - transport = [multiaddr.Multiaddr(t) for t in transport_opt] - # TODO wire muxer up with swarm - # muxer = muxer_opt or ["mplex/6.7.0"] + transport = TCP() + + # TODO TransportUpgrader is not doing anything really + # TODO parse muxer and sec to pass into TransportUpgrader + muxer = muxer_opt or ["mplex/6.7.0"] sec = sec_opt or ["secio"] + upgrader = TransportUpgrader(sec, muxer) + peerstore = peerstore_opt or PeerStore() - upgrader = TransportUpgrader(sec, transport) - swarm_opt = Swarm(id_opt, peerstore, upgrader) - tcp = TCP() - swarm_opt.add_transport(tcp) + swarm_opt = Swarm(id_opt, peerstore, upgrader, transport) return swarm_opt diff --git a/libp2p/network/swarm.py b/libp2p/network/swarm.py index b30567b..ff62c52 100644 --- a/libp2p/network/swarm.py +++ b/libp2p/network/swarm.py @@ -12,14 +12,14 @@ from .connection.raw_connection import RawConnection class Swarm(INetwork): # pylint: disable=too-many-instance-attributes, cell-var-from-loop - def __init__(self, peer_id, peerstore, upgrader): + def __init__(self, peer_id, peerstore, upgrader, transport): self.self_id = peer_id self.peerstore = peerstore self.upgrader = upgrader + self.transport = transport self.connections = dict() self.listeners = dict() self.stream_handlers = dict() - self.transport = None # Protocol muxing self.multiselect = Multiselect() @@ -183,10 +183,6 @@ class Swarm(INetwork): return True return False - def add_transport(self, transport): - # TODO: Support more than one transport - self.transport = transport - def create_generic_protocol_handler(swarm): """ Create a generic protocol handler from the given swarm. We use swarm