Marek Miklewicz
127 lines
3.2 KiB
Bash
127 lines
3.2 KiB
Bash
#!/bin/bash
|
|
set -euo pipefail
|
|
|
|
PLUGIN_DIR="$(cd "$(dirname "$0")/../.." && pwd)"
|
|
CUSTOMBUILD_ROOT="${CUSTOMBUILD_ROOT:-/usr/local/directadmin/custombuild}"
|
|
ACTION="${1:-install}"
|
|
SETTINGS_FILE="${MYSQL_PLUGIN_SETTINGS_FILE:-$PLUGIN_DIR/plugin-settings.conf}"
|
|
MARKER="managed by DirectAdmin MySQL plugin CustomBuild hook"
|
|
|
|
log() {
|
|
printf 'mysql custombuild-hooks: %s\n' "$*"
|
|
}
|
|
|
|
die() {
|
|
printf 'mysql custombuild-hooks: %s\n' "$*" >&2
|
|
exit 1
|
|
}
|
|
|
|
read_setting() {
|
|
local key="$1"
|
|
local default="$2"
|
|
local line value
|
|
|
|
[ -r "$SETTINGS_FILE" ] || {
|
|
printf '%s\n' "$default"
|
|
return 0
|
|
}
|
|
|
|
line="$(grep -E "^${key}=" "$SETTINGS_FILE" | tail -n 1 || true)"
|
|
[ -n "$line" ] || {
|
|
printf '%s\n' "$default"
|
|
return 0
|
|
}
|
|
|
|
value="${line#*=}"
|
|
value="${value%%#*}"
|
|
value="${value%\"}"
|
|
value="${value#\"}"
|
|
value="${value%\'}"
|
|
value="${value#\'}"
|
|
value="$(printf '%s' "$value" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')"
|
|
printf '%s\n' "${value:-$default}"
|
|
}
|
|
|
|
custombuild_hooks_enabled() {
|
|
case "$(read_setting CUSTOMBUILD_HOOKS_ENABLED true)" in
|
|
true|yes|on|1) return 0 ;;
|
|
*) return 1 ;;
|
|
esac
|
|
}
|
|
|
|
write_hook() {
|
|
local target="$1"
|
|
local app="$2"
|
|
local repair_script="$3"
|
|
local health_script="$4"
|
|
local repair_name health_name tmp
|
|
|
|
[ -f "$repair_script" ] || die "missing repair script: $repair_script"
|
|
[ -f "$health_script" ] || die "missing health check script: $health_script"
|
|
repair_name="$(basename "$repair_script")"
|
|
health_name="$(basename "$health_script")"
|
|
|
|
mkdir -p "$(dirname "$target")"
|
|
tmp="$(mktemp)"
|
|
cat > "$tmp" <<EOF
|
|
#!/bin/bash
|
|
# $MARKER
|
|
set -euo pipefail
|
|
|
|
PLUGIN_DIR='$PLUGIN_DIR'
|
|
LOG_DIR="\$PLUGIN_DIR/data/logs"
|
|
LOG_FILE="\$LOG_DIR/custombuild-${app}-repair.log"
|
|
|
|
mkdir -p "\$LOG_DIR" 2>/dev/null || true
|
|
|
|
{
|
|
printf '%s\\n' "=== \$(date '+%Y-%m-%d %H:%M:%S') ${app} CustomBuild post hook ==="
|
|
"\$PLUGIN_DIR/scripts/setup/${repair_name}"
|
|
if ! "\$PLUGIN_DIR/scripts/setup/${health_name}"; then
|
|
printf '%s\\n' "${app}: warning: health check failed after repair"
|
|
fi
|
|
} >> "\$LOG_FILE" 2>&1
|
|
EOF
|
|
install -m 0755 "$tmp" "$target"
|
|
rm -f "$tmp"
|
|
log "installed CustomBuild post hook: $target"
|
|
}
|
|
|
|
install_all() {
|
|
custombuild_hooks_enabled || {
|
|
log "disabled by CUSTOMBUILD_HOOKS_ENABLED"
|
|
return 0
|
|
}
|
|
|
|
write_hook \
|
|
"$CUSTOMBUILD_ROOT/custom/hooks/roundcube/post/90-alt-mysql-repair.sh" \
|
|
"roundcube" \
|
|
"$PLUGIN_DIR/scripts/setup/roundcube_repair_config.sh" \
|
|
"$PLUGIN_DIR/scripts/setup/roundcube_health_check.sh"
|
|
|
|
write_hook \
|
|
"$CUSTOMBUILD_ROOT/custom/hooks/phpmyadmin/post/90-alt-mysql-repair.sh" \
|
|
"phpmyadmin" \
|
|
"$PLUGIN_DIR/scripts/setup/phpmyadmin_repair.sh" \
|
|
"$PLUGIN_DIR/scripts/setup/phpmyadmin_health_check.sh"
|
|
}
|
|
|
|
uninstall_all() {
|
|
local hook
|
|
for hook in \
|
|
"$CUSTOMBUILD_ROOT/custom/hooks/roundcube/post/90-alt-mysql-repair.sh" \
|
|
"$CUSTOMBUILD_ROOT/custom/hooks/phpmyadmin/post/90-alt-mysql-repair.sh"
|
|
do
|
|
if [ -f "$hook" ] && grep -Fq "$MARKER" "$hook"; then
|
|
rm -f "$hook"
|
|
log "removed CustomBuild post hook: $hook"
|
|
fi
|
|
done
|
|
}
|
|
|
|
case "$ACTION" in
|
|
install) install_all ;;
|
|
uninstall) uninstall_all ;;
|
|
*) die "unknown action: $ACTION" ;;
|
|
esac
|