#!/bin/sh
# Nick Bernstein https://github.com/nicholasbernstein/install-fbsd-desktop
# most of this comes from the freebsd handbook 5.4.1. Quick Start x-config
set -o pipefail
#set -e
set -x
PS4="$0 $LINENO >"

LOGFILE="installx.log"
ERRLOG="installx.err"
exec 2>"$ERRLOG"
date > "$LOGFILE"

grep -q "kern.vty" /boot/loader.conf || echo "kern.vty=vt" >> /boot/loader.conf

change_pkg_url_to_latest () {
	sed -i 'orig' 's/quarterly/latest/' /etc/pkg/FreeBSD.conf
	grep url /etc/pkg/FreeBSD.conf
}

load_card_readers () {
	# Load MMC/SD card-reader support
	sysrc kld_list+="mmc"
	sysrc kld_list+="mmcsd"
	sysrc kld_list+="sdhci"
}

load_atapi() {
	# Access ATAPI devices through the CAM subsystem
	sysrc kld_list+="atapicam"
}

load_fuse() {
	# Filesystems in Userspace
	fuse_pkgs="fuse fuse-utils"
	extra_pkgs="$extra_pkgs fusefs-lkl e2fsprogs"
	sysrc kld_list+="fusefs"
}

load_coretemp(){
	# Intel Core thermal sensors
	sysrc kld_list+="coretemp"
}

load_amdtemp() {
	# AMD K8, K10, K11 thermal sensors
	if ( sysctl -a | grep -q -i "hw.model" | grep -q AMD ) ; then 
		amdtemp_load="YES"
	fi
}

load_bluetooth() { 
	# most common bluetooth adapters use this
	sysrc kld_list+="ng_ubt"
	sysrc hcsecd_enable="YES"
	sysrc sdpd_enable="YES"
}

enable_ipfw_firewall() {
	# this enables the ipfw firewall with the workstation profile
	# it allows communication w/ other hosts on the network, outgoing traffic
	# and any specific ports (ssh) we choose to enable
	sysrc firewall_type="WORKSTATION"
	sysrc firewall_myservices="22/tcp"
	sysrc firewall_allowservices="any"
	sysrc firewall_enable="YES"
}

enable_tmpfs() {
	# In-memory filesystems
	sysrc kld_list+="tmpfs"
}

enable_async_io() {
	# Asynchronous I/O
	sysrc kld_list+="aio"
}

enable_workstation_pwr_mgmnt() {
	# powerd: hiadaptive speed while on AC power, adaptive while on battery power
	sysrc powerd_enable="YES"
	sysrc powerd_flags="-a hiadaptive -b adaptive"
}

enable_webcam(){
	#this just enables the ability to use webcams
	extra_pkgs="$extra_pkgs webcamd"
	sysrc kld_list+="cuse4bsd"
	sysrc webcamd_enable="YES"
}

enable_cups() {
	#this just enables the ability to use printers
	extra_pkgs="$extra_pkgs cups"
	sysrc cupsd_enable="YES"
}


linuxBaseC7 () {
		sysrc kld_list+="linux"
		kldstat | grep -q linux || kldload linux
		sysrc kld_list+="linux64"
		kldstat | grep -q linux64 || kldload linux64
		sysrc linux_enable="YES"

		mkdir -p /compat/linux/proc /compat/linux/dev/shm /compat/linux/sys
		grep "/compat/linux/proc" /etc/fstab 2>/dev/null || \
			echo "linprocfs   /compat/linux/proc  linprocfs rw 0 0" >> /etc/fstab
		grep "/compat/linux/sys" /etc/fstab 2>/dev/null || \
			echo "linsysfs    /compat/linux/sys   linsysfs  rw 0 0" >> /etc/fstab
		grep "/compat/linux/dev" /etc/fstab 2>/dev/null || \
			echo "tmpfs    /compat/linux/dev/shm  tmpfs rw,mode=1777 0 0" >> /etc/fstab
}

virtualbox-ose-additions() {
		sysrc vboxguest_enable="YES"
		sysrc vboxservice_enable="YES"
		dialog --infobox "Please use VBoxSVGA as the virtualbox display driver for best performance." 0 0
}

adjust_sysctl_buffers() { 
	sysctl net.local.stream.recvspace=65536
	grep "net.local.stream.recvspace" /etc/sysctl.conf || echo "net.local.stream.recvspace=65536" >> /etc/sysctl.conf
	sysctl net.local.stream.sendspace=65536
	grep "net.local.stream.sendspace" /etc/sysctl.conf || echo "net.local.stream.sendspace=65536" >> /etc/sysctl.conf
}

report(){
	# $1 - testname, $2 - $?
        STATUS="OK"
        if [ "$2" -eq 1 ] ; then
                STATUS="FAILED"
        fi
        echo "$1: $STATUS" | tee -a $LOGFILE
}

