#!/bin/bash

# Script to generate file system usage information files (.pkgsize)

# for use of the install-jds script to calculate the disk space

# requirements

if [ $# = 0 -o "x$1" = "x--help" -o "x$1" = "x-h" ]; then

    echo "Usage: $0 directory..."

    echo "Calculate file system usage information files for the given directories"

    exit 1

fi

TEMPFILES=

register_tempfile () {

    TEMPFILES="$TEMPFILES $1"

}

clean_up () {

    IFS=" "

    for f in $TEMPFILES; do

	rm -rf $TEMPFILES

    done

}

trap clean_up HUP INT TERM ERR

TEMPDIR=/tmp/.gen_pkgsize.$$

register_tempfile $TEMPDIR

mkdir -p $TEMPDIR

fatal () {

    for msg in "${@}"; do

	echo $msg 1>&2

    done

    clean_up

    exit 1

}

unpack () {

    cd $TEMPDIR

    gzip -dc "$1" | /bin/tar xf -

}

print_subdir () {

    cd $TEMPDIR/$3

    dummy=$(echo */*)

    echo "$dummy" | grep '\*' > /dev/null

    if [ $? != 0 ]; then

	du -s */* | sed -e "s%^\([0-9]*\)[ 	]*\(.*\)%$1 \1 $2/\2%"

    else

	du -s * | sed -e "s%^\([0-9]*\)[ 	]*\(.*\)%$1 \1 $2/\2%"

    fi

}

print_sizes () {

    pname=`echo "$1" | sed -e 's/\.tar\.gz$//' -e 's/\.tgz$//'`

    cd $TEMPDIR

    test ! -d $pname && fatal "Cannot find directory \"$pname\" after unpacking \"$1\""

    basedir=`grep '^BASEDIR=' $pname/pkginfo | cut -f2 -d=`

    if [ "x$basedir" = x/ ]; then

	basedir=

    fi

    test -d "$pname/reloc" && print_subdir "$pname" "$basedir" "$pname/reloc"

    test -d "$pname/root" && print_subdir "$pname" "$basedir" "$pname/root"

}

gen_pkgsize () {

    echo "Generating .pkgsize in $1"

    dir_pkgs=$(cd "$1"; find . -name 'SUNW*.tar.gz' -print -o -name 'SUNW*.tgz' -print | sed -e 's/^.\///')

    IFS=" 	

"

    sizefile="$1/.pkgsize"

    cat /dev/null > "$sizefile"

    for pkg in $dir_pkgs; do

	unpack "$1/$pkg" || fatal "Failed to unpack $1/$pkg"

	print_sizes "$pkg" >> "$sizefile"

	rm -rf $TEMPDIR

	mkdir -p $TEMPDIR

    done

}

# main ()

IFS=" 	

"

mydir=`pwd`

for dir in "${@}"; do

    case "$dir" in

	/*)

	    gen_pkgsize "$dir"

	    ;;

	*)

	    gen_pkgsize "$mydir/$dir"

	    ;;

    esac

done

clean_up

exit 0