#!/bin/sh

error() {
    cat <<EOF;
usage: mdkinst_stage2_tool [--clean] [--to-clp | --from-clp] <stage2 dir> [<clp file>]
EOF
    exit 1
}

if [ "$1" = "--clean" ]; then
    CLEAN=1
    shift
fi

[ $# = 2 -o $# = 3 ] || error

if [ "$1" = "--to-clp" -o "$1" == "--from-clp" ]; then
    ACTION=$1
    shift
    STAGE2_DIR="$1"
    shift
    LIVE_DIR="$STAGE2_DIR/live"
    if [ -n "$1" ]; then
	CLP=$1
	shift
    else
	CLP="$STAGE2_DIR/mdkinst.clp"
    fi
else
    error
fi

if [ $ACTION = "--to-clp" ]; then
    which create_compressed_fs >/dev/null 2>/dev/null || { echo "missing command create_compressed_fs (from cloop-utils)"; exit 1; }

    [ -d "$LIVE_DIR" ] || error
    echo "Creating $CLP from $LIVE_DIR"
    rm -f $STAGE2_DIR/.room
    mkisofs -quiet -D -R $LIVE_DIR | create_compressed_fs - 65536 $CLP 200 2>/dev/null
    echo foo > $STAGE2_DIR/.room
    if [ -s $STAGE2_DIR/.room ]; then
	rm -f $STAGE2_DIR/.room
	[ -n "$CLEAN" ] && rm -rf $LIVE_DIR 
    else
	echo "not enough space"
	rm -f $CLP
	exit 1
    fi
else
    which extract_compressed_fs >/dev/null 2>/dev/null || { echo "missing command extract_compressed_fs (from cloop-utils)"; exit 1; }

    [ -f "$CLP" ] || error
    echo "Creating $LIVE_DIR from $CLP"

    TMPDIR=${TMPDIR=/tmp}
    MNTPOINT=$TMPDIR/stage2_img

    if [ $EUID != "0" ]; then
	SUDO="sudo"
	PATH="/sbin:/usr/sbin:$PATH"
    fi

    $SUDO umount $MNTPOINT 2>/dev/null
    mkdir -p $MNTPOINT
    extract_compressed_fs $CLP > $CLP.iso 2>/dev/null
    $SUDO mount -o loop $CLP.iso $MNTPOINT || { echo "can not loopback mount $CLP.iso"; exit 1; }
    mkdir $LIVE_DIR || { echo "not enough space"; exit 1; }
    cp -a $MNTPOINT/* $LIVE_DIR/ || { rm -rf $LIVE_DIR; exit 1; }
    $SUDO umount $MNTPOINT
    rm -f $CLP.iso
    [ -n "$CLEAN" ] && rm -f $CLP
fi

exit 0