initial commit

src/snakia/core/rx/async_bindable.py

@@ -0,0 +1,105 @@
+from typing import Any, Awaitable, Callable, Literal, overload
+
+from .base_bindable import BaseBindable, BindableSubscriber, ValueChanged
+
+
+class AsyncBindable[T: Any](BaseBindable[T]):
+    """
+    An asynchronous bindable.
+    """
+
+    def __init__(self, default_value: T | None = None) -> None:
+        super().__init__(default_value)
+        self.__subscribers: list[BindableSubscriber[T, Awaitable[Any]]] = []
+
+    @property
+    def value(self) -> T:
+        return self.__value
+
+    @property
+    def subscribers(
+        self,
+    ) -> tuple[BindableSubscriber[T, Awaitable[Any]], ...]:
+        """Get the subscribers."""
+        return (*self.__subscribers,)
+
+    async def set(self, value: T) -> None:
+        """Set the value."""
+        e = ValueChanged(self.__value, value)
+        self.__value = value
+        for subscriber in self.__subscribers:
+            await subscriber(e)
+
+    @overload
+    def subscribe(
+        self,
+        subscriber: BindableSubscriber[T, Awaitable[Any]],
+        /,
+        run_now: Literal[True],
+    ) -> Awaitable[None]: ...
+
+    @overload
+    def subscribe(
+        self,
+        subscriber: BindableSubscriber[T, Awaitable[Any]],
+        /,
+        run_now: Literal[False] = False,
+    ) -> None: ...
+
+    def subscribe(
+        self,
+        subscriber: BindableSubscriber[T, Awaitable[Any]],
+        /,
+        run_now: bool = False,
+    ) -> None | Awaitable[None]:
+        """Subscribe to an value."""
+        self.__subscribers.append(subscriber)
+        if run_now:
+
+            async def _run() -> None:
+                await subscriber(
+                    ValueChanged(self.__default_value, self.__value)
+                )
+
+            return _run()
+        return None
+
+    def unsubscribe(
+        self, subscriber: BindableSubscriber[T, Awaitable[Any]]
+    ) -> None:
+        """Unsubscribe from an value."""
+        self.__subscribers.remove(subscriber)
+
+    @overload
+    def on(
+        self, run_now: Literal[True]
+    ) -> Callable[
+        [BindableSubscriber[T, Awaitable[Any]]], Awaitable[None]
+    ]: ...
+
+    @overload
+    def on(
+        self, run_now: Literal[False] = False
+    ) -> Callable[[BindableSubscriber[T, Awaitable[Any]]], None]: ...
+
+    def on(self, run_now: bool = False) -> Callable[
+        [BindableSubscriber[T, Awaitable[Any]]],
+        None | Awaitable[None],
+    ]:
+        """Decorator to subscribe to an value."""
+
+        def wrapper(
+            subscriber: BindableSubscriber[T, Awaitable[Any]],
+        ) -> None | Awaitable[None]:
+            self.__subscribers.append(subscriber)
+            if run_now:
+
+                async def _run() -> None:
+                    await subscriber(
+                        ValueChanged(self.__default_value, self.__value)
+                    )
+
+                return _run()
+            return None
+
+        return wrapper