configs/chroot/sbin/remove-trunks.sh

#!/bin/bash
#
# ===========================================================================
# Copyright (C) 2008, 2009, 2010, 2011 Marc H. Thoben
# Copyright (C) 2011 Darshan Shaligram
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
# ===========================================================================
#

set -u

CHROOT="%%DGL_CHROOT%%"
BASE_DIR="%%CHROOT_CRAWL_BASEDIR%%"
GAME="%%GAME%%"

BINPATH="%%CHROOT_CRAWL_BINARY_PATH%%"
VERSIONS_DB="%%VERSIONS_DB%%"
USER_ID="%%DGL_UID%%"
DGL_USER="%%DGL_USER%%"

GAME_BASE="$GAME-"
GAME_GLOB="$GAME_BASE*"

VERBOSE=
PROMPTS_ENABLED=1

# Toss the leading slash:
BASE_DIR=${BASE_DIR#/}
BINPATH=${BINPATH#/}
FORCE=

while getopts vq opt; do
    case $opt in
        v) VERBOSE=1 ;;
        q) PROMPTS_ENABLED= ;;
        f) FORCE=1 ;;
    esac
done
shift $((OPTIND - 1))

verbose() {
    [[ -n "$VERBOSE" ]]
}

prompts-enabled() {
    [[ -n "$PROMPTS_ENABLED" ]]
}

hash-version-detail() {
    local hash="$1"
    sqlite3 "$VERSIONS_DB" <<EOF
SELECT description || ' (' || major || '.' || minor || ')'
FROM versions
WHERE hash='$hash';
EOF
}

canonicalise-version() {
    local version="$1"
    sqlite3 "$VERSIONS_DB" <<EOF
SELECT hash FROM versions
WHERE hash LIKE '$version%' OR description='$version';
EOF
}

saves-in-dir() {
    local folder="$1"
    local -a globs
    globs=(${folder}/saves/{,sprint/,zotdef/}*.{sav,chr,cs})
    if (( $# == 2 )); then
        local suffix="/$2"
        globs=($folder/saves$suffix/*.{sav,chr,cs})
    fi
    /bin/ls -1 ${globs[@]} 2>/dev/null
}

count-saves-in-dir() {
    saves-in-dir "$@" | wc -l
}

strip-save-uid-extension() {
    sed 's/-'${USER_ID}'//g;s/\.\(sav\|chr\|cs\)//g;'
}

savegame-dirs() {
    /bin/ls -1td $GAME_GLOB
}

list_hashes()
{
    cd $CHROOT/$BASE_DIR

    if verbose; then
	echo "Date             Version                       Amount  Players"
	echo "**************************************************************"

	for folder in $(savegame-dirs); do
            local hash=${folder/$GAME_BASE/}
            local version_detail="$(hash-version-detail $hash)"
            local save_count="$(count-saves-in-dir "$folder")"
	    /bin/ls -1tdl ${folder} 2>/dev/null | \
		sed "s/$GAME_BASE//g;" | \
		awk '{ printf "%3s %2s  %s    %17s %6s    ",
                    $6, $7, $8,
                    "'"$version_detail"'",
                    "'"$save_count"'" }'

	    for char in $(saves-in-dir "$folder" | strip-save-uid-extension)
	    do
		echo -n "${char#$folder/saves/} "
	    done

	    echo
	done
    else
	echo "Date             Version                          Players in Games"
	echo "*******************************************************************"

	for folder in $(savegame-dirs)
	do
            local hash=${folder/$GAME_BASE/}
            local version_detail="$(hash-version-detail $hash)"
            local save_count="$(count-saves-in-dir "$folder" '')"
            local sprint_save_count="$(count-saves-in-dir "$folder" 'sprint')"
            local zotdef_save_count="$(count-saves-in-dir "$folder" 'zotdef')"
	    /bin/ls -1tdl ${folder} 2>/dev/null | \
		sed "s/$GAME_BASE//g;" | \
		awk '{ printf "%3s %2s  %s    %17s %6s in trunk, %3s in sprint, %3s in zotdef", $6, $7, $8,
                    "'"$version_detail"'",
                    "'"$save_count"'",
                    "'"$sprint_save_count"'",
                    "'"$zotdef_save_count"'" }'
	    echo
	done
    fi
}

PARAMS="$(echo "$*" | sed 's/[$*\/();|+]//g')"

if test $# -eq 0
then
    if prompts-enabled; then
	echo "Usage: $(basename $0) [-q] [-v] [hash] [hash] ..."
	echo
    fi
    list_hashes
    echo
    exit 0
fi

cd $CHROOT


for raw_version in ${PARAMS}
do
    version="$(canonicalise-version $raw_version)"
    if [[ -z "$version" ]]; then
        echo -e "Revision $raw_version unknown\n";
        exit 1
    fi

    GAME_VER="$GAME-$version"
    GAME_SAVEDIR="$BASE_DIR/$GAME_VER"
    if test -f $BINPATH/$GAME_VER -a -d $GAME_SAVEDIR; then
        if [[ "$(count-saves-in-dir "$GAME_SAVEDIR")" > 0 && -z "$FORCE" ]]
        then
            echo "$GAME_SAVEDIR contains save games, use -f to delete anyway"
            exit 1
        fi
        
	if prompts-enabled; then
	    while { ps -fC $GAME_VER | awk '{ print $1" "$2"\t "$5" "$7"\t "$8" "$9" "$10 }' | grep ^"$DGL_USER"; }
	    do
		echo "There are still active crawl processes running..."
		echo "-- Press RETURN to try again --"
		read
	    done

	    echo "Removing ${version} from repository..."
	    echo "delete from versions where hash=\"${version}\";"| sqlite3 ${VERSIONS_DB}
	    rm $BINPATH/$GAME_VER
	    rm -r $BASE_DIR/$GAME_VER
	else
	    if { ps -fC $GAME_VER | awk '{ print $1" "$2"\t "$5" "$7"\t "$8" "$9" "$10 }' | grep ^"$DGL_USER" >/dev/null; }
	    then
		echo "Revision ${version} is being played..."
	    else
		echo "Removing ${version} from repository..."
		echo "delete from versions where hash=\"${version}\";"| sqlite3 ${VERSIONS_DB}
		rm $BINPATH/$GAME_VER
		rm -r $BASE_DIR/$GAME_VER
	    fi
	fi
    else
	echo "Revision ${version} not found..."
    fi
done

echo
exit 0