gonets/SRV/api/v1/subnet.py

from fastapi import APIRouter, HTTPException, Body
from models.subnet import SubnetScanResult, SubnetScanRequest
from models.wifi import WifiScanResult
from models.request import NoteRequest, WifiConnectRequest, WifiDisconnectRequest
from services.subnet_service import SubnetService
from services.wifi_service import WifiService
from services.connect_service import WifiConnectionService

router = APIRouter()

@router.get("/devices", response_model=SubnetScanResult)
def scan_subnet(request: SubnetScanRequest = Body(...)):
    """Сканирует локальную подсеть и возвращает данные об устройствах.

    Выполняет сканирование указанной подсети для обнаружения активных устройств
    с возможностью проверки указанных портов.

    Args:
        request (SubnetScanResult): Параметры для сканирования подсети.
            - subnet (str, optional): Подсеть для сканирования (например "192.168.1.0/24").
            По умолчанию, если subnet пуст, то сканируется подсеть "192.168.1.0/24".
    Returns:
        SubnetScanResult: Объект с результатами сканирования (устройства и их данные).

    Raises:
        HTTPException: Если произошла ошибка сканирования (status_code=500).
    """
    try:
        subnet_service = SubnetService()
        result = subnet_service.scan_subnet(request.subnet)
        return result
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка сканирования: {str(e)}")

@router.get("/wifi", response_model=WifiScanResult)
def scan_wifi(timeout: int = 30):
    """Сканирует Wi-Fi точки доступа и возвращает данные о них.

    Выполняет сканирование доступных Wi-Fi сетей с указанным таймаутом.

    Args:
        timeout (int, optional): Таймаут сканирования в секундах. По умолчанию 30.

    Returns:
        WifiScanResult: Объект с результатами сканирования (список Wi-Fi сетей).

    Raises:
        HTTPException: Если произошла ошибка сканирования Wi-Fi (status_code=500).
    """
    try:
        wifi_service = WifiService()
        result = wifi_service.scan_wifi(timeout=timeout)
        return result
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка сканирования Wi-Fi: {str(e)}")

@router.post("/devices/{mac}/note")
def add_note(mac: str, request: NoteRequest):
    """Добавляет заметку для устройства по MAC-адресу.

    Привязывает заметку к устройству, идентифицируемому по MAC-адресу.

    Args:
        mac (str): MAC-адрес устройства.
        request (NoteRequest): Объект с полем "note" для текста заметки.

    Returns:
        dict: Словарь с сообщением об успешном добавлении заметки.

    Raises:
        HTTPException: Если устройство не найдено (status_code=404), заметка уже
            существует (status_code=400), или произошла внутренняя ошибка (status_code=500).
    """
    try:
        subnet_service = SubnetService()
        success, error_message = subnet_service.add_note(mac, request.note)
        if not success:
            if "не найдено" in error_message:
                raise HTTPException(status_code=404, detail=error_message)
            elif "уже существует" in error_message:
                raise HTTPException(status_code=400, detail=error_message)
        return {"message": f"Заметка '{request.note}' добавлена для устройства {mac}"}
    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка добавления заметки: {str(e)}")

@router.put("/devices/{mac}/note")
def update_note(mac: str, request: NoteRequest):
    """Обновляет заметку для устройства по MAC-адресу.

    Изменяет существующую заметку, привязанную к устройству.

    Args:
        mac (str): MAC-адрес устройства.
        request (NoteRequest): Объект с полем "note" для нового текста заметки.

    Returns:
        dict: Словарь с сообщением об успешном обновлении заметки.

    Raises:
        HTTPException: Если устройство не найдено (status_code=404), заметка не
            существует (status_code=400), или произошла внутренняя ошибка (status_code=500).
    """
    try:
        subnet_service = SubnetService()
        success, error_message = subnet_service.update_note(mac, request.note)
        if not success:
            if "не найдено" in error_message:
                raise HTTPException(status_code=404, detail=error_message)
            elif "не существует" in error_message:
                raise HTTPException(status_code=400, detail=error_message)
        return {"message": f"Заметка '{request.note}' обновлена для устройства {mac}"}
    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка обновления заметки: {str(e)}")

@router.delete("/devices/{mac}/note")
def delete_note(mac: str):
    """Удаляет заметку для устройства по MAC-адресу.

    Удаляет привязанную заметку для указанного устройства.

    Args:
        mac (str): MAC-адрес устройства.

    Returns:
        dict: Словарь с сообщением об успешном удалении заметки.

    Raises:
        HTTPException: Если устройство не найдено (status_code=404), заметка не
            существует (status_code=400), или произошла внутренняя ошибка (status_code=500).
    """
    try:
        subnet_service = SubnetService()
        success, error_message = subnet_service.delete_note(mac)
        if not success:
            if "не найдено" in error_message:
                raise HTTPException(status_code=404, detail=error_message)
            elif "не существует" in error_message:
                raise HTTPException(status_code=400, detail=error_message)
        return {"message": f"Заметка удалена для устройства {mac}"}
    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}")

