#!/usr/bin/env bash

if [ $# -lt 3 ]; then
	echo "usage: $0 <db-name> <db-user> <db-pass> [db-host] [wp-version]"
	exit 1
fi

DB_NAME=$1
DB_USER=$2
DB_PASS=$3
DB_HOST=${4-localhost}
WP_VERSION=${5}
if [[ ${5} == "latest" ]]; then
	WP_TEST_VERSION="trunk"
else
	WP_TEST_VERSION="tags/${5}"
fi

SCRIPTPATH=`pwd -P`

WP_TESTS_DIR="${SCRIPTPATH}/wordpress-tests-lib"
WP_CORE_DIR="${SCRIPTPATH}/wordpress/"

set -ex

install_wp() {
	rm -Rf "$WP_CORE_DIR"
	mkdir -p "$WP_CORE_DIR"
	if [[ $WP_TEST_VERSION  == "trunk" ]]; then
		wget -nv -O /tmp/wordpress.zip https://wordpress.org/nightly-builds/wordpress-latest.zip
		unzip -d . /tmp/wordpress.zip
	elif [[ $WP_VERSION  == "" ]]; then
		wget -nv -O /tmp/wordpress.tar.gz https://wordpress.org/latest.tar.gz
		tar --strip-components=1 -zxmf /tmp/wordpress.tar.gz -C "$WP_CORE_DIR"
		WP_TEST_VERSION=$(cd "$WP_CORE_DIR"; wp core version --allow-root)
	else
		ARCHIVE_NAME="wordpress-$WP_VERSION"
		wget -nv -O /tmp/wordpress.tar.gz https://wordpress.org/${ARCHIVE_NAME}.tar.gz
		tar --strip-components=1 -zxmf /tmp/wordpress.tar.gz -C "$WP_CORE_DIR"
	fi

	wget -nv -O "${WP_CORE_DIR}/wp-content/db.php" https://raw.github.com/markoheijnen/wp-mysqli/master/db.php
}

install_test_suite() {
	# portable in-place argument for both GNU sed and Mac OSX sed
	if [[ $(uname -s) == 'Darwin' ]]; then
		local ioption='-i .bak'
	else
		local ioption='-i'
	fi

	# set up testing suite
	rm -Rf "$WP_TESTS_DIR"
	mkdir -p "$WP_TESTS_DIR"
	cd "$WP_TESTS_DIR"

	if [[ $WP_TEST_VERSION  == "trunk" ]]; then
		svn co --quiet http://develop.svn.wordpress.org/trunk/tests/phpunit/includes/
		svn co --quiet http://develop.svn.wordpress.org/trunk/tests/phpunit/data/
		wget -nv -O wp-tests-config.php https://develop.svn.wordpress.org/trunk/wp-tests-config-sample.php
	elif [[ $WP_VERSION  == "" ]]; then
		svn co --quiet http://develop.svn.wordpress.org/trunk/tests/phpunit/includes/
		svn co --quiet http://develop.svn.wordpress.org/trunk/tests/phpunit/data/
		wget -nv -O wp-tests-config.php https://develop.svn.wordpress.org/trunk/wp-tests-config-sample.php
	else
		svn co --quiet http://develop.svn.wordpress.org/tags/${WP_VERSION}/tests/phpunit/includes/
		svn co --quiet http://develop.svn.wordpress.org/tags/${WP_VERSION}/tests/phpunit/data/
		wget -nv -O wp-tests-config.php https://develop.svn.wordpress.org/tags/${WP_VERSION}/wp-tests-config-sample.php
	fi

	sed $ioption "s:dirname( __FILE__ ) . '/src/':'${WP_CORE_DIR}':" wp-tests-config.php
	sed $ioption "s/youremptytestdbnamehere/$DB_NAME/" wp-tests-config.php
	sed $ioption "s/yourusernamehere/$DB_USER/" wp-tests-config.php
	sed $ioption "s/yourpasswordhere/$DB_PASS/" wp-tests-config.php
	sed $ioption "s|localhost|${DB_HOST}|" wp-tests-config.php
	echo "\$table_prefix = 'wptests' . ( getenv('TEST_TOKEN') !== false ? getenv('TEST_TOKEN') : '' ) . '_';" >> wp-tests-config.php
}

install_db() {
	# parse DB_HOST for port or socket references
	local PARTS=(${DB_HOST//\:/ })
	local DB_HOSTNAME=${PARTS[0]};
	local DB_SOCK_OR_PORT=${PARTS[1]};
	local EXTRA=""

	if ! [ -z $DB_HOSTNAME ] ; then
		if [[ "$DB_SOCK_OR_PORT" =~ ^[0-9]+$ ]] ; then
			EXTRA=" --host=$DB_HOSTNAME --port=$DB_SOCK_OR_PORT --protocol=tcp"
		elif ! [ -z $DB_SOCK_OR_PORT ] ; then
			EXTRA=" --socket=$DB_SOCK_OR_PORT"
		elif ! [ -z $DB_HOSTNAME ] ; then
			EXTRA=" --host=$DB_HOSTNAME --protocol=tcp"
		fi
	fi

	# create database
	MYSQL=`which mysql`

	$MYSQL --user="$DB_USER" --password="$DB_PASS"$EXTRA -e "DROP DATABASE IF EXISTS $DB_NAME;"
	$MYSQL --user="$DB_USER" --password="$DB_PASS"$EXTRA -e "CREATE DATABASE IF NOT EXISTS $DB_NAME;"
	$MYSQL --user="$DB_USER" --password="$DB_PASS"$EXTRA -e "GRANT USAGE ON *.* TO $DB_USER@$DB_HOSTNAME IDENTIFIED BY '$DB_PASS';"
	$MYSQL --user="$DB_USER" --password="$DB_PASS"$EXTRA -e "GRANT ALL PRIVILEGES ON $DB_NAME.* TO $DB_USER@$DB_HOSTNAME;"
	$MYSQL --user="$DB_USER" --password="$DB_PASS"$EXTRA -e "FLUSH PRIVILEGES;"
	$MYSQL --user="$DB_USER" --password="$DB_PASS"$EXTRA -e "USE $DB_NAME;"

#	mysqladmin create $DB_NAME --user="$DB_USER" --password="$DB_PASS"$EXTRA
}

install_wp
install_test_suite
install_db