py-libp2p/libp2p/transport/upgrader.py

from libp2p.stream_muxer.mplex.mplex import Mplex


class TransportUpgrader:
    # pylint: disable=no-self-use

    def __init__(self, secOpt, muxerOpt):
        self.sec = secOpt
        self.muxer = muxerOpt

    def upgrade_listener(self, transport, listeners):
        """
        Upgrade multiaddr listeners to libp2p-transport listeners
        """

    def upgrade_security(self, conn, peer_id):
        """
        Upgrade conn to be a secured connection
        """
        # TODO: Do exchange to determine security module
        pass

    def upgrade_connection(self, conn, generic_protocol_handler, peer_id):
        """
        Upgrade raw connection to muxed connection
        """

        # For PoC, no security, default to mplex
        # TODO do exchange to determine multiplexer
        return Mplex(conn, generic_protocol_handler, peer_id)