#!/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

unset ROOTFS; [[ -d /usr/lib/ublinux ]] || ROOTFS=.

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}/system; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null
SOURCE=${SYSCONF}/network; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null

exec_hostname(){
    local PARAM="$@"
    [[ -n ${PARAM} ]] && eval "${PARAM%%=*}='${PARAM#*=}'"
    SET_HOSTNAME="${HOSTNAME}"
    [[ -z ${SET_HOSTNAME} ]] && SET_HOSTNAME=ublinux-live

    # Удалить предыдущее имя хоста
    OLD_HOSTNAME=$(tail -1 ${ROOTFS}/etc/hostname); OLD_HOSTNAME=${OLD_HOSTNAME%%.*}
    sed -E -e "s/\s${OLD_HOSTNAME}(\s|$)/ /g" -e "s/\s${OLD_HOSTNAME}\.[-0-9a-zA-Z.]*(\s|$)/ /g" -e "/^\s*[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\s*$/d" -i ${ROOTFS}/etc/hosts

    # Если нет новой строки, то вставить
    tail -c1 -- ${ROOTFS}/etc/hosts | grep -qx $'\n' || echo >> ${ROOTFS}/etc/hosts

    HOSTS_1="127.0.0.1	sb-ssl.l.google.com safebrowsing.clients.google.com safebrowsing.cache.l.google.com"
    grep -q "^\s*${HOSTS_1}\s*$" ${ROOTFS}/etc/hosts || echo "${HOSTS_1}" >> ${ROOTFS}/etc/hosts
    HOSTS_2="127.0.0.1	${SET_HOSTNAME%%.*}"
    grep -q "^\s*${HOSTS_2}\s*$" ${ROOTFS}/etc/hosts || echo "${HOSTS_2}" >> ${ROOTFS}/etc/hosts

    [[ ${SET_HOSTNAME} != ${SET_HOSTNAME#*.} ]] && DOMAIN="${SET_HOSTNAME#*.}"
    if [[ ${DOMAIN} != "" ]]; then
	SET_HOSTNAME="${SET_HOSTNAME%%.*}.${DOMAIN}"
	echo "${SET_HOSTNAME}" > ${ROOTFS}/etc/hostname
	HOSTS_3="127.0.0.1	${SET_HOSTNAME}"
	grep -q "^\s*${HOSTS_3}\s*$" ${ROOTFS}/etc/hosts || echo "${HOSTS_3}" >> ${ROOTFS}/etc/hosts

	## Указать доменных пользователей AccountsService, что-бы были видны для выбора логина в DM
	ls -d ${ROOTFS}/home/*@*/ 2>/dev/null | xargs -ri basename {} | xargs -ri touch ${ROOTFS}/var/lib/AccountsService/users/{}

	## Домен для сервера dhcpd
	if [[ -f ${ROOTFS}/etc/dhcpd.conf ]]; then
	    sed s/"option domain-name .*"/"option domain-name \"${DOMAIN}\";"/g -i ${ROOTFS}/etc/dhcpd.conf
	    sed s/"option nis-domain .*"/"option nis-domain \"${DOMAIN}\";"/g -i ${ROOTFS}/etc/dhcpd.conf
	fi

	## Имя хоста записать в конфиг с постфиксом домена
	[[ -f ${ROOTFS}/etc/ublinux/config ]] && { grep -q "^\s*HOSTNAME=${SET_HOSTNAME}\s*$" ${ROOTFS}/etc/ublinux/config || sed "s/^\s*HOSTNAME=.*/HOSTNAME=${SET_HOSTNAME}/g" -i ${ROOTFS}/etc/ublinux/config; }
	[[ -f ${ROOTFS}/etc/ublinux/system ]] && { grep -q "^\s*HOSTNAME=${SET_HOSTNAME}\s*$" ${ROOTFS}/etc/ublinux/system || sed "s/^\s*HOSTNAME=.*/HOSTNAME=${SET_HOSTNAME}/g" -i ${ROOTFS}/etc/ublinux/system; }
    else
	echo "${SET_HOSTNAME}" > ${ROOTFS}/etc/hostname
    fi
}

exec_hostname_live(){
    # Если выполнение в initrd, то выход
    [[ -z ${ROOTFS} ]] || return 0
    local PARAM="$@"
    [[ -n ${PARAM} ]] && eval "${PARAM%%=*}='${PARAM#*=}'"
    SET_HOSTNAME="${HOSTNAME}"
    [[ -z ${SET_HOSTNAME} ]] && SET_HOSTNAME=ublinux-live
    [[ ${DOMAIN} != "" ]] && SET_HOSTNAME="${SET_HOSTNAME%%.*}.${DOMAIN}"
    hostnamectl set-hostname ${SET_HOSTNAME}

    ## Если меняется имя хоста в запущенных X, то новое имя добавляем в xauth
    who | grep "(:[0-9.]*)$" | cut -d' ' -f1 | xargs -ri su {} -c "xauth list | sed 's|^.*/|su {} -c \\\\\"xauth add ${SET_HOSTNAME}/|;s|$|\\\\\"|'" | xargs -ri sh -c '{}'

    # Если указан домен, то задаём переменную DOMAIN=
    # Примечание: Имя сервера контроллера домена включает постфикс домена, но в домен не входит
    #[[ ${SET_HOSTNAME} != ${SET_HOSTNAME#*.} ]] && DOMAIN="${SET_HOSTNAME#*.}"
    #[[ ${DOMAIN} != "" ]] && ubconfig -q set network DOMAIN="${DOMAIN}"
}

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

    if [[ -z $@ || $1 == set ]]; then
	shift
	KERNEL_HOSTNAME=$(cmdline_value hostname)
	[[ ${KERNEL_HOSTNAME} == "" ]] || HOSTNAME=${KERNEL_HOSTNAME}
	exec_hostname $@
	exec_hostname_live $@
    elif [[ $1 == remove ]]; then
	shift
	exec_hostname $@
	exec_hostname_live $@
    fi
