vesta/bin/v-update-user-counters
2018-11-24 22:43:21 +02:00

218 lines
6.6 KiB
Bash
Executable File

#!/bin/bash
# info: update user usage counters
# options: USER
#
# Function updates usage counters like U_WEB_DOMAINS, U_MAIL_ACCOUNTS, etc.
#----------------------------------------------------------#
# Variable&Function #
#----------------------------------------------------------#
# Argument definition
user=$1
# Includes
source $VESTA/func/main.sh
source $VESTA/conf/vesta.conf
#----------------------------------------------------------#
# Verifications #
#----------------------------------------------------------#
check_args '0' "$#" 'USER'
if [ ! -z "$user" ]; then
is_format_valid 'user'
is_object_valid 'user' 'USER' "$user"
fi
#----------------------------------------------------------#
# Action #
#----------------------------------------------------------#
# Creating user_list
if [ -z "$user" ]; then
user_list=$(ls $VESTA/data/users)
else
user_list="$user"
fi
# Updating user stats
for user in $user_list; do
USER_DATA=$VESTA/data/users/$user
# Cleaning counters
SUSPENDED_USERS=0
SUSPENDED_WEB=0
SUSPENDED_DNS=0
SUSPENDED_MAIL=0
SUSPENDED_DB=0
SUSPENDED_CRON=0
IP_AVAIL=0
IP_OWNED=0
U_USERS=0
U_DISK=0
DISK=0
U_DISK_DIRS=$(get_user_value '$U_DISK_DIRS')
if [ -z "$U_DISK_DIRS" ]; then
U_DISK_DIRS=0
fi
DISK=$((DISK + U_DISK_DIRS))
U_DISK_WEB=0
U_DISK_MAIL=0
U_DISK_DB=0
BANDWIDTH=0
U_BANDWIDTH=0
U_WEB_DOMAINS=0
U_WEB_SSL=0
U_WEB_ALIASES=0
U_DNS_DOMAINS=0
U_DNS_RECORDS=0
U_MAIL_DOMAINS=0
U_MAIL_DKIM=0
U_MAIL_ACCOUNTS=0
U_DATABASES=0
U_CRON_JOBS=0
U_BACKUPS=0
# New IFS
IFS=$'\n'
# Checking users
if [ "$user" = 'admin' ]; then
spnd=$(grep "SUSPENDED='yes'" $VESTA/data/users/*/user.conf | wc -l)
SUSPENDED_USERS=$spnd
U_USERS=$(ls $VESTA/data/users/|wc -l)
fi
# Checking ip
for ip in $(ls $VESTA/data/ips) ;do
source $VESTA/data/ips/$ip
if [ "$OWNER" = "$user" ]; then
IP_OWNED=$((IP_OWNED + 1))
IP_AVAIL=$((IP_AVAIL + 1))
fi
if [ "$OWNER" = 'admin' ] && [ "$STATUS" = 'shared' ]; then
IP_AVAIL=$((IP_AVAIL + 1))
fi
done
if [ "$user" = 'admin' ]; then
IP_AVAIL=$(ls $VESTA/data/ips | wc -l)
fi
# Checking web system
U_WEB_DOMAINS=0
for domain_str in $(cat $USER_DATA/web.conf) ;do
eval $domain_str
U_DISK_WEB=$((U_DISK_WEB + U_DISK))
U_WEB_DOMAINS=$((U_WEB_DOMAINS + 1))
if [ "$SSL" = 'yes' ]; then
U_WEB_SSL=$((U_WEB_SSL + 1))
fi
if [ "$SUSPENDED" = 'yes' ]; then
SUSPENDED_WEB=$((SUSPENDED_WEB + 1))
fi
if [ ! -z "$ALIAS" ]; then
aliases=$(echo "$ALIAS"|tr ',' '\n'| wc -l)
U_WEB_ALIASES=$((U_WEB_ALIASES + aliases))
fi
BANDWIDTH=$((BANDWIDTH + U_BANDWIDTH))
done
DISK=$((DISK + U_DISK_WEB))
# Checking dns system
for domain_str in $(cat $USER_DATA/dns.conf); do
eval $domain_str
U_DNS_DOMAINS=$((U_DNS_DOMAINS + 1))
records=$(wc -l $USER_DATA/dns/$DOMAIN.conf| cut -f 1 -d ' ')
U_DNS_RECORDS=$((U_DNS_RECORDS + records))
if [ "$SUSPENDED" = 'yes' ]; then
SUSPENDED_DNS=$((SUSPENDED_DNS + 1))
fi
update_object_value 'dns' 'DOMAIN' "$DOMAIN" '$RECORDS' "$records"
done
# Checking mail system
for domain_str in $(cat $USER_DATA/mail.conf); do
eval $domain_str
U_DISK_MAIL=$((U_DISK_MAIL + U_DISK))
U_MAIL_DOMAINS=$((U_MAIL_DOMAINS + 1))
accounts=$(wc -l $USER_DATA/mail/$DOMAIN.conf| cut -f 1 -d ' ')
U_MAIL_ACCOUNTS=$((U_MAIL_ACCOUNTS + accounts))
if [ "$DKIM" = 'yes' ]; then
U_MAIL_DKIM=$((U_MAIL_DKIM + 1))
fi
if [ "$SUSPENDED" = 'yes' ]; then
SUSPENDED_MAIL=$((SUSPENDED_MAIL + 1))
fi
update_object_value 'mail' 'DOMAIN' "$DOMAIN" '$ACCOUNTS' "$accounts"
done
DISK=$((DISK + U_DISK_MAIL))
# Checking db system
for db_str in $(cat $USER_DATA/db.conf); do
eval $db_str
U_DISK_DB=$((U_DISK_DB + U_DISK))
U_DATABASES=$((U_DATABASES + 1))
if [ "$SUSPENDED" = 'yes' ]; then
SUSPENDED_DB=$((SUSPENDED_DB + 1))
fi
done
DISK=$((DISK + U_DISK_DB))
# Checking cron system
for cron_str in $(cat $USER_DATA/cron.conf);do
eval $cron_str
U_CRON_JOBS=$((U_CRON_JOBS + 1))
if [ "$SUSPENDED" = 'yes' ]; then
SUSPENDED_CRON=$((SUSPENDED_CRON + 1))
fi
done
# Checking backup
for backup_str in $(cat $USER_DATA/backup.conf); do
U_BACKUPS=$((U_BACKUPS +1))
done
U_DISK=$DISK
U_BANDWIDTH=$BANDWIDTH
update_user_value "$user" '$SUSPENDED_USERS' "$SUSPENDED_USERS"
update_user_value "$user" '$SUSPENDED_WEB' "$SUSPENDED_WEB"
update_user_value "$user" '$SUSPENDED_DNS' "$SUSPENDED_DNS"
update_user_value "$user" '$SUSPENDED_MAIL' "$SUSPENDED_MAIL"
update_user_value "$user" '$SUSPENDED_DB' "$SUSPENDED_DB"
update_user_value "$user" '$SUSPENDED_CRON' "$SUSPENDED_CRON"
update_user_value "$user" '$IP_AVAIL' "$IP_AVAIL"
update_user_value "$user" '$IP_OWNED' "$IP_OWNED"
update_user_value "$user" '$U_USERS' "$U_USERS"
update_user_value "$user" '$U_DISK' "$U_DISK"
update_user_value "$user" '$U_DISK_DIRS' "$U_DISK_DIRS"
update_user_value "$user" '$U_DISK_WEB' "$U_DISK_WEB"
update_user_value "$user" '$U_DISK_MAIL' "$U_DISK_MAIL"
update_user_value "$user" '$U_DISK_DB' "$U_DISK_DB"
update_user_value "$user" '$U_BANDWIDTH' "$U_BANDWIDTH"
update_user_value "$user" '$U_WEB_DOMAINS' "$U_WEB_DOMAINS"
update_user_value "$user" '$U_WEB_SSL' "$U_WEB_SSL"
update_user_value "$user" '$U_WEB_ALIASES' "$U_WEB_ALIASES"
update_user_value "$user" '$U_DNS_DOMAINS' "$U_DNS_DOMAINS"
update_user_value "$user" '$U_DNS_RECORDS' "$U_DNS_RECORDS"
update_user_value "$user" '$U_MAIL_DOMAINS' "$U_MAIL_DOMAINS"
update_user_value "$user" '$U_MAIL_DKIM' "$U_MAIL_DKIM"
update_user_value "$user" '$U_MAIL_ACCOUNTS' "$U_MAIL_ACCOUNTS"
update_user_value "$user" '$U_DATABASES' "$U_DATABASES"
update_user_value "$user" '$U_CRON_JOBS' "$U_CRON_JOBS"
update_user_value "$user" '$U_BACKUPS' "$U_BACKUPS"
done
#----------------------------------------------------------#
# Vesta #
#----------------------------------------------------------#
# Logging
log_event "$OK" "$ARGUMENTS"
exit