You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
233 lines
6.6 KiB
233 lines
6.6 KiB
2 years ago
|
#!/bin/bash
|
||
|
|
# info: update user usage counters
|
||
|
|
# options: USER
|
||
|
|
#
|
||
|
|
# example: v-update-user-counters admin
|
||
|
|
#
|
||
|
|
# Function updates usage counters like U_WEB_DOMAINS, U_MAIL_ACCOUNTS, etc.
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Variables & Functions #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
# Argument definition
|
||
|
|
user=$1
|
||
|
|
|
||
|
|
# Includes
|
||
|
|
# shellcheck source=/etc/hestiacp/hestia.conf
|
||
|
|
source /etc/hestiacp/hestia.conf
|
||
|
|
# shellcheck source=/usr/local/hestia/func/main.sh
|
||
|
|
source $HESTIA/func/main.sh
|
||
|
|
# load config file
|
||
|
|
source_conf "$HESTIA/conf/hestia.conf"
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Verifications #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
check_args '0' "$#" 'USER'
|
||
|
|
if [ -n "$user" ]; then
|
||
|
|
is_format_valid 'user'
|
||
|
|
is_object_valid 'user' 'USER' "$user"
|
||
|
|
fi
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Action #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
# Creating user_list
|
||
|
|
if [ -z "$user" ]; then
|
||
|
|
user_list=$($BIN/v-list-sys-users plain)
|
||
|
|
else
|
||
|
|
user_list="$user"
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Updating user stats
|
||
|
|
for user in $user_list; do
|
||
|
|
USER_DATA=$HESTIA/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
|
||
|
|
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'" $HESTIA/data/users/*/user.conf | wc -l)
|
||
|
|
SUSPENDED_USERS=$spnd
|
||
|
|
U_USERS=$($BIN/v-list-sys-users plain | wc -l)
|
||
|
|
fi
|
||
|
|
|
||
|
|
ips=$(ls $HESTIA/data/ips)
|
||
|
|
# Checking ip
|
||
|
|
for ip in $ips; do
|
||
|
|
source $HESTIA/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 $HESTIA/data/ips | wc -l)
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Checking web system
|
||
|
|
U_WEB_DOMAINS=0
|
||
|
|
if [ -f $USER_DATA/web.conf ]; then
|
||
|
|
for domain_str in $(cat $USER_DATA/web.conf); do
|
||
|
|
parse_object_kv_list "$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 [ -n "$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))
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Checking dns system
|
||
|
|
if [ -f $USER_DATA/dns.conf ]; then
|
||
|
|
for domain_str in $(cat $USER_DATA/dns.conf); do
|
||
|
|
parse_object_kv_list "$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
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Checking mail system
|
||
|
|
if [ -f $USER_DATA/mail.conf ]; then
|
||
|
|
for domain_str in $(cat $USER_DATA/mail.conf); do
|
||
|
|
parse_object_kv_list "$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))
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Checking db system
|
||
|
|
if [ -f $USER_DATA/db.conf ]; then
|
||
|
|
for db_str in $(cat $USER_DATA/db.conf); do
|
||
|
|
parse_object_kv_list "$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))
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Checking cron system
|
||
|
|
if [ -f $USER_DATA/cron.conf ]; then
|
||
|
|
for cron_str in $(cat $USER_DATA/cron.conf); do
|
||
|
|
parse_object_kv_list "$cron_str"
|
||
|
|
U_CRON_JOBS=$((U_CRON_JOBS + 1))
|
||
|
|
if [ "$SUSPENDED" = 'yes' ]; then
|
||
|
|
SUSPENDED_CRON=$((SUSPENDED_CRON + 1))
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Checking backup
|
||
|
|
if [ -f $USER_DATA/backup.conf ]; then
|
||
|
|
for backup_str in $(cat $USER_DATA/backup.conf); do
|
||
|
|
U_BACKUPS=$((U_BACKUPS + 1))
|
||
|
|
done
|
||
|
|
fi
|
||
|
|
|
||
|
|
# Update values
|
||
|
|
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
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Hestia #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
# Logging
|
||
|
|
log_event "$OK" "$ARGUMENTS"
|
||
|
|
|
||
|
|
exit
|