262 lines
8.3 KiB
Bash
262 lines
8.3 KiB
Bash
|
#!/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
|