"""Command line interface for hbd package."""
import argparse

from .config import load_config
from .server import run as run_server

PUSHSRVS = ["all", "pushover", "mattermost"]


def build_parser():
    parser = argparse.ArgumentParser(
        prog="hbd",
        description="HeartBeatDaemon - Wait for heartbeat messages and act on them (or their absence)",
        formatter_class=argparse.RawDescriptionHelpFormatter,
    )
    parser.add_argument("-c", "--config", dest="configfile", help="Config file path (YAML)")
    parser.add_argument("-f", "--foreground", action="store_true", help="Run in foreground")
    parser.add_argument("-v", "--verbose", action="store_true", help="Verbose output")
    parser.add_argument("-p", "--pushsrv", dest="pushsrv", choices=PUSHSRVS, help="Push service to use")
    parser.add_argument("-x", "--debug", action="count", default=0, help="Increase debug level")
    return parser


def main(argv=None):
    parser = build_parser()
    args = parser.parse_args(argv)

    config = load_config(args.configfile)

    # Apply CLI overrides
    if args.foreground:
        config["foreground"] = True
    if args.verbose:
        config["verbose"] = True
    if args.pushsrv:
        config["pushsrv"] = args.pushsrv
    if args.debug:
        config.setdefault("debug", 0)
        config["debug"] += args.debug

    run_server(config)


if __name__ == "__main__":
    main()