5cbe355660
- Replace RSA key exchange with SRP (Secure Remote Password) - Password never transmitted over network - Add unit tests for endpoints - Fix datetime.UTC compatibility for Python < 3.11 - Fix logger.exception usage - Update README with new auth flow diagram
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
import asyncio
|
|
from typing import Optional
|
|
from sanic import Websocket
|
|
from .logger import logger
|
|
|
|
|
|
class ConnectionManager:
|
|
def __init__(self):
|
|
self.active_connections: dict[str, Websocket] = {}
|
|
self._lock = asyncio.Lock()
|
|
|
|
async def connect(self, user_id: str, websocket: Websocket) -> None:
|
|
async with self._lock:
|
|
self.active_connections[user_id] = websocket
|
|
logger.info(f"Client connected: {user_id}")
|
|
|
|
async def disconnect(self, user_id: str) -> None:
|
|
async with self._lock:
|
|
if user_id in self.active_connections:
|
|
del self.active_connections[user_id]
|
|
logger.info(f"Client disconnected: {user_id}")
|
|
|
|
async def broadcast(self, message: str, exclude_user: Optional[str] = None) -> None:
|
|
async with self._lock:
|
|
disconnected = []
|
|
for user_id, connection in list(self.active_connections.items()):
|
|
if exclude_user and user_id == exclude_user:
|
|
continue
|
|
try:
|
|
await connection.send(message)
|
|
except Exception:
|
|
logger.exception(f"Failed to send message to {user_id}")
|
|
disconnected.append(user_id)
|
|
|
|
for user_id in disconnected:
|
|
if user_id in self.active_connections:
|
|
del self.active_connections[user_id]
|
|
|
|
async def send_personal(self, user_id: str, message: str) -> bool:
|
|
async with self._lock:
|
|
if connection := self.active_connections.get(user_id):
|
|
try:
|
|
await connection.send(message)
|
|
return True
|
|
except Exception:
|
|
logger.exception(f"Failed to send personal message to {user_id}")
|
|
return False
|
|
return False
|