Source code for services.ip_service

"""
Сервис проверки и применения IP‑настроек через NetworkManager (D‑Bus).

Назначение:
- Перед применением конфигурации проверяет занятость IP адреса, сканируя сегмент
  сети средствами ARP (через `NetworkScanner`).
- Применяет изменения профиля подключения через `sdbus_async.networkmanager` и
  выполняет переподключение интерфейса.

Требования/ограничения:
- Доступ к системной шине D‑Bus и права на изменения сетевых профилей.
- Структура входных данных должна соответствовать схеме `ipv4_updates` для
  корректного обновления вложенных полей у профиля.
"""
from typing import Dict, Any
from sdbus_async.networkmanager import (
    NetworkManager,
    NetworkManagerSettings,
    NetworkConnectionSettings,
    NetworkDeviceGeneric,
)
import sdbus
from sdbus_async.networkmanager.settings import ConnectionProfile
from services.arp_service import NetworkScanner
from utils.i18n import get_translations
from babel.support import Translations
from fastapi import Depends



class IPService:
    """Сервис проверки доступности IP-адреса и применения настроек сети.

    Предоставляет асинхронные методы для проверки занятости IP-адреса с использованием
    ARP-сканирования и применения настроек через NetworkManager.
    """

    def __init__(self):
        """Инициализирует сервис IP.

        Создаёт объект ARP-сканера и устанавливает системную шину D-Bus как
        по умолчанию.
        """
        self.arp_scanner = NetworkScanner()
        self.bus = sdbus.sd_bus_open_system()
        sdbus.set_default_bus(self.bus)

    def _reset_bus(self):
        """Перезапускает соединение с системной шиной D-Bus.

        Закрывает текущее соединение и создаёт новое, устанавливая его как
        по умолчанию.
        """
        new_bus = sdbus.sd_bus_open_system()
        sdbus.set_default_bus(new_bus)
        if hasattr(self, "bus") and self.bus is not None:
            self.bus = None
        self.bus = new_bus



    async def check_ip_config(self, args: Dict[str, Any], translations: Translations = Depends(get_translations())) -> Dict[str, Any]:
        """Проверяет доступность IP-адреса перед его применением.

        Выполняет ARP-сканирование сети для проверки, занят ли указанный IP-адрес.
        Перезапускает шину D-Bus перед выполнением.

        Args:
            args (Dict[str, Any]): Словарь с данными, включая "ipv4_updates"
                с полем "ipv4.address-data" для нового IP-адреса.
            translations (Translations): Объект переводов.

        Returns:
            Dict[str, Any]: Словарь с ключами:
                - "available" (bool): Доступен ли IP-адрес.
                - "message" (str): Сообщение о результате проверки.

        Raises:
            Exception: Если произошла ошибка при сканировании ARP.
        """
        self._reset_bus()
        clients = self.arp_scanner.scan_network()
        new_ip = args["ipv4_updates"]["ipv4.address-data"][0]["address"]
        if any(client["ip"] == new_ip for client in clients):
            return {
                "available": False,
                "message": translations.gettext("IP-адрес {ip} уже занят.").format(ip=new_ip)
            }
        return {
            "available": True,
            "message": translations.gettext("Настройки сетевого профиля доступны для использования.")
        }




    async def apply_ip_config(self, args: Dict[str, Any], translations: Translations = Depends(get_translations())) -> None:
        """Применяет проверенные настройки IP к указанному соединению.

        Перезапускает шину D-Bus, обновляет настройки соединения через NetworkManager,
        отключает и переподключает устройство к сети.

        Args:
            args (Dict[str, Any]): Словарь с данными, включая:
                - "connection_id" (str): Идентификатор соединения.
                - "ipv4_updates" (Dict): Обновления настроек IPv4.
            translations (Translations): Объект переводов.

        Raises:
            ValueError: Если соединение с указанным ID не найдено.
            RuntimeError: Если устройство для интерфейса не найдено.

        Note:
            Требуются права для изменения настроек через NetworkManager.
        """
        self._reset_bus()
        nm = NetworkManager()
        settings = NetworkManagerSettings()
        connection_paths = await settings.get_connections_by_id(
            args["connection_id"])

        if not connection_paths:
            raise ValueError(translations.gettext("Подключение {connection_id} не найдено").format(connection_id=args["connection_id"]))

        connection = NetworkConnectionSettings(connection_paths[0])
        profile = await connection.get_profile()
        settings_dict = profile.to_settings_dict()

        for key_path, value in args["ipv4_updates"].items():
            section, *props = key_path.split(".")
            current = settings_dict[section]
            for prop in props[:-1]:
                current = current.setdefault(prop, {})
            current[props[-1]] = value

        new_profile = ConnectionProfile.from_settings_dict(settings_dict)
        await connection.update(new_profile.to_dbus())
        print(f"Updated settings for {args['connection_id']}")

        interface_name = settings_dict["connection"]["interface-name"]
        devices = await nm.get_devices()
        for device_path in devices:
            device = NetworkDeviceGeneric(device_path)
            if await device.interface == interface_name:
                await device.disconnect()
                print(f"Disconnected from {interface_name}")
                await nm.activate_connection(connection=connection_paths[0], device=device_path)
                print(f"Reconnected to {args['connection_id']}")
                return

        raise RuntimeError(translations.gettext("Устройство для интерфейса {interface} не найдено").format(interface=interface_name))