#!/bin/sh -e
#
#  Copyright (C) 2015 ScyllaDB

print_usage() {
    echo "scylla_ntp_setup --subdomain centos"
    echo "  --subdomain specify subdomain of pool.ntp.org (ex: centos, fedora or amazon)"
    exit 1
}

while [ $# -gt 0 ]; do
    case "$1" in
        "--subdomain")
            DOMAIN="$2"
            shift 2
            ;;
        *)
            print_usage
            ;;
    esac
done

. /etc/os-release
if [ "$NAME" = "Ubuntu" ]; then
    apt-get install -y ntp ntpdate
    service ntp stop
    ntpdate `cat /etc/ntp.conf |grep "^server"|head -n1|awk '{print $2}'`
    service ntp start
else
    yum install -y ntp ntpdate || true
    if [ "$DOMAIN" != "" ]; then
        sed -e "s#\..*\.pool\.ntp\.org#.$DOMAIN.pool.ntp.org#" /etc/ntp.conf > /tmp/ntp.conf
        mv /tmp/ntp.conf /etc/ntp.conf
    fi
    if [ "`systemctl is-active ntpd`" = "active" ]; then
        systemctl stop ntpd.service
    fi
    ntpdate `cat /etc/ntp.conf |grep "^server"|head -n1|awk '{print $2}'`
    systemctl enable ntpd.service
    systemctl start ntpd.service
fi
