78 lines
1.8 KiB
Bash
Executable File
78 lines
1.8 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
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="${TARGET_DIR%/}"
|
|
|
|
if $ALL; then
|
|
DEPTH=999
|
|
fi
|
|
|
|
sort_flag="-rh"
|
|
if $REVERSE; then
|
|
sort_flag="-h"
|
|
fi
|
|
|
|
if [[ "$OSTYPE" == "darwin"* ]]; then
|
|
result=$(du -h -d "$DEPTH" "$TARGET_DIR" 2>/dev/null | grep -v "^.*${TARGET_DIR}$" | sort $sort_flag)
|
|
else
|
|
result=$(du -h --max-depth="$DEPTH" "$TARGET_DIR" 2>/dev/null | grep -v "^.*${TARGET_DIR}$" | 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"
|