96 lines
1.9 KiB
Bash
Executable File
96 lines
1.9 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -eu
|
|
|
|
usage() {
|
|
cat <<'EOF'
|
|
用法: dirsize [选项] [目录]
|
|
|
|
查询目录大小，默认扫描当前目录的直接子目录。
|
|
|
|
选项:
|
|
-a, --all 递归显示所有子目录的大小
|
|
-d, --depth N 指定递归深度 (默认 1)
|
|
-n, --top N 只显示前 N 名 (默认全部)
|
|
-r, --reverse 按从小到大排序 (默认从大到小)
|
|
-h, --help 显示帮助信息
|
|
|
|
示例:
|
|
dirsize 当前目录的直接子目录大小
|
|
dirsize -n 10 当前目录的直接子目录，取前 10 名
|
|
dirsize -a 递归显示所有子目录
|
|
dirsize -d 2 -n 5 深度 2，取前 5 名
|
|
dirsize -n 5 /var/log 查看 /var/log 的子目录，取前 5 名
|
|
EOF
|
|
exit 0
|
|
}
|
|
|
|
DEPTH=1
|
|
TOP=0
|
|
REVERSE=false
|
|
ALL=false
|
|
TARGET_DIR="."
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-a | --all)
|
|
ALL=true
|
|
shift
|
|
;;
|
|
-d | --depth)
|
|
DEPTH="$2"
|
|
shift 2
|
|
;;
|
|
-n | --top)
|
|
TOP="$2"
|
|
shift 2
|
|
;;
|
|
-r | --reverse)
|
|
REVERSE=true
|
|
shift
|
|
;;
|
|
-h | --help) usage ;;
|
|
-*)
|
|
echo "未知选项: $1" >&2
|
|
usage
|
|
;;
|
|
*)
|
|
TARGET_DIR="$1"
|
|
shift
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ ! -d "$TARGET_DIR" ]]; then
|
|
echo "错误: '$TARGET_DIR' 不是有效目录" >&2
|
|
exit 1
|
|
fi
|
|
|
|
TARGET_DIR="$(cd "$TARGET_DIR" && pwd -P)"
|
|
|
|
if $ALL; then
|
|
DEPTH=999
|
|
fi
|
|
|
|
sort_flag="-rh"
|
|
if $REVERSE; then
|
|
sort_flag="-h"
|
|
fi
|
|
|
|
if [[ "$OSTYPE" == "darwin"* ]]; then
|
|
result=$(sudo du -h -d "$DEPTH" "$TARGET_DIR" 2>/dev/null | awk -v dir="$TARGET_DIR" '$2 != dir {print}' | sort $sort_flag)
|
|
else
|
|
result=$(sudo du -h --max-depth="$DEPTH" "$TARGET_DIR" 2>/dev/null | awk -v dir="$TARGET_DIR" '$2 != dir {print}' | sort $sort_flag)
|
|
fi
|
|
|
|
if [[ "$TOP" -gt 0 ]]; then
|
|
result=$(echo "$result" | head -n "$TOP")
|
|
fi
|
|
|
|
if [[ -z "$result" ]]; then
|
|
echo "目录为空或无子目录"
|
|
exit 0
|
|
fi
|
|
|
|
echo "$result"
|