#!/bin/bash
# Version: 0.3.1-proftpd
# Copyright (c) 2005-2026:
#   Darren 'Tadgy' Austin <darren (at) afterdark.org.uk>
# Licensed under the terms of the GNU General Public License version 3.

# shellcheck disable=SC2016,SC2034

PROFTPD_EXEC="/usr/sbin/proftpd"
PROFTPD_ARGS=()
PROFTPD_PIDFILE="/run/proftpd.pid"
DAEMON_EXEC="/usr/bin/daemon"
DAEMON_ARGS=('-N' '-n' "lumberjack-${0##*rc.}" '-r' '-a' '60' '-A' '5' '-L' '1800' '-M' '3' '-l' 'daemon.err' '-b' 'daemon.debug' '-o' 'daemon.info' '--')
LUMBERJACK_EXEC="/opt/bin/lumberjack"
LUMBERJACK_ARGS=('-u' '${LUMBERJACK_RUNUSER:-logger}' '-g' '${LUMBERJACK_RUNGROUP:-logging}' '-z' '-i' '${LUMBERJACK_SOCK:-/run/proftpd-log.sock}' '-o' '${LUMBERJACK_RUNUSER:-logger}:ftp' '-mp' '006' '-l' '{}/logs/ftpd-access.log' '${LUMBERJACK_BASEDIR:-/data/sites}' '{}/logs/%Y/%m/ftpd-access.log')
LUMBERJACK_SOCK="/run/proftpd-log.sock"
LUMBERJACK_BASEDIR="/data/sites"

# Allow configuration in /etc/default to override.
# Additional available variables:
#   PROFTPD_ENVIRONMENT=()	# Extra environment passed to $PROFTPD_EXEC.  Must be an array.
#   DAEMON_ENVIRONMENT=()	# Extra environment passed to $DAEMON_EXEC.  Must be an array.
#   LUMBERJACK_RUNUSER=""	# User name to run lumberjack as.  Default: logger.
#   LUMBERJACK_RUNGROUP=""	# Group name to run lumberjack as.  Default: logging.
#   LUMBERJACK_SOCK=""		# Location of the socket for lumberjack to read data.  Default: /run/proftpd-log.sock.
#   LUMBERJACK_BASEDIR=""	# The base directory where lumberjack should write logs.  Default: /data/sites.
#   SLAY_DELAY=""		# Delay between the SIGTERM and SIGKILL on a 'stop'.  Default: 2s.
#   RESTART_DELAY=""		# Delay between stopping and starting on a 'restart'.  Default: 2s.
# shellcheck disable=SC1090
[[ -e "/etc/default/${0##*rc.}" ]] && { source "/etc/default/${0##*rc.}" || return 1 2>/dev/null || exit 1; }

# Now all possible variable changes are complete, expand out the embedded variables.
eval "$(declare -p LUMBERJACK_ARGS | sed -re 's/\\\$/$/g')"

error() {
  printf "%s: %s\\n" "${BASH_SOURCE[0]##*/}" "$*" >&2
}

checkconfigured() {
  # This function can be used to perform any pre-start tests; hopfully to insure the daemon
  # can start correctly, before actually trying to start it.  A return value of 0 means the
  # tests were passed and the daemon should be started.  Any other value prevents the
  # daemon from being started, and an error message will be emitted.
  getent passwd "${LUMBERJACK_RUNUSER:-logger}" >/dev/null 2>&1 || return 1
  getent group "${LUMBERJACK_RUNGROUP:-logging}" >/dev/null 2>&1 || return 1

  [[ ! -e "${LUMBERJACK_BASEDIR:-/data/sites}" ]] && return 1
  [[ ! -d "${LUMBERJACK_BASEDIR:-/data/sites}" ]] && return 1
  ${PROFTPD_ENVIRONMENT:+declare ${PROFTPD_ENVIRONMENT[*]};} "$PROFTPD_EXEC" -t >/dev/null 2>&1 || return 1
  return 0
}

checkstatus() {
  # shellcheck disable=SC2155
  local RET=0 RUNPIDS="$({ pgrep -f "$PROFTPD_EXEC"; pgrep -F "$PROFTPD_PIDFILE" 2>/dev/null; } | sort -u )"
  if "$DAEMON_EXEC" --running -n "lumberjack-${0##*rc.}"; then
    printf "%s: %s: %s\\n" "${BASH_SOURCE[0]##*/}" "${LUMBERJACK_EXEC##*/}" "running"
  else
    printf "%s: %s: %s\\n" "${BASH_SOURCE[0]##*/}" "${LUMBERJACK_EXEC##*/}" "stopped"
    RET=1
  fi
  if [[ -n "$RUNPIDS" ]]; then
    printf "%s: %s: %s" "${BASH_SOURCE[0]##*/}" "${PROFTPD_EXEC##*/}" "running"
    if [[ -n "$PROFTPD_PIDFILE" ]]; then
      if [[ ! -e "$PROFTPD_PIDFILE" ]]; then
        printf "%s" ", but .pid file does not exist"
      elif ! grep "\<$(<"$PROFTPD_PIDFILE")\>" <<<"$RUNPIDS" >/dev/null 2>&1; then
        printf "%s" ", but .pid file is stale"
      fi
    fi
    printf "\\n"
  else
    printf "%s: %s: %s\\n" "${BASH_SOURCE[0]##*/}" "${PROFTPD_EXEC##*/}" "stopped"
    RET=$(( RET + 2))
  fi
  return $RET
}

