#!/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"