Files
q-tools/files/fastdu

96 lines
2.3 KiB
Bash
Executable File

#!/bin/bash
function helpexit() {
echo "Display file/folder sizes like 'du', but faster and actual byte sizes"
echo "Returns only the size as number"
echo " -h Human readable units"
echo " -s Summary mode, otherwise each argument folder calculated separate"
echo " --help This help"
exit
}
function filesize() {
# Return a human readable size from integer of bytes
# Usage: filesize 10000
[ "$1" = "0" ] && {
echo "0 B"
return 0
}
awk 'BEGIN{ x = '$1'
split("B kB MB GB TB PB EB ZB",type)
for(i=7;y < 1;i--)
y = x / (2^(10*i))
str=int(y*10)/10 " " type[i+2]
if (x==0) { str = "0 B" }
print str
}' || return $?
}
function siprefix() {
# Return a SI prefixed string from integer
[ "$1" = "0" ] && {
echo "0"
return 0
}
awk 'BEGIN{ x = '$1'
split("_ k M G T P E Z",type)
for(i=7;y < 1;i--)
y = x / (10^(3*i))
str=int(y*10)/10 " " type[i+2]
if (x==0) { str = "0" }
if (i==-1) { str = x }
print str
}' || return $?
}
function processfolder() {
if [[ "$COUNT" -eq 1 ]]; then
countfiles "$@"
else
displaysize "$@"
fi
}
function countfiles() {
SIZE=$( find "$@" -type f -printf %s"\n" 2>/dev/null | awk '{ sum += 1 } END { print int(sum)+0 }' )
[[ "$HUMAN" -eq 1 ]] && {
SIZE=$( siprefix "$SIZE" )
}
echo "${SIZE}"
}
function displaysize() {
SIZE=$( find "$@" -type f -printf %s"\n" 2>/dev/null | awk '{ sum += $1 } END { print int(sum)+0 }' )
[[ "$HUMAN" -eq 1 ]] && {
SIZE=$( filesize "$SIZE" )
}
echo "$SIZE"
}
# Get options
what=()
for (( i=1; i<=$#; i++ )); do
[[ "${!i}" = "--help" ]] && helpexit
[[ "${!i}" = "-h" ]] && { HUMAN=1; continue; }
[[ "${!i}" = "-c" ]] && { COUNT=1; continue; }
[[ "${!i}" = "-s" ]] && { SUMMARY=1; continue; }
what+=( "${!i}" )
done
[[ -z "${what[@]}" ]] && what="."
# If only one entry, dont print the name
[[ "${#what[@]}" -eq 1 ]] && SUMMARY=1
if [[ "$SUMMARY" -eq 1 ]]; then
# Display one line
processfolder "${what[@]}"
else
# One size for each argument
for dir in "${what[@]}"; do
printf "%s\t%s\n" "$( processfolder "$dir" )" "$dir"
done
fi