start

#! /usr/bin/env bash
set -e

export STELLAR_HOME="/opt/stellar"
export PGHOME="$STELLAR_HOME/postgresql"
export SUPHOME="$STELLAR_HOME/supervisor"
export COREHOME="$STELLAR_HOME/core"
export HZHOME="$STELLAR_HOME/horizon"
export BRIDGEHOME="$STELLAR_HOME/bridge"
export FRIENDBOTHOME="$STELLAR_HOME/friendbot"

export PGBIN="/usr/lib/postgresql/9.6/bin"
export PGDATA="$PGHOME/data"
export PGUSER="stellar"
export PGPORT=5432

QUICKSTART_INITIALIZED=false
CURRENT_POSTGRES_PID=""


# NOTE: not configurable, generated by network and updated in init_stellar_core
ROOT_ACCOUNT_SEED=""

function main() {
	echo ""
	echo "Starting Stellar Quickstart"
	echo ""

	process_args $*

	echo "mode: $STELLAR_MODE"
	echo "network: $NETWORK ($NETWORK_PASSPHRASE)"

    # custom supervisord scripts
    init_supervisor_custom

	copy_defaults
	init_db
	init_stellar_core
	init_bridge_server # must come after init_stellar_core!
	init_horizon
	init_friendbot
	copy_pgpass

	stop_postgres  # this gets started in init_db

	echo "**************************************************************"
	echo "Starting Stellar services..."
	echo ""
	echo "Stellar Core    v10.0.0-685-1fc018b4"
	echo "Horizon         v0.15.1"
	echo "Bridge          v0.0.31"
	echo "Friendbot       v0.0.1"
	echo "**************************************************************"

	# launch services
	exec_supervisor
}

function process_args() {
	while [[ -n "$1" ]]; do
  	ARG="$1"
	  shift

	  case "${ARG}" in
	  --testnet)
	    NETWORK="testnet"
	    ;;
	  --pubnet)
	    NETWORK="pubnet"
	    ;;
	  --integrationnet)
        NETWORK="integrationnet"
        ;;
	  *)
	    echo "Unknown container arg $ARG" >&2
	    exit 1
	  esac
	done

	# TODO: ask for what network to use
	if [ -z "$NETWORK" ]; then
		NETWORK="testnet"
	fi

	case "$NETWORK" in
	testnet)
    export NETWORK_PASSPHRASE="Test SDF Network ; September 2015"
    ;;
	pubnet)
    export NETWORK_PASSPHRASE="Public Global Stellar Network ; September 2015"
    ;;
    integrationnet)
    export NETWORK_PASSPHRASE="Integration Test Network ; zulucrypto"
    ;;
	*)
		echo "Unknown network: '$NETWORK'" >&2
		exit 1
	esac

	# Are we ephemeral or persistent?
	if [ -z "$STELLAR_MODE" ]; then
		if [ -f "/opt/stellar/.docker-ephemeral" ]; then
			STELLAR_MODE="ephemeral"
		else
			STELLAR_MODE="persistent"
		fi
	fi
}

function set_pg_password() {
	# use a random password when ephemeral (or some other unknown mode)
	if [ "$STELLAR_MODE" != "persistent" ]; then
		PGPASS=$(head /dev/urandom | tr -dc 'a-zA-Z0-9' |  head -c 16)
		echo "postgres password: $PGPASS"
		return 0
	fi

	if [ -n "$PGPASS" ]; then
		echo "postgres password: $PGPASS"
		return 0
	fi

	# ask for a password when persistent
	read -s -p "Enter New Postgresql Password: " PGPASS
	echo ""
	read -s -p "Confirm: " PGPASS_CONFIRMATION
	echo ""

	if [ -z "$PGPASS" ]; then
		echo "Password empty" >&2
		exit 1
	fi

	if [ "$PGPASS" != "$PGPASS_CONFIRMATION" ]; then
		echo "Password mistmach" >&2
		exit 1
	fi

}

