#!/bin/sh

if [ "$PG_PORT" = "" ]; then
  PG_PORT=5432
fi

if [ "$MIN_POOL_SIZE" = "" ]; then
  MIN_POOL_SIZE=0
fi

if [ "$DEFAULT_POOL_SIZE" = "" ]; then
  DEFAULT_POOL_SIZE=20
fi

if [ "$RESRVE_POOL_SIZE" = "" ]; then
  RESERVE_POOL_SIZE=0
fi

if [ "$MAX_CONN" = "" ]; then
  MAX_CONN=1024
fi

echo "Starting pgbouncer for $PG_USER@$PG_HOST:$PG_PORT/$PG_DATABASE"
echo "Stats user: $STATS_USER"
echo

mkdir -p /etc/pgbouncer

cat > /etc/pgbouncer/pgbouncer.ini <<EOF
[databases]
$PG_DATABASE = dbname=$PG_DATABASE host=$PG_HOST port=$PG_PORT user=$PG_USER password=$PG_PASS
stats        = dbname=$PG_DATABASE host=$PG_HOST port=$PG_PORT user=$STATS_USER password=$STATS_PASS

[pgbouncer]
pool_mode = transaction
listen_port = 5432
listen_addr = 0.0.0.0
auth_type = md5
auth_file = /etc/pgbouncer/users
admin_users = $PG_USER
stats_users = $STATS_USER
max_client_conn = $MAX_CONN
min_pool_size = $MIN_POOL_SIZE
default_pool_size = $DEFAULT_POOL_SIZE
reserve_pool_size = $RESERVE_POOL_SIZE
ignore_startup_parameters = extra_float_digits
EOF

cat > /etc/pgbouncer/users <<EOF
"$PG_USER" "$PG_PASS"
"$STATS_USER" "$STATS_PASS"
EOF

exec /usr/bin/pgbouncer /etc/pgbouncer/pgbouncer.ini