#!/bin/bash

# The MIT License (MIT)
#
# Copyright (c) 2014-2018 Greg Messner, greg@messners.com
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

#
# This is the service utility script for running MySQL in a docker container.  It is intended
# to be called from upstart or systemd service scripts, but can also be used from the command line
# to start, stop, and check the status of a running MySQL container.
#

# Load the docker utility functions
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
source $SCRIPT_DIR/docker-functions.sh

#
# Print the help information for the MySQL service container.
#
function help {

	echo
	echo "  This script performs one or more actions on the MySQL service container."
	echo
	echo "  Usage : $(basename $0) action ..."
	echo
	echo "  Available actions"
	echo "  ------------------------------------------------------------------------"
	echo "  attach  Attaches the current process to the running MySQL service,"
	echo "          usually only used by actual OS services"
	echo "  cleanup Cleans up (removes) the MySQL service container"
	echo "  destroy Stops and removes the MySQL service container, and deletes"
	echo "          the associated database data (docker volume)"
	echo "  help    Prints this help information and exits"
	echo "  start   Starts the MySQL service container. When starting the service"
	echo "          the following environment variables must be set:"
	echo "          MYSQL_DATABASE, MYSQL_USER, and MYSQL_PASSWORD"
	echo "  status  Returns a one word status for the MySQL service"
	echo "  stop    Stops the MySQL service container"
	echo
	echo "  NOTE: Multiple actions may be combined on a single command line, for example:"
	echo "          $(basename $0) start attach"
	echo
}


#
# Stops the service container if it is running and deletes the associated docker volume.
#
destroy() {

	# Make sure the container is stopped and removed
	df_kill_container ${MYSQL_CONTAINER}

	# Remove the docker volume if it is present
	if docker volume inspect ${MYSQL_DATA_VOLUME} >/dev/null 2>&1 ; then
		echo -n "Removing ${MYSQL_DATA_VOLUME} data volume..."
		docker volume rm --force ${MYSQL_DATA_VOLUME} >/dev/null 2>&1
		echo "done"
	fi
}


#
# Launches the container and waits for it to be fully started.  After calling 'start', you can
# call the 'attach' method to attach to the log output of the container.
#
start() {

	# Make sure the database name, username, and password is set, if not, abort
	ENV_ERRORS="false"
	if [ -z "$MYSQL_DATABASE" ]; then
		echo "MYSQL_DATABASE env variable is not set!" 1>&2
		ENV_ERRORS="true"
	fi

	if [ -z "$MYSQL_USER" ]; then
		echo "MYSQL_USER is env variable not set!" 1>&2
		ENV_ERRORS="true"
	fi

	if [ -z "$MYSQL_PASSWORD" ]; then
		echo "MYSQL_PASSWORD env variable is not set!" 1>&2
		ENV_ERRORS="true"
	fi

	if [ "$ENV_ERRORS" == "true" ]; then
		exit 1
	fi

	# If the data volume does not exist, create it
	if ! docker volume inspect ${MYSQL_DATA_VOLUME} >/dev/null 2>&1 ; then
		echo "Creating ${MYSQL_DATA_VOLUME} data volume"
		docker volume create ${MYSQL_DATA_VOLUME} >/dev/null 2>&1
	fi

	# Make sure that docker does not have the service container present
	df_kill_container ${MYSQL_CONTAINER}

	# Run the docker container
	CONTAINER_ID=$(docker run --name "${MYSQL_CONTAINER}" \
		--detach --restart=no \
		--health-cmd="mysqladmin ping --silent" \
		-p ${MYSQL_PORT}:3306 \
		-e MYSQL_RANDOM_ROOT_PASSWORD=yes \
		-e MYSQL_USER=${MYSQL_USER} \
		-e MYSQL_PASSWORD="${MYSQL_PASSWORD}" \
	        -e MYSQL_DATABASE=${MYSQL_DATABASE} \
		-v ${MYSQL_DATA_VOLUME}:/var/lib/mysql \
		${MYSQL_IMAGE_NAME}:${MYSQL_IMAGE_VERSION})

	if [ $? -ne 0 ]; then
		echo "Error starting ${MYSQL_CONTAINER}"
		exit 1
	fi

	# Wait a maximum of 60 seconds for the container to be fully started and healthy
	echo "Started ${MYSQL_CONTAINER}, container ID=${CONTAINER_ID}"
	echo "Waiting for ${MYSQL_CONTAINER} to become healthy"
	df_wait_for_container 60 ${MYSQL_CONTAINER}
}


#############################################################################

# Need at least a single argumet (action), if none are provided putput usage info and exit with error.
if [ $# -lt 1 ]; then
	echo "Usage : $(basename $0) attach|cleanup|destroy|help|status|start|stop"
	exit 1
fi

if [ "$1" == "help" ]; then
	help
	exit 0
fi

# Default the env variables if they are not set.
MYSQL_IMAGE_NAME=${MYSQL_IMAGE_NAME:-mysql}
MYSQL_IMAGE_VERSION=${MYSQL_IMAGE_VERSION:-"5.7.22"}
MYSQL_CONTAINER=${MYSQL_CONTAINER:-mysql-db}
MYSQL_DATA_VOLUME=${MYSQL_DATA_VOLUME:-mysql-db-data}
MYSQL_PORT=${MYSQL_PORT:-3306}

#
# Loop thru the arguments (actions). Some actions may be paired together, for example:
#
#	start attach  -  Will start the container and attach the current process to it
#	stop cleanup  -  Will stop the container and then clean it up (remove it)
#
for ARG in "$@"
do
	case "$ARG" in

		"attach")
			df_attach_container $MYSQL_CONTAINER
			;;

		"cleanup")
			df_cleanup_container $MYSQL_CONTAINER
			;;

		"destroy")
			destroy
			;;

		"start")
			start
			;;

		"status")
			df_container_status $MYSQL_CONTAINER
			;;

		"stop")
			df_stop_container $MYSQL_CONTAINER
			;;

		*)
			echo "'$1' is not a valid $(basename $0 '.sh') command, aborting!"
			exit 1
			;;
	esac

	RESULTS=$?
	if [ $RESULTS -ne 0 ]; then
		exit $RESULTS
	fi
done

# vim: syntax=sh ts=4 sw=4 sts=4 sr noet