#!/sbin/openrc-run
# Copyright 1999-2024 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2

depend() {
	use net.lo
	# localmount needed for $basedir
	need localmount s6-svscan
}

get_config() {
	my_print_defaults --defaults-file="$1" mysqld server mariadb |
	sed -n -e "s/^--$2=//p"
}

mysql_svcname() {
	local ebextra=
	case "${svc_name}" in
		mysql*) ;;
		*) ebextra=" (mysql)" ;;
	esac
	echo "${svc_name}${ebextra}"
}

stringContain() { [ -z "${2##*$1*}" ] && [ -z "$1" -o -n "$2" ]; }

bootstrap_galera() {
	MY_ARGS="--wsrep-new-cluster ${MY_ARGS}"
	mark_service_starting
	if start_pre && start ; then
		mark_service_started
		return 0
	else
		mark_service_stopped
		return 1
	fi
}


extra_commands="checkconfig"
extra_stopped_commands="bootstrap_galera"

description_checkconfig="Verify the server's configuration"
description_bootstrap_galera="Start a new Galera cluster with this server as the initial node"
supervisor=s6
name=$(mysql_svcname)
s6_service_timeout_stop="$((1000*${STOP_TIMEOUT:-120}))"
#s6_svwait_options_start="-U -t $((1000*${STARTUP_EARLY_TIMEOUT:-1000}))"
svc_name=${RC_SVCNAME%-s6}
s6_service_path=/var/svc.d/${svc_name}

start_pre() {
	# Check the config or die
	if [ ${RC_CMD} != "restart" ] ; then
		checkconfig || return 1
	fi

	MY_CNF="${MY_CNF:-/etc/${svc_name}/my.cnf}"

	if [ ! -r "${MY_CNF}" ] ; then
		eerror "Cannot read the configuration file \`${MY_CNF}'"
		return 1
	fi

	# tail -n1 is critical as these we only want the last instance of the option
	local basedir=$(get_config "${MY_CNF}" basedir | tail -n1)
	local pidfile=$(get_config "${MY_CNF}" 'pid[_-]file' | tail -n1)
	local socket=$(get_config "${MY_CNF}" socket | tail -n1)
	local chroot=$(get_config "${MY_CNF}" chroot | tail -n1)
	local wsrep="$(get_config "${MY_CNF}" 'wsrep[_-]on' | tail -n1 | awk '{print tolower($0)}')"
	local wsrep_new=$(get_config "${MY_CNF}" 'wsrep-new-cluster' | tail -n1)

	if [ -n "${chroot}" ] ; then
		socket="${chroot}/${socket}"
		pidfile="${chroot}/${pidfile}"
	fi

	# Galera: Only check datadir if not starting a new cluster and galera is enabled
	# wsrep_on is not on or wsrep-new-cluster exists in the config or MY_ARGS
	[ "${wsrep}" = "1" ] && wsrep="on"
	if [ "${wsrep}" != "on" ] || [ -n "${wsrep_new}" ] || stringContain 'wsrep-new-cluster' "${MY_ARGS}" ; then

		local datadir=$(get_config "${MY_CNF}" datadir | tail -n1)
		if [ ! -d "${datadir}" ] ; then
			eerror "MySQL datadir \`${datadir}' is empty or invalid"
			eerror "Please check your config file \`${MY_CNF}'"
			return 1
		fi

		if [ ! -d "${datadir}"/mysql ] ; then
			# find which package is installed to report an error
			local EROOT=$(portageq envvar EROOT)
			local DBPKG_P=$(portageq match ${EROOT} $(portageq expand_virtual ${EROOT} virtual/mysql | head -n1))
			if [ -z ${DBPKG_P} ] ; then
				eerror "You don't appear to have a server package installed yet."
			else
				eerror "You don't appear to have the mysql database installed yet."
				eerror "Please run \`emerge --config =${DBPKG_P}\` to have this done..."
			fi
			return 1
		fi
	fi

	local piddir="${pidfile%/*}"
	checkpath -d --owner mysql:mysql --mode 0755 "$piddir"
	rc=$?
	if [ $rc -ne 0 ]; then
		eerror "Directory $piddir for pidfile does not exist and cannot be created"
		return 1
	fi

	# Prepare env files to source
	mkdir -p "/var/svc.d/${svc_name}"
	echo "MY_CNF=\"${MY_CNF}\"" > "/var/svc.d/${svc_name}/env"
	echo "MY_ARGS=\"${MY_ARGS}\"" >> "/var/svc.d/${svc_name}/env"
	echo "basedir=\"${basedir}\"" >> "/var/svc.d/${svc_name}/env"
	mkdir -p "/var/svc.d/${svc_name}/log"
	echo "SVCNAME=\"${svc_name}\"" > "/var/svc.d/${svc_name}/log/env"
	echo "S6_LOG_OPTIONS=\"${S6_LOG_OPTIONS}\"" > "/var/svc.d/${svc_name}/log/env"
}

start_post() {
	local socket=$(get_config "${MY_CNF}" socket | tail -n1)
	local chroot=$(get_config "${MY_CNF}" chroot | tail -n1)
	local startup_timeout=${STARTUP_TIMEOUT:-900}

	if [ -n "${chroot}" ] ; then
		socket="${chroot}/${socket}"
	fi
	ewaitfile ${startup_timeout} "${socket}"
}

checkconfig() {
	local my_cnf="${MY_CNF:-/etc/${svc_name}/my.cnf}"
	local basedir=$(get_config "${my_cnf}" basedir | tail -n1)
	local svc_name=$(mysql_svcname)
	ebegin "Checking mysqld configuration for ${svc_name}"

	if [ ${RC_CMD} = "checkconfig" ] ; then
		# We are calling checkconfig specifically.  Print warnings regardless.
		"${basedir}"/sbin/mysqld --defaults-file="${my_cnf}" --help --verbose > /dev/null
	else
		# Suppress output to check the return value
		"${basedir}"/sbin/mysqld --defaults-file="${my_cnf}" --help --verbose > /dev/null 2>&1

		# If the above command does not return 0,
		# then there is an error to echo to the user
		if [ $? -ne 0 ] ; then
			"${basedir}"/sbin/mysqld --defaults-file="${my_cnf}" --help --verbose > /dev/null
		fi
	fi

	eend $? "${svc_name} config check failed"
}

stop_pre() {
	if [ ${RC_CMD} = "restart" ] ; then
		checkconfig || return 1
	fi
}

# vim: filetype=gentoo-init-d sw=2 ts=2 sts=2 noet: