import argparse
import logging
import sys
from types import ModuleType
from typing import Optional

from . import config, create, custom_logger, machines, secrets, vms, webui
from .errors import ClanError
from .ssh import cli as ssh_cli

log = logging.getLogger(__name__)

argcomplete: Optional[ModuleType] = None
try:
    import argcomplete  # type: ignore[no-redef]
except ImportError:
    pass


def create_parser(prog: Optional[str] = None) -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(prog=prog, description="cLAN tool")

    parser.add_argument(
        "--debug",
        help="Enable debug logging",
        action="store_true",
    )

    subparsers = parser.add_subparsers()

    parser_create = subparsers.add_parser(
        "create", help="create a clan flake inside the current directory"
    )
    create.register_parser(parser_create)

    parser_config = subparsers.add_parser("config", help="set nixos configuration")
    config.register_parser(parser_config)

    parser_ssh = subparsers.add_parser("ssh", help="ssh to a remote machine")
    ssh_cli.register_parser(parser_ssh)

    parser_secrets = subparsers.add_parser("secrets", help="manage secrets")
    secrets.register_parser(parser_secrets)

    parser_machine = subparsers.add_parser(
        "machines", help="Manage machines and their configuration"
    )
    machines.register_parser(parser_machine)

    parser_webui = subparsers.add_parser("webui", help="start webui")
    webui.register_parser(parser_webui)

    parser_vms = subparsers.add_parser("vms", help="manage virtual machines")
    vms.register_parser(parser_vms)

    if argcomplete:
        argcomplete.autocomplete(parser)

    if len(sys.argv) == 1:
        parser.print_help()
    return parser


# this will be the entrypoint under /bin/clan (see pyproject.toml config)
def main() -> None:
    parser = create_parser()
    args = parser.parse_args()

    if args.debug:
        custom_logger.register(logging.DEBUG)
        log.debug("Debug logging enabled")
    else:
        custom_logger.register(logging.INFO)

    if not hasattr(args, "func"):
        log.error("No argparse function registered")
        return
    try:
        args.func(args)
    except ClanError as e:
        log.exception(e)
        sys.exit(1)


if __name__ == "__main__":
    main()