#!/bin/bash

set -o errexit
set -o pipefail
set -o nounset


timeout=${WAIT_POSTGRES_TIMEOUT:-300}


if [ -z "${POSTGRES_USER}" ]; then
    base_postgres_image_default_user='postgres'
    export POSTGRES_USER="${base_postgres_image_default_user}"
fi
export DATABASE_URL="postgres://${POSTGRES_USER}:${POSTGRES_PASSWORD}@${POSTGRES_HOST}:${POSTGRES_PORT}/${POSTGRES_DB}"

postgres_ready() {
python3 << END
import sys
import psycopg2
try:
    psycopg2.connect(
        dbname="${POSTGRES_DB}",
        user="${POSTGRES_USER}",
        password="${POSTGRES_PASSWORD}",
        host="${POSTGRES_HOST}",
        port="${POSTGRES_PORT}",
    )
except psycopg2.OperationalError:
    sys.exit(-1)
sys.exit(0)
END
}

until postgres_ready; do
  >&2 echo 'Waiting for PostgreSQL to become available...'
  timeout=$((timeout-1))
  if [ $timeout -eq 0 ]; then
    >&2 echo 'PostgreSQL is NOT available'
    exit 1
  fi
  sleep 1
done
>&2 echo 'PostgreSQL is available'

exec "$@"
