#!/usr/bin/env python3

import argparse
import os
import sys
import re

VARIANTS = [
    "debian",
    "freebsd",
    "centos",
    "aliyun",
    "anolis",
    "fedora",
    "rhel",
    "sles",
    "opensuse",
    "ubuntu",
    "almalinux",
    "rocky",
    "unknown"]

if "avoid-pep8-E402-import-not-top-of-file":
    _tdir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
    sys.path.insert(0, _tdir)
    from cloudinit import templater
    from cloudinit import util
    from cloudinit.atomic_helper import write_file


def is_aliyun_qboot():
    check_files = ('/etc/image-id',)
    reg = re.compile("Quick Boot")
    for fn in check_files:
        if os.path.exists(fn):
            with open(fn, "r") as fd:
                out = reg.search(fd.read())
                if out:
                    return 'qboot'
    return 'normal'


def main():
    parser = argparse.ArgumentParser()
    platform = util.system_info()
    parser.add_argument(
        "--variant", default=platform['variant'], action="store",
        help="define the variant.", choices=VARIANTS)
    parser.add_argument(
        "template", nargs="?", action="store",
        default='./config/cloud.cfg.tmpl',
        help="Path to the cloud.cfg template")
    parser.add_argument(
        "output", nargs="?", action="store", default="-",
        help="Output file.  Use '-' to write to stdout")

    args = parser.parse_args()

    with open(args.template, 'r') as fh:
        contents = fh.read()
    qboot = is_aliyun_qboot()
    tpl_params = {'variant': args.variant, 'qboot': qboot}
    contents = (templater.render_string(contents, tpl_params)).rstrip() + "\n"
    util.load_yaml(contents)
    if args.output == "-":
        sys.stdout.write(contents)
    else:
        write_file(args.output, contents, omode="w")


if __name__ == '__main__':
    main()
