#!/usr/bin/env bash
#
# Initial script for Live operating system
# This script are launching before starting init from linux-live script.
# Current dir allways must be set to root (/)
# All system path must be relative, except initrd dirs

ENABLED=yes
[ "$ENABLED" != "yes" ] && exit 0
DEBUGMODE=no

PATH=.:/:/usr/bin:/usr/local/bin:/usr/local/sbin

[[ -d /usr/lib/ublinux ]] && { unset ROOTFS; unset CMD_CHROOT; } || { ROOTFS='.'; CMD_CHROOT='chroot . '; }
SOURCE=${ROOTFS}/usr/lib/ublinux/functions; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null || exit 0
SOURCE=${ROOTFS}/usr/lib/ublinux/os-config; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null || exit 0
debug_mode "$0" "$@"

SYSCONF="${ROOTFS}/${SYSCONF}"
SOURCE=${SYSCONF}/config; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null
SOURCE=${SYSCONF}/network; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null

    [[ -n $1 ]] && NTPSERVERS=$1
    [[ -n $2 ]] && ETH_INTERFACE="-$2"
    # List NTP servers: PART_SYSTEMDPATH/Service_Name Process_Name
    NTP_SERVICES_PROCESS="
        sysinit.target.wants/systemd-timesyncd.service systemd-timesyncd
        multi-user.target.wants/ntpd.service ntpd
        multi-user.target.wants/ntpdate.service ntpdate
        multi-user.target.wants/chronyd.service chronyd
        multi-user.target.wants/openntpd.service ntpd
        multi-user.target.wants/ptp4l.service ptp4l
        multi-user.target.wants/phc2sys.service phc2sys"

    ISSYSTEMD=$(readlink -fq ${ROOTFS}/usr/bin/init | grep "lib/systemd/systemd$")
    if [[ ${NTPSERVERS,,} == "stop" ]]; then
	# Для systemd
	if [[ -n ${ISSYSTEMD} ]]; then
	    while IFS=' ' read ITEM_NTP_SERVICE ITEM_NTP_PROCESS; do
		[[ -z ${ITEM_NTP_SERVICE} || -z ${ITEM_NTP_PROCESS} ]] && continue
		if [[ -n ${ROOTFS} ]]; then
		    ${CMD_CHROOT} systemctl --quiet disable ${ITEM_NTP_SERVICE##*/} &>/dev/null 
		else
		    systemctl --quiet disable --now ${ITEM_NTP_SERVICE##*/} &>/dev/null || pkill -9 ${ITEM_NTP_PROCESS}
		fi
	    done < <(tr -d '\t' <<< ${NTP_SERVICES_PROCESS})
	fi
    # Вызывая из NetworkManager скрипта, переменна ${NTPSERVERS} перезаписана на сервера NTP
    elif [[ -n ${NTPSERVERS} && ${NTPSERVERS,,} != "dhcp" ]]; then 
	[[ ${NTPSERVERS,,} == "default" ]] && NTPSERVERS=${NTPSERVERS_DEFAULT}
    	if [[ -n ${ISSYSTEMD} && ${NTP_SYSTEMSERVICE} == "systemd-timesyncd.service" && -f ${ROOTFS}/usr/lib/systemd/system/systemd-timesyncd.service ]]; then
    	    CONFIG_TIMESYNCD=${ROOTFS}/etc/systemd/timesyncd.conf.d/ulinux${ETH_INTERFACE}.conf
    	    mkdir -p ${ROOTFS}/etc/systemd/timesyncd.conf.d
    	    NTPSERVERS=$(tr ',;' ' ' <<< ${NTPSERVERS})
    	    cat <<-EOF > "${CONFIG_TIMESYNCD}"
[Time]
NTP=${NTPSERVERS}
FallbackNTP=${NTPSERVERS_FALLBACK}
EOF
	    if [[ -n ${ROOTFS} ]]; then
	        ${CMD_CHROOT} systemctl --quiet enable ${NTP_SYSTEMSERVICE} &>/dev/null 
	    else
	        systemctl --quiet enable --now ${NTP_SYSTEMSERVICE} &>/dev/null
	    fi
    	fi
    fi