"""
Provides the `cmdeploy` entry point function,
along with command line option and subcommand parsing.
"""
import importlib.resources
import argparse
import shutil
import subprocess
import os
from pathlib import Path

import iniconfig

from termcolor import colored
from chatmaild.config import read_config, write_initial_config


class Out:
    """Convenience print output printer providing coloring."""

    def red(self, msg):
        print(colored(msg, "red"))

    def green(self, msg):
        print(colored(msg, "green"))

    def __call__(self, msg, red=False, green=False):
        color = "red" if red else ("green" if green else None)
        print(colored(msg, color))


description = """\
Setup your chatmail server configuration and
deploy it via SSH to your remote location.
"""


def add_config_option(parser):
    parser.add_argument(
        "--config",
        dest="chatmail_ini",
        action="store",
        default=Path("chatmail.ini"),
        type=Path,
        help="path to the chatmail.ini file",
    )


def add_subcommand(subparsers, func):
    name = func.__name__
    assert name.endswith("_cmd")
    name = name[:-4]
    doc = func.__doc__.strip()
    p = subparsers.add_parser(name, description=doc, help=doc)
    p.set_defaults(func=func)
    add_config_option(p)
    return p


def get_parser():
    """Return an ArgumentParser for the 'cmdeploy' CLI."""
    parser = argparse.ArgumentParser(description=description)
    subparsers = parser.add_subparsers(
        title="subcommands",
    )

    init_parser = add_subcommand(subparsers, init_cmd)
    init_parser.add_argument(
        "chatmail_domain",
        action="store",
        help="fully qualified DNS domain name for your chatmail instance",
    )

    install_parser = add_subcommand(subparsers, run_cmd)
    install_parser.add_argument(
        "--dry-run",
        dest="dry_run",
        action="store_true",
        help="don't actually modify the server",
    )

    add_subcommand(subparsers, webdev_cmd)

    add_subcommand(subparsers, test_cmd)

    return parser



def init_cmd(args, out):
    """Initialize chatmail config file."""
    if args.chatmail_ini.exists():
        out.red(f"Path exists, not modifying: {args.chatmail_ini}")
        raise SystemExit(1)
    write_initial_config(args.chatmail_ini, args.chatmail_domain)
    out.green(f"created config file for {args.chatmail_domain} in {args.chatmail_ini}")


def run_cmd(args, out):
    """Deploy chatmail services on the remote server."""
    import pyinfra

    try:
        config = read_config(args.chatmail_ini)
    except Exception as ex:
        out.red(ex)
        raise SystemExit(1)

    popen_args = ["pyinfra"]
    if args.dry_run:
        popen_args.append("--dry")
    popen_args.extend(["--ssh-user", "root", config.mailname])
    popen_args.append("deploy-chatmail/src/deploy_chatmail/deploy.py")

    out(f"{os.getcwd()} $ {' '.join(popen_args)}")
    env = os.environ.copy()
    env["CHATMAIL_DOMAIN"] = config.mailname
    subprocess.check_call(popen_args, env=env)


def webdev_cmd(args, out):
    """Run web development loop for static local web pages."""
    from .www import main

    main()


def test_cmd(args, out):
    """run Run web development loop for static local web pages."""

    tox = shutil.which("tox")
    subprocess.check_call([tox, "-c", "chatmaild"])
    subprocess.check_call([tox, "-c", "deploy-chatmail"])

    pytest_path = shutil.which("tox")
    subprocess.check_call([pytest_path, "tests/online", "-rs", "-x", "-vrx", "--durations=5"])


def main(args=None):
    """Provide main entry point for 'xdcget' CLI invocation."""
    parser = get_parser()
    args = parser.parse_args(args=args)
    if not hasattr(args, "func"):
        return parser.parse_args(["-h"])
    out = Out()
    args.func(args, out)


if __name__ == "__main__":
    main()