AlternC/src/update_quota_mail.sh

107 lines
3.3 KiB
Bash
Executable File

#!/bin/bash
. /usr/lib/alternc/functions.sh
#You can call this script either without arguments, inwich case each maildir quotas will be recalculated
#or you can call it with a directory reffering to a maildir to just sync one mailbox
function panic() {
echo "$1" >&2
exit 1
}
# FIXME: storing THAT amount of data in MAILDIRS (on BIG install like Lautre.net) may crash the shell?
# Generate the $maildirs list based on the arguments
while getopts "am:d:c:" optname
do
case "$optname" in
"a")
# All mails
maildirs=$(mysql_query "select path from mailbox")
;;
"m")
# An email
if [[ ! "$OPTARG" =~ ^[^\@]*@[^\@]*$ ]] ; then
panic "bad mail address provided"
fi
if [[ ! "$(mysql_query "select userdb_home from dovecot_view where user = '$OPTARG'")" ]]; then
panic "non existant mail address"
fi
maildirs=$(mysql_query "select userdb_home from dovecot_view where user = '$OPTARG'")
;;
"d")
# Expecting a domain
# Check if domain is well-formed
if [[ ! "$OPTARG" =~ ^[a-z0-9\-]+(\.[a-z\-]+)+$ ]] ; then
panic "bad domain provided"
fi
# Attemp to get from database.
if [[ ! "$(mysql_query "select domaine from domaines where domaine = '$OPTARG'")" ]]; then
# Seem to be empty
panic "non existant domain"
fi
maildirs=$(mysql_query "select userdb_home from dovecot_view where user like '%@$OPTARG'")
;;
"c")
# An account
if [[! "$OPTARG" =~ ^[a-z0-9]*$ ]] ; then
panic "bad account provided"
fi
if [[! "$(mysql_query "select domaine from domaines where domaine = '$1'")" ]]; then
panic "non existant account"
fi
maildirs=$(mysql_query "select userdb_home from dovecot_view where userdb_uid = $OPTARG")
;;
"?")
panic "Unknown option $OPTARG - stop processing"
;;
":")
panic "No argument value for option $OPTARG - stop processing"
;;
*)
# Should not occur
echo
panic "Unknown error while processing options"
;;
esac
done
# Now we have $maildirs, we can work on it
# FIXME add check if maildir is empty
#Then we loop through every maildir to get the maildir size
for i in $maildirs ; do
if [ ! -d "$i" ];then
echo "The maildir $i does not exists. It's quota won't be resync"
continue
fi
if [ ! -d "$i" ];then
echo "The maildir $i does not exists. It's quota won't be resync"
continue
fi
# We grep only mails, not the others files
mails=`find $i -type f | egrep "(^$i)*[0-9]+\.M"`
# This part count the total mailbox size (mails + sieve scripts + ...)
size=`du -b -s $i|awk '{print $1}'`
mail_count=`echo $mails|wc -w`
echo "folder : "$i
echo "mail count : "$mail_count
echo "dir size : "$size
echo ""
#update the mailbox table accordingly
mysql_query "UPDATE mailbox SET bytes=$size WHERE path='$i' ; "
mysql_query "UPDATE mailbox SET messages=$mail_count WHERE path='$i' ; "
done
# may cause a problem, let's fix this here :)
mysql_query "UPDATE mailbox SET quota=0 WHERE quota IS NULL;"