#!/bin/sh
# run-postresql -- start up postgresql postmaster

# default settings -- argos-env will change based on config settings
ARGOS_ROOT=/bio/argos
ARGOS_POSTGRES=7302
setpath=0
quiet=0
myshell=$SHELL

## fix from conf files ..?
# FLYBASE_PGDATA=$ARGOS_ROOT/flybase/indices/pgsql
# EUGENES_PGDATA=$ARGOS_ROOT/eugenes/indices/pgsql
# DAPHNIA_PGDATA=$ARGOS_ROOT/daphnia/indices/pgsql

# -----------------------------------------------------------------------------

# find_homedir - resolving softlink in $0
# assume real $0 is MY_HOME/bin/script_name
# where MY_HOME/conf and other goodies are located - scarfed from ant/bin/ant

find_MY_HOME()
{
  PRG="$0"
  while [ -h "$PRG" ] ; do
    ls=`ls -ld "$PRG"`
    link=`expr "$ls" : '.*-> \(.*\)$'`
    if expr "$link" : '/.*' > /dev/null; then
      PRG="$link"
    else
      PRG=`dirname "$PRG"`/"$link"
    fi
  done
  
  MY_HOME=`dirname "$PRG"`/..
  # make it fully qualified
  MY_HOME=`cd "$MY_HOME" && pwd`
  PRG=`basename $0`
  # export MY_HOME
  # export PRG
}

set_pgpaths()
{ 
  # echo "# set_pgpaths"
  
  if [ -z "$POSTGRES_HOME" ] ; then
    POSTGRES_HOME="$ARGOS_ROOT/common/servers/postgresql"; 
  fi
    
  if [ ! -r "${POSTGRES_HOME}/bin/pg_ctl" ]; then
    echo "# The POSTGRES_HOME environment variable is not defined correctly"
    echo "# ${POSTGRES_HOME}/bin/pg_ctl   is missing "
    exit 1
  fi

  export POSTGRES_HOME
  
  PGLIB=${POSTGRES_HOME}/lib

  ## ! for Argos should put PGDATA in local ROOT/indices/pgsql 
  ## not in distributed server folder!
  if [ -z "$PGDATA" ] ; then
    PGDATA=${POSTGRES_HOME}/data
    export PGDATA
  fi

  pglog=$ARGOS_ROOT/logs/pgsql.log
  pgpath=${POSTGRES_HOME}/bin:${PATH}
  
### only if desired .. this isnt useful unless we source / . this into shell
### add to argos-env as option?
  if [ $setpath -eq 1 ]; then
   PATH=${pgpath} ; export PATH
  fi
 
  if [ -z "$MANPATH" ] ; then
    MANPATH=${POSTGRES_HOME}/man 
  else 
    MANPATH="${MANPATH}:${POSTGRES_HOME}/man" ; 
  fi
  export MANPATH
  
  ## fixme for multiple OSes
  
  if [ -z "$LD_LIBRARY_PATH" ] ; then
    LD_LIBRARY_PATH=${PGLIB} 
  else 
    LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:${PGLIB}" ; 
  fi
  export LD_LIBRARY_PATH
  if [ -z "$DYLD_LIBRARY_PATH" ] ; then
    DYLD_LIBRARY_PATH=${PGLIB} 
  else 
    DYLD_LIBRARY_PATH="${DYLD_LIBRARY_PATH}:${PGLIB}" ; 
  fi
  export DYLD_LIBRARY_PATH

}

do_showenv()
{
  if [ $quiet -eq 0 ]; then
cat <<HERE
Using POSTGRES_HOME:  $POSTGRES_HOME
      PGPORT:         $ARGOS_POSTGRES
Logging to:           $pglog
Also useful:          set path=(${POSTGRES_HOME}/bin \$path)
                      LD_LIBRARY_PATH=$LD_LIBRARY_PATH
                      MANPATH=$MANPATH
HERE
  fi
}

do_setenv()
{
  dosh="csh"
  case "$myshell" in
    *"csh"|*"tcsh") dosh="csh" ;;
    *"bash"|*"sh") dosh="sh" ;;
    *) dosh="print" ;;
  esac
  tofile="/tmp/argosenvpg$$.$dosh"
  touch $tofile

  if [ $dosh = "sh" ]; then
cat >> $tofile <<HERE
pg=$POSTGRES_HOME
POSTGRES_HOME=$POSTGRES_HOME ; export POSTGRES_HOME
PGDATA=$PGDATA ; export PGDATA
PGPORT=$ARGOS_POSTGRES ; export PGPORT
LD_LIBRARY_PATH=$LD_LIBRARY_PATH ; export LD_LIBRARY_PATH
MANPATH=$MANPATH ; export MANPATH
PATH="${POSTGRES_HOME}/bin:$PATH" ; export PATH
/bin/rm $tofile
HERE
  echo $tofile

  elif [ $dosh = "csh" ]; then
