Working development server (no encryption now)

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

@@ -0,0 +1,52 @@
+from .connection import Connection
+from .room import Room
+from typing import Dict
+
+from ..objects.webmessage import (
+    webmessage_error_message_literal
+)
+
+
+class Service(object):
+    rooms: Dict[str, Room] = {}
+
+    async def get_room(self, name: str) -> Room:
+        if name in self.rooms.keys():
+            return self.rooms[name]
+
+        self.rooms[name] = Room()
+        return self.rooms[name]
+
+    async def broadcast_notification(self, message: str):
+        for room in self.rooms.values():
+            await room.broadcast_notification(message)
+
+    async def broadcast_error(
+            self,
+            error_message: webmessage_error_message_literal
+    ):
+        for room in self.rooms.values():
+            await room.broadcast_error(error_message)
+
+    async def get_room_by_connection(self, connection: Connection) -> Room:
+        for room in self.rooms.values():
+            if connection in room.connections.values():
+                return room
+
+    async def get_connection_by_attribute(
+            self, attribute: str, value: str
+    ) -> Connection:
+        for room in self.rooms.values():
+            if connection := await room.get_connection_by(attribute, value):
+                return connection
+
+    async def close_room(self, room_name: str, reason: str = 'Unknown reason'):
+        room = self.rooms.get(room_name)
+        if room is None:
+            return
+
+        for connection in room.connections.values():
+            await room.disconnect(
+                connection=connection,
+                close_reason=f'Room is closed: {reason}'
+            )