startdaemon() {
  checkconfigured || {
    error "${PROFTPD_EXEC##*/}: not started - pre-start checks failed"
    return 2
  }
  # shellcheck disable=SC2048,SC2086
  ${DAEMON_ENVIRONMENT:+declare ${DAEMON_ENVIRONMENT[*]};} "$DAEMON_EXEC" ${DAEMON_ARGS[*]} "$LUMBERJACK_EXEC" ${LUMBERJACK_ARGS[*]}
  # shellcheck disable=SC2181
  (( $? != 0 )) && {
    error "error starting '${DAEMON_EXEC##*/}'"
    return 2
  }
  # shellcheck disable=SC2048,SC2086
  ${PROFTPD_ENVIRONMENT:+declare ${PROFTPD_ENVIRONMENT[*]};} "$PROFTPD_EXEC" ${PROFTPD_ARGS[*]}
  # shellcheck disable=SC2181
  if (( $? != 0 )); then
    error "error starting '${PROFTPD_EXEC##*/}'"
    return 2
  else
    return 0
  fi
}

stopdaemon() {
  kill -TERM "$(pgrep -f "$PROFTPD_EXEC" | tr $'\n' ' ')" >/dev/null 2>&1
  [[ -e "$PROFTPD_PIDFILE" ]] && {
    sleep 0.5
    kill -TERM "$(<"$PROFTPD_PIDFILE")" >/dev/null 2>&1
  }
  sleep "${SLAY_DELAY:-2}"
  checkstatus >/dev/null && {
    error "${PROFTPD_EXEC##*/}: failed to stop gracefully - slaying"
    kill -KILL "$({ cat "$PROFTPD_PIDFILE"; pgrep -f "$PROFTPD_EXEC"; } 2>/dev/null | sort -u | tr $'\n' ' ')" >/dev/null 2>&1
  }
  sleep 0.5
  "$DAEMON_EXEC" -n "lumberjack-${0##*rc.}" --stop 2>/dev/null
  sleep "${SLAY_DELAY:-2}"
  checkstatus >/dev/null && {
    error "${DAEMON_EXEC##*/}: failed to stop gracefully - slaying"
    "$DAEMON_EXEC" -n "lumberjack-${0##*rc.}" --signal=kill 2>/dev/null
  }
  return 0
}

for EXEC in "$PROFTPD_EXEC" "$DAEMON_EXEC" "$LUMBERJACK_EXEC"; do
  [[ ! -e "$EXEC" ]] && {
    error "$EXEC: not found"
    return 1 2>/dev/null || exit 1
  }
  [[ ! -x "$EXEC" ]] && {
    error "$EXEC: not executable"
    return 1 2>/dev/null || exit 1
  }
done

case "$1" in
  'start')
    checkstatus >/dev/null
    ERR=$?
    if (( ERR == 3 )); then
      startdaemon
      RET=$?
    elif (( ERR == 2 )); then
      error "${DAEMON_EXEC##*/}: already running"
      error "${PROFTPD_EXEC##*/}: stopped"
      printf "  %s\\n" "Try: ${BASH_SOURCE[0]} restart" >&2
      RET=1
    elif (( ERR == 1 )); then
      error "${DAEMON_EXEC##*/}: stopped"
      error "${PROFTPD_EXEC##*/}: already running"
      printf "  %s\\n" "Try: ${BASH_SOURCE[0]} restart" >&2
      RET=1
    else
      error "${DAEMON_EXEC##*/}: already running"
      error "${PROFTPD_EXEC##*/}: already running"
      printf "  %s\\n" "Try: ${BASH_SOURCE[0]} status" >&2
      RET=1
    fi
    ;;
  'stop')
    checkstatus >/dev/null
    if (( $? == 3 )); then
      error "${DAEMON_EXEC##*/}: not running"
      error "${PROFTPD_EXEC##*/}: not running"
      printf "  %s\\n" "Try: ${BASH_SOURCE[0]} status" >&2
      RET=1
    else
      stopdaemon
      RET=$?
    fi
    ;;
  'restart')
    checkstatus >/dev/null
    (( $? != 3 )) && {
      stopdaemon >/dev/null 2>&1
      sleep "${RESTART_DELAY:-2}"
    }
    startdaemon
    RET=$?
    ;;
  'status')
    checkstatus
    RET=$?
    ;;
  *)
    printf "%s\\n" "Usage: ${BASH_SOURCE[0]} <start|stop|restart|status>" >&2
    RET=1
    ;;
esac

return $RET 2>/dev/null || exit $RET
