#!/bin/sh
#
# $FreeBSD: news/sabnzbdplus/files/sabnzbd.in 319436 2013-05-30 17:18:21Z swills $
#
# PROVIDE: sabnzbd
# KEYWORD: shutdown
#
# Add the following lines to /etc/rc.conf.local or /etc/rc.conf
# to enable this service:
#
# sabnzbd_enable (bool):        Set to NO by default.
#                       Set it to YES to enable it.
# sabnzbd_conf_dir:     Directory where sabnzbd configuration
#                       data is stored.
#                       Default: /usr/local/sabnzbd
# sabnzbd_user:         The user account sabnzbd daemon runs as what
#                       you want it to be. It uses '_sabnzbd' user by
#                       default. Do not sets it as empty or it will run
#                       as root.
# sabnzbd_group:        The group account sabnzbd daemon runs as what
#                       you want it to be. It uses '_sabnzbd' group by
#                       default. Do not sets it as empty or it will run
#                       as wheel.

. /etc/rc.subr

name=sabnzbd
rcvar=sabnzbd_enable
load_rc_config ${name}

: ${sabnzbd_enable:=NO}
: ${sabnzbd_user:=_sabnzbd}
: ${sabnzbd_group:=_sabnzbd}
: ${sabnzbd_conf_dir="/usr/local/sabnzbd"}

required_dirs=${sabnzbd_conf_dir}

start_cmd="${name}_start"
status_cmd="${name}_status"
stop_cmd="${name}_stop"
start_precmd="${name}_prestart"

sabnzbd_prestart()
{
	PATH=${PATH}:/usr/local/bin:/usr/local/sbin
	if [ ! -f "${required_dirs}" -a ! -d "${required_dirs}" -a ! -L "${required_dirs}" ]; then
		install -d -o ${sabnzbd_user} -g ${sabnzbd_group} ${required_dirs}
	fi
}

sabnzbd_start()
{
	if [ ! -f "${sabnzbd_pid}" ]; then
		su -m ${sabnzbd_user} -c "/usr/local/bin/SABnzbd.py --daemon -f ${sabnzbd_conf_dir}/sabnzbd.ini"
		echo "Starting ${name}."
	else
		GETPROCESSPID=`/bin/ps -auxw | /usr/bin/awk '/SABnzbd.py/ && !/awk/ && !/sh/ {print $2}'`
		PIDFROMFILE=`cat ${sabnzbd_pid}`
		if [ "$GETPROCESSPID" = "$PIDFROMFILE" ]; then
			echo "${name} already running with PID: ${PIDFROMFILE} ?"
			echo "Remove ${sabnzbd_pid} manually if needed."
		else
			rm -f ${sabnzbd_pid}
			su -m ${sabnzbd_user} -c "/usr/local/bin/SABnzbd.py --daemon -f ${sabnzbd_conf_dir}/sabnzbd.ini"
			echo "Starting ${name}."
		fi
	fi
}

# SABnzbd can only be cleanly stopped by calling the http api
sabnzbd_stop()
{
	echo "Stopping $name"
	if [ -f "${sabnzbd_conf_dir}/sabnzbd.ini" ]; then
		apikey=`grep ^api_key ${sabnzbd_conf_dir}/sabnzbd.ini | tr -d " _"`
		host=`grep -m1 -E '^host\ =\ [0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}' ${sabnzbd_conf_dir}/sabnzbd.ini | tr -dc '[0-9].'`
		if [ ${host} = "0.0.0.0" ] ; then 
			host="localhost" ;
		fi
		port=`grep -m1 ^port ${sabnzbd_conf_dir}/sabnzbd.ini | tr -dc '[0-9]'`
		fetch -o /dev/null "http://${host}:${port}/api?mode=shutdown&${apikey}" > /dev/null 2>&1
	else
		sabnzbd_pid=`ps -U ${sabnzbd_user} | grep "python.*SABnzbd.py.*--daemon" | grep -v 'grep' | awk '{print $1}'`
	        if [ -n "${sabnzbd_pid}" ]; then
			kill ${sabnzbd_pid}
		fi
	fi
}

sabnzbd_status()
{
	sabnzbd_pid=`ps -U ${sabnzbd_user} | grep "python.*SABnzbd.py.*--daemon" | grep -v 'grep' | awk '{print $1}'`
	if [ -n "${sabnzbd_pid}" ]; then
		echo "$name is running as ${sabnzbd_pid}"
	else
		echo "$name is not running"
	fi
}

run_rc_command "$1"
