#!/usr/bin/env xonsh
from argparse import ArgumentParser
from pathlib import Path
import sys
sys.path.append(str(Path(__file__).parent / "lib"))
from support import Support, BASE


support = Support()
parser = ArgumentParser("Build a box")
parser.add_argument("--distro", "-d", help="Name of the distro/ folder")
parser.add_argument("--version", "-v", help="Version of the distro to build")
parser.add_argument("--build", "-b", help="Name of the target build, e.g. qemu.x86_64, virtualbox-iso.x86_64")
parser.add_argument("--list", "-l", help="List all build targets")
parser.add_argument("--all", "-a", help="Build all applicable targets", action="store_true")
parser.add_argument("--upload", "-u", help="Force doing uploads", action="store_true")
parser.add_argument("--headless", help="Run builds headless", action="store_true")
args = parser.parse_args()

# Now provide menus if we don't have defaults
def process_all():
    if args.distro:
        process_distro(support.distros[args.distro])
    else:
        for name, distro in support.distros.items():
            process_distro(distro)

def process_distro(distro):
    if args.version:
        process_builds(distro.builds_for_version(args.version))
    else:
        process_builds(distro.builds)

def process_builds(builds):
        for build in builds:
            if args.build:
                name, arch = args.build.split(".")
                if build.provider.name == name and build.provider.arch == arch:
                    do_build(build)
            else:
                do_build(build)

def do_build(build):
    if args.upload:
        upload = ""
    else:
        upload = "-except=upload"
    if args.headless:
        headless = "headless=true"
    else:
        headless = "headless=false"
    packer build -var @(headless) @(build.var_file) @(build.only) @(upload) @(BASE / "sources")

if args.all or (args.distro and args.version and args.build):
    process_all()
else:
    print("Doing menu stuff")