[WEBSOCKET]: Channels

Author: amadolid

jac-cloud/jac_cloud/jaseci/__init__.py

@@ -62,7 +62,7 @@ async def lifespan(app: _FaststAPI) -> AsyncGenerator[None, _FaststAPI]:
             populate_yaml_specs(cls.__app__)
 
             from .routers import healthz_router, sso_router, user_router
-            from ..plugin.jaseci import walker_router, websocket_router
+            from ..plugin.implementation import walker_router, websocket_router
 
             for router in [
                 healthz_router,

jac-cloud/jac_cloud/jaseci/dtos/__init__.py

@@ -8,6 +8,14 @@
     UserResetPassword,
     UserVerification,
 )
+from .websocket import (
+    ChannelEvent,
+    ConnectionEvent,
+    UserEvent,
+    WalkerEvent,
+    WalkerEventData,
+    WebSocketEvent,
+)
 
 
 __all__ = [
@@ -18,4 +26,10 @@
     "UserRequest",
     "UserResetPassword",
     "UserVerification",
+    "ChannelEvent",
+    "ConnectionEvent",
+    "UserEvent",
+    "WalkerEvent",
+    "WalkerEventData",
+    "WebSocketEvent",
 ]

jac-cloud/jac_cloud/jaseci/dtos/websocket.py

@@ -0,0 +1,51 @@
+"""Jaseci SSO DTOs."""
+
+from typing import Annotated, Any, Literal, Union
+
+from pydantic import BaseModel, Field
+
+
+class ConnectionEvent(BaseModel):
+    """Connection Event Model."""
+
+    type: Literal["connection"]
+
+
+class WalkerEventData(BaseModel):
+    """Walker Event Data Model."""
+
+    walker: str
+    node: str | None = None
+    context: dict[str, Any]
+
+
+class WalkerEvent(BaseModel):
+    """Walker Event Model."""
+
+    type: Literal["walker"]
+    data: WalkerEventData
+
+
+class UserEvent(BaseModel):
+    """Walker Event Model."""
+
+    type: Literal["user"]
+    root_id: str
+    data: dict
+
+
+class ChannelEvent(BaseModel):
+    """Walker Event Model."""
+
+    type: Literal["channel"]
+    channel_id: str
+    data: dict
+
+
+class WebSocketEvent(BaseModel):
+    """WebSocket Event."""
+
+    event: Annotated[
+        Union[ConnectionEvent, WalkerEvent, UserEvent, ChannelEvent],
+        Field(discriminator="type"),
+    ]

jac-cloud/jac_cloud/plugin/__init__.py

@@ -1,6 +1,6 @@
 """Jaseci Plugins."""
 
-from .jaseci import specs
+from .implementation import WEBSOCKET_MANAGER, specs
 
 
-__all__ = ["specs"]
+__all__ = ["WEBSOCKET_MANAGER", "specs"]

jac-cloud/jac_cloud/plugin/implementation/__init__.py

@@ -0,0 +1,6 @@
+"""Jaseci Plugin Implementations."""
+
+from .api import specs, walker_router
+from .websocket import WEBSOCKET_MANAGER, websocket_router
+
+__all__ = ["specs", "walker_router", "WEBSOCKET_MANAGER", "websocket_router"]