main.py

"""
``clang_tools.main``
--------------------

The module containing main entrypoint function.
"""

import argparse

from .install import install_clang_tools, uninstall_clang_tools
from . import RESET_COLOR, YELLOW
from .util import Version


def get_parser() -> argparse.ArgumentParser:
    """Get and parser to interpret CLI args."""
    parser = argparse.ArgumentParser()

    parser.add_argument(
        "-i",
        "--install",
        metavar="VERSION",
        help="Install clang-tools about a specific version. This can be in the form of"
        " a semantic version specification (``x.y.z``, ``x.y``, ``x``). NOTE: A "
        "malformed version specification will cause a silent failure.",
    )
    parser.add_argument(
        "-t",
        "--tool",
        nargs="+",
        default=["clang-format", "clang-tidy"],
        metavar="TOOL",
        help="Specify which tool(s) to install.",
    )
    parser.add_argument(
        "-d",
        "--directory",
        default="",
        metavar="DIR",
        help="The directory where the clang-tools are installed.",
    )
    parser.add_argument(
        "-f",
        "--overwrite",
        action="store_true",
        help="Force overwriting the symlink to the installed binary. This will only "
        "overwrite an existing symlink.",
    )
    parser.add_argument(
        "-b",
        "--no-progress-bar",
        action="store_true",
        help="Do not display a progress bar for downloads.",
    )
    parser.add_argument(
        "-u",
        "--uninstall",
        metavar="VERSION",
        help="Uninstall clang-tools with specific version. "
        "This is done before any install.",
    )
    return parser


def main():
    """The main entrypoint to the CLI program."""
    parser = get_parser()
    args = parser.parse_args()

    if args.uninstall:
        uninstall_clang_tools(args.uninstall, args.directory)
    elif args.install:
        version = Version(args.install)
        if version.info != (0, 0, 0):
            install_clang_tools(
                version,
                args.tool,
                args.directory,
                args.overwrite,
                args.no_progress_bar,
            )
        else:
            print(
                f"{YELLOW}The version specified is not a semantic",
                f"specification{RESET_COLOR}",
            )
    else:
        print(
            f"{YELLOW}Nothing to do because `--install` and `--uninstall`",
            f"was not specified.{RESET_COLOR}",
        )
        parser.print_help()


if __name__ == "__main__":
    main()