function copy_defaults() {
	local CP="rsync -a"

	if [ -d $PGHOME/etc ]; then
		echo "postgres: config directory exists, skipping copy"
	else
		$CP /opt/stellar-default/common/postgresql/ $PGHOME
	fi

	if [ -d $SUPHOME/etc ]; then
		echo "supervisor: config directory exists, skipping copy"
	else
		$CP /opt/stellar-default/common/supervisor/ $SUPHOME
	fi

	if [ -d $COREHOME/etc ]; then
		echo "stellar-core: config directory exists, skipping copy"
	else
		$CP /opt/stellar-default/common/core/ $COREHOME
		$CP /opt/stellar-default/$NETWORK/core/ $COREHOME
	fi

	if [ -d $HZHOME/etc ]; then
		echo "horizon: config directory exists, skipping copy"
	else
		$CP /opt/stellar-default/common/horizon/ $HZHOME
	fi

	if [ -d $BRIDGEHOME/etc ]; then
        echo "bridge: config directory exists, skipping copy"
    else
        $CP /opt/stellar-default/common/bridge/ $BRIDGEHOME
        $CP /opt/stellar-default/$NETWORK/bridge/ $BRIDGEHOME
    fi
}

function copy_pgpass() {
	local CP="rsync -a"

	$CP /opt/stellar/postgresql/.pgpass /root/
	chmod 0600 /root/.pgpass

	$CP /opt/stellar/postgresql/.pgpass /home/stellar
	chmod 0600 /home/stellar/.pgpass
	chown stellar:stellar /home/stellar/.pgpass
}

function init_supervisor_custom() {
    local CP="rsync -a"

    if [ "$NETWORK" == "integrationnet" ]; then
        echo "init-supervisor-custom: copying integrationnet scripts"
        $CP /opt/stellar-default/$NETWORK/supervisor.d/ /etc/supervisor.d/
        $CP /opt/stellar-default/$NETWORK/home/fixtures/ /home/stellar/fixtures/
    fi
}

