post-install

#!/usr/bin/env bash

#	enable periodic SSD trim (once a week)
#	sudo systemctl start fstrim.timer

# Check if user is root / sudo and if true, exit. "yay" install needs to run as normal user.
if [[ $(id -u) = 0 ]]
	then printf "\nPlease run as non-root user\n\n"
	exit 1
fi

# Bold colour
BRed='\033[1;31m'         # Red
BGreen='\033[1;32m'       # Green
BYellow='\033[1;33m'      # Yellow

# Colour Reset
Colour_Off='\033[0m'       # Text Reset

### Password required for SUDO
printf "\n>>>   The non-root user password is required for SUDO\n\n"
while true; do
	read -r -s -p "Please enter the user password for [${USER}]: " user_password1
	printf "\n"
	read -r -s -p "Confirm the user password for [${USER}]: " user_password
	printf "\n"

	[ "${user_password1}" = "${user_password}" ] && break
	printf "\n%bPasswords DO NOT match%b\nPlease try again...\n\n" "${BRed}" "${Colour_Off}"
done

printf "\n\n-------------------------------------\n"
printf "\n>>>   This sets the country to be used by the [reflector] script\n"
printf ">>>   Reflector is a Python script which can retrieve the most up-to-date package mirrors\n"
printf ">>>   See: https://github.com/BashTux1/arch-auto-install/blob/master/README.md#reflector-country-list\n"
printf "\nEnter Reflector Country [Australia]: "
read -r reflector_country
reflector_country=${reflector_country:-Australia}

### This asks if you want to install UFW and sets the variable accordingly
printf "\n-------------------------------------\n"
printf "\n>>>   %bOptional:%b [UFW] (Uncomplicated Firewall) with some default config\n" "${BYellow}" "${Colour_Off}"
printf ">>>   Allow SSH (Port 22)\n"
while true; do
	printf "\n>>>   Would you like to install UFW [Y]: "
	read -r ufw
	ufw=${ufw:-Y}

	case $ufw in
		[yY][eE][sS]|[yY])
			break
		;;
		[nN][oO]|[nN])
			break
		;;
		*)
			echo ">>>   Invalid input... Valid entries are E.g. [Y / N] or [y / n] or [Yes / No]"
		;;
	esac
done

### This asks if you want to install Yay and sets the variable accordingly
printf "\n-------------------------------------\n"
printf "\n>>>   %bOptional:%b [yay] {Yet another yogurt}, Pacman wrapper and AUR helper \n" "${BYellow}" "${Colour_Off}"
while true; do
	printf "\n>>>   Would you like to install Yay [Y]: "
	read -r yay
	yay=${yay:-Y}

	case $yay in
		[yY][eE][sS]|[yY])
			break
		;;
		[nN][oO]|[nN])
			break
		;;
		*)
			echo ">>>   Invalid input... Valid entries are E.g. [Y / N] or [y / n] or [Yes / No]"
		;;
	esac
done

### This asks if you want to install ZSH and sets the variable accordingly
printf "\n-------------------------------------\n"
printf "\n>>>   %bOptional:%b [Zsh] with [Oh My Zsh] and apply some customisations\n" "${BYellow}" "${Colour_Off}"
while true; do
	printf "\n>>>   Would you like to install Zsh [Y]: "
	read -r zsh
	zsh=${zsh:-Y}

	case $zsh in
		[yY][eE][sS]|[yY])
			break
		;;
		[nN][oO]|[nN])
			break
		;;
		*)
			echo ">>>   Invalid input... Valid entries are E.g. [Y / N] or [y / n] or [Yes / No]"
		;;
	esac
done

printf "\n\n\n######   Syncing repos and updating packages   ######\n"
echo "${user_password}" | sudo -S pacman -Syu --noconfirm

printf "######   Installing reflector and Applying Custom Mirrors   ######\n"
sudo pacman -S --noconfirm reflector

sudo sed -i 's/^--/# --/g' /etc/xdg/reflector/reflector.conf
sudo tee -a /etc/xdg/reflector/reflector.conf << EOF

##### Config Added via post-install.sh #####

--save /etc/pacman.d/mirrorlist
--country $reflector_country
--protocol https
--latest 10
--sort rate
EOF

sudo systemctl enable reflector.timer
sudo systemctl start reflector.service

### If answered yes, install UFW, else do not install
if [[ "$ufw" =~ ^([yY][eE][sS]|[yY])$ ]]
then
	printf "######   Installing and configuring UFW   ######\n"
	sudo pacman -S --noconfirm ufw
	sudo systemctl enable ufw
	sudo systemctl start ufw
	sudo ufw default deny incoming
	sudo ufw default allow outgoing
	sudo ufw limit ssh
#	sudo ufw allow 80
#	sudo ufw allow 443
	echo "y" | sudo ufw enable
fi

### If answered yes, install yay, else do not install
if [[ "$yay" =~ ^([yY][eE][sS]|[yY])$ ]]
then
	printf "######   Installing yay   ######\n"
	git clone https://aur.archlinux.org/yay.git
	cd yay || exit
	makepkg -si --noconfirm
	cd ..
	rm -rf yay-bin
