feat: add dirsize

2026-05-13 22:44:10 +08:00
parent 68e6780238
commit 27e5c11ea9
5 changed files with 160 additions and 78 deletions
@@ -1,25 +0,0 @@
-#!/bin/bash
-
-# 检查是否输入了目录参数，如果没有，默认使用当前目录
-TARGET_DIR=${1:-"."}
-
-echo "--- 正在扫描目录: $TARGET_DIR ---"
-echo "--- 正在查找最大的 10 个文件/文件夹 ---"
-echo "----------------------------------------"
-
-# 执行查找逻辑
-# -d 1: 只查看当前目录层级（不展开子目录的所有细碎文件）
-# -h: 以人类可读格式显示 (K, M, G)
-# sort -rh: 按人类可读的数字反向排序 (需要支持 -h 的 sort，Mac/Linux 通用)
-# head -n 10: 取前 10 名
-
-if [[ "$OSTYPE" == "darwin"* ]]; then
-    # Mac (BSD 版本的 du 略有不同)
-    du -sh "$TARGET_DIR"/* | sort -rh | head -n 10
-else
-    # Linux (GNU 版本)
-    du -sh "$TARGET_DIR"/* --max-depth=1 2>/dev/null | sort -rh | head -n 10
-fi
-
-echo "----------------------------------------"
-echo "扫描完成。"
@@ -0,0 +1,77 @@
+#!/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"