#!/bin/sh

install_ems_on_ubuntu () {
     echo "#!/bin/sh
DIR=/opt/openbaton/ems
PROG="/usr/bin/python"
ARG="/usr/local/bin/openbaton-ems"
DAEMON_USER=root
PIDFILE=/var/run/ems.pid
. /lib/lsb/init-functions
do_start() {
        log_daemon_msg \"Starting EMS\"
        start-stop-daemon --start --background --pidfile \$PIDFILE --make-pidfile --user \$DAEMON_USER --chuid \$DAEMON_USER --exec \"\$PROG\" -- \$ARG
}
do_stop() {
        log_daemon_msg \"Stopping EMS\"
        start-stop-daemon --stop --pidfile \$PIDFILE
}
do_restart() {
        log_daemon_msg \"Restarting EMS\"
        do_stop
        do_start
}
    case \"\$1\" in
        "start" )
            do_start ;;
        "stop" )
            do_stop ;;
        "restart" )
            do_restart;;
        * )
            usage;;
    esac
" > /etc/init.d/openbaton-ems
     chmod +x /etc/init.d/openbaton-ems
     update-rc.d openbaton-ems defaults
}

install_ems_on_centos () {
    echo "# Open Baton EMS init script for the 'Systemd' System and Service Manager
# Copyright (c) 2016 Open Baton (http://www.openbaton.org)
#
# Licensed 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.
[Unit]
Description=Open Baton Element Management System (EMS) Systemd Unit
After=network.target
[Service]
Type=simple
User=root
ExecStart=/usr/bin/python /usr/bin/openbaton-ems
[Install]
WantedBy=runlevel2.target runlevel4.target multi-user.target" > /usr/lib/systemd/system/openbaton-ems.service

    systemctl daemon-reload
    systemctl enable openbaton-ems

}

if [ $(cat /etc/os-release | grep -i "ubuntu" | wc -l) -gt 0 ]; then
    os=ubuntu
elif [ $(cat /etc/os-release | grep -i "centos" | wc -l) -gt 0 ]; then
    os=centos
else
    os=undefined
fi

case ${os} in
    ubuntu)
	    install_ems_on_ubuntu
	    ;;
    centos)
        install_ems_on_centos
	    ;;
    *)
	    echo "OS not recognized"
	    exit 1
	    ;;
esac