elastalert

#! /bin/sh
### BEGIN INIT INFO
# Provides:          elastalert 
# Required-Start:    $remote_fs $syslog
# Required-Stop:     $remote_fs $syslog
# Default-Start:     2 3 4 5
# Default-Stop:      0 1 6
# Short-Description: ElastAlert service
# Description:       ElastAlert run at system startup
### END INIT INFO

# Author: fabian lee
#
# Do NOT "set -e"

# PATH should only include /usr/* if it runs after the mountnfs.sh script
PATH=/sbin:/usr/sbin:/bin:/usr/bin
DESC="ElastAlert service"
NAME=elastalert
EA_DIR=/tmp/elastalert
DAEMON=$EA_DIR/bin/$NAME
LOG_DIR=/var/log/$NAME
EA_USER=elastalert
EA_GROUP=$EA_USER

CONFIG_NAME=config.yaml
# if RULE_OPTS is empty, then rules_folder from config is used
RULE_OPTS=""
#RULE_OPTS="--rule $EA_DIR/cpu_high.yaml"
DAEMON_ARGS="--config $EA_DIR/$CONFIG_NAME $RULE_OPTS --debug --verbose"

PID_DIR="/var/run/$NAME"
PID_FILE="$PID_DIR/$NAME.pid"
SCRIPTNAME=/etc/init.d/$NAME

# Exit if elastalert is not installed in expected location
if [ ! -d "$EA_DIR" ]; then
  echo "$EA_DIR folder does not exist, check EA_DIR variable in script"
  exit 3
fi
if [ ! -f "$DAEMON" ]; then
  echo "$DAEMON executable does not exist, was expecting $EA_DIR/bin/$NAME"
  exit 3
fi

# Read configuration variable file if it is present
[ -r /etc/default/$NAME ] && . /etc/default/$NAME

# Load the VERBOSE setting and other rcS variables
. /lib/init/vars.sh

# Define LSB log_* functions.
# Depend on lsb-base (>= 3.2-14) to ensure that this file is present
# and status_of_proc is working.
. /lib/lsb/init-functions

#
# Function that starts the daemon/service
#
do_start()
{

	log_daemon_msg "Starting $DESC"

        pid=`pidofproc -p $PID_FILE $NAME`
        if [ -n "$pid" ] ; then
                log_daemon_msg "Already running."
                log_end_msg 0
                exit 0
        fi

        # Prepare environment
        mkdir -p "$LOG_DIR" "$PID_DIR" && chown "$EA_USER":"$EA_GROUP" "$LOG_DIR" "$PID_DIR"
	touch -f "$LOG_DIR/$NAME.log" && chown "$EA_USER":"$EA_GROUP" "$LOG_DIR/$NAME.log"

	# Return
	#   0 if daemon has been started
	#   1 if daemon was already running
	#   2 if daemon could not be started
	start-stop-daemon -d $EA_DIR --background --no-close --start --user $EA_USER -c $EA_USER -m --pidfile $PID_FILE --exec $DAEMON -- $DAEMON_ARGS >> $LOG_DIR/$NAME.log 2>&1


	return=$?
	log_daemon_msg "return was $return (0=started,1=already running,2=error)"
	# Add code here, if necessary, that waits for the process to be ready
	# to handle requests from services started subsequently which depend
	# on this one.  As a last resort, sleep for some time.
	log_end_msg $return
	exit $return
}

#
# Function that stops the daemon/service
#
do_stop()
{
	log_daemon_msg "Stopping $DESC"

        if [ -f "$PID_FILE" ]; then
                start-stop-daemon --stop --pidfile "$PID_FILE" \
                        --user "$EA_USER" \
                        --quiet \
                        --retry forever/TERM/20 > /dev/null
                if [ $? -eq 1 ]; then
                        log_daemon_msg "$DESC is not running but pid file exists, cleaning up"
                elif [ $? -eq 3 ]; then
                        PID="`cat $PID_FILE`"
                        log_daemon_msg "Failed to stop $DESC (pid $PID)"
                        exit 1
                fi
                rm -f "$PID_FILE"
        else
                log_daemon_msg "(not running)"
        fi

        log_end_msg 0
}

#
# Function that sends a SIGHUP to the daemon/service
#
do_reload() {
	#
	# If the daemon can reload its configuration without
	# restarting (for example, when it is sent a SIGHUP),
	# then implement that here.
	#
	start-stop-daemon --stop --signal 1 --quiet --pidfile $PIDFILE --name $NAME
	return 0
}

case "$1" in
  start)
	[ "$VERBOSE" != no ] && log_daemon_msg "Starting $DESC" "$NAME"
	do_start
	case "$?" in
		0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
		2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
	esac
	;;
  stop)
	[ "$VERBOSE" != no ] && log_daemon_msg "Stopping $DESC" "$NAME"
	do_stop
	case "$?" in
		0|1) [ "$VERBOSE" != no ] && log_end_msg 0 ;;
		2) [ "$VERBOSE" != no ] && log_end_msg 1 ;;
	esac
	;;
  status)
	status_of_proc "$DAEMON" "$NAME" && exit 0 || exit $?
	;;
  #reload|force-reload)
	#
	# If do_reload() is not implemented then leave this commented out
	# and leave 'force-reload' as an alias for 'restart'.
	#
	#log_daemon_msg "Reloading $DESC" "$NAME"
	#do_reload
	#log_end_msg $?
	#;;
  restart|force-reload)
	#
	# If the "reload" option is implemented then remove the
	# 'force-reload' alias
	#
	log_daemon_msg "Restarting $DESC" "$NAME"
	do_stop
	case "$?" in
	  0|1)
		do_start
		case "$?" in
			0) log_end_msg 0 ;;
			1) log_end_msg 1 ;; # Old process is still running
			*) log_end_msg 1 ;; # Failed to start
		esac
		;;
	  *)
		# Failed to stop
		log_end_msg 1
		;;
	esac
	;;
  *)
	echo "Usage: $SCRIPTNAME {start|stop|status|restart|force-reload}" >&2
	exit 3
	;;
esac

: