#!/bin/sh

prog=`basename $0`
maxshowdepth=0

case "$TREEDU" in
"" ) cmd=du ;;
* ) cmd="$TREEDU" ;;
esac

usage() {
cat <<END
Runs "du", and displays the output as a tree of file space used.

Use:
    $prog [-h] [-maxdepth (1..9)] [-du_opts] [--] [path ...]

The default path is ".", the current directory.
The default command is "du", but this can be overridden by setting the
environment variable TREEDU to another du-ish command.

Options (must not be combined):
    -h:		print this help message, then exit.
    -maxdepth:	only show dirs maxdepth levels deep.  -1 or -2 is very useful.
    -du_opts:	any other options (e.g. -k) are passed to the du command.
END
}

extraopts=
while true
do
    case "$1" in
    -h ) usage; exit ;;
    -[0-9] ) maxshowdepth=`expr 0 - $1` ;;
    -- ) shift ; break ;;
    -* ) extraopts="$extraopts $1" ;;
    * ) break ;;
    esac
    shift
done

test $# = 0 && set -- .
startdir=`pwd`
for d in ${1+"$@"} ; do
    cd $startdir
    cd $d
    case $maxshowdepth in
	0 ) echo "File space tree for `pwd`" ;;
	* ) echo "File space tree for `pwd` " \
		    "(max depth shown: $maxshowdepth)" ;;
    esac

    $cmd $extraopts | sort +1df | sed sx\\.xx | \
	    awk -F/ '
		BEGIN {
		    'maxshowdepth=$maxshowdepth'
		}

		NR == 1 {
		    printf " %8d\n   =======|====|====== \n", 0+$1
		}

		NR > 1  && (NF-1 <= maxshowdepth || maxshowdepth <= 0) {
		    printf " %8d | %2d | ", 0+$1, NF-1
		    for (i=2; i<NF; ++i)
			printf "    "
		    print $NF
		}' -
    echo .
done