"""Практика к уроку про разрешение имени: три ответа и одна цена.

ЗАЧЕМ ОТДЕЛЬНЫЙ ФАЙЛ. Задачи урока проверяются сборкой
(scripts/validate-practice.mjs): показанный читателю код обязан построчно быть
в скрипте, верный вариант — дословно встречаться в записи прогона, а
заявленное число — печататься самой программой.

ЧТО ЗДЕСЬ ПРОВЕРЯЕТСЯ. Два места, где ошибаются чаще всего. Первое: сколько
имён на самом деле спрашивается, когда в resolv.conf есть поисковая область, —
и что с этим делает точка на конце. Второе: во что обходится имя, которого
нет, при настройках по умолчанию — не «примерно долго», а точное число секунд,
которое считается из трёх величин.

Сервер имён и подмена resolv.conf берутся из `resolve.py`, чтобы задача и
разбор урока стояли на одном и том же коде.

ЗАПУСК: sudo python3 bench/nameres/practice.py
Вывод: runs/practice.txt
"""

import os
import shutil
import socket
import sys
import time

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from resolve import (  # noqa: E402
    BACKUP,
    PROBE,
    RESOLV,
    SilentNameServer,
    write_resolv,
)

SEARCH = "search svc.cluster.local cluster.local"


def asked_for(server: SilentNameServer, name: str, ndots: int) -> list[str]:
    """Какие имена libc спросила у сервера, пока искала одно."""
    write_resolv(
        f"nameserver 127.0.0.1\n{SEARCH}\noptions timeout:1 attempts:1 ndots:{ndots}\n"
    )
    server.seen.clear()
    try:
        socket.getaddrinfo(name, 80, socket.AF_INET, socket.SOCK_STREAM)
    except OSError:
        pass
    return [n for n in server.seen if PROBE in n]


def family_queries(server: SilentNameServer) -> int:
    """Сколько запросов уходит на одно имя, когда семейство не задано."""
    write_resolv("nameserver 127.0.0.1\noptions timeout:1 attempts:1 ndots:15\n")
    server.seen.clear()
    try:
        socket.getaddrinfo(f"{PROBE}.invalid", 80, socket.AF_UNSPEC, socket.SOCK_STREAM)
    except OSError:
        pass
    return len([n for n in server.seen if PROBE in n])


def default_cost(server: SilentNameServer) -> float:
    """Цена имени, которого нет, при настройках по умолчанию — в секундах.

    ПОЧЕМУ ИМЕННО ЭТИ ТРИ ЧИСЛА. `timeout:2 attempts:3` и два сервера — это
    то, что лежит в resolv.conf у контейнера, откуда снят урок. Ответ считает
    не редакция, а прогон: секунды печатаются замером, а не умножением в уме.
    """
    write_resolv(
        "nameserver 127.0.0.1\nnameserver 127.0.0.1\noptions timeout:2 attempts:3\n"
    )
    server.seen.clear()
    started = time.perf_counter()
    try:
        socket.getaddrinfo(f"{PROBE}.invalid", 80, socket.AF_INET, socket.SOCK_STREAM)
    except OSError:
        pass
    return time.perf_counter() - started


def main() -> None:
    print(f"Python {sys.version.split()[0]} · Linux {os.uname().release}")
    print()

    try:
        shutil.copyfile(RESOLV, BACKUP)
        server = SilentNameServer()
    except OSError as exc:
        print(f"not permitted: {type(exc).__name__}")
        print("run as root: the exercises need :53 and /etc/resolv.conf")
        return

    try:
        # --- Часть 1: три ответа о поведении --------------------------------
        relative = asked_for(server, f"{PROBE}.internal", ndots=5)
        absolute = asked_for(server, f"{PROBE}.internal.", ndots=5)
        families = family_queries(server)
        print(len(relative))
        print(len(absolute))
        print(families)

        # --- Часть 2: цена имени, которого нет -------------------------------
        #
        # ПОЧЕМУ СЕКУНДЫ, А НЕ МИЛЛИСЕКУНДЫ. Здесь нет ни сети, ни диска:
        # всё время состоит из выдержек, которые libc отсчитывает по часам.
        # Разброс между запусками — единицы миллисекунд на двенадцать секунд,
        # поэтому округление до десятых честно, а не удобно.
        seconds = default_cost(server)
        print()
        print(f"resolv.conf: timeout                 2")
        print(f"resolv.conf: attempts                3")
        print(f"resolv.conf: nameservers             2")
        print(f"one name that does not resolve, s    {seconds:.1f}")
    finally:
        shutil.copyfile(BACKUP, RESOLV)
        server.close()


if __name__ == "__main__":
    main()