#!/bin/bash
# 统计各虚拟主机站点目录占用，写入 /tmp 供面板读取（面板 open_basedir 限制无法直接读 /home）
DB=/vhs/kangle/etc/vhs.db
OUT=/tmp/kangle_web_quota.txt
TMP=$OUT.$$
: > "$TMP"
sqlite3 "$DB" "select name||' '||doc_root from vhost;" 2>/dev/null | while read vh doc; do
  [ -n "$vh" ] || continue
  [ -d "$doc" ] || continue
  mb=$(du -sm "$doc" 2>/dev/null | awk '{print $1}')
  [ -z "$mb" ] && mb=0
  echo "$vh $mb" >> "$TMP"
done
mv "$TMP" "$OUT"
chmod 644 "$OUT"