manual_generate.py

# SPDX-FileCopyrightText: 2021 Diego Elio Pettenò
#
# SPDX-License-Identifier: 0BSD

import asyncio
import pathlib
from collections.abc import Sequence

import aiodns
import click
import click_log

from ipv6_in_real_life import data_input, json, observability, render

click_log.basic_config()


async def amain(
    output_directory: pathlib.Path,
    json_only: bool,
    nameserver: Sequence[str],
) -> None:
    # Initialize the start timestamp.
    observability.Metrics.get()

    source = data_input.load_packaged_data()

    if not nameserver:
        nameserver = None

    resolver = aiodns.DNSResolver(nameserver)
    await source.resolve_all(resolver)

    try:
        (output_directory / "ipv6-in-real-life.json").write_text(
            json.generate_json(source)
        )
        if not json_only:
            (output_directory / "index.html").write_text(render.index(source))
            (output_directory / "details.html").write_text(
                render.details(source)
            )
    finally:
        observability.Metrics.get().write_out(output_directory / "metrics.json")


@click.command()
@click_log.simple_verbosity_option()
@click.option(
    "--output-directory",
    type=click.Path(dir_okay=True, file_okay=False, exists=True, writable=True),
    default="./out",
)
@click.option("--json-only", type=bool, is_flag=True, default=False)
@click.option("--nameserver", type=str, multiple=True)
def main(
    output_directory: str,
    json_only: bool,
    nameserver: Sequence[str],
) -> None:
    loop = asyncio.SelectorEventLoop()
    asyncio.set_event_loop(loop)

    asyncio.run(amain(pathlib.Path(output_directory), json_only, nameserver))


if __name__ == "__main__":
    main()