from typing import Awaitable, Callable, Union

from libp2p.peer.id import ID

from .pb import rpc_pb2

SyncValidatorFn = Callable[[ID, rpc_pb2.Message], bool]
AsyncValidatorFn = Callable[[ID, rpc_pb2.Message], Awaitable[bool]]
ValidatorFn = Union[SyncValidatorFn, AsyncValidatorFn]

UnsubscribeFn = Callable[[], Awaitable[None]]