#!/usr/bin/env bash

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

SOURCE=/usr/lib/ublinux/functions; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null || exit 0
SOURCE=/usr/lib/ublinux/default; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null || exit 0
debug_mode "$0" "$@"

#. $HARDCONF
#echo "$XDRIVER" | grep -q "nvidia" || rmmod nvidia
#STEAMNODPMS=yes

SOURCE=${SYSCONF}/config; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null
SOURCE=${SYSCONF}/video; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null
SOURCE=${SYSCONF}/desktop; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null
SOURCE=${SYSCONF}/keyboard; [[ -f ${SOURCE} ]] && . ${SOURCE} 2>/dev/null

    [[ ${DISPLAYMANAGER} == "none" ]] && exit

    [[ -n ${VGADRV_AUTO} || -n ${XORG_EXT} || -n ${XORG_MONITOR[@]} ]] && mkdir -p /etc/X11/xorg.conf.d
    rm -f /etc/X11/xorg.conf.d/*-*-ubconfig.conf

    # Auto VGA driver loader
    if [[ -n ${VGADRV_AUTO} ]]; then
	while read LSPCI_DEVICE; do
	    while read KERNEL_MODULE; do
		[[ ${KERNEL_MODULE} != "nouveau" ]] && modprobe ${KERNEL_MODULE} 2>/dev/null && break
	    done < <(lspci -ks ${LSPCI_DEVICE%% *} | grep 'Kernel modules:' | cut -d: -f2 | tr ',' '\n' | tac)
	done < <(lspci | grep -Ei 'vga|3d')
	if lsmod | grep nvidia &>/dev/null; then
	    #[[ -f /usr/bin/nvidia-xconfig ]] && nvidia-xconfig -s --no-logo --no-composite --no-dynamic-twinview --force-generate --output-xconfig=/etc/X11/xorg.conf
	    [[ -e /etc/X11/xorg.conf.d/20-nvidia-ubconfig.conf ]] || ln -sf /usr/lib/ublinux/xorg.conf.d/20-nvidia-ubconfig.conf /etc/X11/xorg.conf.d/
	    #[[ -e /etc/udev/rules.d/70-nvidia-ubconfig.rules ]] || ln -sf /usr/lib/ublinux/udev/70-nvidia-ubconfig.rules /etc/udev/rules.d/
	    /usr/bin/nvidia-modprobe -c0 -umsl
	    modprobe nvidia_drm
	    modprobe nvidia_uvm
	    modprobe nvidia_modeset
	    #modprobe drm_kms_helper
	    #modprobe drm
	    #echo -e nvidia_drm\\nvidia_uvm\\nnvidia_modeset > usr/lib/modules-load.d/nvidia-dkms.conf
	fi
#    else
	#rm -f /etc/udev/rules.d/70-nvidia-ubconfig.rules
    fi


    #[ ! -z "$VGAID" ] && grep -q $VGAID /usr/share/ublinux/hwdata/deleteconf && rm -f /etc/X11/xorg.conf
    [[ -n "${DELETE_XORG_CONF}" ]] && rm -f /etc/X11/xorg.conf

## Управление дополнительными возможностями XORG
    if [[ -n ${XORG_EXT} ]]; then
	while read I_EXT; do
	    XORG_FILE_EXT="/etc/X11/xorg.conf.d/110-${I_EXT}-ubconfig.conf"
	    if grep -Eiq "^nodpms$" <<< "${I_EXT}"; then
		cat > ${XORG_FILE_EXT} <<EOF
Section "Extensions"
    Option "DPMS" "Disable"
EndSection

Section "ServerFlags"
    Option "StandbyTime" "0"
    Option "SuspendTime" "0"
    Option "OffTime" "0"
    Option "BlankTime" "0"
EndSection
EOF
	    elif grep -Eiq "^dpms$" <<< "${I_EXT}"; then
		cat > ${XORG_FILE_EXT} <<EOF
Section "Extensions"
    Option "DPMS" "Enable"
EndSection
EOF
	    fi
	done <  <(tr ',' '\n' <<< "${XORG_EXT}")
    fi

## Parametrs:
## cvt | gtf | reduced|r | dpms | nodpms | enable | disable | ignore | primary |
## lo|LeftOf:{name} | ro|RightOf:{name} | ab|Above:{name} | be|Below:{name} |
## rotate:normal | rotate:left | rotate:right | rotate:invert
    grep -q xres= /proc/cmdline && XORG_MONITOR=$(cat /proc/cmdline | awk -F xres= '{print $2}' | awk '{print $1}')
    if [[ "${SETXORGRES}" != "no" ]] && [[ -n "${XORG_MONITOR[@]}" ]]; then
	XORG_FILE_MONITOR="/etc/X11/xorg.conf.d/120-monitor-ubconfig.conf"
	for VIDEO_PORT in "${!XORG_MONITOR[@]}"; do
	    SOFT_MODELINE="gtf"
	    SOFT_MODELINE_R=
	    PREFERRED_MODE=
	    XORG_STR_MONITOR=
	    while read VIDEO_PORT_PARAM; do
		if grep -Eiq "^[0-9]*x[0-9]*$|^[0-9]*x[0-9]*x[0-9]*$" <<< "${VIDEO_PORT_PARAM}"; then
		    if [[ -n ${SOFT_MODELINE_R} ]]; then
			XORG_STR_MONITOR+="$(cvt -r $(sed 's/\([0-9]* [0-9]*\) .*/\1/' <<< ${VIDEO_PORT_PARAM//x/ }))\n"
			SOFT_MODELINE_R=
		    elif [[ ${SOFT_MODELINE} == "cvt" ]]; then 
			XORG_STR_MONITOR+="$(cvt ${VIDEO_PORT_PARAM//x/ })\n"
		    else
			if grep -Eiq "^[0-9]*x[0-9]*$" <<< "${VIDEO_PORT_PARAM}"; then
			    XORG_STR_MONITOR+="$(gtf ${VIDEO_PORT_PARAM//x/ } 60 | grep -v '^\s*$')\n"
			else
			    XORG_STR_MONITOR+="$(gtf ${VIDEO_PORT_PARAM//x/ } | grep -v '^\s*$')\n"
			fi
		    fi
		elif grep -Eiq "^cvt$" <<< "${VIDEO_PORT_PARAM}"; then
		    SOFT_MODELINE="cvt"
		elif grep -Eiq "^gtf$" <<< "${VIDEO_PORT_PARAM}"; then
		    SOFT_MODELINE="gtf"
		elif grep -Eiq "^reduced|r$" <<< "${VIDEO_PORT_PARAM}"; then
		    SOFT_MODELINE_R="reduced"
		elif grep -Eiq "^nodpms$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+='  Option "DPMS" "false"\n'
		elif grep -Eiq "^dpms$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+='  Option "DPMS" "true"\n'
		elif grep -Eiq "^enable$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+='  Option "Enable" "true"\n'
		elif grep -Eiq "^disable$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+='  Option "Enable" "false"\n'
		elif grep -Eiq "^ignore$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+='  Option "Ignore" "true"\n'
		elif grep -Eiq "^primary$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+='  Option "Primary" "true"\n'
		elif grep -Eiq "^lo:[[:alnum:]]*$|^LeftOf:[[:alnum:]]*$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+="  Option \"LeftOf\" \"$(cut -d: -f2 <<< ${VIDEO_PORT_PARAM^^})\"\n"
		elif grep -Eiq "^ro:[[:alnum:]]*$|^RightOf:[[:alnum:]]*$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+="  Option \"RightOf\" \"$(cut -d: -f2 <<< ${VIDEO_PORT_PARAM^^})\"\n"
		elif grep -Eiq "^ab:[[:alnum:]]*$|^Above:[[:alnum:]]*$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+="  Option \"Above\" \"$(cut -d: -f2 <<< ${VIDEO_PORT_PARAM^^})\"\n"
		elif grep -Eiq "^be:[[:alnum:]]*$|^Below:[[:alnum:]]*$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+="  Option \"Below\" \"$(cut -d: -f2 <<< ${VIDEO_PORT_PARAM^^})\"\n"
		elif grep -Eiq "^rotate:[[:alpha:]]*$" <<< "${VIDEO_PORT_PARAM}"; then
		    XORG_STR_MONITOR+="  Option \"Rotate\" \"$(cut -d: -f2 <<< ${VIDEO_PORT_PARAM^^})\"\n"
		fi
		[[ -z ${PREFERRED_MODE} ]] && PREFERRED_MODE=$(grep Modeline <<< "${XORG_STR_MONITOR}" | cut -d'"' -f2)
	    done <  <(tr ',' '\n' <<< "${XORG_MONITOR[${VIDEO_PORT}]}")
	    if [[ -n ${XORG_STR_MONITOR} && ${VIDEO_PORT} != 0 ]]; then
	        XORG_STR+='Section "Monitor"\n'
	        XORG_STR+="  Identifier \"${VIDEO_PORT^^}\"\n"
		XORG_STR+="${XORG_STR_MONITOR}"
		[[ -n ${PREFERRED_MODE} ]] && XORG_STR+="  Option \"PreferredMode\" \"${PREFERRED_MODE}\"\n"
		XORG_STR+='EndSection\n\n'
	    fi
	done
	[[ -n ${XORG_STR} ]] && echo -e "${XORG_STR}" > "${XORG_FILE_MONITOR}"
    fi


    if [[ ${MKKBDFXORG} != "no" ]] ;then
	XORG_FILE_KEYBOARD="/etc/X11/xorg.conf.d/130-keyboard-ubconfig.conf"
	if ! grep -sqi XkbOptions /etc/X11/xorg.conf; then
	    cat > ${XORG_FILE_KEYBOARD} <<EOF
Section "InputClass"
        Identifier "system-keyboard"
        MatchIsKeyboard "on"
        Option "XkbLayout" "${XKBLAYOUT}"
        Option "XkbModel" "${XKBMODEL}"
        Option "XkbVariant" "${XKBVARIANT}"
        Option "XkbOptions" "${XKBOPTIONS}"
EndSection
EOF
	fi
    fi
