py-libp2p/transport/tcp/tcp.py

import asyncio
from transport.transport_interface import ITransport
from transport.listener_interface import IListener
from network.connection.raw_connection import RawConnection

class TCP(ITransport):

    def __init__(self):
        self.listener = self.Listener()

    class Listener(IListener):

        def __init__(self, handler_function=None):
            self.multiaddrs = []
            self.server = None
            self.handler = staticmethod(handler_function)

        def listen(self, multiaddr):
            """
            put listener in listening mode and wait for incoming connections
            :param multiaddr: multiaddr of peer
            :return: return True if successful
            """
            # TODO check for exceptions
            if "ipfs" in multiaddr.get_protocols():
                # ipfs_id = multiaddr.get_ipfs_id()
                _multiaddr = multiaddr.remove_protocol("ipfs")

            self.multiaddrs.append(multiaddr)
            multiaddr_dict = _multiaddr.to_options()
            loop = asyncio.get_event_loop()
            coroutine = asyncio.start_server(self.handler, multiaddr_dict.host,\
                multiaddr_dict.port, loop=loop)
            self.server = loop.run_until_complete(coroutine)
            return True

        def get_addrs(self):
            """
            retrieve list of addresses the listener is listening on
            :return: return list of addrs
            """
            # TODO check if server is listening
            return self.multiaddrs

        def close(self, options=None):
            """
            close the listener such that no more connections
            can be open on this transport instance
            :param options: optional object potential with timeout
            a timeout value in ms that fires and destroy all connections
            :return: return True if successful
            """
            if self.server is None:
                return False
            self.server.close()
            _loop = asyncio.get_event_loop()
            _loop.run_until_complete(self.server.wait_closed())
            _loop.close()
            self.server = None
            return True

    def dial(self, multiaddr, options=None):
        """
        dial a transport to peer listening on multiaddr
        :param multiaddr: multiaddr of peer
        :param options: optional object
        :return: True if successful
        """
        _multiaddr_dict = multiaddr.to_dict()
        host = _multiaddr_dict.host
        port = _multiaddr_dict.port
        reader, writer = open_conn(host, port)
        return RawConnection(host, port, reader, writer)

    def create_listener(self, handler_function, options=None):
        """
        create listener on transport
        :param options: optional object with properties the listener must have
        :param handler_function: a function called when a new conntion is received
        that takes a connection as argument which implements interface-connection
        :return: a listener object that implements listener_interface.py
        """
        return self.Listener(handler_function)

async def open_conn(host, port):
    reader, writer = await asyncio.open_connection(host, port)
    return reader, writer