#!/bin/sh

# Written by: Rich Greene
# Contains code from: Dylan Charpentier
# Contains code from: http://easyos.net/articles/bsd/freebsd/output_control_in_freebsd_console

# Reset
NC='\033[0m'       # Text Reset

# Regular Colors
Black='\033[0;30m'        # Black
Red='\033[0;31m'          # Red
Green='\033[0;32m'        # Green
Yellow='\033[0;33m'       # Yellow
Blue='\033[0;34m'         # Blue
Purple='\033[0;35m'       # Purple
Cyan='\033[0;36m'         # Cyan
White='\033[0;37m'        # White
COOL='\033[1;41m'         # COOL


clear;

echo -e "${White}This script will install KDE Plasma and a bunch of useful stuff"
echo -e "to make FreeBSD easy fun to use!${NC}"
echo ""
echo ""

# prompt user to proceed or not
echo -e "${Yellow}Would you like to proceed? Yes / No?${NC}"
while true; do
    read yn
    case $yn in
        [Yy]* ) break;;
        [Nn]* ) exit;;
        * ) echo "Please answer Y or N.";;
    esac
done

# set vmware sysrc entries to yes or no
echo ""
echo ""
echo -e "${Yellow}Are you installing this to a VMware virtual machine? Yes / No?${NC}"
while true; do
    read yn
    case $yn in
        [Yy]* ) VM="YES"; break;;
        [Nn]* ) VM="NO"; break;;
        * ) echo "Please answer Y or N.";;
    esac
done

# prompt for username of regular user
echo ""
echo ""
echo -e "${Yellow}Please type a username for a regular user."
echo -e "If it doesn't exist, we'll create it${NC}"
read -p 'Username: ' USERNAME

# Check if user exists, if so, continue, if not force you to create
USER_EXISTS=$(id -u $USERNAME > /dev/null 2>&1; echo $?)
while [ $USER_EXISTS -eq 1 ]
do
  echo -e "${Yellow}$USERNAME doesn't exist, Creating User: $USERNAME.${NC}"
  pw useradd -n $USERNAME -c "$USERNAME" -G wheel -s /bin/sh -m
  echo -e "${Yellow}Type new password for $USERNAME...${NC}"
  passwd $USERNAME
  USER_EXISTS=$(id -u $USERNAME > /dev/null 2>&1; echo $?)
done

# fetch and install FreeBSD updates
echo -e "${Yellow}Installing FreeBSD Updates${NC}"
env PAGER=cat freebsd-update fetch;
freebsd-update install;

# install package manager and install packages
echo -e "${Yellow}Setting up package manager and installing packages${NC}"
env ASSUME_ALWAYS_YES=yes pkg bootstrap;
env ASSUME_ALWAYS_YES=yes pkg install wget nano sudo htop kde5 hal dbus xorg xterm bash zsh sddm screenFetch octopkg chromium;

if [ "$VM" = "YES" ] ; then
  echo -e "${Yellow}Installing VM support packages and drivers${NC}"
  env ASSUME_ALWAYS_YES=yes pkg install open-vm-tools xf86-video-vmware xf86-input-vmmouse;
fi

# update rc.conf with required variables
echo -e "${Yellow}Setting rc.conf variables${NC}"
sysrc hald_enable="YES";
sysrc dbus_enable="YES";
sysrc sddm_enable="YES";

if [ "$VM" = "YES" ]; then
  echo -e "${Yellow}Setting rc.conf open-vm-tools variables${NC}"
  sysrc vmware_guest_vmblock_enable="$VM";
  sysrc vmware_guest_vmhgfs_enable="$VM";
  sysrc vmware_guest_vmmemctl_enable="$VM";
  sysrc vmware_guest_vmxnet_enable="$VM";
  sysrc vmware_guestd_enable="$VM";
fi

# change provided user's shell, add to wheel for sudo access, and set up empty .zshrc user file
echo -e "${Yellow}Changing $USERNAME's shell to zsh${NC}"
chsh -s /usr/local/bin/zsh $USERNAME;
echo -e "${Yellow}Adding $USERNAME to the wheel group${NC}"
pw usermod -n $USERNAME -G wheel;
echo -e "${Yellow}Creating $USERNAME's empty zshrc file${NC}"
cd /home/$USERNAME;
echo "#Empty file, settings inherited from /usr/local/etc/zshrc" > .zshrc;
chown $USERNAME:$USERNAME .zshrc;

#Download customized zshrc from grml and customize terminal further
echo -e "${Yellow}Downloading an customizing GRML's zshrc file${NC}"
cd /usr/local/etc;
wget -O zshrc https://git.grml.org/f/grml-etc-core/etc/zsh/zshrc;
echo "screenFetch" >> zshrc;
sed -i bak "s/alias ll='command ls -l/alias ll='command ls -lahF/" zshrc;
sed -i bak 's/alias ll="command ls -l/alias ll="command ls -lahF/' zshrc;

# allow wheel group to use sudo
echo -e "${Yellow}Giving wheel group sudo privileges"
sed 's/# %wheel ALL=(ALL) ALL/%wheel ALL=(ALL) ALL/' sudoers > sudoers.new;
chmod 640 sudoers;
mv sudoers sudoers.old;
mv sudoers.new sudoers;
chmod 440 sudoers;
chmod 440 sudoers.old;

echo ""
echo -e "${White}Installation Complete! Machine will reboot in"
echo -e "20 seconds for the changes to take effect!"
echo -e "Press Ctrl-C to restart later.${NC}"
echo ""
echo ""

# Start Countdown
progress_bar() {
  progress=$(( $1 * 30 / 20 ))
  bar=`printf "\33[41m% ${progress}s\33[m" ""`
  printf "\33[37D[%- 38s] %- 4s" "${bar}" "$1"
}
i=20
printf "Rebooting in 20 seconds: \33[37C"
while [ $i -ge 0 ];
do
  progress_bar $i
  i=$(( $i - 1 ))
  sleep 1
done
echo
# End Countdown

# reboot machine
reboot