refactored GatewayContext

Author: eadwinCode

ellar/socket_io/context.py

@@ -0,0 +1,51 @@
+import typing as t
+
+from engineio import AsyncServer
+
+from ellar.common_types import TReceive, TScope, TSend
+from ellar.core import App, IExecutionContext
+from ellar.core.context import IHTTPHostContext, IWebSocketHostContext
+from ellar.di import EllarInjector
+
+if t.TYPE_CHECKING:  # pragma: no cover
+    from .model import GatewayBase
+
+
+class GatewayContext(IExecutionContext):
+    def __init__(
+        self,
+        server: AsyncServer,
+        sid: str,
+        message: t.Any,
+        context: "IExecutionContext",
+        environment: t.Dict,
+    ) -> None:
+        self.server = server
+        self.sid = sid
+        self.message = message
+        self._context = context
+        self.environment = environment
+
+    def get_handler(self) -> t.Callable:
+        return self._context.get_handler()
+
+    def get_class(self) -> t.Optional[t.Type["GatewayBase"]]:  # type: ignore[override]
+        return t.cast(t.Type["GatewayBase"], self._context.get_class())
+
+    def get_service_provider(self) -> EllarInjector:  # pragma: no cover
+        return self._context.get_service_provider()
+
+    def switch_to_http_connection(self) -> IHTTPHostContext:  # pragma: no cover
+        return self._context.switch_to_http_connection()
+
+    def switch_to_websocket(self) -> IWebSocketHostContext:  # pragma: no cover
+        return self._context.switch_to_websocket()
+
+    def get_app(self) -> App:  # pragma: no cover
+        return self._context.get_app()
+
+    def get_type(self) -> str:  # pragma: no cover
+        return self._context.get_type()
+
+    def get_args(self) -> t.Tuple[TScope, TReceive, TSend]:  # pragma: no cover
+        return self._context.get_args()

ellar/socket_io/model.py

@@ -1,22 +1,9 @@
-import typing as t
-
-from engineio import AsyncServer
-
-if t.TYPE_CHECKING:
-    from ellar.core import IExecutionContext
+from .context import GatewayContext
 
 
 class GatewayType(type):
     pass
 
 
-class GatewayContext(t.NamedTuple):
-    server: AsyncServer
-    sid: str
-    message: t.Any
-    context: "IExecutionContext"
-    environment: t.Dict
-
-
 class GatewayBase(metaclass=GatewayType):
     context: GatewayContext