#!/bin/bash
#
# Copyright 2013 Red Hat, Inc.
# All Rights Reserved.
#
# 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.

set -eu

usage() {
    echo "Usage: os-svc-restart -n SERVICENAME"
    echo ""
    echo "  -a             Use alternate svc-map instead of map-services"
    echo "  -h             Show help and exit"
    echo "  -n SERVICENAME Name of job/service file."
    echo ""
    exit $1
}

SERVICENAME=${SERVICENAME:-""}
MAPPING_COMMAND=map-services

nshift=0
while getopts "hn:" opt; do
    case "$opt" in
        n) SERVICENAME=$OPTARG;;
        h) usage 0;;
        a) MAPPING_COMMAND=svc-map;;
        \?) usage 0;;
        :) usage 1;;
    esac
done

shift $(($OPTIND-1))
if [ -z "$SERVICENAME" ] ; then
    usage 1
fi

function restart_upstart_service {
    local name="$1"
    service "$name" restart
}

function restart_systemd_service {
    local name="$1"
    systemctl restart "$($MAPPING_COMMAND "$name").service"
}

# TODO: SysV init fallback support
DIB_INIT_SYSTEM=$(dib-init-system)
if [ "$DIB_INIT_SYSTEM" == "upstart" ]; then
    restart_upstart_service "$SERVICENAME"
elif [ "$DIB_INIT_SYSTEM" == "systemd" ]; then
    restart_systemd_service "$SERVICENAME"
fi
