import json
from datetime import datetime
from typing import Set
from fastapi import WebSocket

from app.models import Position, Event


class ConnectionManager:
    def __init__(self):
        self.active_connections: Set[WebSocket] = set()

    async def connect(self, websocket: WebSocket):
        await websocket.accept()
        self.active_connections.add(websocket)

    def disconnect(self, websocket: WebSocket):
        self.active_connections.discard(websocket)

    async def broadcast(self, message: dict):
        """Отправить сообщение всем подключенным клиентам"""
        disconnected = set()
        for connection in self.active_connections:
            try:
                await connection.send_json(message)
            except Exception:
                disconnected.add(connection)

        # Remove disconnected clients
        self.active_connections -= disconnected

    async def broadcast_position(self, position: Position):
        """Отправить обновление позиции"""
        message = {
            "type": "position_update",
            "data": {
                "vehicle_id": position.vehicle_id,
                "lat": position.lat,
                "lon": position.lon,
                "speed": position.speed,
                "heading": position.heading,
                "timestamp": position.timestamp.isoformat()
            }
        }
        await self.broadcast(message)

    async def broadcast_event(self, event: Event):
        """Отправить событие"""
        message = {
            "type": "event",
            "data": {
                "id": event.id,
                "vehicle_id": event.vehicle_id,
                "type": event.type,
                "payload": event.payload,
                "timestamp": event.timestamp.isoformat()
            }
        }
        await self.broadcast(message)


# Global instance
manager = ConnectionManager()