#!/usr/bin/env bash
# TA-Proxmenu - Proxmox Setup Scripts for TA Use


[ "${2}" != "q" ] && source /opt/idssys/defaults/colors.inc
source /opt/idssys/defaults/default.inc
source /opt/idssys/ta-proxmenu/defaults.inc

INSTALL_PULSE() {
	echo
	curl -fsSL https://github.com/rcourtman/Pulse/releases/latest/download/install.sh | bash
	echo
	echo -e "\n${idsCL[Green]}Pulse has been installed${idsCL[Default]}"
	[ ${action-x} ] && exit 0 || ENTER2CONTINUE
}

INSTALL_ACRONIS() {
	read -n 1 -p "Are you sure you wish to install Acronis (Y/n)?" choice
	case "$choice" in
		[Nn]) MAIN_MENU;;
		* )
			echo
			cd /tmp
			wget "https://us5-cloud.acronis.com/bc/api/ams/links/agents/redirect?language=multi&channel=CURRENT&system=linux&architecture=64&productType=enterprise&login=010180ae-63c4-4495-bed0-4ec934c25af9&white_labeled=0" -O ./acronisinstall
			chmod +x ./acronisinstall
			./acronisinstall
			rm -f ./acronisinstall
			echo
			echo -e "\n${idsCL[Green]}Acronis has been installed${idsCL[Default]}"
			[ ${action-x} ] && exit 0 || ENTER2CONTINUE
		;;
	esac
}

INSTALL_PROXMENUX() {
#	read -n 1 -p "Are you sure you wish to install ProxMenux (Y/n)?" choice
#	case "$choice" in 
#	[Nn]) MAIN_MENU;;
#	* )
#		echo
		bash -c "$(wget -qLO - https://raw.githubusercontent.com/MacRimi/ProxMenux/main/install_proxmenux.sh)"
		systemctl disable --now proxmenux-monitor
		menu
#		echo -e "\n${idsCL[Green]}ProxMenux has been installed${idsCL[Default]}"
#		[ ${action-x} ] && exit 0 || ENTER2CONTINUE
#	esac
}

PROXMENUX_POST_INSTALL() {
	PMFLDR='/usr/local/share/proxmenux/scripts/post_install'
	[ ! -f ${PMFLDR}/customizable_post_install.sh ] && INSTALL_PROXMENUX
	bash ${PMFLDR}/customizable_post_install.sh
	
	touch /opt/.PROXMENUX_POST_INSTALL
	[ -s /etc/apt/sources.list ] && cat /dev/null > /etc/apt/sources.list
	
}

INSTALL_GLANCES() {
	read -n 1 -p "Are you sure you wish to install Glances (Y/n)?" choice
	case "$choice" in 
	[Nn]) echo;;
	* )
		echo
		apt install glances -y
		echo -e "\n${idsCL[Green]}Glances has been installed${idsCL[Default]}"
		[ ${action-x} ] && exit 0 || ENTER2CONTINUE
	esac
}

INSTALL_SCREENCONNECT() {
	read -n 1 -p "Are you sure you wish to install ScreenConnect (Y/n)?" choice
	case "$choice" in 
	[Nn]) echo;;
	* )
		echo
		echo -en "\n${idsCL[LightYellow]}Paste the URL provided from the Build Installer: ${idsCL[Default]}"
		read -e SCURL    
		wget "${SCURL}" -O /tmp/scinstall
		dpkg -i /tmp/scinstall
		apt install --fix-broken -y
		apt remove "connectwis*" -y
		dpkg -i /tmp/scinstall
		rm -f /tmp/scinstall
		systemctl disable --now proxmenux-monitor
		echo -e "\n${idsCL[Green]}ScreenConnect has been installed${idsCL[Default]}"
		[ ${action-x} ] && exit 0 || ENTER2CONTINUE
	esac
}

