WSMessage object

hopper/models/ws_dto.py

@@ -1,6 +1,10 @@
 from __future__ import annotations
 
+import json
+from typing import TypeVar, Generic
+
 from pydantic import Field
+from pydantic.generics import GenericModel
 
 from hopper.api.dto import BaseModel, BoardDto, DestinationDto, PlayerDto, PositionDto
 from hopper.enums import ObjectType
@@ -15,6 +19,7 @@ class LayerDto(BaseModel):
     name: str
     objects: list[LayerObjectDto]
 
+
 class GameDumpPlayerDto(PlayerDto):
     ...
 
@@ -24,3 +29,22 @@ class GameDumpDto(BaseModel):
     destination: DestinationDto
     players: list[GameDumpPlayerDto]
     layers: list[LayerDto]
+
+
+TMessageData = TypeVar("TMessageData", bound=BaseModel)
+
+
+class WSMessage(GenericModel):
+    message: str
+    data: TMessageData
+
+    def __str__(self) -> str:
+        return self.to_str()
+
+    def to_str(self) -> str:
+        return json.dumps(self.dict())
+
+
+class WSGameDumpMessage(WSMessage):
+    message: str = "game_dump"
+    data: GameDumpDto