#! /usr/bin/env bash

# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

### BEGIN INIT INFO
# Provides:		zookeeper
# Required-Start:	$remote_fs $syslog
# Required-Stop:	$remote_fs $syslog
# Default-Start:	2 3 4 5
# Default-Stop:		
# Short-Description:	Apache ZooKeeper server
### END INIT INFO

set -e

# /etc/init.d/zookeeper: start and stop the Apache ZooKeeper daemon

umask 022

. /usr/libexec/zkEnv.sh

. /lib/lsb/init-functions

ZOOPIDDIR=/var/lib/zookeeper/data
ZOOPIDFILE=${ZOOPIDDIR}/zookeeper_server.pid

check_privsep_dir() {
    # Create the PrivSep empty dir if necessary
    if [ ! -d ${ZOOPIDDIR} ]; then
	mkdir -p ${ZOOPIDDIR}
        chown zookeeper:hadoop ${ZOOPIDDIR}
	chmod 0775 ${ZOOPIDDIR} 
    fi
}

# Are we running from init?
run_by_init() {
    ([ "$previous" ] && [ "$runlevel" ]) || [ "$runlevel" = S ]
}

check_for_no_start() {
    # forget it if we're trying to start, and /etc/zookeeper/zookeeper_not_to_be_run exists
    if [ -e /etc/zookeeper/zookeeper_not_to_be_run ]; then 
	if [ "$1" = log_end_msg ]; then
	    log_end_msg 0
	fi
	if ! run_by_init; then
	    log_action_msg "Apache ZooKeeper server not in use (/etc/zookeeper/zookeeper_not_to_be_run)"
	fi
	exit 0
    fi
}

export PATH="${PATH:+$PATH:}/usr/sbin:/usr/bin"

case "$1" in
  start)
	check_for_no_start
	check_privsep_dir
	log_daemon_msg "Starting Apache ZooKeeper server" "zookeeper"
	if start-stop-daemon --start --quiet --oknodo --pidfile ${ZOOPIDFILE} -c zookeeper -x ${ZOOKEEPER_PREFIX}/sbin/zkServer.sh start; then
	    log_end_msg 0
	else
	    log_end_msg 1
	fi
	;;
  stop)
	log_daemon_msg "Stopping Apache ZooKeeper server" "zookeeper"
	if start-stop-daemon --stop --quiet --oknodo --pidfile ${ZOOPIDFILE}; then
	    log_end_msg 0
	else
	    log_end_msg 1
	fi
	;;

  restart)
	check_privsep_dir
	log_daemon_msg "Restarting Apache ZooKeeper server" "zookeeper"
	start-stop-daemon --stop --quiet --oknodo --retry 30 --pidfile ${ZOOPIDFILE}
	check_for_no_start log_end_msg
	if start-stop-daemon --start --quiet --oknodo --pidfile ${ZOOPIDFILE} -c zookeeper -x ${ZOOKEEPER_PREFIX}/sbin/zkServer.sh start; then
	    log_end_msg 0
	else
	    log_end_msg 1
	fi
	;;

  try-restart)
	check_privsep_dir
	log_daemon_msg "Restarting Apache ZooKeeper server" "zookeeper"
	set +e
	start-stop-daemon --stop --quiet --retry 30 --pidfile ${ZOOPIDFILE}
	RET="$?"
	set -e
	case $RET in
	    0)
		# old daemon stopped
		check_for_no_start log_end_msg
		if start-stop-daemon --start --quiet --oknodo --pidfile ${ZOOPIDFILE} -c zookeeper -x ${ZOOKEEPER_PREFIX}/sbin/zkServer.sh start; then
		    log_end_msg 0
		else
		    log_end_msg 1
		fi
		;;
	    1)
		# daemon not running
		log_progress_msg "(not running)"
		log_end_msg 0
		;;
	    *)
		# failed to stop
		log_progress_msg "(failed to stop)"
		log_end_msg 1
		;;
	esac
	;;

  status)
	status_of_proc -p ${ZOOPIDFILE} ${JAVA_HOME}/bin/java zookeeper && exit 0 || exit $?
	;;

  *)
	log_action_msg "Usage: /etc/init.d/zookeeper {start|stop|restart|try-restart|status}"
	exit 1
esac

exit 0
