#!/bin/sh

# MANUALLY move cork.rc to specified location, and create HOME directory

# Original Author: edolg.net

domain='example.com'

webdir='/var/www'


# check whether the drive that holds a given directory is marked as "nodev"
# directories marked as "nodev" cannot contain special devices (e.g. /dev/null)
is_nodev() {
	dir="$1"

	# fstab dir that contains (or is) $dir
	mntp=$(mountp $dir)

	if grep "[[:space:]]$mntp[[:space:]]" </etc/fstab | grep -E '(,|[[:space:]])nodev(,|[[:space:]])' >/dev/null
	then
		return 0
	fi

	return 1
}

# find the innermost mount point which contains the given directory from /etc/fstab entries
mountp() {
	dir="$1"

	mntps=$(awk '{ print $2 }' </etc/fstab)
	while :
	do
		echo "$mntps" | grep "^$dir$" >/dev/null
		test $? -eq 0 && break
		test $dir = '/' && return 1	# avoid infinite loop
		dir=$(dirname $dir)
	done

	echo $dir
}

lncp() {
	last=$(eval echo $"$#")
	rest=$(for arg in $(seq 1 $(($# - 1))); do eval echo $"$arg"; done)
	for f in $rest
	do
		ln "$f" "$last" 2>/dev/null || cp "$f" "$last"
	done
}

lsso() {
	ldd "$@" | awk '{ print $7 }' | grep '^/.*\.so' | sort | uniq
}

preinst() {
	if [ $(uname) != "OpenBSD" ]
	then
		echo "$0: operating system is not OpenBSD" >&2
		return 1
	fi

	if [ $(whoami) != "root" ]
	then
		echo "$0: root user required" >&2
		return 1
	fi

	# check webdir's value
	echo "$webdir" | grep -E '^(/[^[:cntrl:]]+)+$' >/dev/null
	if [ $? -eq 1 ]
	then
		echo "$0: invalid chroot directory" >&2
		return 1
	fi
}

httpdconf() {
	# backup current httpd.conf
	if [ -r /etc/httpd.conf ]
	then
		cp /etc/httpd.conf /etc/httpd.conf.bk
		echo "$0: /etc/httpd.conf exists, backed up to /etc/httpd.conf.bk" >&2
	fi

	echo "$httpdconffile" >/etc/httpd.conf
}

fstabconf() {
	if is_nodev $webdir
	then
		cp /etc/fstab /etc/fstab.bk

		# remove "nodev" from $webdir in /etc/fstab to make /dev/null
		# this requires a reboot to be effective
		mntp=$(mountp $webdir)
		oldline=$(grep "[[:space:]]$mntp[[:space:]]" /etc/fstab)
		newline=$(echo "$oldline" | sed 's/nodev//;s/,,/,/')
		oldfile=$(cat /etc/fstab)
		echo "$oldfile" | sed "s!$oldline!$newline!" >/etc/fstab
		echo "$0: /etc/fstab changed, reboot required to apply"
	fi
}

# make the plan 9 environment in the new root
mk9env() {
	# install plan9port in $webdir
	pkg_add git || return 1
	git clone https://github.com/9fans/plan9port $webdir$p9pdir || return 1
	( cd $webdir$p9pdir ; ./INSTALL -r $p9pdir ) || return 1

	# lncp libraries required by plan9port into the chroot environment
	libs=$(lsso $(find $webdir$p9pdir -type f) 2>/dev/null)
	for l in $libs
	do
		d=$(dirname $l)
		mkdir -p $webdir/$d
		lncp $l $webdir/$d
	done

	# all programs need to be in $webdir/bin and some are missing
	rm -Rf $webdir/bin
	mv $webdir$p9pdir/bin $webdir/bin
	lncp /bin/{pwd,mv} $webdir/bin

	# create devices
	mkdir $webdir/dev
	( cd $webdir/dev ; /dev/MAKEDEV std )

}

services() {
	rcctl enable slowcgi httpd
}


all() {
	if ! preinst
	then
		echo "$0: could not complete pre-installation checks" >&2
		exit 1
	fi

	if ! httpdconf
	then
		echo "$0: could not configure httpd" >&2
		exit 1
	fi

	if ! fstabconf
	then
		echo "$0: could not configure /etc/fstab" >&2
		exit 1
	fi

	if ! inst
	then
		echo "$0: could not install cork" >&2
		exit 1
	fi

	if ! mk9env
	then
		echo "$0: could not add files and directories to $webdir" >&2
		exit 1
	fi

	if ! services
	then
		echo "$0: could not enable required services" >&2
		exit 1
	fi

	echo
	echo "$0: setup completed!"
	echo "$0: you may need to reboot (see prior messages); otherwise, you can start httpd and slowcgi"
}

domain=${domain:-"example.com"}
webdir=${webdir:-"/var/www"}

# other useful variables
p9pdir='/plan9'	# after chroot, full is $webdir$p9pdir
siteroot="$webdir/sites/$domain"
httpdconffile='server "'$domain'" {

	# see httpd.conf(5) to enable ssl/tls
        listen on * port 80
        connection request timeout 4
        location "/img/*" {
                root "/sites/'$domain'"
        }
        location found "/*" {
                root "/sites/'$domain'/home"
        }
        location not found "/*" {
                root "/"
                fastcgi {
                        param PATH "/bin"
                        param PLAN9 "/usr/local/plan9"
                        param SCRIPT_FILENAME "/sites/'$domain'/cork.rc"
                        socket "/run/slowcgi.sock"
                }
        }
}

types {
	include "/usr/share/misc/mime.types"
}'

if [ $# -ne 0 ]
then
	for f
	do
		$f
	done
else
	all
fi