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.
80 lines
2.2 KiB
80 lines
2.2 KiB
2 years ago
|
#!/bin/bash
|
||
|
|
# info: update domains bandwidth usage
|
||
|
|
# options: USER
|
||
|
|
#
|
||
|
|
# example: v-update-web-domains-traff bob
|
||
|
|
#
|
||
|
|
# This function recalculates bandwidth usage for all user webdomains.
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# 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
|
||
|
|
# shellcheck source=/usr/local/hestia/func/domain.sh
|
||
|
|
source $HESTIA/func/domain.sh
|
||
|
|
# load config file
|
||
|
|
source_conf "$HESTIA/conf/hestia.conf"
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Verifications #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
check_args '1' "$#" 'USER'
|
||
|
|
is_format_valid 'user'
|
||
|
|
is_system_enabled "$WEB_SYSTEM" 'WEB_SYSTEM'
|
||
|
|
is_object_valid 'user' 'USER' "$user" "$user"
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Action #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
for domain in $(search_objects 'web' 'SUSPENDED' "no" 'DOMAIN'); do
|
||
|
|
# Reset BW counter on the start of the month
|
||
|
|
if [ "$(date +%d)" = '01' ]; then
|
||
|
|
update_object_value 'web' 'DOMAIN' "$domain" '$U_BANDWIDTH' '0'
|
||
|
|
fi
|
||
|
|
|
||
|
|
log_file="/var/log/$WEB_SYSTEM/domains/$domain.bytes"
|
||
|
|
bytes=0
|
||
|
|
|
||
|
|
# Parsing log
|
||
|
|
while read line; do
|
||
|
|
if [[ "$line" =~ ^[0-9]+$ ]]; then
|
||
|
|
line=${line#0}
|
||
|
|
if [ -n "$line" ]; then
|
||
|
|
bytes=$(($bytes + $line))
|
||
|
|
fi
|
||
|
|
fi
|
||
|
|
done < $log_file
|
||
|
|
|
||
|
|
# Converting to Mb
|
||
|
|
mb=$(echo "$bytes / 1024 / 1024" | bc)
|
||
|
|
|
||
|
|
# Nulling log
|
||
|
|
echo > $log_file
|
||
|
|
|
||
|
|
get_domain_values 'web'
|
||
|
|
bandwidth=$((U_BANDWIDTH + mb))
|
||
|
|
|
||
|
|
# Updating bandwidth value in config
|
||
|
|
update_object_value 'web' 'DOMAIN' "$domain" '$U_BANDWIDTH' "$bandwidth"
|
||
|
|
|
||
|
|
done
|
||
|
|
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
# Hestia #
|
||
|
|
#----------------------------------------------------------#
|
||
|
|
|
||
|
|
# Recalculating user bandwidth
|
||
|
|
recalc_user_bandwidth_usage
|
||
|
|
|
||
|
|
exit
|