# this is mainly just to make sure pkg has been bootstrapped
export ASSUME_ALWAYS_YES=yes
pkg update | tee -a "$LOGFILE"
report "pkg bootstrapping" "$?"

# this will help with performance of desktop applications
# and may help perf during install, so I'm doing it early
adjust_sysctl_buffers
report "add sync buffers" "$?"

add_user_to_video() {
	# Your user needs to be in the video group to use video acceleration
	default_user=`grep 1001 /etc/passwd | awk -F: '{ print $1 }'`
	VUSER=`dialog --title "Video User" --clear  --inputbox "What user should be added to the video group?" 0 0  $default_user --stdout`

	pw groupmod video -m $VUSER && echo "added $VUSER to group: video"
	report "add $VUSER to video group" "$?"
	pw groupmod wheel -m $VUSER && echo "added $VUSER to group: wheel" 
	report "add $VUSER to wheel group" "$?"

	# probably not necessary, logging into an x session as root isn't recommended.
	pw groupmod video -m root
	report "add root to wheel group" "$?"
}

add_user_to_video

gen_xinit() {
	# the following creates a .xinitrc file in the user's home directory that will launch
	# the installed windowmanager as well as allow the slim display manager to pass it as
	# an argument. 
	if [ ! $1 ] ; then 
		echo "argument needed by gen_xinit" 
		return 0 
	else
		xinittxt="#!/bin/sh\n mywm="$1"\n if [ \$1 ] ; then\n \tcase \$1 in \n \t\tdefault) exec \$mywm ;;\n \t\t*) exec \$1 ;;\n \tesac\n else\n \texec \$mywm\n fi"
		#echo -e $xinittxt > /home/$VUSER/.xinitrc && chown $VUSER:$VUSER /home/$VUSER/.xinitrc
		echo -e $xinittxt > /home/$VUSER/.xinitrc && chown $VUSER /home/$VUSER/.xinitrc
		test -d /etc/skel || mkdir /etc/skel
		echo -e $xinittxt > /etc/skel/.xinitrc
	fi
}


# pick our desktop environment. SDDM is going to be used as the login 
# manager instead of slim since it "works out of the box" w/o .xinitrc stuff

set_login_mgr() { 
	mywm="sddm"
	if ( uname -r | grep -q  "11" ) ; then
		mywm="slim"
		slim_extra_pkgs="slim-freebsd-dark-theme"
		pwd_mkdb -p /etc/master.passwd
	fi
	echo "login manager: $mywm"
	sysrc "${mywm}_enable"="YES"
}
set -x
set_login_mgr
report "set login manager" "$?"
set +x

dialog --title "Rolling Release" --yesno "Change pkg to use 'latest' packages instead of quarterly? Recommended for workstations. This prevents potential missing firefox package in 13.1 quarterly" 0 0

rolling=$?

if [ "$rolling" -eq 0  ] ; then 
	change_pkg_url_to_latest
	report "quarterly->latest changed" "$?"
fi


desktop=$(dialog --clear --title "Select Desktop" \
        --menu "Select desktop environment to be installed:" 0 0 0 \
        "KDE"  "KDE (FBSD 12+ only)" \
        "LXDE"  "The Lightweight X Desktop Environment" \
		"LXQT" "Lightweight QT Desktop (FBSD 12+ only)" \
        "GNOME" "The modern GNOME desktop" \
        "Xfce4" "Lightweight XFCE desktop" \
        "WindowMaker" "Bringing neXt back" \
        "awesome" "A tiling window manager" \
        "MATE"  "The MATE dekstop, a fork of GNOME 2" --stdout)

# for any additional entries, please add a case statement below

case $desktop in
  KDE)
      gen_xinit "startkde"
      DESKTOP_PGKS="kde5 plasma5-plasma plasma5-plasma-disks plasma5-plasma-systemmonitor kde-baseapps ${mywm}" 
      sysrc ${mywm}_enable="YES"
      ;;
  windowmaker)
      gen_xinit "/usr/local/bin/wmaker"
      DESKTOP_PGKS="windowmaker wmakerconf ${mywm}" 
      sysrc ${mywm}_enable="YES"
