arch-install.sh

#!/bin/bash

# https://github.com/prmsrswt/arch-install.sh

function ascii {

	#Added ASCII Art cause why not
	echo	' $$$$$$\                      $$\             $$$$$$\                       $$\               $$\ $$\ '
	echo	'$$  __$$\                     $$ |            \_$$  _|                      $$ |              $$ |$$ |'
	echo	'$$ /  $$ | $$$$$$\   $$$$$$$\ $$$$$$$\          $$ |  $$$$$$$\   $$$$$$$\ $$$$$$\    $$$$$$\  $$ |$$ |'
	echo	'$$$$$$$$ |$$  __$$\ $$  _____|$$  __$$\         $$ |  $$  __$$\ $$  _____|\_$$  _|   \____$$\ $$ |$$ |'
	echo	'$$  __$$ |$$ |  \__|$$ /      $$ |  $$ |        $$ |  $$ |  $$ |\$$$$$$\    $$ |     $$$$$$$ |$$ |$$ |'
	echo	'$$ |  $$ |$$ |      $$ |      $$ |  $$ |        $$ |  $$ |  $$ | \____$$\   $$ |$$\ $$  __$$ |$$ |$$ |'
	echo	'$$ |  $$ |$$ |      \$$$$$$$\ $$ |  $$ |      $$$$$$\ $$ |  $$ |$$$$$$$  |  \$$$$  |\$$$$$$$ |$$ |$$ |'
	echo	'\__|  \__|\__|       \_______|\__|  \__|      \______|\__|  \__|\_______/    \____/  \_______|\__|\__|'
	echo	'                                                                                                      '
	echo	'                                                                                                      '

}

function br {
	# Just output a bunch of crap, but it looks cool so..
	for ((i=1; i<=`tput cols`; i++)); do echo -n -; done
}

function cont {
	read -r -p "[SUCCESS] Continue to next step? [Y/n] " contin
	case $continue in
		[Nn][oO]|[nN] )
			exit
			;;
		*)
			;;
	esac
}

function set-time {
	echo "Setting time...."
	# This command fixes different time reporting when dual booting with windows.
	timedatectl set-local-rtc 1 --adjust-system-clock
}

function partion {
	br
	read -r -p "Do you want to do partioning? [y/N] " resp
	case "$resp" in
		[yY][eE][sS]|[yY])
			echo "gdisk will be used for partioning"
			read -r -p "which drive you want to partition (exapmle /dev/sda)? " drive
			# Using gdisk for GPT, if you want to use MBR replace it with fdisk
			gdisk $drive
			;;
		*)
			;;
	esac
	cont
}

function mounting {
	br
	read -r -p "which is your root partition? " rootp
	mkfs.ext4 $rootp
	mount $rootp /mnt
	mkdir /mnt/boot
	read -r -p "which is your boot partition? " bootp
	read -r -p "Do you want to format your boot partition? [y/N] " response
	case "$response" in
		[yY][eE][sS]|[yY])
			mkfs.fat -F32 $bootp
			;;
		*)
			;;
	esac
	mount $bootp /mnt/boot
	read -r -p "Do you want to use a seperate home partition? [y/N] " responsehome
	case "$responsehome" in
		[yY][eE][sS]|[yY])
			read -r -p "which is your home partition? " homep
			read -r -p "Do you want to format your home partition? [y/N] " rhome
			case "$rhome" in
				[yY][eE][sS]|[yY])
					mkfs.ext4 $homep
					;;
				*)
					;;
			esac
			mount $homep /mnt/home
			;;
		*)
			;;
	esac
	cont
}

function base {
	br
	echo "Starting installation of packages in selected root drive..."
	sleep 1
	pacstrap /mnt \
				base \
				diffutils \
				e2fsprogs \
				inetutils \
				less \
				linux \
				linux-firmware \
				logrotate \
				man-db \
				man-pages \
				nano \
				texinfo \
				usbutils \
				which \
				base-devel \
				networkmanager \
				sudo \
				bash-completion \
				git \
				vim \
				exfat-utils \
				ntfs-3g \
				grub \
				os-prober \
				efibootmgr \
				htop \
				vlc \
				pacman-contrib \
				ttf-hack
	genfstab -U /mnt >> /mnt/etc/fstab
	cont
}

function install-gnome {
	pacstrap /mnt gnome gnome-tweaks papirus-icon-theme
	arch-chroot /mnt bash -c "systemctl enable gdm && exit"
	# Editing gdm's config for disabling Wayland as it does not play nicely with Nvidia
	arch-chroot /mnt bash -c "sed -i 's/#W/W/' /etc/gdm/custom.conf && exit"
}

function install-deepin {
	pacstrap /mnt deepin lightdm gedit
	arch-chroot /mnt bash -c "systemctl enable lightdm && exit"
}

function install-kde {
	pacstrap /mnt xorg plasma sddm
	arch-chroot /mnt bash -c "systemctl enable sddm && exit"
	pacstrap /mnt ark dolphin ffmpegthumbs gwenview kaccounts-integration kate kdialog kio-extras konsole ksystemlog okular print-manager
}

function de {
	br
	echo -e "Choose a Desktop Environment to install: \n"
	echo -e "1. GNOME \n2. Deepin \n3. KDE \n4. None"
	read -r -p "DE: " desktope
	case "$desktope" in
		1)
			install-gnome
			;;
		2)
			install-deepin
			;;
		3)
			install-kde
			;;
		*)
			;;
	esac
	cont
}

