db-sample-data

#!/usr/bin/env bash

#---------------------------------------------------------------------
usage ()
{
	cat <<EOT

${0##*/}
    Dumps frequently updated data from production to a web-accessible
    file for downloading and importing into a local development
    database. Prompts with the download URL and waits for a response
    from the user to delete the file. Pass the names of the DB tables
    you wish to download as arguments. Without arguments, will dump
    the entire \$default database.

Usage:
    bin/${0##*/} [table] [table2] [table3]

Environment:
    DB_SAMPLE_DATA_URL - If this environment variable is set, it will
                         be used when displaying the download URL for
                         the temporary zip file.

EOT

	exit ${1:-0}  # Exit with code 0 unless an arg is passed to the method.
}
if [ "$1" = '-h' ]; then
	usage
fi

if [ -z "$1" ]; then
	TABLES=''
else
	TABLES="--tables "$*""
fi


# Configuration vars. Set these appropriately for your app and environment.
PUBLIC_URL=${DB_SAMPLE_DATA_URL:-}
WEBROOT_PATH="webroot/"


# Internal var setup.
eval $( bin/db-credentials ) # Yuck, but nothing else works!
OPTIONS="--skip-add-drop-table --no-create-info"
TMP_PATH="tmp"
DESTINATION_NAME="${DB_DATABASE}_sample_data_$(date +%Y-%m-%d)"


# Generate the dump command.
if [ -n "$DB_PORT" ]; then
	PORT_CLAUSE="--port=$DB_PORT"
fi
CMD="mysqldump --host=${DB_HOST} ${PORT_CLAUSE} --user=${DB_USERNAME} ${OPTIONS} ${DB_DATABASE} $TABLES"


# Run the dump command.
# Handle all the different ways to provide (or not provide) a password.
pushd "${TMP_PATH}" >/dev/null
if [[ $DB_PASSWORD =~ " |'" ]]; then
	$CMD -p"${DB_PASSWORD}" > "${DESTINATION_NAME}.sql"
elif [ -n "${DB_PASSWORD}" ]; then
	$CMD -p$DB_PASSWORD > "${DESTINATION_NAME}.sql"
else
	$CMD > "${DESTINATION_NAME}.sql"
fi
popd >/dev/null


# Zip the file and remove the original .sql.
zip -rq9 "${WEBROOT_PATH}${DESTINATION_NAME}.zip" "${TMP_PATH}/${DESTINATION_NAME}.sql"
rm -f "${TMP_PATH}/${DESTINATION_NAME}.sql"


# Prompt to download the file, then delete it.
echo "## Download URL: ${PUBLIC_URL}/${DESTINATION_NAME}.zip"
read -p "?? Delete ${PUBLIC_URL}/${DESTINATION_NAME}.zip? [Y/n]: " ARG_DELETEFILE
case $ARG_DELETEFILE in
	[Yy]*|*)
		rm -f "${WEBROOT_PATH}${DESTINATION_NAME}.zip"
		echo "## File deleted."
		;;
	[Nn]*)
		echo "## File REMAINS! Please clean it up when you are done to prevent data leaks."
		;;
esac