import asyncio
import logging
from contextlib import asynccontextmanager
from typing import Optional, List

from fastapi import FastAPI, HTTPException

from app.core.discovery import DiscoveryService
from app.core.state import state_manager
from app.drivers.wiz import WizDriver

logging.basicConfig(
    level=logging.INFO, format="%(asctime)s | %(levelname)s | %(message)s"
)
logger = logging.getLogger(__name__)

discovery = DiscoveryService()
wiz = WizDriver()


@asynccontextmanager
async def lifespan(app: FastAPI):
    """
    Жизненный цикл приложения: выполняется при старте и остановке.
    """

    # Задача для периодического сканирования сети
    async def periodic_discovery():
        logger.info("🚀 Запущена фоновая служба Discovery")

        while True:
            try:
                found_devices = await discovery.scan_network()

                for dev_data in found_devices:
                    state_manager.update_device(dev_data)

                logger.info(
                    f"📡 Сеть просканирована. Устройств в памяти: {len(state_manager.devices)}"
                )
            except Exception as e:
                logger.error(f"❌ Ошибка в цикле Discovery: {e}")

            # Интервал сканирования (например, раз в 10 минут)
            await asyncio.sleep(600)

    discovery_task = asyncio.create_task(periodic_discovery())

    yield

    # Логика при завершении работы сервера
    discovery_task.cancel()
    logger.info("🛑 Фоновая служба Discovery остановлена")


app = FastAPI(
    title="Ignis Core API",
    description="Ядро системы умного дома для управления лампами WiZ/Gauss",
    version="1.0.0",
    lifespan=lifespan,
)


@app.get("/health")
async def health():
    """Проверка работоспособности сервиса."""
    return {"status": "online", "devices_discovered": len(state_manager.devices)}


@app.get("/devices")
async def get_all_devices():
    """Возвращает список всех обнаруженных устройств из стейта."""
    return state_manager.devices


@app.post("/control/group/{group_id}")
async def control_group(
    group_id: str,
    state: Optional[bool] = None,
    brightness: Optional[int] = None,
    scene: Optional[str] = None,
):
    """
    Управление группой ламп (люстрой).
    Параметры передаются в JSON или как query-params.
    """
    # 1. Получаем IP-адреса всех ламп в группе из стейт-менеджера
    ips = state_manager.get_group_ips(group_id)

    if not ips:
        raise HTTPException(
            status_code=404, detail=f"Группа '{group_id}' не найдена или пуста"
        )

    # 2. Формируем параметры команды для WiZ
    params = {}
    if state is not None:
        params["state"] = state
    if brightness is not None:
        params["dimming"] = brightness
    if scene and scene in wiz.SCENES:
        params["sceneId"] = wiz.SCENES[scene]

    if not params:
        raise HTTPException(
            status_code=400,
            detail="Не указаны параметры управления (state, brightness или scene)",
        )

    # 3. Отправляем команды всем лампам группы параллельно
    tasks = [wiz.set_pilot(ip, params) for ip in ips]
    results = await asyncio.gather(*tasks, return_exceptions=True)

    return {
        "group_id": group_id,
        "affected_ips": ips,
        "status": "commands_sent",
        "params": params,
    }


@app.post("/control/device/{device_id}")
async def control_device(device_id: str, state: bool):
    """Прямое управление устройством по его ID (MAC-адресу)."""
    device = state_manager.devices.get(device_id)

    if not device:
        raise HTTPException(
            status_code=404, detail="Устройство не найдено в текущем стейте"
        )

    result = await wiz.set_pilot(device.ip, {"state": state})
    return {"device_id": device_id, "ip": device.ip, "result": result}


if __name__ == "__main__":
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=8000)