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

from . import admin, config, machines, secrets, webui
from .errors import ClanError
from .ssh import cli as ssh_cli

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")
    subparsers = parser.add_subparsers()

    parser_admin = subparsers.add_parser("admin", help="administrate a clan")
    admin.register_parser(parser_admin)

    # DISABLED: this currently crashes if a flake does not define .#clanOptions
    if os.environ.get("CLAN_OPTIONS_FILE") is not None:
        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)

    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 not hasattr(args, "func"):
        return
    try:
        args.func(args)
    except ClanError as e:
        print(f"{sys.argv[0]}: {e}")
        sys.exit(1)


if __name__ == "__main__":
    main()