from abc import ABC, abstractmethod


class IHost(ABC):

    @abstractmethod
    def get_id(self):
        """
        :return: peer_id of host
        """
        pass

    @abstractmethod
    def get_network(self):
        """
        :return: network instance of host
        """
        pass

    @abstractmethod
    def get_mux(self):
        """
        :return: mux instance of host
        """
        pass

    @abstractmethod
    def set_stream_handler(self, protocol_id, stream_handler):
        """
        set stream handler for host
        :param protocol_id: protocol id used on stream
        :param stream_handler: a stream handler function
        :return: true if successful
        """
        pass

    # protocol_id can be a list of protocol_ids
    # stream will decide which protocol_id to run on
    @abstractmethod
    def new_stream(self, peer_id, protocol_id):
        """
        :param peer_id: peer_id that host is connecting
        :param proto_id: protocol id that stream runs on
        :return: true if successful
        """
        pass