#!/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 Proxy base init script

# Do nothing unless $SDP_INSTANCE is defined.
export SDP_INSTANCE="${1:-Unset}"
export OS_INIT_MECHANISM="${OS_INIT_MECHANISM:-Unset}"
declare -i StartDelay="${SDP_START_DELAY:-2}"
declare -i ExitCode=0
declare -i UseSystemd=0
declare ServiceName=
declare OpMode="${2:-usage}"

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

# Load SDP controlled shell environment.
# shellcheck disable=SC1091
source /p4/common/bin/p4_vars "$SDP_INSTANCE"
ExitCode=$?

if [[ "$ExitCode" -ne 0 ]]; then
   echo -e "\\nError: Failed to load SDP environment for instance $SDP_INSTANCE.\n"
   exit 1
fi

# shellcheck disable=SC1090
source "$P4CBIN/ps_functions.sh"
ExitCode=$?

if [[ "$ExitCode" -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 "$P4PBIN" ]]; then
   echo -e "\\nError: $P4PBIN is not executable."
   exit 2;
fi

# Add and override values from p4_vars
export P4LOG=$LOGS/p4p.log

# Use '-u $P4SERVICEUSER' if required.
export P4P_FLAGS="-d -v proxy.monitor.level=$PROXY_MON_LEVEL -p $PROXY_PORT -r $P4PCACHE -t $PROXY_TARGET -e 3072 -L $P4LOG"

# Determine if systemd is configured for this instance.
if [[ -n "$(command -v systemctl)" ]]; then
   ServiceName="${P4PBIN##*/}"
   if systemctl cat "$ServiceName" > /dev/null 2>&1; then
      UseSystemd=1
      MustUseSystemdMsg="\\nError: It appears an attempt was made to start or stop the $ServiceName\\nservice without using systemd. When a systemd unit file is configured, only the\\nsystemd init mechanism can be used to start and stop the service, so that\\nsystemd maintains a more reliable indication of the status of the service. Run\\nas root or as $OSUSER with sudo. For example:\\n\\n\\tsudo systemctl start $ServiceName\\n\\tsudo systemctl stop $ServiceName\\n\\tsystemctl status $ServiceName\\n\\nNote that the 'status' check does not require sudo.\\n\\nThe systemd unit file is:\\n\\n$(systemctl cat "$ServiceName")"
   fi
fi

# If running with systemd, we drop the '-d' flag and let systemd handle the
# process forking.
if [[ "$UseSystemd" -eq 1 ]]; then
   export P4P_FLAGS="${P4P_FLAGS/-d / }"
fi

case "$OpMode" in
   (start)
      if [[ "$UseSystemd" -eq 1 && "$OS_INIT_MECHANISM" != "systemd" ]]; then
         echo -e "$MustUseSystemdMsg"
         exit 1
      fi

      # Delay start $StartDelay seconds.
      sleep "$StartDelay"

      echo "Starting ${P4PBIN} ${P4P_FLAGS}"
      # shellcheck disable=SC2086
      "${P4PBIN}" ${P4P_FLAGS}
   ;;

   (status)
      p4p_pids=$(get_pids "$P4PBIN")
      if [[ -n "$p4p_pids" ]]; then
         echo "${P4PBIN} is running as pids: $p4p_pids."
         exit 0
      else
         echo "${P4PBIN} is NOT running."
         exit 1
      fi
   ;;

   (stop)
      p4p_pids=$(get_pids "$P4PBIN")

      if [[ -n "$p4p_pids" ]]; then
         if [[ "$UseSystemd" -eq 1 && "$OS_INIT_MECHANISM" != "systemd" ]]; then
            echo -e "$MustUseSystemdMsg"
            exit 1
         fi

         while [[ -n "$p4p_pids" ]]; do
            echo "Shutting down ${P4PBIN} pids: $p4p_pids."
            # shellcheck disable=SC2086
            kill $p4p_pids 2>&1 | $GREP -v -i "No such"
            sleep 1
            p4p_pids=$(get_pids "$P4PBIN")
         done
         echo "${P4PBIN} stopped."
         exit 0
      else
         echo "${P4PBIN} was NOT running."
         exit 1
      fi
   ;;
   (restart)
      $0 "$SDP_INSTANCE" stop
      $0 "$SDP_INSTANCE" start
   ;;
   (*)
      echo -e "\\nUsage: $0 SDP_INSTANCE {start|stop|status|restart}\n"
      exit 1
   ;;
esac