#!/usr/bin/env bash
#
# Author: Dmitry Razumov <asmeron@ublinux.com>
# Copyright (c) 2021-2025 UBLinux <support@ublinux.com>
#
# Initial script for Linux UBLinux
# This script are launching before starting init from initrd 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 ]] && { ROOTFS= ; CMD_CHROOT= ; } || { [[ -d /sysroot ]] && ROOTFS="/sysroot" || ROOTFS="."; CMD_CHROOT="chroot ${ROOTFS}"; }
SOURCE=${ROOTFS}/usr/lib/ublinux/functions; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null || exit 0
SOURCE=${ROOTFS}/usr/lib/ublinux/default; [[ -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

    # 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"

    CONFIG_TIMESYNCD="${ROOTFS}/etc/systemd/timesyncd.conf.d/ubconfig.conf"

exec_ntp_servers_set(){
    restart_systemd_ntp(){
        if [[ -n ${ROOTFS} ]]; then
            ${CMD_CHROOT} /usr/bin/systemctl --quiet enable ${NTP_SYSTEMSERVICE} &>/dev/null
        else
	    /usr/bin/systemctl daemon-reload
            /usr/bin/systemctl --quiet enable ${NTP_SYSTEMSERVICE} &>/dev/null
            /usr/bin/systemctl --quiet restart ${NTP_SYSTEMSERVICE} &>/dev/null
        fi
    }
    ISSYSTEMD=$(readlink -fq ${ROOTFS}/usr/bin/init | grep "lib/systemd/systemd$")
    if [[ ${NTPSERVERS,,} == "dhcp" ]]; then
	# Вызывая из NetworkManager скрипта, переменную ${NTPSERVERS} перезаписать на сервера NTP из DHCP
	[[ -n $1 ]] && local NTPSERVERS="$1"
    fi
    if [[ ${NTPSERVERS,,} == "disable" ]]; then
    # Не настраивать сервер времени
	true
    elif [[ ${NTPSERVERS,,} == "stop" ]]; then
    # Остановить и отключить сервисы синхронизации времени
	exec_ntp_servers_stop
    elif [[ ${NTPSERVERS,,} == "dhcp" ]]; then
	[[ -n ${ISSYSTEMD} ]] && restart_systemd_ntp
    elif [[ -n ${NTPSERVERS} ]]; then
	[[ ${NTPSERVERS,,} == "default" ]] && NTPSERVERS=${NTPSERVERS_DEFAULT}
	[[ ${NTPSERVERS,,} == "ntp-ru" ]] && NTPSERVERS=${NTPSERVERS_RU}
	# Активная systemd и выбран сервис NTP_SYSTEMSERVICE=systemd-timesyncd.service
	if [[ -n ${ISSYSTEMD} && ${NTP_SYSTEMSERVICE} == "systemd-timesyncd.service" && -f ${ROOTFS}/usr/lib/systemd/system/${NTP_SYSTEMSERVICE} ]]; then
	    NTPSERVERS=$(tr ',;' ' ' <<< ${NTPSERVERS})
	    [[ -d ${CONFIG_TIMESYNCD%/*} ]] || mkdir -p ${CONFIG_TIMESYNCD%/*}
	    find ${CONFIG_TIMESYNCD%/*}/ -name "ubconfig*.conf" -delete 2>/dev/null
	    cat <<-EOF > "${CONFIG_TIMESYNCD}"
[Time]
NTP=${NTPSERVERS}
FallbackNTP=${NTPSERVERS_FALLBACK}
EOF
	fi
	[[ -n ${ISSYSTEMD} ]] && restart_systemd_ntp
    fi
}
exec_ntp_servers_stop(){
    # Определяем активную systemd
    ISSYSTEMD=$(readlink -fq ${ROOTFS}/usr/bin/init | grep "lib/systemd/systemd$")
    while IFS=' ' read ITEM_NTP_SERVICE ITEM_NTP_PROCESS; do
        [[ -z ${ITEM_NTP_SERVICE} || -z ${ITEM_NTP_PROCESS} ]] && continue
	if [[ -n ${ISSYSTEMD} && -n ${ROOTFS} ]]; then
	    ${CMD_CHROOT} /usr/bin/systemctl --quiet disable ${ITEM_NTP_SERVICE##*/} &>/dev/null
	elif [[ -n ${ISSYSTEMD} ]]; then
	    /usr/bin/systemctl --quiet is-enabled ${ITEM_NTP_SERVICE##*/} && /usr/bin/systemctl --quiet disable ${ITEM_NTP_SERVICE##*/} &>/dev/null
	    /usr/bin/systemctl --quiet is-active ${ITEM_NTP_SERVICE##*/} && { /usr/bin/systemctl --quiet stop ${ITEM_NTP_SERVICE##*/} &>/dev/null || pkill -f ${ITEM_NTP_PROCESS}; }
	    [[ ${NTP_SYSTEMSERVICE} == "systemd-timesyncd.service" ]] && find ${CONFIG_TIMESYNCD%/*}/ -name "ubconfig*.conf" -delete 2>/dev/null
	fi
    done < <(tr -d '\t' <<< ${NTP_SERVICES_PROCESS})
    true	# Для выполнения из ubconfig со статусом ОК
}

################
##### MAIN #####
################

    if [[ -z $@ || $1 == @("set"|"set="|"set+="|"set++=") ]]; then
	shift
	exec_ntp_servers_set $@
    elif [[ $1 == @("set-="|"set--="|"remove") ]]; then
	shift
	exec_ntp_servers_stop $@
	exec_ntp_servers_set $@
    fi