misc / begin CAH work, much cleanup to do / conceptualizing so far only

endpoints/cah.py

@@ -0,0 +1,102 @@
+#!/usr/bin/env python3.12
+from fastapi import FastAPI, WebSocket, WebSocketDisconnect
+
+import time
+import uuid
+import json
+import random
+from cah.constructors import CAHClient
+from cah.websocket_conn import ConnectionManager
+
+class CAH(FastAPI):
+    """CAH Endpoint(s)"""   
+    def __init__(self, app: FastAPI, util, constants, glob_state):  # pylint: disable=super-init-not-called
+        self.app = app
+        self.util = util
+        self.constants = constants
+        self.glob_state = glob_state
+
+        self.ws_endpoints = {
+            "cah": self.cah_handler,
+            #tbd
+            }
+
+        self.endpoints = {
+            #tbd if any non-WS endpoints
+        }
+
+        self.connection_manager = ConnectionManager()
+
+
+        for endpoint, handler in self.ws_endpoints.items():
+            print(f"Adding websocket route: {endpoint} @ {handler}")
+            app.add_api_websocket_route(f"/{endpoint}/", handler)
+
+        for endpoint, handler in self.endpoints.items():
+            app.add_api_route(f"/{endpoint}/", handler, methods=["POST"])
+    
+    async def cah_handler(self, websocket: WebSocket):
+        """/cah WebSocket"""
+        await self.connection_manager.connect(websocket)
+        await websocket.send_json({
+            "event": "connection_established",
+            "ts": int(time.time()),
+        })
+
+        try:
+            while True:
+                data = await websocket.receive_json()
+                if data.get('event') == 'handshake':
+                    await self.cah_handshake(websocket,
+                                                    data)
+        except WebSocketDisconnect:
+            disconnected = self.connection_manager.get_connection_by_ws(websocket)
+            self.connection_manager.disconnect(websocket)
+            await self.connection_manager.broadcast({
+                "event": "client_disconnected",
+                "ts": time.time(),
+                "data": {
+                    "disconnected_resource": disconnected.get('client').resource,
+                }
+            })
+
+                
+
+    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({
+                "err": "WTF",
+            })
+            return await websocket.close()
+
+        csid = str(data.get('csid'))
+        resource = data.get('resource')
+        platform = data.get('platform')
+        
+        if not csid in self.constants.VALID_CSIDS:
+            await websocket.send_json({
+                "err": "Unauthorized",
+            })
+            return await websocket.close()
+        
+        client = CAHClient(
+            resource=resource,
+            platform=platform,
+            csid=csid,
+            connected_at=time.time(), # fix
+        )
+        await self.connection_manager.handshake_complete(websocket, csid, client)
+        
+        await websocket.send_json({
+            "event": "handshake_response",
+            "ts": int(time.time()),
+            "data": {
+                "success": True,
+                "resource": resource,
+                "platform": platform,
+                "games": [str(uuid.uuid4()) for x in range(0, 11)],
+            },
+        })

endpoints/counters.py

@@ -2,7 +2,6 @@
 
 #!/usr/bin/env python3.12
 
-import importlib
 from fastapi import FastAPI
 from pydantic import BaseModel
 

endpoints/ws_test.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+
+import asyncio
+import json
+import random
+import uuid
+from websockets.client import connect
+
+async def hello():
+    async with connect("wss://api.codey.lol/cah/") as websocket:
+        message = await websocket.recv()
+        x = str(uuid.uuid4())
+        print(f"Init Message: {message}")
+        await websocket.send(json.dumps({
+            "event": "handshake",
+            "data": {
+                "resource": f"Test Client UUID {x}",
+                "platform": "Discord",
+                "csid": "2bd60a53-023d-49a5-a668-ce8fa8f6ec7f",
+            }
+        }))
+        while True:
+            message = await websocket.recv()
+            print(message)
+            await asyncio.sleep(random.uniform(20, 35))
+            await websocket.close()
+        # for x in range(0, 200):
+        #     await websocket.send(f"Hello world! {x}")
+        #     message = await websocket.recv()
+        #     print(f"Received: {message}")
+        #     await asyncio.sleep(0.2)
+            
+asyncio.get_event_loop().create_task(hello())
+asyncio.get_event_loop().run_forever()