VestaCP/bin/v_update_database_disk

#!/bin/bash
# info: update database disk usage
# options: user database
#
# The function recalculates disk usage for speciefic database.


#----------------------------------------------------------#
#                    Variable&Function                     #
#----------------------------------------------------------#

# Argument defenition
user=$1
database=$2

# Includes
source $VESTA/conf/vesta.conf
source $VESTA/func/main.sh
source $VESTA/func/db.sh


#----------------------------------------------------------#
#                    Verifications                         #
#----------------------------------------------------------#

check_args '2' "$#" 'user database'
validate_format 'user' 'database'
is_system_enabled "$DB_SYSTEM"
is_object_valid 'user' 'USER' "$user"
is_object_unsuspended 'user' 'USER' "$user"
is_object_valid 'db' 'DB' "$database"
is_object_unsuspended 'db' 'DB' "$database"


#----------------------------------------------------------#
#                       Action                             #
#----------------------------------------------------------#

# Get database values
get_database_values

# Switching on db type
case $TYPE in
    mysql) get_mysql_disk_usage ;;
    pgsql) get_pgsql_disk_usage ;;
esac


#----------------------------------------------------------#
#                       Vesta                              #
#----------------------------------------------------------#

# Updating disk usage
update_object_value 'db' 'DB' "$database" '$U_DISK' "$usage"
recalc_user_disk_usage

# Logging
log_event "$OK" "$EVENT"

exit