INSTALL_RMM() {
	read -n 1 -p "Are you sure you wish to install RMM (Y/n)?" choice
	case "$choice" in 
	[Nn]) echo;;
	* )
		echo
    
		echo -en "\n${idsCL[LightYellow]}Paste the Linux Server URL provided from the Download Agent screen: ${idsCL[Default]}"
		read -e RMMURL
		wget "${RMMURL}" -O /tmp/rmminstall
		TOKEN="$(echo ${RMMURL} | awk -F 'TKN' '{print $2}' | awk -F '/RUN' '{print $1}')"
		CMD="TOKEN=${TOKEN} bash /tmp/rmminstall"
		eval ${CMD}
		systemctl restart ITSPlatform
		# rm -f /tmp/rmminstall
    
		echo -e "\n${idsCL[Green]}RMM has been installed${idsCL[Default]}"
		[ ${action-x} ] && exit 0 || ENTER2CONTINUE
	esac
}

INSTALL_S1() {
	echo
	echo -en "${idsCL[LightYellow]}Paste the customers SentinelOne Site Token: ${idsCL[Default]}"
	read -e s1token
	cd /tmp
	wget "https://git.schroedercity.com/TAI/TA-ProxMenu/raw/branch/main/SentinelAgent_linux_x86_64_v25_3_2_11.deb"
	dpkg -i ./SentinelAgent_linux_x86_64*.deb
	/opt/sentinelone/bin/sentinelctl management token set ${s1token}
	/opt/sentinelone/bin/sentinelctl control start
	rm -f ./SentinelAgent_linux_x86_64*.deb
	
	echo -e "\n${idsCL[Green]}SentinelOne Agent has been installed. Make sure its added to a \"DETECT ONLY\" policy${idsCL[Default]}"
	[ ${action-x} ] && exit 0 || ENTER2CONTINUE
}

INSTALL_OMSA() {
	read -n 1 -p "Are you sure you wish to install Dell OpenManage Administrator (Y/n)?" choice
	case "$choice" in 
	[Nn]) echo;;
	* )
		echo
		mkdir /tmp/omsa
		cd /tmp/omsa
		apt install -y gnupg libcurl4t64 libncurses6 libxslt1.1 libgpm2 libtinfo6
		mkdir -p /etc/apt/keyrings
		wget -qO - https://linux.dell.com/repo/pgp_pubkeys/0x1285491434D8786F.asc | gpg --dearmor -o /etc/apt/keyrings/linux.dell.com.gpg
		chmod +r /etc/apt/keyrings/linux.dell.com.gpg
		echo "deb [signed-by=/etc/apt/keyrings/linux.dell.com.gpg] http://linux.dell.com/repo/community/openmanage/11000/jammy jammy main" > /etc/apt/sources.list.d/linux.dell.com.list
		apt update
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/o/openwsman/libwsman-curl-client-transport1_2.6.5-0ubuntu16_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/o/openwsman/libwsman-client4t64_2.6.5-0ubuntu16_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/o/openwsman/libwsman1t64_2.6.5-0ubuntu16_amd64.deb
		wget -c http://http.us.debian.org/debian/pool/main/libx/libxml2/libxml2-16_2.15.1+dfsg-2+b1_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/o/openwsman/libwsman-server1t64_2.6.5-0ubuntu16_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/s/sblim-sfcc/libcimcclient0_2.2.8-0ubuntu2_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/o/openwsman/openwsman_2.6.5-0ubuntu16_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/multiverse/c/cim-schema/cim-schema_2.48.0-0ubuntu1_all.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/s/sblim-sfc-common/libsfcutil0_1.0.1-0ubuntu4_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/multiverse/s/sblim-sfcb/sfcb_1.4.9-0ubuntu7_amd64.deb
		wget -c http://archive.ubuntu.com/ubuntu/pool/universe/s/sblim-cmpi-devel/libcmpicppimpl0_2.0.3-0ubuntu2_amd64.deb
		wget -c http://ftp.us.debian.org/debian/pool/main/o/openssl/libssl1.1_1.1.1w-0+deb11u1_amd64.deb
		dpkg -i libwsman-curl-client-transport1_2.6.5-0ubuntu16_amd64.deb
		dpkg -i libwsman-client4t64_2.6.5-0ubuntu16_amd64.deb
		dpkg -i libxml2-16_2.15.1+dfsg-2+b1_amd64.deb
		dpkg -i libwsman1t64_2.6.5-0ubuntu16_amd64.deb
		dpkg -i libwsman-server1t64_2.6.5-0ubuntu16_amd64.deb
		dpkg -i libcimcclient0_2.2.8-0ubuntu2_amd64.deb
		dpkg -i openwsman_2.6.5-0ubuntu16_amd64.deb
		dpkg -i cim-schema_2.48.0-0ubuntu1_all.deb
		dpkg -i libsfcutil0_1.0.1-0ubuntu4_amd64.deb
		dpkg -i sfcb_1.4.9-0ubuntu7_amd64.deb
		dpkg -i libcmpicppimpl0_2.0.3-0ubuntu2_amd64.deb
		dpkg -i libssl1.1_1.1.1w-0+deb11u1_amd64.deb
		apt install -y srvadmin-all
		/opt/dell/srvadmin/sbin/srvadmin-services.sh start
		rm -Rf /tmp/omsa
		
		echo -e "\n${idsCL[Green]}Dell OMSA has been installed${idsCL[Default]}"
		echo -e "\n${idsCL[LightCyan]}Available at: ${idsCL[LightGreen]}https://${RNIP}:1311${idsCL[Default]}"
		[ ${action-x} ] && exit 0 || ENTER2CONTINUE
	esac
}

