Migrated to dragonion_core

dragonion_server/modules/server/handlers/managers/connection.py

@@ -1,6 +1,6 @@
 from attrs import define
 from fastapi import WebSocket
-from ..objects.webmessage import (
+from dragonion_core.proto.web import (
     webmessages_union,
     webmessage_error_message_literal,
     WebErrorMessage,

dragonion_server/modules/server/handlers/managers/room.py

@@ -3,7 +3,7 @@ from .connection import Connection
 from typing import Dict
 from fastapi import WebSocket
 
-from ..objects.webmessage import (
+from dragonion_core.proto.web import (
     webmessages_union,
     WebMessageMessage,
     WebNotificationMessage,

dragonion_server/modules/server/handlers/managers/service.py

@@ -2,7 +2,7 @@ from .connection import Connection
 from .room import Room
 from typing import Dict
 
-from ..objects.webmessage import (
+from dragonion_core.proto.web import (
     webmessage_error_message_literal
 )
 

dragonion_server/modules/server/handlers/objects/__init__.py

@@ -1 +0,0 @@
-pass

dragonion_server/modules/server/handlers/objects/webmessage.py

@@ -1,69 +0,0 @@
-from typing import Literal, Final, Union
-from dataclasses_json import dataclass_json
-from dataclasses import dataclass
-
-
-webmessage_type_literal = Literal[
-    "connect", "message", "disconnect", "error", "notification"
-]
-webmessage_error_message_literal = Literal[
-    "unknown", "username_exists", "invalid_webmessage"
-]
-
-
-@dataclass_json
-@dataclass
-class _WebAnyMessage:
-    username: str | None = None
-    type: webmessage_type_literal = "message"
-    message: str | None = None
-    error_message: webmessage_error_message_literal | None = None
-
-
-@dataclass_json
-@dataclass
-class WebMessageMessage:
-    username: str
-    message: str
-    type: Final = "message"
-
-
-@dataclass_json
-@dataclass
-class WebErrorMessage:
-    error_message: webmessage_error_message_literal
-    type: Final = "error"
-
-
-@dataclass_json
-@dataclass
-class WebUserMessage:
-    type: Literal["connect", "disconnect"]
-    username: str
-
-
-@dataclass_json
-@dataclass
-class WebNotificationMessage:
-    message: str
-    type: Final = "notification"
-
-
-webmessages_union = Union[
-    WebMessageMessage,
-    WebErrorMessage,
-    WebUserMessage,
-    WebNotificationMessage
-]
-
-
-class WebMessage:
-    @staticmethod
-    def from_json(data) -> webmessages_union:
-        return {
-            "connect": WebUserMessage.from_json,
-            "disconnect": WebUserMessage.from_json,
-            "message": WebMessageMessage.from_json,
-            "error": WebErrorMessage.from_json,
-            "notification": WebNotificationMessage.from_json
-        }[_WebAnyMessage.from_json(data).type](data)

dragonion_server/modules/server/handlers/websocket_server.py

@@ -1,6 +1,6 @@
 from fastapi import WebSocket, WebSocketDisconnect
 from .managers.service import Service
-from .objects.webmessage import (
+from dragonion_core.proto.web import (
     webmessages_union,
     WebMessage
 )

dragonion_server/modules/server/integration.py

@@ -1,7 +1,7 @@
 import sys
 
 from dragonion_server.utils.onion import Onion
-from dragonion_server.utils.generated_auth.db import AuthFile
+from dragonion_core.proto.file import AuthFile
 from dragonion_server.utils.config.db import services
 
 from rich import print