#!/bin/sh

# The MIT License (MIT)
#
# Copyright (c) 2026 pacman64
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.


# filesizes [options...] [files/folders...]
#
# Show all filesizes as TSV lines, after a header/column line. You can give
# it a mix of files and folders, for convenience.
#
# The options are, available both in single and double-dash versions
#
#   -h, -help             show this help message
#   -s, -sort, -sorted    reverse-sort by filesize


sorted=0

case "$1" in
    -h|--h|-help|--help)
        awk '/^# +filesizes /, /^$/ { gsub(/^# ?/, ""); print }' "$0"
        exit 0
    ;;

    -s|--s|-sort|--sort|-sorted|--sorted)
        sorted=1
        shift
    ;;
esac

[ "$1" = '--' ] && shift

lbuf=''
if { [ -p /dev/stdout ] || [ -t 1 ]; } && [ -e /usr/bin/stdbuf ]; then
    lbuf='stdbuf -oL'
fi

swap='{ print $2, $1 }'
to_tsv='
    {
        gsub(/^ *| *\r$/, "")
        gsub(/^([^ ])*/, "&\t")
        gsub(/ *\t */, "\t")
        print
    }
'

printf "file\tbytes\n"

for arg in "${@:-.}"; do
    if [ -f "${arg}" ]; then
        wc -c "${arg}"
        continue
    fi

    if [ ! -d "${arg}" ]; then
        continue
    fi

    ${lbuf} find "${arg}" -type f -exec wc -c {} + | ${lbuf} head -n -1
done |
    ${lbuf} awk "${to_tsv}" | ${lbuf} awk -F "\t" -v OFS="\t" "${swap}" |
if [ "${sorted}" -eq 1 ]; then
    sort -t "$(printf '\t')" -rnk2,2 -k1,1
else
    cat
fi