docker-entrypoint.sh

#!/bin/bash
set -e

# usage: file_env VAR [DEFAULT]
#    ie: file_env 'XYZ_DB_PASSWORD' 'example'
# (will allow for "$XYZ_DB_PASSWORD_FILE" to fill in the value of
#  "$XYZ_DB_PASSWORD" from a file, especially for Docker's secrets feature)
file_env() {
	local var="$1"
	local fileVar="${var}_FILE"
	local def="${2:-}"
	if [ "${!var:-}" ] && [ "${!fileVar:-}" ]; then
		echo >&2 "error: both $var and $fileVar are set (but are exclusive)"
		exit 1
	fi
	local val="$def"
	if [ "${!var:-}" ]; then
		val="${!var}"
	elif [ "${!fileVar:-}" ]; then
		val="$(< "${!fileVar}")"
	fi
	export "$var"="$val"
	unset "$fileVar"
}

if [ "${1:0:1}" = '-' ]; then
	set -- postgres "$@"
fi

if [ "$1" = 'postgres' ]; then
	mkdir -p "$PGDATA"
	chmod 700 "$PGDATA"
	chown -R postgres "$PGDATA"

	mkdir -p /run/postgresql
	chmod g+s /run/postgresql
	chown -R postgres /run/postgresql

	# look specifically for PG_VERSION, as it is expected in the DB dir
	if [ ! -s "$PGDATA/PG_VERSION" ]; then
		file_env 'POSTGRES_INITDB_ARGS'
		eval "gosu postgres initdb $POSTGRES_INITDB_ARGS"

		# check password first so we can output the warning before postgres
		# messes it up
		file_env 'POSTGRES_PASSWORD'
		if [ "$POSTGRES_PASSWORD" ]; then
			pass="PASSWORD '$POSTGRES_PASSWORD'"
			authMethod=md5
		else
			# The - option suppresses leading tabs but *not* spaces. :)
			cat >&2 <<-'EOWARN'
				****************************************************
				WARNING: No password has been set for the database.
				         This will allow anyone with access to the
				         Postgres port to access your database. In
				         Docker's default configuration, this is
				         effectively any other container on the same
				         system.

				         Use "-e POSTGRES_PASSWORD=password" to set
				         it in "docker run".
				****************************************************
			EOWARN

			pass=
			authMethod=trust
		fi

		{ echo; echo "host all all all $authMethod"; } | gosu postgres tee -a "$PGDATA/pg_hba.conf" > /dev/null

		# internal start of server in order to allow set-up using psql-client
		# does not listen on external TCP/IP and waits until start finishes
		gosu postgres pg_ctl -D "$PGDATA" \
			-o "-c listen_addresses='localhost'" \
			-w start

		file_env 'POSTGRES_USER' 'postgres'
		file_env 'POSTGRES_DB' "$POSTGRES_USER"

		psql=( psql -v ON_ERROR_STOP=1 )

		if [ "$POSTGRES_DB" != 'postgres' ]; then
			"${psql[@]}" --username postgres <<-EOSQL
				CREATE DATABASE "$POSTGRES_DB" ;
			EOSQL
			echo
		fi

		if [ "$POSTGRES_USER" = 'postgres' ]; then
			op='ALTER'
		else
			op='CREATE'
		fi
		"${psql[@]}" --username postgres <<-EOSQL
			$op USER "$POSTGRES_USER" WITH SUPERUSER $pass ;
		EOSQL
		echo

		psql+=( --username "$POSTGRES_USER" --dbname "$POSTGRES_DB" )

		echo
		for f in /docker-entrypoint-initdb.d/*; do
			case "$f" in
				*.sh)     echo "$0: running $f"; . "$f" ;;
				*.sql)    echo "$0: running $f"; "${psql[@]}" -f "$f"; echo ;;
				*.sql.gz) echo "$0: running $f"; gunzip -c "$f" | "${psql[@]}"; echo ;;
				*)        echo "$0: ignoring $f" ;;
			esac
			echo
		done

		## add by baruch@brillix.co.il
		# add replication user to all postgres master and slave 
        file_env 'PASSWORD'
		file_env 'MASTER'
		gosu postgres psql -c "CREATE USER replicator REPLICATION LOGIN ENCRYPTED PASSWORD '${PASSWORD}';"
		echo "*:*:*:replicator:${PASSWORD}" > /.pgpass
		echo "${MASTER}:5432:repliction:replicator:${PASSWORD}" >> /.pgpass
		gosu postgres pg_ctl -D "$PGDATA" -m fast -w stop
		# add by baruch@brillix.co.il to support master slave
		#chack if the server is SLAVE 
		# 1. recover from master 
		# 2. replace config file 
		# 3. create recover file 
		# 4. start pg
		# !notice the value of MASTER id the master IP !
		if [ "$MASTER" ];then
			cat >&2 <<-'EOWARN'
				****************************************************
				Configuring postgres as slave
				****************************************************
			EOWARN
			##clear datadir
      gosu postgres bash -c 'cp ${PGDATA}/*.conf /tmp/'
			gosu postgres bash -c 'rm -r ${PGDATA}/*'
			# recover from master 
			gosu postgres bash -c 'export PGPASSWORD=${PASSWORD} ; pg_basebackup  -h ${MASTER} -p 5432  -D ${PGDATA} -U replicator -v -P'
			# replace config files
			gosu postgres bash -c 'cp /tmp/*.conf ${PGDATA}/'
			##recovery file creation 
			echo '' > ${PGDATA}/recovery.conf
			chown postgres.postgres ${PGDATA}/recovery.conf
			echo "standby_mode = 'on' " >> ${PGDATA}/recovery.conf
			echo "primary_conninfo = 'host=${MASTER} port=5432 user=replicator password=${PASSWORD} sslmode=require '" >> ${PGDATA}/recovery.conf
			echo "trigger_file = '/tmp/postgresql.trigger' " >>  ${PGDATA}/recovery.conf
		fi

		echo
		echo 'PostgreSQL init process complete; ready for start up.'
		echo
	fi

	exec gosu postgres "$@"
fi

exec "$@"