#!/usr/bin/env bash

CORFUDBBIN="${BASH_SOURCE-$0}"
CORFUDBBIN="$(dirname "${CORFUDBBIN}")"
CORFUDBBINDIR="$(cd "${CORFUDBBIN}"; pwd)"

. "$CORFUDBBINDIR"/corfuDBEnv.sh 

# handle commnad line params
# ...............................
usage="--unit {sequencer|configmaster|logunit.port} --cmd {start|stop|restart|status} [--configdir <conf-dir>] "

unit=""
cmd=""
cfgdir=$CORFUDBCFGDIR

# parse arguments
while  [ $# -ge 2 ] 
do
	if [ "--unit" == "$1" ]
	then
		shift
		unit=$1
		shift
	elif [ "--configdir" == "$1" ]
	then
		shift
		cfgdir=$1
		shift
	elif [ "--cmd" == "$1" ]
	then
		shift
		cmd=$1
		shift
	else
		echo "unrecognized command-line switch: $1" >&2
		echo "Usage: $0 $usage" >&2
		exit 1
	fi
done

echo "running unit $unit cmd $cmd config-dir $cfgdir" >&2
if [ X$unit = X"" ] || [ X$cmd = X"" ] 
then
	echo "Usage: $0 $usage" >&2
	exit 1
fi
# ............................

CORFUDBCFG="$cfgdir/corfudb.$unit.yml"
CORFUDB_DAEMON_OUT="/var/log/corfudb.$unit.log"
CORFUDBMAIN="org.corfudb.infrastructure.ConfigParser"

if $cygwin
then
    CORFUDBCFG=`cygpath -wp "$CORFUDBCFGDIR/corfudb.$unit.yml"`
    CORFUDB_DAEMON_OUT=`cygpath -wp "/var/log/corfudb.$unit.log"`
    KILL=/bin/kill
else
    KILL=kill
fi

if [ -z "$CORFUDBPIDFILE" ]; then
    CORFUDBPIDFILE="/var/run/corfudb.$unit.pid"
else
    mkdir -p "$(dirname $CORFUDBPIDFILE)"
fi

echo "Using config: $CORFUDBCFG" >&2
echo "Logging to: $CORFUDB_DAEMON_OUT" >&2
echo "PID to: $CORFUDBPIDFILE" >&2

case $cmd in
start)
    echo -n "Starting CorfuDB unit $unit..." >&2
    if [ -f "$CORFUDBPIDFILE" ]; then
	echo "trying to kill previous ..."
        if kill -0 `cat $CORFUDBPIDFILE` > /dev/null 2>&1; then
            echo $unit already running as process `cat "$CORFUDBPIDFILE"`.
            exit 0
        fi
    fi
    nohup "$JAVA" "-Dorg.slf4j.simpleLogger.defaultLogLevel=${CORFUDB_LOG4J_PROP}" \
    -cp "$CLASSPATH" $JVMFLAGS "$CORFUDBMAIN" "$CORFUDBCFG" > "$CORFUDB_DAEMON_OUT" 2>&1 < /dev/null &
    if [ $? -eq 0 ]
    then
        if /bin/echo -n $! > "$CORFUDBPIDFILE"
        then
            #sleep 1
            # The server may have failed to start. Let's make sure it did
            if kill -0 $! > /dev/null 2>&1;
            then
                echo Started
            else
                echo Failed to start, log was:
                cat "$CORFUDB_DAEMON_OUT"
            fi
        else
            echo Failed to write PID
            exit 1
        fi
    else
        echo Server failed to start
        exit 1
    fi
    ;;
stop)
    echo -n "Stopping CorfuDB unit $unit..."
    if [ ! -f "$CORFUDBPIDFILE" ];
    then
        echo "Could not find a PID file to stop..."
        exit 0 #should this be exit 1, maybe?
    else
        $KILL $(cat "$CORFUDBPIDFILE")
        rm "$CORFUDBPIDFILE"
        echo Stopped
        exit 0
    fi
    ;;
restart)
    "$0" "$1" stop
    sleep 1
    "$0" "$1" start
    ;;
status)
    if [ ! -f "$CORFUDBPIDFILE" ];
    then
        echo "Could not find a PID file..."
        exit 0 #should this be exit 1, maybe?
    else
        echo -n "CorfuDB unit $unit running as PID "
        echo $(cat "$CORFUDBPIDFILE")
        echo "Last 100 lines of log:"
        cat $CORFUDB_DAEMON_OUT | tail -100
        exit 0
    fi
    ;;
*)
    echo "Usage: $0 [--config <conf-dir>] <node-name> {start|stop|restart|status}" >&2
    ;;
esac