wordpress-tools/local/bin/wp-update-config-db-info.sh
2023-09-27 15:05:45 -07:00

262 lines
8.3 KiB
Bash
Executable File

#!/bin/bash
#
# wordpress-tools
# https://git.stack-source.com/msb/wordpress-tools
# Copyright (c) 2023 Matthew Saunders Brown <matthewsaundersbrown@gmail.com>\
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# require root
if [ "${EUID}" -ne 0 ]; then
echo "This script must be run as root"
exit
fi
virtualhost_basedir="/srv/www"
virtualhost_htdocsdir="html"
declare -a input_keys_index
input_keys_index=(s d u p x)
declare -A input_keys_array
input_keys_array[s]="MySQL Hostname (Server)"
input_keys_array[d]="MySQL Database"
input_keys_array[u]="MySQL Username"
input_keys_array[p]="MySQL Password"
input_keys_array[x]="MySQL Table Prefix"
declare -A input_values_array
help()
{
thisfilename=$(basename -- "$0")
echo "$thisfilename - Update wp-config.php DB info."
echo ""
echo "Usage: $thisfilename [-v <virtualhost>] [-s <hostname>] [-d <database>] [-u <username>] [-p <password>] [-x <table_prefix>]"
echo ""
echo " -v <virtualhost> VirtualHost of WordPress config that will be updated. If not set attempt to autodect based on current/working dir."
echo " -s <hostname> MySQL Hostname, defaults to 127.0.0.1"
echo " -d <database> MySQL Database, defaults to 'VirtualHost' with dots & dashes changed from symbols to words if not specified."
echo " -u <username> MySQL Username, defaults to owner of html dir if not specified."
echo " -p <password MySQL Password, random password is generated if not set."
echo " -x <table_prefix> WordPress db table prefix, defaults to wp_"
echo " -f Force update - skips confirmation step, for automation."
echo " -h Print this help."
echo ""
echo " You will be prompted to any options that were not"
echo " specified on the command line and given a chance"
echo " to confirm before performing the install unless the"
echo " -f option is specified."
exit
}
# force (-f) defaults to false
force=false
# set any options that were passed
while getopts "v:s:d:u:p:x:fh" opt; do
case "${opt}" in
h )
help
exit;;
v )
input_values_array[v]=${OPTARG}
;;
s )
input_values_array[s]=${OPTARG}
;;
d )
input_values_array[d]=${OPTARG}
;;
u )
input_values_array[u]=${OPTARG}
;;
p )
input_values_array[p]=${OPTARG}
;;
x )
input_values_array[x]=${OPTARG}
;;
f )
force=true
;;
\? )
echo "Invalid option: $OPTARG" 1>&2
exit 1;;
: )
echo "Invalid option: $OPTARG requires an argument" 1>&2
exit 1;;
esac
done
# v - virtualhost
function validate_v() {
if [[ -d "$virtualhost_basedir/${input_values_array[v]}" ]]; then
if [[ ! -d "$virtualhost_basedir/${input_values_array[v]}/$virtualhost_htdocsdir" ]]; then
echo "${input_values_array[v]} is installed, but DocumentRoot for ${input_values_array[v]} does not exist."
echo "$virtualhost_basedir/${input_values_array[v]}/$virtualhost_htdocsdir"
exit
fi
else
echo "VirtualHost ${input_values_array[v]} is not installed on this server."
echo "$virtualhost_basedir/${input_values_array[v]}/$virtualhost_htdocsdir"
unset input_values_array[v]
fi
}
if [[ -n "${input_values_array[v]}" ]] ; then
validate_v
else
cwd=`pwd`
echo "cwd = $cwd"
if echo $cwd |grep -q ^/srv/www/ ; then
input_values_array[v]=`echo $cwd |cut -d '/' -f 4`
validate_v
else
while [ -z "${input_values_array[v]}" ]; do
read -p "Enter the VirtualHost of WordPress config " virtualhost
if [ -n "$virtualhost" ] ; then
input_values_array[v]=$virtualhost
validate_v
fi
done
fi
fi
# s - Shell Username
# get shell user value, user can't override
shell_username=$(stat -c '%U' $virtualhost_basedir/${input_values_array[v]})
# m - MySQL Hostname
if [[ -z "${input_values_array[s]}" ]] ; then
if [[ -f "$virtualhost_basedir/${input_values_array[v]}/.my.cnf" ]]; then
input_values_array[s]=`grep host /srv/www/${input_values_array[v]}/.my.cnf |cut -d = -f 2`
else
input_values_array[s]="127.0.0.1"
fi
fi
# d - MySQL Database
if [[ -z "${input_values_array[d]}" ]] ; then
if [[ -f "$virtualhost_basedir/${input_values_array[v]}/.my.cnf" ]]; then
input_values_array[d]=`grep database /srv/www/${input_values_array[v]}/.my.cnf |cut -d = -f 2`
else
mysql_database=${input_values_array[v]}
mysql_database=${mysql_database//./dot}
mysql_database=${mysql_database//-/dash}
input_values_array[d]=$mysql_database
fi
fi
# u - MySQL Username
if [[ -z "${input_values_array[u]}" ]] ; then
if [[ -f "$virtualhost_basedir/${input_values_array[v]}/.my.cnf" ]]; then
input_values_array[u]=`grep user /srv/www/${input_values_array[v]}/.my.cnf |cut -d = -f 2`
else
input_values_array[u]="$shell_username@${input_values_array[v]}"
fi
fi
# p - MySQL Password
if [[ -z "${input_values_array[p]}" ]] ; then
if [[ -f "$virtualhost_basedir/${input_values_array[v]}/.my.cnf" ]]; then
input_values_array[p]=`grep password /srv/www/${input_values_array[v]}/.my.cnf |cut -d = -f 2`
fi
fi
# p - MySQL Password
function validate_p() {
mysql_username_count=`mysql -s -N -e "SELECT COUNT(*) FROM mysql.user WHERE User='${input_values_array[n]}'"`
if [[ "$mysql_username_count" -gt 0 ]]; then
mysql_password_verification=`mysql -s -A -N -e "SELECT COUNT(1) Password_is_OK FROM mysql.user WHERE user='${input_values_array[n]}' AND password=PASSWORD('${input_values_array[w]}')"`
if [[ "$mysql_password_verification" -ne 1 ]]; then
read -p "ERROR: Invalid MySQL Password for existing MySQL User ${input_values_array[n]}, please enter current password: " mysql_password
if [ -n "$mysql_password" ] ; then
input_values_array[p]=$mysql_password
restart_script
fi
fi
fi
}
# x - MySQL DB Prefix
function validate_x() {
if mysql -e "USE ${input_values_array[d]}" 2> /dev/null; then
mysql_db_prefix=`echo ${input_values_array[x]} |sed 's|_|\\\_|g'`
mysql_db_prefix_count=`mysql -s -N -e "SELECT COUNT(*) FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = '${input_values_array[d]}' AND TABLE_NAME LIKE '$mysql_db_prefix%'"`
if [[ "$mysql_db_prefix_count" -gt 0 ]]; then
read -p "One or more MySQL tables with DB Prefix '${input_values_array[x]}' already exist, please enter new prefix: " mysql_db_prefix
if [ -n "$mysql_db_prefix" ] ; then
input_values_array[x]=$mysql_db_prefix
restart_script
fi
fi
fi
}
if [ -z "${input_values_array[x]}" ] ; then
input_values_array[x]="wp_"
fi
# display confirmation
function display_confirmation() {
echo
for key in "${!input_keys_index[@]}" ; do
value=${input_keys_index[$key]}
echo "$value - ${input_keys_array[$value]}: ${input_values_array[$value]}"
done
echo
echo "i - install now with these settings"
echo "q - quit without installing"
echo
read -p "Select option to edit or install/quit: " option
echo
}
function restart_script() {
# set command line args
cli_args=
for key in "${!input_values_array[@]}" ; do
value=${input_values_array[$key]}
cli_args="$cli_args -$key '$value'"
done
# echo "${0} $cli_args"
echo
eval ${0} $cli_args
exit
}
if [[ "$force" == "false" ]]; then
display_confirmation
else
option="i"
fi
if [[ "$option" == "i" ]]; then
# validate MySQL DB Prefix
validate_x
# validate MySQL User Password
validate_p
echo
echo "Updating WP config db info for VirtualHost ${input_values_array[v]} now..."
# switch to Shell User and update config
# sudo -u ${input_values_array[s]} --shell /bin/bash << EOF
echo cd $virtualhost_basedir/${input_values_array[v]}/$virtualhost_htdocsdir
echo wp config set DB_HOST ${input_values_array[s]}
echo wp config set DB_NAME ${input_values_array[d]}
echo wp config set DB_USER ${input_values_array[u]}
echo wp config set DB_PASSWORD ${input_values_array[p]}
# EOF
elif [[ "$option" == "q" ]]; then
echo "Quiting without updating config."
elif [[ " ${input_keys_index[@]} " =~ " $option " ]]; then
read -p "Enter new value for ${input_keys_array[$option]} (${input_values_array[$option]}): " new_value
if [ -n "$new_value" ] ; then
input_values_array[$option]=$new_value
fi
restart_script
else
echo "ERROR: Invalid entry, try again."
fi