function init_db() {
	if [ -f $PGHOME/.quickstart-initialized ]; then
		echo "postgres: already initialized"
		return 0
	fi
	pushd $PGHOME

	# workaround!!!! from: https://github.com/nimiq/docker-postgresql93/issues/2
	mkdir /etc/ssl/private-copy; mv /etc/ssl/private/* /etc/ssl/private-copy/; rm -r /etc/ssl/private; mv /etc/ssl/private-copy /etc/ssl/private; chmod -R 0700 /etc/ssl/private; chown -R postgres /etc/ssl/private
	# end workaround

	echo "postgres user: $PGUSER"

	set_pg_password

	run_silent "finalize-pgpass" sed -ri "s/__PGPASS__/$PGPASS/g" /opt/stellar/postgresql/.pgpass

	mkdir -p $PGDATA
	chown postgres:postgres $PGDATA
	chmod 0700 $PGDATA

	run_silent "init-postgres" sudo -u postgres $PGBIN/initdb -D $PGDATA

	start_postgres
	run_silent "create-horizon-db" sudo -u postgres createdb horizon
	run_silent "create-core-db" sudo -u postgres createdb core
	run_silent "create-bridge-db" sudo -u postgres createdb bridge

	run_silent "stellar-postgres-user" sudo -u postgres psql <<-SQL
		CREATE USER $PGUSER WITH PASSWORD '$PGPASS';
		GRANT ALL PRIVILEGES ON DATABASE horizon to $PGUSER;
		GRANT ALL PRIVILEGES ON DATABASE core to $PGUSER;
		GRANT ALL PRIVILEGES ON DATABASE bridge to $PGUSER;
	SQL

	touch .quickstart-initialized
	popd
}

function init_stellar_core() {
	if [ -f $COREHOME/.quickstart-initialized ]; then
		echo "core: already initialized"
		return 0
	fi
	pushd $COREHOME

	run_silent "chown-core" chown stellar:stellar .
	run_silent "finalize-core-config" sed -ri "s/__PGPASS__/$PGPASS/g" etc/stellar-core.cfg

	start_postgres
	run_silent "init-core-db" sudo -u stellar stellar-core --newdb --conf etc/stellar-core.cfg

	# get master seed from the file that run_silent generates
	ROOT_ACCOUNT_SEED=$(grep "Root account seed" /tmp/run_silent.out | cut -d' ' -f 8)
	echo "Root Seed: ${ROOT_ACCOUNT_SEED}"

	# integration test network is standalone and needs additional setup
    if [ "$NETWORK" == "integrationnet" ]; then
        run_silent "init-core-scp" sudo -u stellar stellar-core --forcescp --conf etc/stellar-core.cfg
    fi

	touch .quickstart-initialized
	popd
}

# NOTE: must be run after init_stellar_core since it requires $ROOT_ACCOUNT_SEED
function init_bridge_server() {
    if [ -f $BRIDGEHOME/.quickstart-initialized ]; then
        echo "bridge: already initialized"
        return 0
    fi
    pushd $BRIDGEHOME

    run_silent "chown-bridge" chown stellar:stellar .

    sed -ri \
    		-e "s/__PGPASS__/$PGPASS/g" \
    		-e "s/__NETWORK__/$NETWORK_PASSPHRASE/g" \
    		-e "s/__ROOT_ACCOUNT_SEED__/$ROOT_ACCOUNT_SEED/g" \
    		etc/bridge.cfg

    touch .quickstart-initialized
    popd
}

function init_horizon() {
	if [ -f $HZHOME/.quickstart-initialized ]; then
		echo "horizon: already initialized"
		return 0
	fi
	pushd $HZHOME

	run_silent "chown-horizon" chown stellar:stellar .

	sed -ri \
		-e "s/__PGPASS__/$PGPASS/g" \
		-e "s/__NETWORK__/$NETWORK_PASSPHRASE/g" \
		etc/horizon.env

	start_postgres
	run_silent "init-horizon-db" sudo -u stellar ./bin/horizon db init

	touch .quickstart-initialized
	popd
}

function init_friendbot() {
	if [ -f $FRIENDBOTHOME/.quickstart-initialized ]; then
		echo "friendbot: already initialized"
		return 0
	fi

	cp -R /opt/stellar-default/common/friendbot/ $FRIENDBOTHOME

	pushd $FRIENDBOTHOME

	run_silent "chown-friendbot" chown stellar:stellar .

	touch .quickstart-initialized
	popd
}

function exec_supervisor() {
	echo "starting suervisor"
	exec supervisord -n -c $SUPHOME/etc/supervisord.conf
}

# run_silent is a utility function that runs a command with an abbreviated
# output provided it succeeds.
function run_silent() {
	local LABEL=$1
	shift
	local COMMAND=$1
	shift
	local ARGS=$@
	local OUTFILE="/tmp/run_silent.out"

	echo -n "$LABEL: "
	set +e

	$COMMAND $ARGS &> $OUTFILE

	if [ $? -eq 0 ]; then
    echo "ok"
	else
	  echo "failed!"
		echo ""
		cat $OUTFILE
		exit 1
	fi

	set -e
}

function start_postgres() {
	if [ ! -z "$CURRENT_POSTGRES_PID" ]; then
		return 0
	fi

	sudo -u postgres $PGBIN/postgres -D $PGDATA -c config_file=$PGHOME/etc/postgresql.conf &> /dev/null &
	CURRENT_POSTGRES_PID=$!

	while ! sudo -u postgres psql -c 'select 1' &> /dev/null ; do
	  echo "Waiting for postgres to be available..."
	  sleep 1
	done

	echo "postgres: up"
}

function stop_postgres() {
    echo "Stopping: '$CURRENT_POSTGRES_PID'"
	if [ -z "$CURRENT_POSTGRES_PID" ]; then
		return 0
	fi

	# wait for postgres to die
	while kill -0 "$CURRENT_POSTGRES_PID" &> /dev/null; do
	    # If kill doesn't work, use "killall" on subsequent runs since the PID may be invalid
	    killall postgres
		sleep 5
	done
	echo "postgres: down"
}

pushd () {
    command pushd "$@" > /dev/null
}

popd () {
    command popd "$@" > /dev/null
}

main $@