cat >> $tofile <<HERE
set pg=$POSTGRES_HOME
setenv POSTGRES_HOME $POSTGRES_HOME
setenv PGDATA $PGDATA
setenv PGPORT $ARGOS_POSTGRES
setenv LD_LIBRARY_PATH $LD_LIBRARY_PATH
setenv MANPATH $MANPATH
setenv PATH "${POSTGRES_HOME}/bin:$PATH"
/bin/rm $tofile
HERE
  echo $tofile

  else
cat  <<HERE
set pg=$POSTGRES_HOME
setenv POSTGRES_HOME $POSTGRES_HOME
setenv PGDATA $PGDATA
setenv PGPORT $ARGOS_POSTGRES
setenv LD_LIBRARY_PATH $LD_LIBRARY_PATH
setenv MANPATH $MANPATH
setenv PATH "${POSTGRES_HOME}/bin:$PATH"
HERE
  fi
}

# ----- Execute The Requested Command -----------------------------------------

do_command()
{
  # get_catargs

  case "$command" in
    "reload") 
      ${POSTGRES_HOME}/bin/pg_ctl -D $PGDATA reload ;;
    
    "restart" )
      ${POSTGRES_HOME}/bin/pg_ctl -o "-i -F -p $ARGOS_POSTGRES" -D $PGDATA restart -m fast  ;;
      
    "start"   ) 
      ${POSTGRES_HOME}/bin/pg_ctl -o "-i -F -p $ARGOS_POSTGRES" -D $PGDATA -l $pglog  start ;;
      
    "stop"    ) 
      ${POSTGRES_HOME}/bin/pg_ctl -D $PGDATA stop -m fast ;;

    "kill"    ) 
      ${POSTGRES_HOME}/bin/pg_ctl -D $PGDATA stop -m immediate ;;
      
    "status"    ) 
      ${POSTGRES_HOME}/bin/pg_ctl -D $PGDATA status ;;

    "setenv"    ) quiet=1; do_setenv ;;
      
    "help"*|*)  
    do_showenv; 
    cat <<HERE
Usage: $0 ( commands ... )
commands:
  start     Start PostgreSQL server
  stop      Stop PostgreSQL
  kill      Stop PostgreSQL immediately
  restart   Re-Start PostgreSQL server
  reload    Reload PostgreSQL configurations
  status    report status of PostgreSQL server
  setenv    print PostgreSQL shell environ settings 
                (e.g source \`$0 setenv\`)

do command for each server in "$POSTGRES_SERVERS" 
options:
    -p              Set PATH  to include Postgres/bin
    -s server_name  Use only this server
  
HERE

    exit 2
    ;;
    
  esac
}


#------------------------------------
# main script

ERROR=0

find_MY_HOME
CONFDIR="$MY_HOME/conf"

## argos-env now loads all of standard config env variables
. `$MY_HOME/bin/argos-env  sh pgenv.conf`

## override (possibly) above settings
# CONFDIR="$MY_HOME/conf"
# runconf="install.conf"
# if [ -r "$CONFDIR"/$runconf ]; then
#   . "$CONFDIR"/$runconf
# fi
# if [ -r "$CONFDIR"/${runconf}.local ]; then
#   . "$CONFDIR"/${runconf}.local
# fi
# if [ -r "$CONFDIR"/pgenv.sh ]; then
#   . "$CONFDIR"/pgenv.sh
# fi

pgserver=
getarg=
appargs=
command=
ARGV=$@
if [ "x$ARGV" = "x" ] ; then 
    ARG2="help"
fi

for ARG in $@ $ARG2  
{
  case "$ARG" in
    "-debug"*) debug="-debug=1" ;;
    "-p"*) setpath=1;;
    "-sh"*) getarg="shell" ;;
    "-s"*) getarg="server" ;;
    "-verbose"|"-v"*) quiet=0 ;;

    *) 
      if [ "$getarg" = "server" ] ; then getarg=""; POSTGRES_SERVERS="$ARG"; 
      elif [ "$getarg" = "shell" ] ; then getarg=""; myshell="$ARG"; 
#       elif [ "$getarg" = "home" ] ; then getarg=""; APACHE_HOME="$ARG"; 
      elif [ -z "$command" ] ; then command="$ARG"; 
      else appargs="$appargs $ARG"; 
      fi 
      ;;
      
  esac
}

set_pgpaths

do_command 

exit $ERROR

### use PG initlocation for service-specific data sets
# initdb --no-locale -L $POSTGRES_HOME/share -D $PGDATA
# initlocation FLYBASE_PGDATA
# pg_ctl -o "-i -F -p $ARGOS_POSTGRES" -D $PGDATA -l $ARGOS_ROOT/logs/pgsql.log start
# createuser -A -P fbwork  [FlyBase password]
# createdb -D FLYBASE_PGDATA gadfly3_chado gadfly3_dump_05_29apr03