import argparse
import os
import subprocess
import tempfile

import yaml

import generator
from utils import query_yes_no


def main():
    argparser = argparse.ArgumentParser()
    argparser.add_argument('-c', '--config', default="config.yml", type=str, metavar="PATH")
    argparser.add_argument('-n', '--dry-run', action='store_true')
    argparser.add_argument('--reset', action='store_true')
    argparser.add_argument('--override-ip', type=str)
    argparser.add_argument('files', type=str, nargs="+", metavar="NAME")
    args = argparser.parse_args()

    dry_run = args.dry_run
    cfg = yaml.load(open(args.config, "rt"), Loader=yaml.FullLoader)

    host = cfg["host"]
    has_flash = cfg.get("has_flash", False)

    if args.override_ip is not None:
        host = args.override_ip

    files = args.files

    orders = [float(os.path.basename(x).split("-")[0].replace("_", ".")) for x in files]
    if orders != list(sorted(orders)):
        print("mixed up order")
        exit(1)

    if args.reset and orders[0] != 0:
        print("reset must start with 0_0")
        exit(1)

    if not args.reset and orders[0] == 0:
        print("not reset can't start with 0_0")
        exit(1)

    if not dry_run and args.reset:
        if not query_yes_no("Are you sure you want to reset configuration?", "no"):
            exit(1)

    def gen(x):
        s = f'\n/log info message="starting {x}..."\n'
        s += generator.render_file(x, cfg.get("include_dirs", []), cfg.get("variables", {}))
        s += f'\n/log info message="finished {x}"\n'
        return s

    script = "\n".join(gen(x) for x in files)
    if args.reset:
        script = ":delay 7s\n" + script

    script += "\n/log info message=\"CONFIGURATION DONE\"\n"

    base_path = "flash/" if has_flash else ""
    script_name = "output.rsc"

    print(script)

    with tempfile.NamedTemporaryFile(mode="wt") as f:
        f.write(script)
        f.flush()

        cargs = [
            "scp",
            "-o", "StrictHostKeyChecking=false",
            "-o", "UserKnownHostsFile=/dev/null",
            "-o", "PubkeyAcceptedKeyTypes=+ssh-rsa",
            f.name,
            f"admin@{host}:{base_path}{script_name}"
        ]
        print(" ".join(cargs))
        if not dry_run:
            subprocess.check_call(cargs)

        if args.reset:
            cmd = f"/system reset-configuration no-defaults=yes skip-backup=yes run-after-reset={base_path}{script_name}"
        else:
            cmd = f"/import file={base_path}{script_name}"
        cargs = [
            "ssh",
            "-o", "StrictHostKeyChecking=false",
            "-o", "UserKnownHostsFile=/dev/null",
            "-o", "PubkeyAcceptedKeyTypes=+ssh-rsa",
            f"admin@{host}",
            cmd,
        ]
        print(" ".join(cargs))
        if not dry_run:
            if args.reset:
                subprocess.run(cargs)
            else:
                out = subprocess.check_output(cargs).decode("utf-8")

                if "Script file loaded and executed successfully" not in out:
                    print("Script error")
                    exit(1)


main()