# Copyright 2018 John Reese
# Licensed under the MIT license

import inspect
from typing import Awaitable, Union

from typing_extensions import Protocol

from .types import T


class Orderable(Protocol):  # pragma: no cover
    def __lt__(self, other):
        ...

    def __gt__(self, other):
        ...


async def maybe_await(object: Union[Awaitable[T], T]) -> T:
    if inspect.isawaitable(object):
        return await object  # type: ignore
    return object  # type: ignore