#! /bin/sh

# $NetBSD$

# Copyright (c) 2013 Alistair Crooks <agc@NetBSD.org>
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#

# Usage: du [-r] [-s] file...
# 	-r == reverse sort, largest first, smallest last
#	-s == silent execution of du, don't recurse into subdirs

revsort=""
silent=""
args=$(getopt "rs" $*)
set -- $args
while [ $# -gt 0 ]; do
	case "$1" in
	-r)	revsort="-r" ;;
	-s)	silent="-s" ;;
	*)	break;
	esac
	shift
done

awk=/usr/bin/awk
du=/usr/bin/du
sort=/usr/bin/sort

${du} -h ${silent} $@ | ${awk} '
BEGIN {
	n = 1; sufs = "BKMGT";
	for (i = 1 ; i <= length(sufs) ; i++) {
		muls[substr(sufs, i, 1)] = n;
		n *= 1024
	}
}
$1 ~ /[0-9.]+[BKMGT]/ {
	bytes = substr($1, 1, length($1) - 1) * muls[substr($1, length($1), 1)]
}
{
	out[outc++] = sprintf("%d %s", bytes, $0)
}
END {
	for (o in out)
		print out[o]
}' | ${sort} -n ${revsort} | ${awk} '{ print $2, $3 }'