fi

# printf "######   Installing fonts   ######\n"
# sudo pacman -S --noconfirm ttf-roboto ttf-roboto-mono ttf-droid ttf-opensans ttf-dejavu ttf-liberation ttf-hack noto-fonts ttf-fira-code ttf-fira-mono ttf-font-awesome noto-fonts-emoji ttf-hanazono

printf "######   Ricing bash   ######\n"
mv ~/.bashrc .bashrc_original
touch ~/.bashrc
tee -a ~/.bashrc << 'EOF'
#
# ~/.bashrc
#

[[ $- != *i* ]] && return

[ -r /usr/share/bash-completion/bash_completion ] && . /usr/share/bash-completion/bash_completion

PS1='\[\033[01;32m\][\u@\h\[\033[01;37m\] \W\[\033[01;32m\]]\$\[\033[00m\] '

alias ls='ls --color=auto'
alias ll='ls -l --color=auto'
alias la='ls -la --color=auto'
alias grep='grep --colour=auto'
alias egrep='egrep --colour=auto'
alias fgrep='fgrep --colour=auto'

alias cp="cp -i"                          # confirm before overwriting something
alias df='df -h'                          # human-readable sizes
alias free='free -m'                      # show sizes in MB
alias more=less

complete -cf sudo
EOF

### If answered yes, install Zsh, else do not install
if [[ "$zsh" =~ ^([yY][eE][sS]|[yY])$ ]]
then
	printf "######   Installing Zsh   ######\n"
	sudo pacman -S --noconfirm zsh zsh-completions

	printf "######   Installing 'oh my zsh'   ######\n"
	sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended

	printf "######   Making Zsh the default shell   ######\n"
	echo "${user_password}" | chsh -s /bin/zsh
		while [ $? -eq 1 ]; do
			printf "You entered the incorrect password for user: %s, try again! \n" "${USER}"
			chsh -s /bin/zsh
		done

	printf "######   Installing Custom Zsh plugins   ######\n"
	## zsh-syntax-highlighting
	git clone https://github.com/zsh-users/zsh-syntax-highlighting.git ~/.oh-my-zsh/custom/plugins/zsh-syntax-highlighting
	## zsh-autosuggestions
	git clone https://github.com/zsh-users/zsh-autosuggestions ~/.oh-my-zsh/custom/plugins/zsh-autosuggestions
	## zsh-history-substring-search
	git clone https://github.com/zsh-users/zsh-history-substring-search ~/.oh-my-zsh/custom/plugins/zsh-history-substring-search

	printf "######   Enabling Custom Zsh plugins   ######\n"
	sudo sed -i 's/^plugins=(git)/plugins=(git archlinux zsh-syntax-highlighting zsh-autosuggestions zsh-history-substring-search)/g' ~/.zshrc

	printf "######   Changing Zsh default theme   ######\n"
	sudo sed -i 's/^ZSH_THEME="robbyrussell"/ZSH_THEME="lukerandall"/g' ~/.zshrc

	printf "######   Adding Zsh key bindings for Keypad   ######\n"
	tee -a ~/.zshrc <<-'EOF'

	## Keypad Bindings

	## The actual codes (for example ^[Oq) may be different on your system.
	## You can press Ctrl+v followed by the key in question to get the code for your terminal.

	# 0 . Enter
	bindkey -s "^[Op" "0"
	bindkey -s "^[On" "."
	bindkey -s "^[OM" "^M"
	# 1 2 3
	bindkey -s "^[Oq" "1"
	bindkey -s "^[Or" "2"
	bindkey -s "^[Os" "3"
	# 4 5 6
	bindkey -s "^[Ot" "4"
	bindkey -s "^[Ou" "5"
	bindkey -s "^[Ov" "6"
	# 7 8 9
	bindkey -s "^[Ow" "7"
	bindkey -s "^[Ox" "8"
	bindkey -s "^[Oy" "9"
	# + -  * /
	bindkey -s "^[Ol" "+"
	bindkey -s "^[OS" "-"
	bindkey -s "^[OR" "*"
	bindkey -s "^[OQ" "/"
	EOF
fi

printf "######   Creating user's folders   ######\n"
sudo pacman -S --noconfirm xdg-user-dirs

printf "######   %bAll done with Post Installation%b   ######\n\n" "${BGreen}" "${Colour_Off}"

# Asks to reboot
#--------------------------------------------

printf "\n-------------------------------------\n"
while true; do
	read -r -p ">>>   Would you like to Reboot [Y]: " reboot
	reboot=${reboot:-Y}

	case $reboot in
		[yY][eE][sS]|[yY])
			break
		;;
		[nN][oO]|[nN])
			break
		;;
		*)
			echo ">>>   Invalid input... Valid entries are E.g. [Y / N] or [y / n] or [Yes / No]"
		;;
	esac
done

if [[ "${reboot}" =~ ^([yY][eE][sS]|[yY])$ ]]; then
	sudo reboot
fi