#!/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 ThisScript=${0##*/}
declare Version=5.0.3
declare CmdLine="$0 $*"
declare -i StartDelay="${SDP_START_DELAY:-2}"
declare -i ExitCode=0
declare -i UseSystemd=0
declare ServiceName=
declare OpMode="${2:-usage}"
declare ProxyServerArgs=

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 disable=SC1091
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

# Determine if systemd is configured for this instance.
if [[ -n "$(command -v systemctl)" ]]; then
   ServiceName="${P4PBIN##*/}"
   if [[ -n "$(systemctl is-enabled "$ServiceName" 2>/dev/null)" ]]; 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<<<<<<<<\\n$(sudo systemctl cat "$ServiceName")\\n>>>>>>>\\n\\nIf you used 'systemctl' and still got this message, then please check if OS_INIT_MECHANISM is defined\\nin the above file. See /p4/sdp/Server/Unix/setup/systemd/p4d_1.service for an example."
   fi
fi

#------------------------------------------------------------------------------
# Override P4LOG value from p4_vars to be suitable for a proxy server.
export P4LOG="$LOGS/p4p.log"

# Build ProxyServerArgs, with some conditional logic. Start with required flags.
ProxyServerArgs+="-p $PROXY_PORT -t $PROXY_TARGET -r $P4PCACHE -L $P4LOG"

# Add user-specified '-v' and other options defined as PROXY_V_FLAGS in the
# Instance Config file.
if [[ -n "${PROXY_V_FLAGS:-}" ]]; then
   ProxyServerArgs+=" ${PROXY_V_FLAGS}"
fi

# Ensure proxy.monitor.level is set to a reasonable default if not
# explicitly defined in the environment.
[[ "${ProxyServerArgs:-}" =~ proxy.monitor.level ]] || \
   ProxyServerArgs+=" -v proxy.monitor.level=${PROXY_MON_LEVEL:-3}"

# Use the '-d' option to do process forking unless running with systemd,
# in which case we do not use '-d' and let systemd handle the process
# forking.
[[ "$UseSystemd" -eq 1 ]] || ProxyServerArgs+=" -d"

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"

      if [[ "$PROXY_TARGET" =~ ^ssl[46]*: ]]; then
         # Quietly attempt a trust of the target server. This might fail if the
         # target server is not reachable, but it doesn't hurt to try.
         echo "${P4BIN}" -p "$PROXY_TARGET" trust -f -y
         "${P4BIN}" -p "$PROXY_TARGET" trust -f -y > /dev/null 2>&1
      fi

      # shellcheck disable=SC2164
      cd "$P4PCACHE"
      if [[ -n "$(command -v ulimit)" && "$(ulimit -c)" != "0" ]]; then
         ulimit -c unlimited
      fi

      echo -e "\\n$(date) Called $ThisScript v$Version with command line:\\n$CmdLine"
      echo "Starting ${P4PBIN} ${ProxyServerArgs}"
      # shellcheck disable=SC2086
      "${P4PBIN}" ${ProxyServerArgs}
   ;;

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

         # Quietly attempt a trust of the local proxy port.
         if [[ "$PROXY_PORT" =~ ^ssl[46]*: ]]; then
            "${P4BIN}" -p "$PROXY_PORT" trust -f -y > /dev/null 2>&1
         fi

         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