from dev import count, webconf, Reg, Alarm, Event, run
from net import ping

# Список хостов для проверки доступности
hosts = [
    {'host': 'ya.ru'},
    {'host': 'mail.ru'},
    {'host': '192.168.0.1'},
]

# Определение констант с помощью функции const
INTERVAL     = const(30)      # Интервал проверки в секундах
PING_COUNT   = const(3)       # Количество пингов для каждого хоста
PING_SIZE    = const(64)      # Размер пакета пинга
PING_TIMEOUT = const(0.2)     # Таймаут ожидания ответа в секундах

# Вычисление общего количества хостов
hosts_len = len(hosts)

# Инициализация счетчиков и конфигурация веб-интерфейса
count(
    Reg=hosts_len * 2,    # Два регистра на каждый хост (Время ответа и Потери связи)
    Ai=0,                 # Количество аналоговых входов
    Ao=0,                 # Количество аналоговых выходов
    Di=0,                 # Количество цифровых входов
    Do=0,                 # Количество цифровых выходов
    Event=hosts_len,      # Количество событий
    Alarm=hosts_len       # Количество тревог
)
webconf(Reg=True, Pin=False, Display=False, Console=True)

# Инициализация объектов Alarm, Event и Reg для каждого хоста
for i, item in enumerate(hosts):
    host = item['host']
    # Инициализация тревоги для недоступности хоста
    item['alarm'] = Alarm(i).conf(name=f'Недоступен {host}')
    # Инициализация события восстановления связи
    item['event'] = Event(i).conf(name=f'Связь восстановлена {host}')
    # Инициализация регистра времени ответа
    item['rTime'] = Reg(i).conf(name=f'Время ответа {host}', type='uint', access='r')
    # Инициализация регистра счетчика потерь связи
    item['rFault'] = Reg(hosts_len + i).conf(name=f'Потери связи с {host}', type='uint', access='r')


def ping_ext(addr):
    """
    Функция для выполнения серии пингов к заданному адресу.
    :param addr: Адрес хоста для пинга.
    :return: Среднее время отклика и статус успешности.
    """
    time_total = 0
    success_count = 0
    
    for _ in range(PING_COUNT):
        t, res = ping(addr, size=PING_SIZE, timeout=PING_TIMEOUT)
        print(f'ping {addr} time {t}ms res {res}')
        
        if res == 0:
            time_total += t
            success_count += 1
    
    if success_count == 0:
        return 0, False  # Все пинги неудачные
    else:
        average_time = int(time_total / success_count)
        return average_time, True  # Среднее время отклика и успешность


def step():
    """
    Основная функция, выполняемая с заданным интервалом.
    Пингует каждый хост и обновляет соответствующие регистры и состояния тревог/событий.
    """
    for i, host in enumerate(hosts):
        addr = host['host']
        avg_time, success = ping_ext(addr)
        if success:
            # Обновление регистра времени ответа
            host['rTime'].val(avg_time)
            # Если ранее был установлен тревожный сигнал, снимаем его и добавляем событие восстановления
            if host['alarm'].val():
                host['alarm'].off()
                host['event'].add()
        else:
            # Установка тревожного сигнала и инкремент счетчика потерь связи
            host['alarm'].on()
            host['rFault'].inc()


# Запуск основной функции с указанным интервалом
run(step, INTERVAL)