#!/bin/bash

# Copyright (C) 2009 Andy Spencer
# See ../COPYING for terms

# GLobals
COMMAND=boot
TESTING=false
if [ $$ == 1 ]; then
	DAEMON=true
else
	DAEMON=false
fi

PATH=/lib/mkinit/bin:/bin:/sbin:/usr/bin:/usr/sbin
export PATH

# Functions 
function usage {
	echo 'usage: '$0' [options] [command]'
	echo 
	echo 'Options:'
	echo '  -h,--help      Print usage information'
	echo '  -t,--test      Fake all commands'
	echo '  -d,--daemon    For spawning stdin listener'
	echo ''
	echo 'Command:'
	echo '  boot           Execute boot-up procedures'
	echo '  halt           Execute shutdown procedures'
	echo '  reload         Re-execute init process'
	echo '  mk <rule>      Execute mk rule'
	echo '  start|stop|restart|zap|status <service>'
	echo '                 Start,stop,restart,zap or query status of service'
	echo '  eval <cmd>     Execute command in mkinit process'
	exit
}

# Handle arguments
function doopts {
	TMP=`getopt -n "$0" -o htd -l help,test,daemon -- "$@"`
	[ $? != 0 ] &&
		usage
	eval set -- "$TMP"

	# Parse options
	while [ ! "$1" == "--" ]; do
		case "$1" in
		-h|--help )   usage ;;
		-t|--test )   TESTING=true ;;
		-d|--daemon ) DAEMON=true ;;
		esac
		shift
	done
	shift
	if [ "$*" ]; then
		COMMAND=($@)
	fi

	# Debugging output
	if $TESTING; then
		echo 'Options'             1>&2
		echo '  TESTING:' $TESTING 1>&2
		echo '  DAEMON:'  $DAEMON  1>&2
		echo '  COMMAND:' $COMMAND 1>&2
	fi
}

# Run mk on the init scripts
function runamk {
	if $TESTING; then
		export P=echo
		export MKINIT_STATE=/tmp/mkinit_state
		mkdir -p $MKINIT_STATE
	fi
	/usr/lib/plan9/bin/mk \
		-f /etc/init.mk \
		-i -k "$@"
}

# Process one command
function process {
	cmd=$1
	shift
	echo mkinit -- running "$cmd" "$@"
	case "$cmd" in
	boot )
		runamk -a "$@"
		;;
	restart )
		if [ "$*" ]; then
			runamk "$@"-stop ||
				runamk "$@"-zap
			runamk "$@"-start
		fi
		;;
	start|stop|zap|status )
		if [ "$*" ]; then
			runamk "$@-$cmd"
		fi
		;;
	reload )
		$TESTING && 
			opt=-t
		exec $0 -r $opt
		;;
	poweroff|reboot|kexec|halt)
		( runamk "$cmd" "$@" & )
		;;
	eval )
		eval "$@"
		;;
	?* )
		runamk "$cmd" "$@"
		;;
	esac
}

# Process arguments
doopts "$@"

# Run whatever was requested
process "${COMMAND[@]}"

# Fork console listener
while $DAEMON; do
	while read -ep "mkinit> " line; do
		process $line
		history -s $line
	done
	$TESTING && exit
	exec 0</dev/console 1>/dev/console 2>&1
	echo "Respawning on /dev/console.."
	sleep 1
done