89 lines
2.4 KiB
Plaintext
Executable File
89 lines
2.4 KiB
Plaintext
Executable File
#!env xonsh
|
|
# vim: set ft=xonsh:
|
|
import argparse
|
|
import pathlib
|
|
import re
|
|
import sys
|
|
import time
|
|
from shutil import rmtree
|
|
|
|
|
|
def get_vagrant_provider(provider):
|
|
"Vagrant provider is not always the same as Packer provider"
|
|
if provider.startswith("qemu"):
|
|
return "libvirt"
|
|
elif provider.startswith("virtualbox"):
|
|
return "virtualbox"
|
|
elif provider.startswith("vmware"):
|
|
return "vmware"
|
|
elif provider.startswith("hyperv"):
|
|
return "hyperv"
|
|
else:
|
|
return provider
|
|
|
|
|
|
def get_providers():
|
|
"List of providers as defined in sources/"
|
|
providers: set[str] = {
|
|
"qemu.amd64"
|
|
"virtualbox-iso.amd64",
|
|
"vmware-iso.amd64"
|
|
}
|
|
return providers
|
|
|
|
|
|
def get_parser(args, unsupported=set()):
|
|
parser = argparse.ArgumentParser("Builder")
|
|
parser.add_argument("--headless", action="store_true")
|
|
parser.add_argument("--upload", action="store_true")
|
|
parser.add_argument("--provider", action="append")
|
|
parsed = parser.parse_args(args)
|
|
opts = {}
|
|
if parsed.headless:
|
|
opts["headless"] = "headless=true"
|
|
else:
|
|
opts["headless"] = "headless=false"
|
|
if parsed.upload:
|
|
opts["upload"] = ""
|
|
else:
|
|
opts["upload"] = "-except=upload"
|
|
if parsed.provider:
|
|
opts["providers"] = set(parsed.provider)
|
|
else:
|
|
opts["providers"] = get_providers()
|
|
# Filter providers
|
|
opts["providers"] -= unsupported
|
|
return opts
|
|
|
|
|
|
def main():
|
|
# Equivalent of `set -e` in Bash
|
|
$RAISE_SUBPROC_ERROR = True
|
|
if len(sys.argv) == 2:
|
|
dry_run = True
|
|
else:
|
|
dry_run = False
|
|
for box in g`output/**/*.box`:
|
|
p = pathlib.Path(box)
|
|
with open("Vagrantfile.in", "r") as fr:
|
|
content = fr.read()
|
|
content = content.replace("@@BOX@@", p.resolve().as_uri())
|
|
content = content.replace("@@BOX_NAME@@", p.name)
|
|
with open("Vagrantfile", "w") as fw:
|
|
fw.write(content)
|
|
if not dry_run:
|
|
vagrant box add @(box) --name @(p.name) --force
|
|
if str(p.parent.name) in ["virtualbox", "libvirt", "vmware", "hyperv"]:
|
|
vagrant up --provider @(p.parent.name)
|
|
else:
|
|
print(f"Now whatcha wanna go and do that for? {box}")
|
|
continue
|
|
vagrant destroy -f
|
|
vag_folder = pathlib.Path(".vagrant")
|
|
if vag_folder.exists():
|
|
rmtree(vag_folder)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|