#! /bin/sh

# 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:		ambari-agent
# Required-Start:	$remote_fs $syslog
# Required-Stop:	$remote_fs $syslog
# Default-Start:	2 3 4 5
# Default-Stop:		
# Short-Description:	Apache HMS Agent
### END INIT INFO

set -e

# /etc/init.d/ambari-agent: start and stop the Apache HMS Agent daemon

umask 022

if test -f /etc/default/ambari-env.sh; then
    . /etc/default/ambari-env.sh
fi

. /lib/lsb/init-functions

# 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/ambari/ambari-agent_not_to_be_run exists
    if [ -e /etc/ambari/ambari-agent_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 HMS Agent not in use (/etc/ambari/ambari-agent_not_to_be_run)"
	fi
	exit 0
    fi
}

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

case "$1" in
  start)
	check_for_no_start
	log_daemon_msg "Starting Apache HMS Agent" "ambari-agent"
	if start-stop-daemon --start --quiet --oknodo --pidfile ${HMS_PID_DIR}/ambari-agent.pid -x /usr/bin/ambari-agent; then
	    log_end_msg 0
	else
	    log_end_msg 1
	fi
	;;
  stop)
	log_daemon_msg "Stopping Apache HMS Agent" "ambari-agent"
	if start-stop-daemon --stop --quiet --oknodo --pidfile ${HMS_PID_DIR}/ambari-agent.pid; then
	    log_end_msg 0
	else
	    log_end_msg 1
	fi
	;;

  restart)
	check_privsep_dir
	log_daemon_msg "Restarting Apache HMS Agent" "ambari-agent"
	start-stop-daemon --stop --quiet --oknodo --retry 30 --pidfile ${HMS_PID_DIR}/ambari-agent.pid
	check_for_no_start log_end_msg
	if start-stop-daemon --start --quiet --oknodo --pidfile ${HMS_PID_DIR}/ambari-agent.pid -x /usr/bin/ambari-agent; then
	    log_end_msg 0
	else
	    log_end_msg 1
	fi
	;;

  status)
	status_of_proc -p ${HMS_PID_DIR}/ambari-agent.pid /usr/bin/ambari-agent ambari-agent && exit 0 || exit $?
	;;

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

exit 0