#!/bin/bash
# Author(s):
# Sascha Wilde <wilde@intevation.de>

ME=`basename "$0"`
BASEDIR=`dirname "$0"`

usage()
{
  cat <<EOF
$ME [OPION]...

Options:
  -d, --db=NAME    connect to the database NAME
  -p, --port=PORT  connect do the postgresql cluster at PORT
  -D, --demo       install demo accounts and data
      --drop       drop database and all roles
      --help       display this help and exit

EOF
}

fatal()
{
  echo >&2 "$1"
  exit 23
}


# Defaults:

db=gemma
port=5432
demo=0
drop=0

# Parse options:

OPTS=`getopt \
      -l help,demo,db:,port:,drop \
      -o Dd:p: -n "$ME" -- "$@"`
[ $? -eq 0 ] || { usage ; exit 1 ; }

eval set -- "$OPTS"

while true ; do
  case "$1" in
    --db|-d)
      db="$2"
      shift 2
      ;;
    --port|-p)
      port="$2"
      shift 2
      ;;
    --demo|-D)
      demo=1
      shift 1
      ;;
    --drop)
      drop=1
      shift 1
      ;;
    --help)
      { usage ; exit 0 ; }
      ;;
    --)
      shift
      break
      ;;
  esac
done


# Main ------------------------------------------------------------

if [[ drop -eq 0 ]] ; then
  # Default operation: create schema
  psql -p "$port" -f "$BASEDIR/roles.sql"
  createdb -p "$port" "$db"
  psql -p "$port" -d "$db" \
       -f "$BASEDIR/gemma.sql" \
       -f "$BASEDIR/manage_users.sql" \
       -f "$BASEDIR/auth.sql"

  if [[ $demo -eq 1 ]] ; then
    psql -p "$port" -f "$BASEDIR/demo-data/responsibility_areas.sql" \
         -d "$db"
    psql -p "$port" -f "$BASEDIR/demo-data/roles.sql" \
         -f "$BASEDIR/demo-data/users.sql" -d "$db"
  fi
else
  # Evil mode: drop everything gemma
  echo "Really drop database '$db' and alle gemma roles? [type 'yes']: "
  read a
  if [[ $a == "yes" ]] ; then
    dropdb -p "$port" "$db"
    for r in `psql -p 5433 -t -c '\du' | awk -F '|' \
              '$3 ~/waterway_user|waterway_admin|sys_admin/ \
               || $1 ~/waterway_user|waterway_admin|sys_admin/ \
               {print $1}'`
    do
      dropuser -p "$port" "$r"
    done
  else
    echo "No harm done."
  fi
fi