#!/bin/sh
#-
# Copyright (c) 2014, 2015 The FreeBSD Foundation
# All rights reserved.
#
# This software was developed by Glen Barber under sponsorship
# from the FreeBSD Foundation.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
#
# mk-vmimage.sh: Create virtual machine disk images in various formats.
#
# $FreeBSD: stable/11/release/scripts/mk-vmimage.sh 281876 2015-04-23 01:04:14Z gjb $
#

proc usage {
	echo "$(0) usage:"
	echo $(@)
	return 1
}

proc main {
	var arg = ''
	setglobal VMCONFIG = '"/dev/null'"
	while getopts "C:c:d:f:i:o:s:S:" arg {
		match $(arg) {
			with C
				setglobal VMBUILDCONF = $(OPTARG)
				
			with c
				setglobal VMCONFIG = $(OPTARG)
				
			with d
				setglobal DESTDIR = $(OPTARG)
				
			with f
				setglobal VMFORMAT = $(OPTARG)
				
			with i
				setglobal VMBASE = $(OPTARG)
				
			with o
				setglobal VMIMAGE = $(OPTARG)
				
			with s
				setglobal VMSIZE = $(OPTARG)
				
			with S
				setglobal WORLDDIR = $(OPTARG)
				
			with *
				
		}
	}
	shift $shExpr(' ${OPTIND} - 1')

	if test -z $(VMBASE) -o \
		-z $(WORLDDIR) -o \
		-z $(DESTDIR) -o \
		-z $(VMSIZE) -o \
		-z $(VMIMAGE)
	{
		usage || exit 0
	}

	if test -z $(VMBUILDCONF) || test ! -e $(VMBUILDCONF) {
		echo "Must provide the path to vmimage.subr."
		return 1
	}

	source "${VMBUILDCONF}"

	if test ! -z $(VMCONFIG) && test ! -c $(VMCONFIG) {
		source "${VMCONFIG}"
	}

	match $(TARGET):$(TARGET_ARCH) {
		with arm64:aarch64
			setglobal ROOTLABEL = '"ufs'"
			setglobal NOSWAP = '1'
			
		with *
			setglobal ROOTLABEL = '"gpt'"
			
	}

	vm_create_base
	vm_install_base
	vm_extra_install_base
	vm_extra_install_packages
	vm_extra_install_ports
	vm_extra_enable_services
	vm_extra_pre_umount
	vm_extra_pkg_rmcache
	cleanup
	vm_copy_base
	vm_create_disk || return 0
	vm_extra_create_disk

	return 0
}

main @Argv