cat <<EOT >/usr/local/share/xsessions/wmaker.desktop
[Desktop Entry]
Encoding=UTF-8
Name=Windowmaker
Comment=Windowmaker Desktop Environment
Exec=/usr/local/bin/wmaker
Icon=
Type=Application
EOT
      ;;
  LXQT)
      gen_xinit "startlxqt"
      DESKTOP_PGKS="lxqt ${mywm}" 
      sysrc ${mywm}_enable="YES"
      ;;
  LXDE)
      gen_xinit "startlxde"
      DESKTOP_PGKS="lxde-meta lxde-common ${mywm}" 
      sysrc ${mywm}_enable="YES"
      ;;
  GNOME)
      gen_xinit "gnome-session"
      DESKTOP_PGKS="gnome" 
      sysrc gnome_enable="YES"
      sysrc gdm_enable="YES"
      sysrc ${mywm}_enable="NO"
      ;;
  xfce4)
      gen_xinit "startxfce4"
      DESKTOP_PGKS="xfce xfce4-goodies ${mywm}" 
      sysrc ${mywm}_enable="YES"
      ;;
  MATE)
      gen_xinit "mate-session"
      echo $desktop
      DESKTOP_PGKS="mate ${mywm}" 
      sysrc ${mywm}_enable="YES"
      ;;
  awesome)
      gen_xinit "awesome"
      DESKTOP_PGKS="awesome ${mywm}" 
      sysrc ${mywm}_enable="YES"
      ;;
  *)
     echo "$desktop isn't a valid option."
     ;;
esac

report "Desktop Selected" "$?"
# The following are generally needed by most modern desktops

sysrc dbus_enable="YES"
#sysrc hald_enable="YES"

report "DBus Enabled" "$?"
grep "proc /proc procfs" /etc/fstab || echo "proc /proc procfs rw 0 0" >> /etc/fstab

# A number of the more lightweight desktops don't include everything you need
# and anyone coming from linux probably wants bash, sudo & vim. Let's make
# the transition easy for them
extra_pkgs=$(dialog --checklist "Select additional packages to install:" 0 0 0 \
firefox "Firefox Web browser" on \
bash "GNU Bourne-Again SHell" on \
vim "VI Improved" on \
git-lite "Lightweight Git client" on \
sudo "Superuser do" on \
thunderbird "Thunderbird Email Client" off \
obs-studio "OBS-Studio recording/casting" off \
audacity "Audio editor" off \
simplescreenrecorder "Does it need a description?" off \
libreoffice "Open source office suite" off \
vlc "Video player" off \
doas "Simpler alternative to sudo" off \
linux_base-c7 "CentOS v7 linux binary compatiblity layer" off \
virtualbox-ose-additions "VirtualBox guest additions" off \
--stdout)

echo "Extra packages:" "$extra_packages" | tee -a "$LOGFILE"


# by default install the full xorg, but if xorg_minimal is set, override it

echo $extra_pkgs | grep -q linux_base-c7 && linuxBaseC7
echo $extra_pkgs | grep -q virtualbox-ose-additions && virtualbox-ose-additions

# Honestly, shouldn't graphic card configuration be done in the base installer? 
# Even if X isn't enabled, the right drivers should be selected and installed.
# Lets handle the 4 major cases, and hope for the best

dialog --title "Graphics Drivers" --yesno "Would you like to try to install the drivers for your video card?\n\nPlease refer to freebsd handbook for more details:\nhttps://www.freebsd.org/doc/handbook/x-config.html" 0 0
install_dv_drivers=$?

if [ "$install_dv_drivers" -eq 0  ] ; then 

	card=$(dialog --checklist "Select additional packages to install:" 0 0 0 \
	i915kms "most Intel graphics cards" off \
	radeonkms "most OLDER Radeon graphics cards" off \
	amdgpu "most NEWER AMD graphics cards" off \
	nvidia "NVidia Graphics Cards" off \
	vesa 	"Generic driver that may work as a fallback" off \
	scfb 	"Another Generic diver for UEFI and ARM" off \
	other "Anything but the above" off \
	--stdout)

	case $card in
		i915kms) 
			vc_pkgs="drm-kmod"
			sysrc kld_list+="/boot/modules/i915kms.ko"
			;;
		radeonkms) 
			vc_pkgs="drm-kmod xf86-video-ati"
			sysrc kld_list+="/boot/modules/radeonkms.ko"
			;;
		amdgpu) 
			vc_pkgs="drm-kmod xf86-video-amdgpu"
			sysrc kld_list+="amdgpu"
			;;
		nvidia) 
			vc_pkgs="nvidia-driver nvidia-xconfig nvidia-settings"
			nvidia-xconfig
			sysrc kld_list+="nvidia-modeset nvidia"
			;;
		vesa)
			vc_pkgs="xf86-video-vesa"
			;;
		scfb)
			vc_pkgs="xf86-video-scfb"
			;;
		*)
			pciconf=$(pciconf -vl | grep -B3 display)
			dialog --msgbox "You'll need to check the FreeBSD handbook or forums. The following output may be helpful in finding a driber: pciconf -vl | grep -B3 display: $pciconf" 0 0
			;;
	esac

fi 