DOWNLOAD_VIRTIO() {
	echo -e "\n${idsCL[LightCyan]}Current \"Stable\" version available for download: ${idsCL[White]}${VIRTIO_FILE}${idsCL[Default]}"
	if [ -f /var/lib/vz/template/iso/${VIRTIO_FILE} ]; then
		echo -en "\n${idsCL[LightRed]}Removing existing download ... "
		rm -f /var/lib/vz/template/iso/${VIRTIO_FILE}
		echo -e "${idsCL[Red]}Done${idsCL[Default]}"
	fi
	wget -q -F -P /var/lib/vz/template/iso ${VIRTIO_DOWNLOAD_URL} &
	echo -e "\n${idsCL[LightCyan]}Downloading will continue in the background\n"
  	[ ${action-x} ] && exit 0 || ENTER2CONTINUE
}

DETECT_CPU(){
  # if [ ! -f /etc/apt/sources.list.d/proxlb.list ]; then
  #   echo "deb https://repo.gyptazy.com/stable /" > /etc/apt/sources.list.d/proxlb.list
  #   wget -O /etc/apt/trusted.gpg.d/proxlb.asc https://repo.gyptazy.com/repository.gpg
  #   apt-get update
  # fi
	if [ ! -f /etc/apt/sources.list.d/gyptazy.list ]; then
		curl https://git.gyptazy.com/api/packages/gyptazy/debian/repository.key -o /etc/apt/keyrings/gyptazy.asc
		echo "deb [signed-by=/etc/apt/keyrings/gyptazy.asc] https://packages.gyptazy.com/api/packages/gyptazy/debian trixie main" | sudo tee -a /etc/apt/sources.list.d/gyptazy.list
		apt update
	fi
	if [ "$(dpkg -l | awk '/proxclmc/ {print }'|wc -l)" -eq 0 ]; then
	  apt -y install proxclmc
  fi

	echo
	proxclmc
	echo
	echo -en "${idsCL[LightCyan]}Would you like to set '${idsCL[LightGreen]}cpu: $(proxclmc --list-only)${idsCL[LightCyan]}' on all VMs (y/N)?${idsCL[Default]} "
	read -n 1 choice
	case "$choice" in 
		[Yy])
			sed -i "/cpu:/c cpu: $(proxclmc --list-only)" /etc/pve/nodes/*/qemu-server/*.conf
			echo
			echo -e "\n${idsCL[Green]}All VM's have been reconfigured\n${idsCL[LightCyan]}This will require the VM's to be powered off and then turned back on in order to take effect${idsCL[Default]}"
			[ ${action-x} ] && exit 0 || ENTER2CONTINUE
		;;
		*) echo;;
	esac
    
}

RESTART_PVE_SERVICES(){
	echo -en "${idsCL[LightCyan]}Would you like to restart all Proxmox services on the local host (Y/n)?${idsCL[Default]} "
	read -n 1 choice
	case "$choice" in
		[Nn]) echo;;
		*) echo
			echo -en "\n${idsCL[Yellow]}Restarting services ... "
			systemctl restart pve-cluster pvedaemon pvestatd pveproxy
			echo -e "${idsCL[Green]}Done${idsCL[Default]}"
			echo -e "\n${idsCL[Green]}This hosts Proxmox services have been restarted${idsCL[Default]}\n"
			[ ${action-x} ] && exit 0 || ENTER2CONTINUE
		;;
	esac
}

MAINTENANCE_MODE(){
	if ha-manager status | grep "maintenance mode" &> /dev/null; then
		echo -en "${idsCL[LightCyan]}Take the local host out of maintenance mode (Y/n)?${idsCL[Default]} "
	else
		echo -en "${idsCL[LightCyan]}Put the local host into maintenance mode (Y/n)?${idsCL[Default]} "
	fi
	read -n 1 choice
	case "$choice" in
		[Nn]) echo;;
		*) echo
			if ha-manager status | grep "maintenance mode" &> /dev/null; then
				ha-manager crm-command node-maintenance disable $(hostname) &
				echo -e "\n${idsCL[Green]}This host will be taken out of maintenance mode${idsCL[Default]}\n"
			else
				ha-manager crm-command node-maintenance enable $(hostname) &
				echo -e "\n${idsCL[Green]}This host will be entered into maintenance mode${idsCL[Default]}\n"
			fi
			[ ${action-x} ] && exit 0 || ENTER2CONTINUE
		;;
	esac
}

MAIN_MENU() {
  CRES=$(pvesh get /cluster/resources)
	while :
	do
		clear
		echo
		echo -e " ${idsCL[Green]}TA-Proxmenu - Proxmox Setup Scripts${idsCL[Default]}                           ${idsCL[Default]}${VERS}"
		echo -e "${idsCL[Green]}---------------------------------------------------------------------------${idsCL[Default]}"
		echo -e "Hostname:    ${idsCL[Cyan]}$(hostname -s)${idsCL[Default]}"
		echo -e "IP Address:  ${idsCL[Cyan]}${RNIP}${idsCL[Default]}"
		echo -e "---------------------------------------------------------------------------"
		echo
        
		echo -en "${idsCL[White]} [${idsCL[LightYellow]}0${idsCL[Default]}] ${idsCL[White]}Run Post-Install Script${idsCL[Default]}"
		[ -f /opt/.PROXMENUX_POST_INSTALL ] && echo -e "${idsCL[Cyan]} - Has been ran prevously${idsCL[Default]}" || echo
		echo -e "${idsCL[White]} [${idsCL[LightYellow]}1${idsCL[Default]}] ${idsCL[White]}Detect CPU-Arch for Live Migrations${idsCL[Default]}"
		if ! echo ${CRES} | grep -i pulse &> /dev/null ; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}2${idsCL[Default]}] ${idsCL[White]}Install Pulse Monitoring${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [2] Pulse Monitoring is already installed${idsCL[Default]}"
		fi
		if [ -f /var/lib/vz/template/iso/${VIRTIO_FILE} ]; then
			echo -e "${idsCL[DarkGray]} [3] Current VirtIO drivers already downloaded to 'local' on this host${idsCL[Default]}"
		elif [ -f /var/lib/vz/template/iso/virtio*.iso ]; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}3${idsCL[Default]}] ${idsCL[LightGreen]}**${idsCL[White]}Download the available updated Win-VirtIO drivers to 'local' on this host${idsCL[Default]}"
		else
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}3${idsCL[Default]}] ${idsCL[White]}Download the current Win-VirtIO drivers to 'local' on this host${idsCL[Default]}"
		fi
		if ! command -v glances &> /dev/null; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}4${idsCL[Default]}] ${idsCL[White]}Install Glances (CLI Monitor)${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [4] Glances is already installed${idsCL[Default]}"
		fi
		if [ "$(dpkg -l | awk '/srvadmin-all/ {print }'|wc -l)" -eq 0 ]; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}5${idsCL[Default]}] ${idsCL[White]}Install Dell OpenManage Server Administrator${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [5] Dell OMSA is already installed - ${idsCL[Cyan]}https://${RNIP}:1311"
		fi		
		echo
		if [ "$(systemctl is-active ITSPlatform)" != "active" ]; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}6${idsCL[Default]}] ${idsCL[White]}Install ConnectWise RMM Agent${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [6] ConnectWise RMM Agent is already installed${idsCL[Default]}"
		fi	
		if [ "$(dpkg -l | awk '/cyberprotect/ {print }'|wc -l)" -eq 0 ]; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}7${idsCL[Default]}] ${idsCL[White]}Install Acronis Backup Agent${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [7] Acronis Backup is already installed${idsCL[Default]}"
		fi
		if [ "$(dpkg -l | awk '/sentinelagent/ {print }'|wc -l)" -eq 0 ]; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}8${idsCL[Default]}] ${idsCL[White]}Install SentinelOne Agent${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [8] SentinelOne is already installed${idsCL[Default]}"
		fi
		echo
		if [ "$(systemctl is-active connectwise*)" != "active" ]; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}S${idsCL[Default]}] ${idsCL[White]}Install ScreenConnect Agent${idsCL[Default]}"
		else
			echo -e "${idsCL[DarkGray]} [S] ScreenConnect is already installed${idsCL[Default]}"
		fi
		echo
		if ha-manager status | grep "maintenance mode" &> /dev/null; then
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}M${idsCL[Default]}] ${idsCL[White]}Take Host out of Maintenance Mode${idsCL[Default]}"
		else
			echo -e "${idsCL[White]} [${idsCL[LightYellow]}M${idsCL[Default]}] ${idsCL[White]}Put Host into Maintenance Mode${idsCL[Default]}"
		fi
		echo -e "${idsCL[White]} [${idsCL[LightYellow]}R${idsCL[Default]}] ${idsCL[White]}Restart Proxmox Services${idsCL[Default]}"
    echo
		echo -e "${idsCL[White]} [${idsCL[LightYellow]}Q${idsCL[Default]}] ${idsCL[White]}Quit${idsCL[Default]}"
		echo
		echo
		echo -e -n "${idsCL[Yellow]}Enter ${idsCL[LightYellow]}option${idsCL[Yellow]} from above:${idsCL[Default]} "
		read -n 1 opt
		echo
		case $opt in
		  [0]) PROXMENUX_POST_INSTALL;;
		  [1]) DETECT_CPU;;
		  [2]) INSTALL_PULSE;;
		  [3]) DOWNLOAD_VIRTIO;;
		  [4]) INSTALL_GLANCES;;
		  [5]) INSTALL_OMSA;;
		  [6]) INSTALL_RMM;;
		  [7]) INSTALL_ACRONIS;;
		  [8]) INSTALL_S1;;
		  
		  [Ss]) INSTALL_SCREENCONNECT;;
		  [Mm]) MAINTENANCE_MODE;;
		  [Rr]) RESTART_PVE_SERVICES;;
		  [Qq]) EXIT1; exit 0;;
			 *)	echo -e "Thats an invaild option,\nplease select a valid option only."; sleep 1;;
		esac
	done
}


if [ ${action-x} ]; then
	case $action in
		pulse) INSTALL_PULSE;;
		rmm) INSTALL_RMM;;
		omsa) INSTALL_OMSA;;
		glances) INSTALL_GLANCES;;
		acronis) INSTALL_ACRONIS;;
		proxmenux) [ ! -f /usr/local/bin/menu ] && INSTALL_PROXMENUX || /usr/local/bin/menu;;
		screenconnect) INSTALL_SCREENCONNECT;;
		cpu) DETECT_CPU;;
    *) MAIN_MENU;;
	esac
else
	MAIN_MENU
fi

exit 0