#!/bin/bash
#==============================================================================
# Copyright and license info is available in the LICENSE file included with
# the Server Deployment Package (SDP), and also available online:
# https://swarm.workshop.perforce.com/projects/perforce-software-sdp/view/main/LICENSE
#------------------------------------------------------------------------------

# Helix Broker base init script

# Do nothing unless $SDP_INSTANCE is defined.
export SDP_INSTANCE=${1:-Unset}

if [[ $SDP_INSTANCE == Unset ]]; then
   echo -e "\nError: The SDP_INSTANCE is not defined.\n."
   exit 1
fi

# Load SDP controlled shell environment.
source /p4/common/bin/p4_vars "$SDP_INSTANCE"
if [[ $? -ne 0 ]]; then
   echo -e "\nError: Failed to load SDP environment for instance $SDP_INSTANCE.\n"
   exit 1
fi

source $P4CBIN/ps_functions.sh
if [[ $? -ne 0 ]]; then
   echo -e "\nError: Failed to load SDP ps_functions.sh lib.\n"
   exit 1
fi

if [[ $(id -u) -eq 0 ]]; then
   exec su - $OSUSER -c "$0 $*"
elif [[ $(id -u -n) != $OSUSER ]]; then
   echo "$0 can only be run by root or $OSUSER"
   exit 1
fi

if [[ ! -x $P4BROKERBIN ]]; then
   echo -e "\nError: $P4BROKERBIN is not executable."
   exit 1
fi

declare OpMode=${2:-status}
declare -i ExitStatus=0

export P4BROKERCFG=${P4CCFG}/${P4SERVER}.broker.${P4MASTER_ID}.cfg

if [[ ! -r "$P4BROKERCFG" ]]; then
   echo -e "\nError: Missing broker config file $P4BROKERCFG.\n"
   exit 1
fi

case "$OpMode" in
   start)
      echo "Starting ${P4BROKERBIN} ($tag config) using ${P4BROKERCFG}."
      ${P4BROKERBIN} -d -c ${P4BROKERCFG}
   ;;

   status)
      p4broker_pids=$(get_pids "$P4BROKERCFG")
      if [[ -n "$p4broker_pids" ]]; then
         echo "${P4BROKERBIN} ($tag config) is running as pids: $p4broker_pids."
         ExitStatus=0
      else
         echo "${P4BROKERBIN} ($tag config) is NOT running."
         ExitStatus=1
      fi
   ;;

   stop)
      p4broker_pids=$(get_pids "$P4BROKERCFG")

      if [[ -n "$p4broker_pids" ]]; then
         while [[ -n "$p4broker_pids" ]]; do
            echo "Shutting down ${P4BROKERBIN} ($tag config) pids $p4broker_pids."
            kill $p4broker_pids 2>&1 | $GREP -v -i "No such"
            sleep 1
            p4broker_pids=$(get_pids "$P4BROKERCFG")
         done
         echo "${P4BROKERBIN} ($tag config) stopped."
         ExitStatus=0
      else
         echo "${P4BROKERBIN} ($tag config) was NOT running."
         ExitStatus=1
      fi
   ;;
   restart)
      $0 $SDP_INSTANCE stop $tag
      $0 $SDP_INSTANCE start $tag
   ;;
   *)
      echo -e "\nUsage: $0 SDP_INSTANCE {start|stop|status|restart} [<ConfigTag>|all]\n"
      ExitStatus=1
   ;;
esac

exit $ExitStatus