@router.post("/wifi/{ssid}/note")
def add_wifi_note(ssid: str, request: NoteRequest):
    """Добавляет заметку для сети Wi-Fi по SSID.

    Привязывает заметку к Wi-Fi сети, идентифицируемой по SSID.

    Args:
        ssid (str): Идентификатор Wi-Fi сети.
        request (NoteRequest): Объект с полем "note" для текста заметки.

    Returns:
        dict: Словарь с сообщением об успешном добавлении заметки.

    Raises:
        HTTPException: Если сеть не найдена (status_code=404), заметка уже
            существует (status_code=400), или произошла внутренняя ошибка (status_code=500).
    """
    try:
        wifi_service = WifiService()
        success, error_message = wifi_service.add_note(ssid, request.note)
        if not success:
            if "не найдена" in error_message:
                raise HTTPException(status_code=404, detail=error_message)
            elif "уже существует" in error_message:
                raise HTTPException(status_code=400, detail=error_message)
        return {"message": f"Заметка '{request.note}' добавлена для сети с SSID {ssid}"}
    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка добавления заметки: {str(e)}")

@router.put("/wifi/{ssid}/note")
def update_wifi_note(ssid: str, request: NoteRequest):
    """Обновляет заметку для сети Wi-Fi по SSID.

    Изменяет существующую заметку, привязанную к Wi-Fi сети.

    Args:
        ssid (str): Идентификатор Wi-Fi сети.
        request (NoteRequest): Объект с полем "note" для нового текста заметки.

    Returns:
        dict: Словарь с сообщением об успешном обновлении заметки.

    Raises:
        HTTPException: Если сеть не найдена (status_code=404), заметка не
            существует (status_code=400), или произошла внутренняя ошибка (status_code=500).
    """
    try:
        wifi_service = WifiService()
        success, error_message = wifi_service.update_note(ssid, request.note)
        if not success:
            if "не найдена" in error_message:
                raise HTTPException(status_code=404, detail=error_message)
            elif "не существует" in error_message:
                raise HTTPException(status_code=400, detail=error_message)
        return {"message": f"Заметка '{request.note}' обновлена для сети с SSID {ssid}"}
    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка обновления заметки: {str(e)}")

@router.delete("/wifi/{ssid}/note")
def delete_wifi_note(ssid: str):
    """Удаляет заметку для сети Wi-Fi по SSID.

    Удаляет привязанную заметку для указанной Wi-Fi сети.

    Args:
        ssid (str): Идентификатор Wi-Fi сети.

    Returns:
        dict: Словарь с сообщением об успешном удалении заметки.

    Raises:
        HTTPException: Если сеть не найдена (status_code=404), заметка не
            существует (status_code=400), или произошла внутренняя ошибка (status_code=500).
    """
    try:
        wifi_service = WifiService()
        success, error_message = wifi_service.delete_note(ssid)
        if not success:
            if "не найдена" in error_message:
                raise HTTPException(status_code=404, detail=error_message)
            elif "не существует" in error_message:
                raise HTTPException(status_code=400, detail=error_message)
        return {"message": f"Заметка удалена для сети с SSID {ssid}"}
    except HTTPException as e:
        raise e
    except Exception as e:
        raise HTTPException(status_code=500, detail=f"Ошибка удаления заметки: {str(e)}")

@router.post("/wifi/connect")
def connect_to_wifi(request: WifiConnectRequest):
    """Подключается к Wi-Fi сети по SSID и паролю.

    Выполняет подключение к указанной Wi-Fi сети с использованием предоставленных
    учетных данных.

    Args:
        request (WifiConnectRequest): Объект с полями "ssid" и "password".

    Returns:
        dict: Словарь с сообщением об успешном подключении.

    Raises:
        HTTPException: Если подключение не удалось (status_code=400).
    """
    wifi_service = WifiConnectionService()
    success, message = wifi_service.connect(request.ssid, request.password)
    if not success:
        raise HTTPException(status_code=400, detail=message)
    return {"message": f"Успешно подключено к {request.ssid}"}

@router.post("/wifi/disconnect")
def disconnect_from_wifi(request: WifiDisconnectRequest):
    """Отключается от Wi-Fi сети по указанному SSID.

    Выполняет отключение от указанной Wi-Fi сети.

    Args:
        request (WifiDisconnectRequest): Объект с полем "ssid".

    Returns:
        dict: Словарь с сообщением об успешном отключении.

    Raises:
        HTTPException: Если отключение не удалось (status_code=400).
    """
    wifi_service = WifiConnectionService()
    success, message = wifi_service.disconnect(request.ssid)
    if not success:
        raise HTTPException(status_code=400, detail=message)
    return {"message": message or f"Успешно отключено от {request.ssid}"}

@router.get("/wifi/status")
def get_wifi_status():
    """Проверяет статус текущего подключения к Wi-Fi.

    Возвращает информацию о текущем состоянии подключения и SSID, если подключение активно.

    Returns:
        dict: Словарь с ключами:
            - "status" (str): Состояние ("connected" или другое).
            - "ssid" (str, optional): SSID текущей сети, если подключено.

    Raises:
        HTTPException: Если проверка статуса не удалась (status_code=400).
    """
    wifi_service = WifiConnectionService()
    success, message = wifi_service.get_status()
    if not success:
        raise HTTPException(status_code=400, detail=message)
    return {"status": "connected", "ssid": message}