# check to see if we should set the user shell to bash
if ( echo $all_pkgs | grep -q "bash" ) ; then
	dialog --title "Bash" --yesno "Would you like to set the $VUSER user's default shell to bash?" --stdout 0 0
	bash_yes=$?
fi

# check to see if we should allow %wheel to sudo
if ( echo $all_pkgs | grep -q "sudo" ) ; then
	dialog --title "sudo" --yesno "Would you like to make sudo act like the default behavior on linux?\n(wheel group can sudo)" --stdout 0 0
	sudo_yes=$?
fi

# This is opt activities
opt_activities=$(dialog --checklist "Select additional options" 0 0 0 \
	load_card_readers "enable card readers like sd cards" on \
	load_atapi "enable atapi to enable external storage devices like cds" on \
	load_fuse "enable userspace fileystems" on \
	load_coretemp "enable cpu temp sensors for intel (and amd)" off \
	load_amdtemp "enable additional amd temp sensors" off \
	enable_tmpfs "enable in-mem tempfs" on \
	enable_cups "printing" off \
	enable_webcam "enables webcams to be used" off \
	enable_ipfw_firewall "enables workstation firewall profile + allow ssh" off \
	enable_async_io "enable async io for better perf" on \
	enable_workstation_pwr_mgmnt "change pwr on battery/plugged in" on \
	load_bluetooth "enable bluetooth kernel modules" off \
	minimal_xorg "only install minimal xorg packages" off \
	--stdout )

#
# this comment is just to draw attention to
# the fact that this line is doing the package installs
# and making it easy to find by having a big comment block
# above it
#
xorg_minimal="xorg-minimal xauth xorg-libraries xorg-fonts xorg-fonts-truetype xf86-input-keyboard xf86-input-libinput xf86-input-mouse"
xorg_pkgs="xorg"

echo $opt_activities | grep -q load_card_readers && load_card_readers
echo $opt_activities | grep -q load_atapi && load_card_readers
echo $opt_activities | grep -q load_fuse && load_fuse
echo $opt_activities | grep -q load_coretemp && load_card_readers
echo $opt_activities | grep -q load_amdtemp && load_card_readers
echo $opt_activities | grep -q load_bluetooth && load_card_readers
echo $opt_activities | grep -q enable_ipfw_firewall && enable_ipfw_firewall
echo $opt_activities | grep -q enable_tmpfs && enable_tmpfs
echo $opt_activities | grep -q enable_async_io && enable_async_io
echo $opt_activities | grep -q enable_workstation_pwr_mgmnt && enable_workstation_pwr_mgmnt
echo $opt_activities | grep -q load_bluetooth && load_bluetooth
echo $opt_activities | grep -q enable_cups && enable_cups
echo $opt_activities | grep -q enable_webcam && enable_webcam
echo $opt_activities | grep -q minimal_xorg && xorg_pkgs=$xorg_minimal

# this is referred to during the package install, but needs to be up here so we can ask the user things.
all_pkgs="$xorg_pkgs dbus $DESKTOP_PGKS $extra_pkgs $vc_pkgs $mywm $slim_extra_pkgs"
echo "pkg install -y $all_pkgs" | tee -a "$LOGFILE"
pkg install -y $all_pkgs | tee -a "$LOGFILE"
report "package installation: " "$?"

# post install stuff
if [ "slim" = $mywm ] ; then
	sed -i '' -E 's/^current_theme.+$/current_theme		slim-freebsd-dark-theme/' /usr/local/etc/slim.conf
	report "slim.conf dark theme" "$?"
fi

# make sudo behave like default linux setup
if [ "$sudo_yes" -eq 0 ] ; then
	test -e /usr/local/etc/sudoers && echo "%wheel ALL=(ALL) ALL" >> /usr/local/etc/sudoers
	report "created sudoers for wheel" "$?"
fi

# on 11.x w/ mate re-installing fixed a core-dump
if [ $desktop = "mate" ] ; then 
	if ( echo $(uname -r) | grep -q "11" ) ; then 
		pkg install -f gsettings-desktop-schemas
	fi
fi

# Set the user's shell to bash
if [ "$bash_yes" -eq 0 ] ; then
	chpass -s /usr/local/bin/bash $VUSER || echo "failed to change shell to bash"
fi



welcome="Thanks for trying this setup script. If you're new to FreeBSD, it's worth noting that instead of trying to search google for how to do something, you probably want to check the handbook on freebsd.org or read the built-in man pages. \n\n Doing a 'man -k <topic>' will search for any matching documentation, and unlike some, ahem, other *nix operating systems, FreeBSD's built in documentation is really good.\n\n"
dialog --msgbox "$welcome Hopefully that worked. You'll probably want to reboot at this point. Please report any problems to http://bug.freebsddesktop.xyz/ or see the installx.log file created" 0 0