# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

"""
Utility methods to print system info for debugging.

Adapted from ``pandas.show_versions`` and ``sklearn.show_versions``.
"""

import platform
import sys


def _get_sys_info() -> dict[str, str]:
    """
    System information.

    Returns:
        dict: system and Python version information
    """
    python = sys.version.replace("\n", " ")

    blob = [
        ("python", python),
        ("executable", sys.executable),
        ("machine", platform.platform()),
    ]

    return dict(blob)


def _get_deps_info() -> dict[str, str | None]:
    """
    Overview of the installed version of main dependencies.

    This function does not import the modules to collect the version numbers
    but instead relies on standard Python package metadata.

    Returns:
        dict: version information on relevant Python libraries
    """
    deps = sorted(
        [
            "Cython",
            "sqlite3",
            "openai",
            "packaging",
            "pip",
            "numpy",
            "scipy",
            "setuptools",
            "torch",
            "transformers",
        ]
    )

    from pyrit import __version__

    deps_info: dict[str, str | None] = {"pyrit": __version__}

    from importlib.metadata import PackageNotFoundError, version

    for modname in deps:
        try:
            deps_info[modname] = version(modname)
        except PackageNotFoundError:
            deps_info[modname] = None
    return deps_info


def show_versions() -> None:
    """Print useful debugging information."""
    sys_info = _get_sys_info()
    deps_info = _get_deps_info()

    print("\nSystem:")
    for k, stat in sys_info.items():
        print(f"{k:>10}: {stat}")

    print("\nPython dependencies:")
    for k, stat_or_none in deps_info.items():
        print(f"{k:>13}: {stat_or_none}")