#!/bin/sh

#  Licensed to the Apache Software Foundation (ASF) under one
#  or more contributor license agreements.  See the NOTICE file
#  distributed with this work for additional information
#  regarding copyright ownership.  The ASF licenses this file
#  to you under the Apache License, Version 2.0 (the
#  "License"); you may not use this file except in compliance
#  with the License.  You may obtain a copy of the License at
#  
#    http://www.apache.org/licenses/LICENSE-2.0
#    
#  Unless required by applicable law or agreed to in writing,
#  software distributed under the License is distributed on an
#  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
#  KIND, either express or implied.  See the License for the
#  specific language governing permissions and limitations
#  under the License.

# Configure a postgresql master server
here=`echo "import os; print os.path.realpath('$0')" | python`; here=`dirname $here`
mkdir -p $1
root=`echo "import os; print os.path.realpath('$1')" | python`

addr=$2
if [ "$addr" = "" ]; then
    host="localhost"
    listen="*"
    port="5432"
else
    host=`$here/../../modules/http/httpd-addr ip $addr`
    if [ "$host" = "" ]; then
        host="localhost"
        listen="*"
    else
        listen=$host
    fi
    port=`$here/../../modules/http/httpd-addr port $addr`
fi
bport=`expr $port + 1000`

pgsql_prefix=`cat $here/pgsql.prefix`
pgbouncer_prefix=`cat $here/pgbouncer.prefix`

user=`id -un`

mkdir -p $root/sqldb/data
chmod 700 $root/sqldb/data
mkdir -p $root/sqldb/archive
mkdir -p $root/sqldb/backup

if [ -f "$root/sqldb/log.conf" ]; then
    pgsql_log=`cat $root/sqldb/log.conf`
else
    mkdir -p $root/logs
    pgsql_log="cat >>$root/logs/postgresql"
fi
mkdir -p $root/sqldb
mkdir -p $root/sqldb/tmp
echo $pgsql_log >$root/sqldb/logger

# Initialize PostgreSQL
if [ ! -f $root/sqldb/data/postgresql.conf ]; then
    $pgsql_prefix/bin/pg_ctl init -D $root/sqldb/data 2>&1 | sh $root/sqldb/logger
    cp $root/sqldb/data/postgresql.conf $root/sqldb/data/postgresql-init.conf
    cp $root/sqldb/data/pg_hba.conf $root/sqldb/data/pg_hba-init.conf
fi

# Generate server configuration
cp $root/sqldb/data/postgresql-init.conf $root/sqldb/data/postgresql.conf
cat >>$root/sqldb/data/postgresql.conf <<EOF

# Generated by: pgsql-conf $*

# Setup logging
log_min_messages = NOTICE
log_min_error_statement = NOTICE
log_min_duration_statement = -1
log_checkpoints = on
log_connections = off
log_disconnections = off
log_duration = off
log_lock_waits = on
log_statement = none

# Listen
listen_addresses = '$listen'
port = $port
unix_socket_directory = '$root/sqldb/tmp'

# Setup archival
archive_mode = on
archive_command = '$here/pgsql-archive $root $host $bport %p %f'

# Setup hot standby with streaming replication
wal_level = hot_standby
max_wal_senders = 5
wal_keep_segments = 32

EOF

# Generate client auth configuration
cp $root/sqldb/data/pg_hba-init.conf $root/sqldb/data/pg_hba.conf
cat >>$root/sqldb/data/pg_hba.conf <<EOF

# Generated by: pgsql-conf $*
# TYPE  DATABASE        USER            CIDR-ADDRESS            METHOD
host    all             all             samenet                 trust
host    replication     all             samenet                 trust

EOF

# Create the db if it's not created yet
nohup /bin/sh -c "($pgsql_prefix/bin/pg_ctl start -W -D $root/sqldb/data 2>&1 | sh $root/sqldb/logger)" 1>/dev/null 2>/dev/null &
sti=0
while [ $sti -ne 30 ]; do
    st=`$pgsql_prefix/bin/pg_ctl status -D $root/sqldb/data | grep 'server is running'`
    if [ "$st" != "" ]; then
        break
    fi
    sleep 1
    sti=$((sti+1))
done

$pgsql_prefix/bin/createdb -h $host -p $port db 2>&1 | sh $root/sqldb/logger

# Create default user roles
$pgsql_prefix/bin/psql -h $host -p $port -c "create role standby with login replication" db 2>&1 | sh $root/sqldb/logger
$pgsql_prefix/bin/psql -h $host -p $port -c "create role bouncer with login" db 2>&1 | sh $root/sqldb/logger

# Backup the db if there's no backup for it yet
baks=`ls $root/sqldb/archive | sort -r  | grep "\.backup\.tar\.gz$"`
if [ "$baks" = "" ]; then
    $here/pgsql-backup $root $host $port
fi

$pgsql_prefix/bin/pg_ctl stop -w -D $root/sqldb/data 2>&1 | sh $root/sqldb/logger

# Generate database restore script
mkdir -p $root/sqldb/scripts
cat >$root/sqldb/scripts/restore <<EOF
#!/bin/sh
$here/pgsql-restore $root
EOF
chmod 700 $root/sqldb/scripts/restore

# Configure HTTPD to serve backup and archive files
if [ -f "$root/conf/httpd.conf" ]; then
    cat >>$root/conf/httpd.conf <<EOF
# Generated by: pgsql-conf $*

# Serve PostgreSQL backup and WAL archive files
ScriptAlias /pgsql-restore "$root/sqldb/scripts/restore"
Alias /pgsql-archive "$root/sqldb/archive"

EOF

fi

# Configure PgBouncer
mkdir -p $root/logs
id=`id -un`
cat >$root/sqldb/data/pgbouncer.conf <<EOF

[databases]
db = host=$host port=$port dbname=db user=bouncer

[pgbouncer]
pool_mode = session
listen_addr = $listen
listen_port = $bport
unix_socket_dir =
auth_type = trust
auth_file=$root/sqldb/data/pgbouncer-auth.conf
logfile = $root/logs/pgbouncer
pidfile = $root/logs/pgbouncer.pid
max_client_conn = 1000
pool_mode = transaction
server_reset_query =
default_pool_size = 500
min_pool_size = 5
reserve_pool_size = 50
log_connections = 0
log_disconnections = 0
stats_period = 3600
admin_users = $id

EOF

cat >$root/sqldb/data/pgbouncer-auth.conf <<EOF
"$id" "password"

EOF