Implement monitoring lua client

server/server/monitoring.py

@@ -1,6 +1,7 @@
-import binascii
+import json
+from uuid import UUID
 
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, WebSocket, WebSocketDisconnect
 from pydantic import BaseModel
 monitoring = FastAPI()
 
@@ -22,4 +23,16 @@ class Ping(BaseModel):
 async def ping(request: Request, data: Ping):
     print("[PING]")
     for line in data.screen.text:
-        print(line)
+        print(line)
+        
+@monitoring.websocket("/computer/{uuid}/ws")
+async def computer_ws(socket: WebSocket, uuid: UUID):
+    await socket.accept()
+    print(f"[WS] Computer {uuid} connected")
+    while True:
+        try:
+            data = await socket.receive_json()
+            #print(f"[WS] rx {json.dumps(data)}")
+        except WebSocketDisconnect:
+            break
+    print(f"[WS] Computer {uuid} disconnected")