#!/bin/sh
# Copyright © 2008-2009  Jan-Marek Glogowski <glogow@fbihome.de>
#
# schroot is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# schroot is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see
# <http://www.gnu.org/licenses/>.
#
#####################################################################

set -e

. "$SETUP_DATA_DIR/common-data"
. "$SETUP_DATA_DIR/common-functions"

if [ -f "$CHROOT_SCRIPT_CONFIG" ]; then
    . "$CHROOT_SCRIPT_CONFIG"
elif [ "$STATUS" = "ok" ]; then
    fatal "script-config file '$CHROOT_SCRIPT_CONFIG' does not exist"
fi

if [ -n "${CHROOT_UNION_TYPE}" ] && [ "${CHROOT_UNION_TYPE}" != 'none' ]; then

    if [ $STAGE = "setup-start" ]; then
	mkdir "${CHROOT_UNION_OVERLAY_DIRECTORY}"
	if [ ! -d "$CHROOT_UNION_OVERLAY_DIRECTORY" ]; then
	    fatal "$CHROOT_UNION_OVERLAY_DIRECTORY does not exist, and could not be created"
	fi

	mkdir "${CHROOT_UNION_UNDERLAY_DIRECTORY}"
	if [ ! -d "$CHROOT_UNION_UNDERLAY_DIRECTORY" ]; then
	    fatal "$CHROOT_UNION_UNDERLAY_DIRECTORY does not exist, and could not be created"
	fi

    elif [ $STAGE = "setup-recover" ]; then
        if [ ! -d "${CHROOT_UNION_OVERLAY_DIRECTORY}" ]; then
	    fatal "Missing overlay directory for session: can't recover"
        fi
        if [ ! -d "${CHROOT_UNION_UNDERLAY_DIRECTORY}" ]; then
	    fatal "Missing underlay directory for session: can't recover"
        fi

    elif [ $STAGE = "setup-stop" ]; then
        if [ "$CHROOT_SESSION_PURGE" = "true" ]; then
	    info "Purging $CHROOT_UNION_OVERLAY_DIRECTORY"
	    if [ -d "${CHROOT_UNION_OVERLAY_DIRECTORY}" ]; then
		rm -rf "${CHROOT_UNION_OVERLAY_DIRECTORY}"
	    fi

	    # For safety, use rmdir rather than rm -rf in case
	    # umount failed.
	    info "Removing $CHROOT_UNION_UNDERLAY_DIRECTORY"
	    if [ -d "${CHROOT_UNION_UNDERLAY_DIRECTORY}" ]; then
		if ! rmdir "${CHROOT_UNION_UNDERLAY_DIRECTORY}"; then
		    warn "Please unmount any filesystems and remove any files under $CHROOT_UNION_UNDERLAY_DIRECTORY"
		    fatal "Failed to remove $CHROOT_UNION_UNDERLAY_DIRECTORY (directory not empty)."
		fi
	    fi
	fi
    fi

fi

