#!/bin/sh
# 
# Copyright (c) 2018, Christer Edwards <christer.edwards@gmail.com>
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# 
# * Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.
# 
# * 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.
# 
# * Neither the name of the copyright holder nor the names of its
#   contributors may be used to endorse or promote products derived from
#   this software without specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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 COPYRIGHT HOLDER 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.

. /usr/local/share/bastille/colors.pre.sh
. /usr/local/etc/bastille/bastille.conf

## version
BASTILLE_VERSION="0.3.20181107"


usage() {
    cat << EOF
Bastille is a jail automation framework that allows you to quickly and easily
create and manage FreeBSD jails.

Usage:
  bastille command [ALL|glob] [args]

Available Commands:
  bootstrap   Bootstrap a FreeBSD release for container base.
  cmd         Execute arbitrary command on targeted container(s).
  console     Console into a running container.
  cp          cp(1) files from host to targeted container(s).
  create      Create a new container.
  destroy     Destroy a stopped container.
  help        Help about any command
  htop        Interactive process viewer (requires htop).
  list        List containers (running and stopped).
  pkg         Manipulate binary packages within targeted container(s). See pkg(8).
  restart     Restart a running container.
  start       Start a stopped container.
  stop        Stop a running container.
  sysrc       Safely edit rc files within targeted container(s).
  top         Display and update information about the top(1) cpu processes.
  update      Update container base -pX release.
  upgrade     Upgrade container release to X.Y-RELEASE.

Use "bastille -v|--version" for version information.
Use "bastille command -h|--help" for more information about a command.

EOF
    exit 1
}

[ $# -lt 1 ] && usage

CMD=$1
shift
CMD_ENV=

# Handle special-case commands first.
case "${CMD}" in
version|-v|--version)
	echo -e "${COLOR_GREEN}${BASTILLE_VERSION}${COLOR_RESET}"
	exit 0
	;;
help|-h|--help)
	usage
	;;
esac

# Filter out all non-commands
case "${CMD}" in
cmd|console|cp|create|destroy|list|pkg|restart|start|stop|sysrc|verify)
	;;
update|upgrade)
	CMD_ENV="${CMD_ENV} PAGER=cat"
	;;
console|bootstrap|htop|top)
	while read envvar envvalue; do
		case "${envvar}" in
			TERM)
				CMD_ENV="${CMD_ENV} ${envvar}=${envvalue}"
				;;
		esac
	done <<-EOF
	$(env | sed -Ee 's,^([^=]*)=(.*),\1 \2,')
	EOF
	;;
bootstrap|update|upgrade)
	while read envvar envvalue; do
		case "${envvar}" in
			FETCH_BIND_ADDRESS|FTP_*|ftp_*|HTTP_*|http_*|SSL_|NO_PROXY|no_proxy|MAKEOBJDIRPREFIX)
				CMD_ENV="${CMD_ENV} ${envvar}=${envvalue}"
				;;
		esac
	done <<-EOF
	$(env | sed -Ee 's,^([^=]*)=(.*),\1 \2,')
	EOF
	;;
*)
	usage
	;;
esac

SCRIPTPATH="${bastille_sharedir}/${CMD}.sh"

: ${UMASK:=022}
umask ${UMASK}

: ${SH:=sh}

exec env -i ${CMD_ENV} ${SH} "${SCRIPTPATH}" "$@"