function installgrub {
	read -r -p "Install GRUB bootloader? [y/N] " igrub
	case "$igrub" in
		[yY][eE][sS]|[yY])
			echo -e "Installing GRUB.."
			arch-chroot /mnt bash -c "grub-install --target=x86_64-efi --efi-directory=/boot --bootloader-id=Arch && grub-mkconfig -o /boot/grub/grub.cfg && exit"
			;;
		*)
			;;
	esac
	cont
}

function archroot {
	br
	read -r -p "Enter the username: " uname
	read -r -p "Enter the hostname: " hname

	echo -e "Setting up Region and Language\n"
	arch-chroot /mnt bash -c "ln -sf /usr/share/zoneinfo/Asia/Kolkata /etc/localtime && hwclock --systohc && sed -i 's/#en_US.UTF-8/en_US.UTF-8/' /etc/locale.gen && locale-gen && echo 'LANG=en_US.UTF-8' > /etc/locale.conf && exit"

	echo -e "Setting up Hostname\n"
	arch-chroot /mnt bash -c "echo $hname > /etc/hostname && echo 127.0.0.1	$hname > /etc/hosts && echo ::1	$hname >> /etc/hosts && echo 127.0.1.1	$hname.localdomain	$hname >> /etc/hosts && exit"

	echo "Set Root password"
	arch-chroot /mnt bash -c "passwd && useradd --create-home $uname && echo 'set user password' && passwd $uname && groupadd sudo && gpasswd -a $uname sudo && EDITOR=vim visudo && exit"

	echo -e "enabling services...\n"
	arch-chroot /mnt bash -c "systemctl enable bluetooth && exit"
	arch-chroot /mnt bash -c "systemctl enable NetworkManager && exit"
	
	echo -e "enabling paccache timer...\n"
	arch-chroot /mnt bash -c "systemctl enable paccache.timer && exit"

	echo -e "Editing configuration files...\n"
	# Enabling multilib in pacman
	arch-chroot /mnt bash -c "sed -i '93s/#\[/\[/' /etc/pacman.conf && sed -i '94s/#I/I/' /etc/pacman.conf && pacman -Syu && sleep 1 && exit"
	# Tweaking pacman, uncomment options Color, TotalDownload and VerbosePkgList
	arch-chroot /mnt bash -c "sed -i '34s/#C/C/' /etc/pacman.conf && sed -i '35s/#T/T/' /etc/pacman.conf && sed -i '37s/#V/V/' /etc/pacman.conf && sleep 1 && exit"

	cont
}

function browser {
	br
	read -r -p "Install firefox? [y/N] " ff
	case "$ff" in
		[yY][eE][sS]|[yY])
			pacstrap /mnt firefox
			;;
		*)
			;;
	esac
	read -r -p "Install chromium? [y/N] " chrom
	case "$chrom" in
		[yY][eE][sS]|[yY])
			pacstrap /mnt chromium
			;;
		*)
			;;
	esac
	cont
}

function install-amd {
	pacstrap /mnt mesa lib32-mesa xf86-video-amdgpu vulkan-radeon lib32-vulkan-radeon
	pacstrap /mnt libva-mesa-driver lib32-libva-mesa-driver mesa-vdpau lib32-mesa-vdpau
}
function install-nvidia {
	br
	read -r -p "Do you want proprietary nvidia drivers? [y/N] " graphic
	case "$graphic" in
		[yY][eE][sS]|[yY])
			pacstrap /mnt nvidia nvidia-settings nvidia-utils lib32-nvidia-utils
			;;
		*)
			;;
	esac
	cont
}

function graphics {
	br
	echo -e "Choose Graphic card drivers to install: \n"
	echo -e "1. AMD \n2. Nvidia \n3. None"
	read -r -p "Drivers [1/2/3]: " drivere
	case "$drivere" in
		1)
			install-amd
			;;
		2)
			install-nvidia
			;;
		*)
			;;
	esac
	cont
}

function installsteam {
	br
	read -r -p "Do you want to install steam? [y/N] " isteam
	case "$isteam" in
		[yY][eE][sS]|[yY])
			pacstrap /mnt steam steam-native-runtime
			;;
		*)
			;;
	esac
	cont
}

function additional {
	br
	read -r -p "Do you want to install fun stuff? [y/N] " funyes # because why not
	case "$funyes" in
		[yY][eE][sS]|[yY])
			pacstrap /mnt sl neofetch lolcat cmatrix
			;;
		*)
			;;
	esac
}

function full-installation {
	set-time
	partion
	mounting
	base
	archroot
	de
	installgrub
	browser
	graphics
	installsteam
	additional
	echo "Installation complete. Reboot you lazy bastard."
}

function step-installation {
	echo "These steps are available for installion:"
	echo "1. set-time"
	echo "2. partioning"
	echo "3. mounting"
	echo "4. base installation"
	echo "5. archroot"
	echo "6. installing a Desktop Environment"
	echo "7. installing grub"
	echo "8. installing browsers"
	echo "9. graphics drivers"
	echo "10. installing steam"
	echo "11. additional stuff"
	read -r -p "Enter the number of step : " stepno

	array=(set-time partion mounting base archroot de installgrub browser graphics installsteam additional)
	#array=(ascii ascii ascii)
	stepno=$[$stepno-1]
	while [ $stepno -lt ${#array[*]} ]
	do
		${array[$stepno]}
		stepno=$[$stepno+1]
	done
}

function main {
	echo "1. Start full installation"
	echo "2. Start from a specific step"
	read -r -p "What would you like to do? [1/2] " what
	case "$what" in
		2)
			step-installation
			;;
		*)
			full-installation
			;;
	esac
}

ascii
read -r -p "Start Installation? [Y/n] " starti
case "$starti" in
	[nN][oO]|[nN])
		;;
	*)
		main
		;;
esac