moar

endpoints/cah.py

@@ -5,7 +5,7 @@ import time
 import uuid
 import json
 import random
-from cah.constructors import CAHClient
+from cah.constructors import CAHClient, CAHGame
 from cah.websocket_conn import ConnectionManager
 
 class CAH(FastAPI):
@@ -15,6 +15,7 @@ class CAH(FastAPI):
         self.util = util
         self.constants = constants
         self.glob_state = glob_state
+        self.games: list[dict] = []
 
         self.ws_endpoints = {
             "cah": self.cah_handler,
@@ -46,16 +47,29 @@ class CAH(FastAPI):
         try:
             while True:
                 data = await websocket.receive_json()
-                if data.get('event') == 'handshake':
-                    await self.cah_handshake(websocket,
-                                                    data)
-                else:
-                    sender = self.connection_manager.get_connection_by_ws(websocket)
-                    await self.connection_manager.broadcast({
-                        "event": "echo",
-                        "from": sender.get('client').resource,
-                        "data": data,
+                event = data.get('event')
+                if not event:
+                    await websocket.send_json({
+                        'err': True,
+                        'errorText': 'Invalid data received, closing connection.'
                     })
+                    return await websocket.close()
+                print(f"Event: {event}")
+                match event:
+                    case 'handshake':
+                        await self.cah_handshake(websocket,
+                                                 data)
+                    case 'create_game':
+                        await self.create_game(websocket,
+                                               data)   
+                    case _:
+                        sender = self.connection_manager.get_connection_by_ws(websocket)
+                        await self.connection_manager.broadcast({
+                            "event": "echo",
+                            "ts": int(time.time()),
+                            "from": sender.get('client').resource,
+                            "data": data,
+                        })
         except WebSocketDisconnect:
             disconnected = self.connection_manager.get_connection_by_ws(websocket)
             self.connection_manager.disconnect(websocket)
@@ -71,7 +85,6 @@ class CAH(FastAPI):
 
     async def cah_handshake(self, websocket: WebSocket, data):
         """Handshake"""
-        self.connection_manager.connect(websocket)
         data = data.get('data')
         if not data:
             await websocket.send_json({
@@ -93,9 +106,10 @@ class CAH(FastAPI):
             resource=resource,
             platform=platform,
             csid=csid,
-            connected_at=time.time(), # fix
+            connected_at=int(time.time()),
+            current_game=None,
         )
-        await self.connection_manager.handshake_complete(websocket, csid, client)
+        await self.connection_manager.handshake_complete(self, websocket, csid, client)
         
         await websocket.send_json({
             "event": "handshake_response",
@@ -104,6 +118,39 @@ class CAH(FastAPI):
                 "success": True,
                 "resource": resource,
                 "platform": platform,
-                "games": [str(uuid.uuid4()) for x in range(0, 11)],
+                "games": [str(game.id) for game in self.games],
             },
         })
+
+    async def create_game(self, websocket, data: str):
+        data = data.get('data')
+        if not data.get('rounds') or not str(data.get('rounds')).isnumeric():
+            return await websocket.send_json({
+                "event": "create_game_response",
+                "ts": int(time.time()),
+                "data": {
+                    "success": False,
+                    "errorText": "Invalid data",
+                    "recvdData": data,
+                }
+            })
+        client = self.connection_manager.get_connection_by_ws(websocket).get('client')
+        rounds = int(data.get('rounds'))
+        game_uuid = str(uuid.uuid4())
+        game = CAHGame(id=game_uuid,
+                       rounds=rounds,
+                       players=[client,],
+                       created_at=int(time.time()),
+                       state=-1,
+                       started_at=0,
+                       state_changed_at=int(time.time()))
+        self.games.append(game)
+        client.current_game = game
+        await websocket.send_json({
+            "event": "create_game_response",
+            "ts": int(time.time()),
+            "data": {
+                "success": True,
+                "createdGame": client.current_game.id,
+            }
+        })