feat: support coroutine checks in bot,wait_for

Without breaking!

Author: LordOfPolls

interactions/client/client.py

@@ -22,6 +22,7 @@
     Sequence,
     Type,
     Union,
+    Awaitable,
 )
 
 import interactions.api.events as events
@@ -935,6 +936,17 @@ async def stop(self) -> None:
         await self.http.close()
         await self._connection_state.stop()
 
+    async def _process_waits(self, event: events.BaseEvent) -> None:
+        if _waits := self.waits.get(event.resolved_name, []):
+            index_to_remove = []
+            for i, _wait in enumerate(_waits):
+                result = await _wait(event)
+                if result:
+                    index_to_remove.append(i)
+
+            for idx in sorted(index_to_remove, reverse=True):
+                _waits.pop(idx)
+
     def dispatch(self, event: events.BaseEvent, *args, **kwargs) -> None:
         """
         Dispatch an event.
@@ -954,15 +966,7 @@ def dispatch(self, event: events.BaseEvent, *args, **kwargs) -> None:
                         f"An error occurred attempting during {event.resolved_name} event processing"
                     ) from e
 
-        if _waits := self.waits.get(event.resolved_name, []):
-            index_to_remove = []
-            for i, _wait in enumerate(_waits):
-                result = _wait(event)
-                if result:
-                    index_to_remove.append(i)
-
-            for idx in sorted(index_to_remove, reverse=True):
-                _waits.pop(idx)
+        asyncio.create_task(self._process_waits(event))
 
         if "event" in self.listeners:
             # special meta event listener
@@ -976,7 +980,7 @@ async def wait_until_ready(self) -> None:
     def wait_for(
         self,
         event: Union[str, "BaseEvent"],
-        checks: Absent[Optional[Callable[..., bool]]] = MISSING,
+        checks: Absent[Optional[Union[Callable[..., bool], Callable[..., Awaitable[bool]]]]] = MISSING,
         timeout: Optional[float] = None,
     ) -> Any:
         """
@@ -989,7 +993,6 @@ def wait_for(
 
         Returns:
             The event object.
-
         """
         event = get_event_name(event)
 

interactions/models/internal/wait.py

@@ -1,24 +1,29 @@
-from asyncio import Future
-from typing import Callable, Optional
+from asyncio import Future, iscoroutinefunction
+from typing import Callable, Optional, Union, Awaitable
 
 __all__ = ("Wait",)
 
 
 class Wait:
     """Class for waiting for a future event to happen. Internally used by wait_for."""
 
-    def __init__(self, event: str, checks: Optional[Callable[..., bool]], future: Future) -> None:
-        self.event = event
-        self.checks = checks
-        self.future = future
+    def __init__(
+        self, event: str, checks: Optional[Union[Callable[..., bool], Callable[..., Awaitable[bool]]]], future: Future
+    ) -> None:
+        self.event: str = event
+        self.check: Optional[Union[Callable[..., bool], Callable[..., Awaitable[bool]]]] = checks
+        self.future: Future = future
 
-    def __call__(self, *args, **kwargs) -> bool:
+    async def __call__(self, *args, **kwargs) -> bool:
         if self.future.cancelled():
             return True
 
-        if self.checks:
+        if self.check:
             try:
-                check_result = self.checks(*args, **kwargs)
+                if iscoroutinefunction(self.check):
+                    check_result = await self.check(*args, **kwargs)
+                else:
+                    check_result = self.check(*args, **kwargs)
             except Exception as exc:
                 self.future.set_exception(exc)
                 return True