arch.sh

#!/bin/bash

: << DOCS

To run on the fly:
bash <(curl -sL bit.ly/a-install)

DOCS



###  Set error detection  ###
error() {

	local sourcefile=$1
	local lineno=$2
	local errornum=$3
	local command=$4

	echo -e "\e[0;41m\n\nFile: \e[1;2;41m$1\n\e[0;41mError: \e[1;2;41m$3\n\e[0;41mLine:\e[1;2;41m $2\n\e[0;41mCommand:\e[1;2;41m $4\n\e[0;29m\n"

	finished=0

	while [ $finished -eq 0 ]; do

		echo -e "What now?\n\n[e] edit (last position)\n[n] edit (error position)\n[c] continue\n[x] exit\n"

		if [[ "$error_bypass" = '1' ]]; then
			choice=c
			echo -e "\nSkipping error.\n"
		else
			read -p "Choice: " -n 2 choice
		fi

		case $choice in
				e)		$editor $arch_path/$arch_file; exit ;;
				n)		$editor +$2 $arch_path/$arch_file; exit ;;
				c)		set +e; break ;;
				x)		exit ;;
				*) 	finished=0 ;;
		esac

	done

}

trap 'error "${BASH_SOURCE}" "${LINENO}" "$?" "${BASH_COMMAND}"' ERR
trap 'echo;echo; read -s -r -n 1 -p "<ctrl> + c pressed. Press any key to continue... "; set +e' SIGINT


error_check () {

	case $1 in
		0)		set +e; echo "No error detection!" ;;
		1)		set -Eeo pipefail ;;
		2)		set -Eueo pipefail ;;
		3)		echo "All commands issued will be printed."; set -Eeox pipefail ;;
	esac

}

error_check 1
error_bypass=0


# fml, I could swear all developers have 20/20 vision :/
if [ -f /usr/share/kbd/consolefonts/ter-132b.psf.gz ] && [[ ! $(grep -e '^HOOKS|consolefont' /etc/mkinitcpio.conf) = '' ]]; then
	setfont ter-132b
else
	setfont -d
fi


fstype='btrfs'						# btrfs,ext4,bcachefs,f2fs,xfs,jfs,nilfs2

bootOwnPartition='false'		# make separate boot partition (true/false)?
[ $fstype = 'bcachefs' ] && bootOwnPartition=true

# Do we want a separate boot partition (which will be ext2)
if [[ $bootOwnPartition = 'true' ]]; then
	espPartNum=1
	bootPartNum=2
	swapPartNum=3
	rootPartNum=4
else
	bootPartNum=0
	espPartNum=1
	swapPartNum=2
	rootPartNum=3
fi

espPart=$espPartNum
bootPart=$bootPartNum
swapPart=$swapPartNum
rootPart=$rootPartNum

mnt=/mnt
mnt2=/mnt2
mnt3=/mnt3

efi_path=/efi
encrypt='false'				# bcachefs only
encryptLuks='false'			# ext4 (Not Working!)
startSwap='8192Mib'			# 2048,4096,8192,(8192 + 1024 = 9216) 
fsPercent='100'				# What percentage of space should the root drive take?
checkPartitions='true'		# Check that partitions are configured optimally?

subvolPrefix='@'				# eg., '/' or '/@' btrfs and bcachefs only
first_snapshot_name='1'		# Only for btrfs
snapshot_dir='/.snapshots'
subvols=($snapshot_dir /boot/grub /var/log /var/tmp)
rootMount='/@root'				# (ex., @root) Only used for bcachefs

btrfs_mountopts="noatime,discard=async"
bcachefs_mountopts="noatime"
boot_mountopts="noatime"
efi_mountopts="noatime"
fsckBcachefs='true'

backup_install='true'		# say 'true' to do snapshots/rysncs during install (only btrfs/bcachefs)
install_backup='true'  # make a compressed backup file?
copy_user_dir='false'    # should we copy /home/$user ?

if [ $fstype = 'btrfs' ]; then
	backup_type='snapper'
else
	backup_type='rsync'
fi

initramfs='mkinitcpio'		# mkinitcpio, dracut, booster
extra_modules='lz4'			# adds to /etc/mkinitcpio modules
extra_hooks=''					# adds to /etc/mkinitcpio hooks



kernel_ops="nmi_watchdog=0 nowatchdog modprobe.blacklist=iTCO_wdt loglevel=3 rd.udev.log_level=3 zswap.enabled=1 zswap.compressor=zstd zswap.max_pool_percent=20 scsi_mod.use_blk_mq=1"

enable_fallback='false' 	# Enable fallback kernel?

user=user
password='123456'
autologin=true
arch_file=$(basename "$0")
arch_path=$(dirname "$0")
editor="vim"

aur_app=none
aur_path=/home/$user/aur
aur_apps_path=/root/pkgs/
aur_apps_root=''
aur_apps_user=''
#aur_apps_kde='brave-bin'
aur_apps_kde=''

base_install="base linux linux-firmware vim parted gptfdisk arch-install-scripts pacman-contrib tar man-db dosfstools"

user_install="sudo git base-devel"

cage_install="cage firefox"

weston_install="brightnessctl wireplumber pipewire pipewire-pulse weston firefox"

phosh_install="phosh phoc phosh-mobile-settings squeekboard firefox"

gnome_install="gnome-shell polkit nautilus gnome-console xdg-user-dirs dconf-editor gnome-browser-connector gnome-shell-extensions gnome-control-center gnome-weather"

kde_install="plasma-desktop plasma-pa plasma-nm kscreen iio-sensor-proxy dolphin konsole ffmpegthumbs bleachbit ncdu kdiskmark brave-bin networkmanager-openvpn openvpn firefox okular"

#kde_install="plasma-desktop plasma-pa maliit-keyboard plasma-nm kscreen iio-sensor-proxy dolphin konsole ffmpegthumbs bleachbit ncdu kdiskmark brave-bin networkmanager-openvpn openvpn vital-synth"

#kde_install="plasma-desktop plasma-pa maliit-keyboard plasma-nm kscreen iio-sensor-proxy dolphin konsole ffmpegthumbs bleachbit ncdu kdiskmark networkmanager-openvpn openvpn firefox brave-bin code gwenview code helix sudo pacman -S nodejs npm"

ucode=intel-ucode
hostname=Arch
timezone=Canada/Eastern

offline=0
reinstall=0
copy_on_host='1' # Copy packages to host? Set to '0' when installing from an arch iso in ram

wlan="wlan0"
wifi_ssid=""
wifi_pass=""

backup_file=/setup.tar.gz

CONFIG_FILES="

/usr/lib/initcpio/init
/usr/lib/initcpio/hooks/btrfs-rollback
/usr/lib/initcpio/install/btrfs-rollback
/usr/lib/initcpio/hooks/bcachefs-rollback
/usr/lib/initcpio/install/bcachefs-rollback
/usr/lib/initcpio/install/liveroot
/usr/lib/initcpio/hooks/liveroot
/etc/overlayroot.conf
/usr/bin/mount.overlayroot
/usr/lib/initcpio/hooks/overlayroot
/usr/lib/initcpio/install/overlayroot

/etc/booster.yaml
/etc/default/grub-btrfs/config
/etc/dracut.conf.d/
/etc/hostname
/etc/hosts
/etc/iwd/main.conf
/etc/locale.conf
/etc/locale.gen
/etc/localtime
/etc/mkinitcpio.d/linux.preset
/etc/NetworkManager/conf.d/
/etc/NetworkManager/system-connections
/var/lib/NetworkManager/timestamps
/etc/pacman.conf
/etc/pacman-offline.conf
/etc/pacman.d/mirrorlist
/etc/security/limits.conf
/etc/sudoers.d/
/etc/sysctl.d/50-coredump.conf
/etc/sysctl.d/99-cache-pressure.conf
/etc/sysctl.d/99-net-keepalive.conf
/etc/sysctl.d/99-net-timeout.conf
/etc/sysctl.d/99-swappiness.conf
/etc/systemd/coredump.conf.d/custom.conf
/etc/systemd/system/getty@tty1.service.d/autologin.conf
/etc/systemd/system/getty@tty1.service.d/wayland.conf
/etc/wpa_supplicant
/etc/updatedb.conf
/root/.mkshrc
/root/.vimrc
/root/pkgs/
/var/lib/dhcpcd
/var/lib/iwd
/var/spool/cron/root

/home/$user/.bash_profile
/home/$user/.bashrc
/home/$user/.cert/
/home/$user/.config/
/home/$user/.enduin
/home/$user/.hushlogin
/home/$user/.local/
/home/$user/.mkshrc
/home/$user/.mozilla
/home/$user/.profile
/home/$user/.vimrc
/home/$user/.vim/

"



check_pkg () {

	for package in "$@"; do

   	if [ ! "$(pacman -Q $package)" ]; then
      	echo -e "\nInstalling program required to run command: $package...\n"
      	pacman --noconfirm -S $package
   	fi

	done

}



check_viable_disk () {

	if [[ "$disk" == "" ]]; then
		echo -e "\nMissing disk parameter. Exiting.\n"
  		exit
	fi

	if [[ ! "$(lsblk --output=PATH -d -n | grep $disk)" ]]; then
		echo -e "\nNo such disk found ($disk). Exiting.\n"
		sleep 2
		exit
	fi

}



check_on_root () {
  
	if [[ $(mount | grep -G "$disk.*on / type") ]]; then 
		echo -e "\nDevice mounted on root. Will not run. Exiting.\n"
		sleep 2
		exit
	fi

}



unmount_disk () {

	sync_disk

	[ "$mnt" = '' ] && return

	if [[ $(mount | grep -E $disk$espPart | grep -E "on $mnt$efi_path") ]]; then
		echo "Unmounting $mnt$efi_path..."
		umount -n -R $mnt$efi_path
		sleep .1
	fi

	if [[ $(mount | grep -E $disk$bootPart | grep -E "on $mnt/boot") ]] && [[ $bootOwnPartition = true ]]; then
		echo "Unmounting $mnt/boot..."
		umount -n -R $mnt/boot
		sleep .1
	fi

	if [[ $(mount | grep -E $disk$rootPart | grep -E "on $mnt") ]]; then

		# Might need to turn error checking off here

		echo "Unmounting $mnt..."

		# Shouldn't be in directory we're unmounting   
		[[ "$(pwd | grep $mnt)" ]] && cd /

		umount -n -R $mnt
		
		sleep .1 

		# Time to get rugged and tough!
		if [[ "$(mount | grep /mnt)" ]]; then

			echo -e "\nCouldn't unmount. Trying alternative method. Please be patient...\n" 

			mounted=1

			while [[ "$mounted" -eq 1 ]]; do
				
				cd /

				if [[ "$(mount | grep /mnt)" ]]; then
					sleep .1
					umount -l $mnt
				else
					mounted=0
				fi

			done

			echo -e "\nLazy unmount successful.\n"
			systemctl daemon-reload

		fi

	else
		echo -e "\nDisk already unmounted!\n"
	fi

	if [ "$encryptLuks" = 'true' ] && [ "$(mount | grep /dev/mapper/root)" ]; then
echo
		#cryptsetup close root
	fi

}



choose_disk () {

	search_disks=1
	host="$(mount | awk '/ on \/ / { print $1}' | sed 's/p*[0-9]$//g')"

	while [ $search_disks -eq 1 ]; do
		
		mnt=''

		echo -e "\nDrives found (current mount: /):\n"
		
		#rootfs=$(mount | grep ' / ' | sed 's/(.*//; s/\/dev.* type //; s/ //')
		rootfs=$(mount | grep ' / ' | sed 's/(.*//; s/^.*type //; s/ //')

		#[ "$(mount | grep ' on / type overlay' | awk '{print $5}')" ] && echo -e "\n       *** Running in overlay mode! ***\n"
		[ "$rootfs" = "overlay" ] && echo -e "\n       *** Running in overlay mode! ***\n"

		rootMount="$(findmnt -o FSROOT -n / | sed 's#^/##')"
		
		if [ $rootfs = 'btrfs' ]; then
		
			[ "$(snapper list --columns number,read-only | grep '[0-9][-\*].*yes')" ] && echo -e "\n       *** Running in read only mode! ***\n"
			
			#rootSub="$(mount | grep ' / ' | sed 's#.*.subvol=/##; s/)//')"
			defaultSub="$(btrfs su get-default / | awk '{ print $9 }')"

			if [ ! "$rootMount" = "$defaultSub" ]; then
				echo -e "\n       *** NOT mounted on default subvolume ($defaultSub)! ***\n\n"
			fi
		fi


		lsblk --output=PATH,SIZE,MODEL,TRAN -d | grep -P "/dev/sd|nvme|vd" | sed "s#$host.*#&  $rootfs#g" | sed "s#$host.*#& (host) $rootMount#"

		[[ $rootfs = 'btrfs' ]] || [[ $rootfs = 'bcachefs' ]] && extra='snapshots '

		choices='quit '$extra'backup edit $ # '$(lsblk -dpnoNAME|grep -P "/dev/sd|nvme|vd")' / update script logout reboot suspend hibernate poweroff stats benchmark rollback-script'
		

		echo -e "\nWhich drive?\n"

	select disk in $choices
		do
			case $disk in
				quit)			sync_disk; exit ;;
				edit)			$editor $arch_path/$arch_file; exit ;;
				$)				sudo -u $user bash ;;
				\#)			bash ;;
				backup)		backup_config ;;
				script)		download_script ;;
				update)		pacman -Syu ;;
				rollback-script)	$editor /lib/initcpio/hooks/btrfs-rollback; mkinitcpio -P ;;
				logout)		killall systemd ;;
				reboot)		reboot ;;
				suspend)		echo mem > /sys/power/state ;;
				hibernate)	echo disk > /sys/power/state ;;
				poweroff)	poweroff ;;
				benchmark)	benchmark ;;
				stats)	  	
								if [[ "$rootfs" = 'btrfs' ]]; then
									btrfs su list /
								fi

								echo; free -h; echo
								systemd-analyze | sed 's/in .*=/in/;s/graph.*//'
								echo $(systemd-analyze  blame | wc -l) systemd services
								;;
				snapshots)	snapshots_menu ;;
				/)				disk=$host; search_disks=0; break ;;
				*)    		if [[ $disk = '' ]]; then
						   		
									echo -e "\nInvalid option!\n"
								
								else
									
									if [[ "$(mount | awk '/ on \/ / { print $1}' | sed 's/p*[0-9]$//g')" = $disk ]]; then
										mnt=''
									else
										mnt='/mnt'
									fi

									search_disks=0
									break
								
								fi
								;;
				'')   		echo -e "\nInvalid option!\n" ;;
			esac

		done

	if [ "$(echo $disk | grep nvme)" ]; then
		bootPart="p$bootPartNum"
		espPart="p$espPartNum"
		swapPart="p$swapPartNum"
		rootPart="p$rootPartNum"
	else
		bootPart="$bootPartNum"
		espPart="$espPartNum"
     	swapPart="$swapPartNum"
     	rootPart="$rootPartNum"
	fi

	echo -e "\nDisk chosen: $disk (mounted on $mnt/)"
	
	done
}



delete_partitions () {

	check_on_root
	unmount_disk

	echo -e "\nWiping disk...\n"

	wipefs -af $disk

	check_pkg gptfdisk

	sgdisk -Zo $disk

}


fs_packages () {

	case $fstype in
		ext4)			pkg=e2fsprogs ;;
		btrfs)		pkg="btrfs-progs grub-btrfs rsync" ;;
		bcachefs)	pkg="bcachefs-tools rsync" ;;
		xfs)			pkg=xfsprogs ;;
		f2fs)			pkg=f2fs-tools ;;
		jfs)			pkg=jfsutils ;;
		nilfs2)		pkg="pacstrap_install nilfs-utils" ;;
	esac

	echo "$pkg"

}


create_partitions () {

	check_on_root

	delete_partitions
	
	systemctl daemon-reload

	check_pkg parted dosfstools

	if [ $bootOwnPartition = true ]; then

		parted --fix --align optimal -s $disk mklabel gpt \
			mkpart ESP fat32 1Mib 512Mib \
			mkpart BOOT ext2 512Mib 1024Mib \
			mkpart SWAP linux-swap 1024Mib $startSwap \
			set $espPartNum esp on \
			set $swapPartNum swap on
				
			mkfs.ext2 -F -L BOOT $disk$bootPart 
		
	else

		parted --fix --align optimal -s $disk mklabel gpt \
			mkpart ESP fat32 1Mib 512Mib \
			mkpart SWAP linux-swap 1024Mib $startSwap \
			set $espPartNum esp on \
			set $swapPartNum swap on
		
	fi

	if [ $fstype = bcachefs ]; then
		# Parted doesn't recognise bcachefs filesystem
		parted -s $disk mkpart ROOT ext4 $startSwap $fsPercent%
	else
		parted -s $disk mkpart ROOT $fstype $startSwap $fsPercent%
	fi

	parted -s $disk print

	# Won't work without a small delay
	sync_disk
	echo "Sleeping one second..."
	sleep .1

	mkfs.fat -F 32 -n EFI $disk$espPart 
	mkswap -L SWAP $disk$swapPart

	if [ "$encryptLuks" = 'true' ]; then
 		cryptsetup -v luksFormat $disk$rootPart
 		cryptsetup open $disk$rootPart root
	fi

	pacman -S --needed --noconfirm $(fs_packages)

	case $fstype in

		btrfs)		pacman -S --needed --noconfirm btrfs-progs
						mkfs.btrfs -f -n 32k -L ROOT $disk$rootPart ;;
		ext4)			if [ "$encryptLuks" = 'true' ]; then

							mkfs.ext4 /dev/mapper/root

							mount /dev/mapper/root $mnt
							
							# Check that mapping works as intended: 
							umount $mnt
							cryptsetup close root
							cryptsetup open $disk$rootPart root
							mount /dev/mapper/root $mnt

						else

							mkfs.ext4 -F -q -t ext4 -L ROOT $disk$rootPart
							
							echo "Running tune2fs to create fast commit journal area..." 
							tune2fs -O fast_commit $disk$rootPart

						fi

						;;
		xfs)			pacman -S --needed --noconfirm xfsprogs 
						mkfs.xfs -f -L ROOT $disk$rootPart ;;
		jfs)			pacman -S --needed --noconfirm jfsutils
						mkfs.jfs -f -L ROOT $disk$rootPart ;;
		f2fs)			pacman -S --needed --noconfirm f2fs-tools
						mkfs.f2fs -f -l ROOT $disk$rootPart ;;
		nilfs2)		pacman -S --needed --noconfirm nilfs-utils
						mkfs.nilfs2 -f -L ROOT $disk$rootPart ;;
		bcachefs)	pacman -S --needed --noconfirm bcachefs-tools rsync
						
						if [[ $encrypt = true ]]; then

							#You MUST add 'bcachefs' module and hook (after 'filesystem')
							
							bcachefs format -f -L ROOT --encrypted $disk$rootPart
							bcachefs unlock -k session $disk$rootPart

						else
							bcachefs format -f -L ROOT $disk$rootPart
						fi
						;;
	esac
	
	if [[ $checkPartitions = true ]]; then

		echo -e "\nRunning tests to check partitions\n"

		if [[ -f /home/$user/.local/bin/checkpartitionsalignment.sh ]]; then
			/home/user/.local/bin/./checkpartitionsalignment.sh $disk
		fi

	fi
	
	parted -s $disk print

	sync_disk
	echo -e "\nPausing for 1 second...\n"
	sleep .1
	
	cd /

	echo -e "\nMounting $mnt..."


	if [ "$encryptLuks" = 'true' ]; then

		if [ ! "$(mount | grep /dev/mapper/root)" ]; then
			cryptsetup open $disk$rootPart root
		fi

		mount /dev/mapper/root $mnt

	else

		mount -t $fstype --mkdir $disk$rootPart $mnt
	
	fi


	if [ "$fstype" = "btrfs" ]; then

		# https://www.ordinatechnic.com/distribution-specific-guides/Arch/an-arch-linux-installation-on-a-btrfs-filesystem-with-snapper-for-system-snapshots-and-rollbacks

		#btrfs subvolume create $mnt/@
		#btrfs subvolume create $mnt/@$snapshot_dir
			
		#mkdir $mnt/@$snapshot_dir/$first_snapshot_name
		#btrfs subvolume create $mnt/@$snapshot_dir/$first_snapshot_name/snapshot
			
		#mkdir $mnt/@/boot
		#btrfs subvolume create $mnt/@/boot/grub
			
		#mkdir $mnt/@/var
		#btrfs subvolume create $mnt/@/var/log
		#btrfs subvolume create $mnt/@/var/tmp
		
		btrfs subvolume create $mnt/$subvolPrefix
		
		for subvol in "${subvols[@]}"; do

			mkdir -p "$(dirname $mnt/$subvolPrefix$subvol)"
			btrfs su create $mnt/$subvolPrefix$subvol

		done
		
		mkdir -p "$(dirname $mnt/$subvolPrefix$snapshot_dir/$first_snapshot_name/snapshot)"
		btrfs su create $mnt/$subvolPrefix$snapshot_dir/$first_snapshot_name/snapshot

		btrfs subvolume set-default $(btrfs subvolume list $mnt | grep "$subvolPrefix$snapshot_dir/$first_snapshot_name/snapshot" | grep -oP '(?<=ID )[0-9]+') $mnt

		chattr +C $mnt/$subvolPrefix/var/log
		chattr +C $mnt/$subvolPrefix/var/tmp

		echo
		btrfs subvolume list $mnt
		
		echo -e "\nDefault subvolume set to:"
		btrfs subvolume get-default $mnt
		echo


	elif [ "$fstype" = "bcachefs" ]; then

		# https://www.reddit.com/r/bcachefs/comments/1b3uv59/booting_into_a_subvolume_and_rollback/
	
		# Subvolumes not currently being added to /etc/fstab
		for subvol in "${subvols[@]}"; do

			echo -e "Creating subvolume: $mnt$subvolPrefix$subvol..."
			
			# Cannot create the subvolume without dirname path (..)
			mkdir -p "$(dirname $mnt$subvolPrefix$subvol)"

			bcachefs subvolume create "$mnt$subvolPrefix$subvol"
		
		done

	fi

	unmount_disk
	mount_disk

	mkdir -p $mnt/{dev,etc,proc,root,run,sys,tmp,var/cache/pacman/pkg,/var/log,/var/tmp,$aur_apps_path}

	chmod -R 750 $mnt/root
	mkdir -p $mnt/root/.gnupg
	chmod -R 700 $mnt/root/.gnupg
	chmod -R 1777 $mnt/var/tmp

	if [ $fstype = btrfs ]; then

		echo
		btrfs su list $mnt
		echo

		# If not created snapper won't recognise it as a snapshot

		setdate=$(date +"%Y-%m-%d %H:%M:%S")

		cat > $mnt$snapshot_dir/$first_snapshot_name/info.xml << EOF
<?xml version="1.0"?>
<snapshot>
  <type>single</type>
  <num>1</num>
  <date>$setdate</date>
  <description>Initial snapshot</description>
  <cleanup></cleanup>
</snapshot>
EOF

	fi

	genfstab -U $mnt

}


mount_disk () {

	# No need to mount if we're on the main machine
	[ "$mnt" = "" ] && return

	check_on_root

	if [[ ! $(mount | grep -E $disk$rootPart | grep -E "on $mnt") ]]; then

		if [ "$fstype" = "btrfs" ]; then
		
			if [ $(read -t 2 -sn1 -p "Press any key to change root mount." && echo 1) ]; then

				mount -m $disk$rootPart -o "$btrfs_mountopts" $mnt
				echo;echo
				btrfs su list $mnt
				umount -R $mnt
				echo -e "\nPlease enter a path to mount:\n"
				read mntSubvol

			fi

			
			if [ "$mntSubvol" ]; then
				mount -m $disk$rootPart -o "$btrfs_mountopts,subvol=$mntSubvol" $mnt
			else
				# btrfs will automatically mount the 'set-default' subvolume
				mount -m $disk$rootPart -o "$btrfs_mountopts" $mnt
			fi
			
			echo -e "\nDefault subvolume = $(btrfs su get-default $mnt | sed 's/ID.*path //')\n"
			echo mount -m $disk$rootPart -o "$btrfs_mountopts,subvol=$(mount | grep "$disk$rootPart on $mnt " | sed 's#/dev/.*subvol=##; s/)//; s#^/##')" $mnt
			

			#mount -m $disk$rootPart -o "$btrfs_mountopts",subvol=@$snapshot_dir $mnt$snapshot_dir 
			#mount -m $disk$rootPart -o "$btrfs_mountopts",subvol=@/boot/grub $mnt/boot/grub	
			#mount -m $disk$rootPart -o "$btrfs_mountopts",subvol=@/var/log,nodatacow $mnt/var/log
			#mount -m $disk$rootPart -o "$btrfs_mountopts",subvol=@/var/tmp,nodatacow $mnt/var/tmp
			

			for subvol in "${subvols[@]}"; do
				
				echo mount -m $disk$rootPart -o "$btrfs_mountopts",subvol=$subvolPrefix$subvol $mnt$subvol 
				mount -m $disk$rootPart -o "$btrfs_mountopts",subvol=$subvolPrefix$subvol $mnt$subvol 

			done
			


		elif [[ $fstype = bcachefs ]]; then

			# mount: /dev/sda4: Input/output error
			# [ERROR src/commands/mount.rs:395] Mount failed: Input/output error

			# Not sure if this goes before or after encrypt for encrypted device
			
			#echo "Running fsck on disk... Please be patient."
			if [ $fsckBcachefs = 'true' ]; then
				
				echo -e "\nWould you like to run fsck first? (type 'y' to run)\n"
				read -sn1 key

				[ "$key" = 'y' ] && bcachefs fsck -p $disk$rootPart

			fi

			if [[ $encrypt = true ]]; then
				keyctl link @u @s
				bcachefs unlock -k session $disk$rootPart
			fi
			

			mount -t $fstype --mkdir -o "$bcachefs_mountopts" $disk$rootPart $mnt

			if [ "$(ls $mnt | grep '@root')" ]; then
				
				echo -e "\nThere is a drive on $rootMount. Type 'y' to mount it.\n"
				read -sn1 key

				[ "$key" = 'y' ] && mount --bind -o "$bcachefs_mountopts" $mnt$rootMount $mnt

			fi
			
			for subvol in "${subvols[@]}"; do
			
				#if [ ! "$subvol" = '.snapshots' ]; then
					
					echo mount --bind -o "$bcachefs_mountopts" $mnt$subvolPrefix$subvol $mnt$subvolPrefix$subvol
					mount --bind --mkdir -o "$bcachefs_mountopts" $mnt$subvolPrefix$subvol $mnt$subvolPrefix$subvol
				#fi

			done
			
		else
	
			if [ "$encryptLuks" = 'true' ]; then

				if [ ! "$(mount | grep /dev/mapper/root)" ]; then
					cryptsetup open $disk$rootPart root
				fi

				mount /dev/mapper/root $mnt

			else

				# For some reason ext4 prefers a plain mount
				if [ "$fstype" = "ext4" ]; then		
					mount $disk$rootPart $mnt	
				else
					mount -t $fstype --mkdir -o $mountops $disk$rootPart $mnt	
				fi

			fi

		fi

	fi
	
	if [[ ! $(mount | grep -E $disk$bootPart | grep -E "on $mnt/boot") ]] && [[ $bootOwnPartition = true ]]; then
		echo mount -m $disk$bootPart -o $boot_mountopts $mnt/boot
		mount -m $disk$bootPart -o $boot_mountopts $mnt/boot
	fi

	if [[ ! $(mount | grep -E $disk$espPart | grep -E "on $mnt$efi_path") ]]; then
		echo mount -m $disk$espPart -o $efi_mountopts $mnt$efi_path
		mount -m $disk$espPart -o $efi_mountopts $mnt$efi_path
	fi

}



install_base () {

	mount_disk

	echo '[options]
HoldPkg     = pacman glibc
Architecture = auto

CheckSpace
ParallelDownloads = 5
Color

[custom]
SigLevel = Optional TrustAll
Server = file:///var/cache/pacman/pkg/
' > /etc/pacman-offline.conf
	
	if [ ! $mnt = '' ]; then

		cp /etc/pacman-offline.conf $mnt/etc/pacman-offline.conf


		if [ "$offline" -eq 1 ]; then
			echo "Copying database files..."
			mkdir -p $mnt/var/lib/pacman/sync
			cp -r /var/lib/pacman/sync/*.db $mnt/var/lib/pacman/sync/
		fi

	fi

	reset_keys

	check_pkg arch-install-scripts

	#check_pkg reflector
	#echo -e "\nRunning reflector...\n"
	#reflector > /etc/pacman.d/mirrorlist

	[ "$(cat /etc/pacman.d/mirrorlist)" = "" ] && update_mirrorlist


	mkdir -p $mnt/etc/pacman.d
	[ ! $mnt = '' ] && cp /etc/pacman.d/mirrorlist $mnt/etc/pacman.d/mirrorlist

	pacstrap_install "$(fs_packages) $base_install"

	cp $mnt/etc/pacman.conf $mnt/etc/pacman.conf.pacnew

	sed -i 's/#Color/Color/' $mnt/etc/pacman.conf
	sed -i 's/CheckSpace/#CheckSpace/' $mnt/etc/pacman.conf
	sed -i 's/#ParallelDownloads.*$/ParallelDownloads = 5/' $mnt/etc/pacman.conf
	#sed -i 's/SigLevel    = .*$/SigLevel    = TrustAll/' $mnt/etc/pacman.conf

}


auto_login () {

	login_user=$1

	echo -e "\nCreating auto-login for $login_user.\n"

	mkdir -p $mnt/etc/systemd/system/getty@tty1.service.d
	cat > $mnt/etc/systemd/system/getty@tty1.service.d/autologin.conf << EOF
[Service]
Type=simple
ExecStart=
ExecStart=-/sbin/agetty --skip-login --nonewline --noissue --autologin $login_user --noclear %I 38400 linux
EOF

}


hypervisor_setup () {

	choices=(exit auto kvm vmware oracle microsoft)

   select choice in "${choices[@]}"; do
		case $choice in
			auto)			hypervisor=$(systemd-detect-virt) 
							echo "Detected: $hypervisor" ;;
			kvm)			pacstrap_install qemu-guest-agent
							systemctl --root=$mnt enable qemu-guest-agent ;;
			vmware)		pacstrap_install open-vm-tools
							systemctl --root=$mnt enable vmtoolsd
							systemctl --root=$mnt enable vmware-vmblock-fuse ;;
			oracle)		pacstrap_install virtualbox-guest-utils
							systemctl --root=$mnt enable vboxservice ;;
			microsoft)	pacstrap_install hyperv
							systemctl --root=$mnt enable hv_fcopy_daemon
							systemctl --root=$mnt enable hv_kvp_daemon
							systemctl --root=$mnt enable hv_vss_daemon ;;
			exit)			break ;;
			*)    		echo -e "\nInvalid option!\n" ;;
			'')   		echo -e "\nInvalid option!\n" ;;
		esac
	done

}



setup_fstab () {

	mount_disk

	if [ $fstype = 'btrfs' ]; then
	
		sed -i 's#^Q /var/lib/machines 0700 - - -#\#&#' $mnt/usr/lib/tmpfiles.d/systemd-nspawn.conf
		sed -i 's#^Q /var/lib/portables 0700#\#&#' $mnt/usr/lib/tmpfiles.d/portables.conf

		[ "$(btrfs su list $mnt/ | grep var/lib/portables)" ] && btrfs su delete $mnt/var/lib/portables
		[ "$(btrfs su list $mnt/ | grep var/lib/machines)" ] && btrfs su delete $mnt/var/lib/machines

	fi

	echo -e "\nCreating new /etc/fstab file...\n"

	genfstab -U $mnt/ > $mnt/etc/fstab
	grep $disk /proc/mounts > $mnt/etc/fstab.bak

	###  Tweak the resulting /etc/fstab generated  ###

	SWAP_UUID=$(blkid -s UUID -o value $disk$swapPart)

	# genfstab will generate a swap drive. we're using a swap file instead
	sed -i '/LABEL=SWAP/d; /none.*swap.*defaults/d' $mnt/etc/fstab

	sed -i '/portal/d' $mnt/etc/fstab


	# Bad idea to use subids when rolling back 
	#sed -i 's/subvolid=.*,//g' $mnt/etc/fstab

	sed -i 's/relatime/noatime/g' $mnt/etc/fstab
	
	#sed -i 's/\/.*ext4.*0 1/\/      ext4    rw,noatime,commit=60      0 1/' $mnt/etc/fstab

	# Make /efi read-only
	#sed -i 's/\/efi.*vfat.*rw/\/efi     vfat     ro/' $mnt/etc/fstab

	if [ $fstype = bcachefs ]; then

		# bcachefs mounts are not added automatically
		for subvol in "${subvols[@]}"; do
			
			echo "$subvolPrefix$subvol                $subvolPrefix$subvol          none            rw,$bcachefs_mountopts,rw,noshard_inode_numbers,bind  0 0" >> $mnt/etc/fstab 
		
		done

		sed -i 's#^/.snapshots.*#\#&#' $mnt/etc/fstab

	fi

	if [ "$fstype" = 'btrfs' ]; then

		# Or the system might boot into the incorrect subvol after default is changed
		#sed -i 's#,subvol=/@/.snapshots/.*/snapshot.*0 0#      0 0#' $mnt/etc/fstab
	echo	
		# modification is necessary because otherwise GRUB will always look for the kernel in /@/boot instead of /@/.snapshots/
		#sed -i 's#rootflags=subvol=${rootsubvol} ##' $mnt/etc/grub.d/10_linux
		#sed -i 's#rootflags=subvol=${rootsubvol} ##' $mnt/etc/grub.d/20_linux_xen

	fi

	# Remount to test	
	mount -a

	systemctl daemon-reload

	cat $mnt/etc/fstab

}



choose_initramfs () {

	mount_disk

	if [ "$1" ]; then
		choice=$1
	else

		echo -e "\nWhich initramfs would you like to install?\n
1. quit 2. mkinitcpio 3. dracut 4. booster\n"

		read choice
	fi

	case $choice in
		quit|1)			;;
		mkinitcpio|2)	pacstrap_install mkinitcpio 

							sed -i "s/PRESETS=.*/PRESETS=('default')/" $mnt/etc/mkinitcpio.d/linux.preset

		cat /etc/mkinitcpio.conf | awk -F'[=()]' -v srch="filesystems" -v repl="liveroot" -v var="HOOKS=" '$0 ~ var { sub(srch,repl" "srch,$0) }; { print $0 }'



							cat > $mnt/etc/mkinitcpio.conf <<EOF
MODULES=($extra_modules)
BINARIES=()
FILES=()

#HOOKS=(base udev autodetect microcode modconf kms keyboard keymap consolefont block filesystems fsck)
HOOKS=(base udev autodetect microcode modconf keyboard block filesystems $extra_hooks)

COMPRESSION="lz4"

#COMPRESSION_OPTIONS=()

MODULES_DECOMPRESS="no"
EOF

							if [ $encrypt = true ] && [ $fstype = bcachefs ]; then
								add_hooks MODULES bcachefs 
                     	add_hooks HOOKS bcachefs
							fi
							
							if [ ! "$(flash_drive)" ]; then
								add_hooks HOOKS resume
							fi
	
							if [ $enable_fallback = 'false' ]; then

								sed -i "s/ 'fallback'//" $mnt/etc/mkinitcpio.d/linux.preset
								rm -rf $mnt/boot/initramfs-linux-fallback.img
								
								if [[ $mnt = '' ]]; then
                        	grub-mkconfig -o $mnt/boot/grub/grub.cfg
                     	else
                        	arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
                     	fi

							fi

							cat $mnt/etc/mkinitcpio.conf

							if [[ $mnt = '' ]]; then
								mkinitcpio -p linux
							else
								arch-chroot $mnt mkinitcpio -P
							fi

							;;
		dracut|3)		pacstrap_install dracut 

							echo 'hostonly="yes"
compress="lz4"
add_drivers+=" "
omit_dracutmodules+=" "' > $mnt/etc/dracut.conf.d/myflags.conf

							if [[ $mnt = '' ]]; then
								dracut -f $mnt/boot/initramfs-linux.img
								#dracut --regenerate-all $mnt/boot/initramfs-linux.img
								dracut -f $mnt/boot/initramfs-linux-fallback.img
								grub-mkconfig -o $mnt/boot/grub/grub.cfg
							else
								arch-chroot $mnt dracut -f /boot/initramfs-linux.img
								#arch-chroot $mnt dracut --regenerate-all /boot/initramfs-linux.img
								arch-chroot $mnt dracut -f /boot/initramfs-linux-fallback.img
								arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
							fi

							;;
		booster|4)		pacstrap_install booster 
							
							# manual build
							booster build $mnt/boot/booster-linux.img --force

							if [[ $mnt = '' ]]; then	
								grub-mkconfig -o $mnt/boot/grub/grub.cfg
							else
								arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
							fi
							;;
		*)					echo -e "Invalid option!" ;;
	esac

}



install_Refind () {

	mount_disk

	pacstrap_install refind 
	
	if [ "$mnt" = '/' ]; then
		refind-install --usedefault $disk$espPart --alldrivers
	else
		arch-chroot $mnt refind-install --usedefault $disk$espPart --alldrivers
	fi

	mkdir -p $mnt$efi_path/EFI/refind/drivers_x64/

	cp -r /usr/share/refind/drivers_x64/ $mnt$efi_path/EFI/refind/drivers_x64/

	mkrlconf

	sed -i 's/#enable_touch/enable_touch/' $mnt/efi/EFI/BOOT/refind.conf

	

	return



	VOLUME_UUID=$(blkid $disk | awk -F\" '{ print $2 }')
	SWAP_UUID=$(blkid -s UUID -o value $disk$swapPart)
	ROOT_UUID=$(blkid -s UUID -o value $disk$rootPart)

	if [ "$fstype" = "btrfs" ]; then
		rootflags='rootflags=subvol=@'
	else
		rootflags=''
	fi


	echo "\"Boot with standard options\"  \"root=UUID=$ROOT_UUID rw $rootflags $kernel_ops resume=UUID=$SWAP_UUID\"" > $mnt/boot/refind_linux.conf
	echo "\"Boot nomodeset\"  \"root=UUID=$ROOT_UUID rw $rootflags $kernel_ops resume=UUID=$SWAP_UUID systemd.unit=multi-user.target nomodeset\"" >> $mnt/boot/refind_linux.conf
	echo "\"Boot acpi=off\"  \"root=UUID=$ROOT_UUID rw $rootflags $kernel_ops resume=UUID=$SWAP_UUID systemd.unit=multi-user.target acpi=off\"" >> $mnt/boot/refind_linux.conf
	echo "\"Boot read only\"  \"root=UUID=$ROOT_UUID ro $rootflags $kernel_ops resume=UUID=$SWAP_UUID\"" >> $mnt/boot/refind_linux.conf
	echo "\"Boot no options\"  \"root=UUID=$ROOT_UUID rw $rootflags resume=UUID=$SWAP_UUID\"" >> $mnt/boot/refind_linux.conf

#	sed -i 's/#textonly/textonly/g; s/timeout .*/timeout 3/g; s/#also_scan_dirs boot,@/also_scan_dirs +,boot,@/g; s/#scan_all_linux_kernels false/scan_all_linux_kernels false/g' $mnt$efi_path/EFI/BOOT/refind.conf

	mkdir -p $mnt$efi_path/EFI/BOOT/
cat > $mnt$efi_path/EFI/BOOT/refind.conf <<EOF
timeout 2 
#scan_all_linux_kernels off
#also_scan_dirs +,boot,@/boot
showtools install, shell, bootorder, gdisk, memtest, mok_tool, about, hidden_tags, reboot, exit, firmware, fwupdate
#enable_touch
textonly
EOF

	rm -rf /boot/grub

	echo -e "\nYou should have a fully bootable system now. Feel free to test it.\n"

}


install_limine () {

	mount_disk

	pacstrap_install limine

	mkdir -p $mnt$efi_path/EFI/limine

	cp $mnt/usr/share/limine/BOOTX64.EFI $mnt$efi_path/EFI/limine/
	
	efibootmgr \
      --create \
      --disk $disk \
      --part $espPartNum \
      --label "Arch Linux Limine Bootloader" \
      --loader '\EFI\limine\BOOTX64.EFI' \
      --unicode \
      --verbose
	  	
	ROOT_UUID=$(blkid -s UUID -o value $disk$rootPart)
 

	cat > $mnt$efi_path/EFI/limine/limine.conf << EOF

timeout: 5

/Arch Linux
    protocol: linux
    path: boot():/vmlinuz-linux
    cmdline: root=UUID=$ROOT_UUID rw
    module_path: boot():/initramfs-linux.img 
   
EOF

		
}


flash_drive () {

	[ "$(echo $disk | grep /dev/sd)" ] && echo true

}


install_grub () {

	mount_disk

	pacstrap_install grub os-prober efibootmgr inotify-tools lz4


	[[ $fstype = bcachefs ]] && extra_ops='rootfstype=bcachefs'

	if [ "$encryptLuks" = 'true' ]; then

		ENCRYPT_UUID=$(blkid | grep /dev/sda4 | awk -F\" '{print $2}')
		echo "Encrypt UUID: $ENCRYPT_UUID"

		extra_ops="$extra_ops cryptdevice=UUID=$ENCRYPT_UUID:root root=/dev/mapper/root"

	fi

	# Check if we're on a flash drive
	if [ ! "$(flash_drive)" ]; then
		
		SWAP_UUID=$(blkid -s UUID -o value $disk$swapPart)
		extra_ops="$extra_ops resume=UUID=$SWAP_UUID"
	
	fi

	grub-install --target=x86_64-efi --efi-directory=$mnt$efi_path --bootloader-id=GRUB --removable --recheck $disk --boot-directory=$mnt/boot

	cat > $mnt/etc/default/grub << EOF2

GRUB_DISTRIBUTOR=""
GRUB_DEFAULT=saved
GRUB_SAVEDEFAULT=true
GRUB_DISABLE_SUBMENU=true
GRUB_TERMINAL_OUTPUT="console"
GRUB_CMDLINE_LINUX="$kernel_ops $extra_ops"
GRUB_DISABLE_RECOVERY="true"
#GRUB_HIDDEN_TIMEOUT=1
GRUB_RECORDFAIL_TIMEOUT=1
GRUB_TIMEOUT=1
GRUB_DISABLE_OS_PROBER=true

# Update grub with:
# grub-mkconfig -o /boot/grub/grub.cfg

EOF2


	# Remove grub os-prober message
	#sed -i 's/grub_warn/#grub_warn/g' $mnt/etc/grub.d/30_os-prober

	###  Offer readonly grub booting option  ###
	#cp $mnt/etc/grub.d/10_linux $mnt/etc/grub.d/10_linux-readonly
	#sed -i 's/\"\$title\"/\"\$title \(readonly\)\"/g' $mnt/etc/grub.d/10_linux-readonly
	#sed -i 's/ rw / ro /g' $mnt/etc/grub.d/10_linux-readonly

	#cp $mnt/etc/grub.d/10_linux $mnt/etc/grub.d/10_linux-nomodeset
	#sed -i 's/\"\$title\"/\"\$title \(nomodeset\)\"/g' $mnt/etc/grub.d/10_linux-nomodeset
	#sed -i 's/ rw / rw nomodeset /g' $mnt/etc/grub.d/10_linux-nomodeset


	if [[ $mnt = '' ]]; then
		grub-mkconfig -o $mnt/boot/grub/grub.cfg
	else
		arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
	fi

	# btrfs cannot store saved default so will result in spare file error
	if [ "$fstype" = "btrfs" ] && [ "$bootOwnPartition" = "false" ]; then
		sed -i 's/^GRUB_DEFAULT=.*/GRUB_DEFAULT=0/' $mnt/etc/default/grub	
		sed -i 's/^GRUB_SAVEDEFAULT=.*/GRUB_SAVEDEFAULT=false/' $mnt/etc/default/grub
	fi


	# So systemd won't remount as 'rw'
	#systemctl --root=$mnt mask systemd-remount-fs.service

	echo -e "\nYou should have a fully bootable system now. Feel free to test it.\n"

}



install_EFISTUB () {

	echo "TODO."

	return 0

	mount_disk

	SWAP_UUID=$(blkid -s UUID -o value $disk$swapPart)
	ROOT_UUID=$(blkid -s UUID -o value $disk$rootPart)


	echo '# mkinitcpio preset file for the 'linux' package

#ALL_config="/etc/mkinitcpio.conf"
ALL_kver="/boot/vmlinuz-linux"

PRESETS=('default' 'fallback')

#default_config="/etc/mkinitcpio.conf"
default_image="/boot/initramfs-linux.img"
#default_uki="/efi/EFI/Linux/arch-linux.efi"
#default_options="--splash /usr/share/systemd/bootctl/splash-arch.bmp"

#fallback_config="/etc/mkinitcpio.conf"
fallback_image="/boot/initramfs-linux-fallback.img"
#fallback_uki="/efi/EFI/Linux/arch-linux-fallback.efi"
fallback_options="-S autodetect"
' > $mnt/etc/mkinitcpio.d/linux.preset

	#[ ! -f $mnt/usr/bin/efibootmgr ] && pacstrap -K $mnt efibootmgr
	pacstrap_install efibootmgr

	arch-chroot $mnt /bin/bash -e << EOF

	mkinitcpio -p linux

	efibootmgr --create --disk $disk --part $espPart --label "Arch Linux" --loader "/vmlinuz-linux" --unicode "root=$ROOT_UUID rootflags=subvol=@ resume=$SWAP_UUID rw initrd=\EFI\boot\initramfs-linux.img"

	# An example of deleting entries:
	# efibootmgr -B 0001 -L 'Arch Linux 2'
	efibootmgr --unicode


EOF

}



install_uki () {

	echo "TODO."

	return 0	

	mount_disk

	SWAP_UUID=$(blkid -s UUID -o value $disk$swapPart)
	ROOT_UUID=$(blkid -s UUID -o value $disk$rootPart)

	echo "layout=uki" > $mnt/etc/kernel/install.conf

	echo "root=UUID=$ROOT_UUID rw resume=UUID=$SWAP_UUID" > $mnt/etc/kernel/cmdline

	echo 'ALL_config="/etc/mkinitcpio.conf"
ALL_kver="/boot/vmlinuz-linux"
ALL_microcode=(/boot/*-ucode.img)

PRESETS=("default")

default_image="/efi/EFI/arch/initramfs-linux.img"
#default_image="/boot/initramfs-linux.img"
default_uki="/efi/EFI/Linux/arch-linux.efi"' > $mnt/etc/mkinitcpio.d/linux.preset

	mkdir -p $mnt/efi/EFI/{Linux,arch}

	arch-chroot $mnt /bin/bash -e << EOF
mkinitcpio -p linux

efibootmgr --create --disk $disk --part $espPart --label "Arch Linux" --loader '\EFI\Linux\arch-linux.efi' --unicode -o 0007,0004,0003

EOF

}



install_SYSTEMDBOOT () {

	echo "TODO"

	exit

	#arch-chroot $mnt bootctl --esp-path=/efi --boot-path=/boot install
	#arch-chroot $mnt bootctl --esp-path=/efi install
	arch-chroot $mnt bootctl install
	arch-chroot $mnt bootctl update
	#arch-chroot $mnt systemctl enable systemd-boot-update.service 
 	systemctl enable systemd-boot-update.service --root=/$mnt

}



general_setup () {

	mount_disk

	#	echo "$password" | passwd --stdin && echo "Root password created."
	echo "root:$password" | chpasswd --root=$mnt/ && echo "Root password created."

	echo -e 'en_US.UTF-8 UTF-8\nen_US ISO-8859-1' > $mnt/etc/locale.gen  
	echo 'LANG=en_US.UTF-8' > $mnt/etc/locale.conf
	echo 'Arch-Linux' > $mnt/etc/hostname
	echo 'KEYMAP=us' > $mnt/etc/vconsole.conf
	ln -sf $mnt/usr/share/zoneinfo/$timezone $mnt/etc/localtime

	echo "127.0.0.1   localhost
::1         localhost
127.0.1.1   $hostname.localdomain   $hostname" > $mnt/etc/hosts

	echo '[[ $- != *i* ]] && return
alias ls="ls --color=auto"
alias grep="grep --color=auto"
alias vi="vim"
alias hx="helix"
alias arch="sudo /usr/local/bin/arch.sh"
setfont -d
PS1="# "' > $mnt/root/.bashrc

	if [[ $mnt = '' ]]; then
		hwclock --systohc
		locale-gen
	else
		arch-chroot $mnt hwclock --systohc
		arch-chroot $mnt locale-gen
	fi

	cat > $mnt/root/.vimrc << EOF

au BufReadPost *
    \ if line("'\"") > 0 && line("'\"") <= line("$") && &filetype != "gitcommit" | 
    \ execute("normal \`\"") | 
    \ endif

set mouse=c

syntax on

set tabstop=3
set shiftwidth=3
set autoindent
set smartindent

EOF

	auto_login root

}



setup_user () {

	mount_disk

	pacstrap_install sudo

	if [ "$(grep -c "^$user" $mnt/etc/passwd)" -eq 0 ]; then
		useradd --root=$mnt/ -m $user -G wheel -p "$password"
		echo "$user:$password" | chpasswd --root=$mnt/ && echo "User password created."
	fi
	
	if [ "$(grep -c "^$user" $mnt/etc/passwd)" -eq 0 ]; then
		echo -e "\nUser was not created. Exiting.\n"
		exit
	fi


	rm -rf $mnt/home/$user/.cache

	# TODO: find a way to create symbolic link from host
	if [[ $mnt = '' ]]; then
		#ln -s $mnt/tmp $mnt/home/$user/.cache
		ln -s $mnt/run/user/1000 $mnt/home/$user/.cache
		visudo -c
	else
		#arch-chroot $mnt ln -s /tmp /home/$user/.cache
		arch-chroot $mnt ln -s /run/user/1000 /home/$user/.cache
		arch-chroot $mnt visudo -c
	fi

	chown -R $user:$user $mnt/home/$user/.cache


	mkdir -p -m 750 $mnt/etc/sudoers.d
	echo '%wheel ALL=(ALL:ALL) ALL' > $mnt/etc/sudoers.d/1-wheel
	echo "$user ALL = NOPASSWD: /usr/local/bin/arch.sh" > $mnt/etc/sudoers.d/10-arch
	chmod 0440 $mnt/etc/sudoers.d/{1-wheel,10-arch}
	
	
	sudo -u $user mkdir -p $mnt/home/$user/{.local/bin,Documents,Downloads}


	echo '
# If running bash
if [ -n "$BASH_VERSION" ]; then

        # include .bashrc if it exists
        if [ -f "$HOME/.bashrc" ]; then
                . "$HOME/.bashrc"
   fi
fi

export PATH="$HOME/.local/bin:$PATH"
export EDITOR=/usr/bin/vim
#export QT_QPA_PLATFORM=wayland
export QT_IM_MODULE=Maliit
export QT_AUTO_SCREEN_SCALE_FACTOR=1
export KWIN_IM_SHOW_ALWAYS=1
export PATH="$HOME/.local/bin:$PATH"

manager=""
default_manager=kde-dbus

if [[ -z $DISPLAY && $(tty) == /dev/tty1 && $XDG_SESSION_TYPE == tty ]]; then

#	read -t 2 -s -n1 -p "Press <w> to choose a window manager..." key;echo;echo;

	if [ "$key" = "w" ]; then
		if [ "$manager" = "choice" ] || [ "$manager" = "" ]; then
			echo -e "Choose a window manager:
		1. none
		2. weston
		3. kde
		4. kde-mobile
		5. gnome
		6. phosh
		7. kde (dbus)\n"
         read manager
      fi
	fi

	[ "$manager" = "" ] && manager=$default_manager

	case $manager in
		1|none)			;;
		2|weston)		exec weston --shell=desktop ;;
		3|kde)			startplasma-wayland ;;
		4|kde-mobile)	startplasmamobile ;;
		5|gnome)			MOZ_ENABLE_WAYLAND=1 gnome-session --session=gnome-wayland ;;
		6|phosh)			phosh-session ;;
 		7|kde-dbus)		/usr/lib/plasma-dbus-run-session-if-needed /usr/bin/startplasma-wayland ;;
	esac

fi

' > $mnt/home/$user/.bash_profile

	touch $mnt/home/$user/.hushlogin
	
	echo '[[ $- != *i* ]] && return
	
alias ls="ls --color=auto"
alias grep="grep --color=auto"
alias vi="vim"
alias hx="helix"
alias arch="sudo /usr/local/bin/arch.sh"
PS1="$ "' > $mnt/home/$user/.bashrc

	cp $mnt/root/.vimrc $mnt/home/$user/.vimrc

	chown $user:$user $mnt/home/$user/{.vimrc,.hushlogin,.bash_profile,.bashrc}

	ls -la $mnt/home/$user
	
	auto_login user

}



setup_iwd () {

	mount_disk


   #setup_dhcp


	pacstrap_install iw iwd
	
	mkdir -p $mnt/etc/iwd $mnt/var/lib/iwd

	echo '[General]
EnableNetworkConfiguration=true

[Scan]
DisablePeriodicScan=true' > $mnt/etc/iwd/main.conf

	if [[ ! $mnt = '' ]]; then
		[ -d /var/lib/dhcpcd ] && cp -r /var/lib/dhcpcd $mnt/var/lib/
		[ -d /var/lib/iwd ] && cp -r /var/lib/iwd $mnt/var/lib/
	fi

	echo "Enabling network services..."
	systemctl --root=$mnt enable iwd.service

}

setup_networkmanager () {

   mount_disk

   #setup_dhcp

   pacstrap_install networkmanager 
                              
   systemctl --root=$mnt enable NetworkManager.service
	
	#	echo "[device]
	#wifi.backend=iwd" > $mnt/etc/NetworkManager/conf.d/wifi_backend.conf
	
	if [[ -d /etc/NetworkManager ]] && [[ ! $mnt = '' ]]; then
		cp -r /etc/NetworkManager $mnt/etc/ 
	fi
}


setup_wpa () {

	mount_disk

	#setup_dhcp


	pacstrap_install iw wpa_supplicant


	systemctl --root=$mnt enable wpa_supplicant.service

	mkdir -p $mnt/etc/wpa_supplicant
	
	if [[ -d /etc/wpa_supplicant ]] && [[ ! $mnt = '' ]]; then
		cp -r /etc/wpa_supplicant $mnt/etc/ 
	fi

	systemctl --root=$mnt enable wpa_supplicant@wlo1.service

}



setup_dhcp () {

	mount_disk


	pacstrap_install dhcpcd


	# Helps with slow booting caused by waiting for a connection
	mkdir -p $mnt/etc/systemd/system/dhcpcd@.service.d/
	echo '[Service]
ExecStart=
ExecStart=/usr/bin/dhcpcd -b -q %I' > $mnt/etc/systemd/system/dhcpcd@.service.d/no-wait.conf

	#[ "$(cat $mnt/etc/dhcpcd.conf | grep noarp)" ] && echo noarp >> $mnt/etc/dhcpcd.conf
	[ "$(grep noarp $mnt/etc/dhcpcd.conf)" ] && echo noarp >> $mnt/etc/dhcpcd.conf

	echo "Enabling dhcp services..."
	systemctl --root=$mnt enable dhcpcd.service

}


connect_wireless () {


   if [[ "$wifi_ssid" = "" ]] || [[ "$wifi_pass" = "" ]]; then
   	read -p "Please enter SSID: " $wifi_ssid
   	read -p "Please enter password: " $wifi_pass

	fi

   echo -e "\nWhich network manager would you like to connect to?\n"

   net_choices=(quit dhcpcd iwd wpa_supplicant networkmanager)
   select net_choice in "${net_choices[@]}"
   do
      case $net_choice in
         "quit")           break ;;
         "dhcp")           dhcpcd ;;
         "iwd")            

	iwctl station wlan0 scan

	while [ "$(iwctl station $wlan get-networks | grep 'No networks available')" ]; do
   	echo "Attempting to connected to $wlan..."
   	iwctl station $wlan get-networks
   	sleep 1
	done

	echo "$wlan connected!"

	echo "Waiting 5 seconds before connecting to $wlan..."
	sleep 5

	iwctl --passphrase $wifi_pass station $wlan connect $wifi_ssid
	echo "Waiting 5 seconds to ping..."
	sleep 5

	echo "Attempting to ping google.ca..."

	ping -c 1 -i 1 -q google.ca

									;;
         "wpa_supplicant") 

	interface=$(iw dev | awk '$1=="Interface"{print $2}')
	#interface=$(cat /proc/net/wireless | perl -ne '/(\w+):/ && print $1')
	
	wpa_passphrase "$wifi_ssid" "$wifi_pass" > /etc/wpa_supplicant/$interface.conf
	cp /etc/wpa_supplicant/$interface.conf /etc/wpa_supplicant/wpa_supplicant-$interface.conf
	#wpa_supplicant -B -i $interface -c <(wpa_passphrase $wifi_ssid $wifi_pass)

	systemctl enable wpa_supplicant@$interface.service
   systemctl start wpa_supplicant@$interface.service

			;;
         "networkmanager") 
										nmcli radio wifi on
										nmcli device wifi connect $wifi_ssid password $wifi_pass 
										;;
         '')               	echo -e "\nInvalid option!\n" ;;
      esac
   done

}



install_aur () {

	check_on_root
	mount_disk

# sudo cp /usr/lib/libalpm.so.15 /usr/lib/libalpm.so.13

	#pacstrap_install git less fakeroot pkg-config
	pacstrap_install git base-devel


	[ "$aur_app" = "paru" ] && pacstrap_install cargo
	#[ "$aur_app" = "none" ] && pacstrap_install cargo


#	arch-chroot $mnt /bin/bash << EOF
#
#cd $aur_path

#sudo -u $user git clone https://aur.archlinux.org/$aur_app.git

#cd $aur_app
#sudo -u $user makepkg -si

#sudo -u $user $aur_app --gendb

#chown -R $user:$user /home/$user/$aur_app

#EOF


	#[ "$aur_app" = "paru" ] && arch-chroot $mnt pacman -R rust
	#[ "$aur_app" = "yay" ] && arch-chroot $mnt pacman -R rust

	#rm -rf $mnt/home/$user/{.cargo,$aur_app/*} $mnt/usr/lib/{go,rustlib}
echo "To install an AUR package:

	sudo git clone https://aur.archlinux.org/<aur-package>.git
	cd <aur-package>
	makepkg -si"

}



setup_acpid () {

	mount_disk

	# Or install will generate an error: acpid: /usr/lib/systemd/system/acpid.service exists in filesystem
	rm -rf /usr/lib/systemd/system/acpid.service
	#pacstrap_install acpid cpupower htop power-profiles-daemon
	pacstrap_install acpid cpupower htop

		echo '[Unit]
Description=AC user power service

[Service]
ExecStart=/home/user/.local/bin/power.sh

[Install]
WantedBy=multi-user.target' > $mnt/etc/systemd/system/user-power.service

	ln -s -f $mnt/etc/systemd/system/user-power.service $mnt/etc/systemd/system/multi-user.target.wants/user-power.service


# Needs to be reimplimented after hibernation

echo '#!/bin/sh

case $1/$2 in
  pre/*)
    echo "Going to $2..."
    ;;
  post/*)
    echo "Waking up from $2..."
         echo 85 > /sys/class/power_supply/BAT0/charge_control_end_threshold
    ;;
esac' > $mnt/usr/lib/systemd/system-sleep/sleep.sh 


	# TODO find a way to activate sysd service
	arch-chroot $mnt systemctl enable acpid.service

	echo 'SUBSYSTEM=="power_supply", ATTR{online}=="0", RUN+="/home/user/.local/bin/pluggedin.sh false"
SUBSYSTEM=="power_supply", ATTR{online}=="1", RUN+="/home/user/.local/bin/pluggedin.sh true"' > $mnt/etc/udev/rules.d/powersave.rules

echo '[Unit]
Description=ACPI event daemon
Documentation=man:acpid(8)

[Service]
ExecStart=/usr/bin/acpid --foreground --netlink

[Install]
WantedBy=multi-user.target' > $mnt/usr/lib/systemd/system/acpid.service

	# Manually enable it
	ln -f -s $mnt/usr/lib/systemd/system/acpid.service $mnt/etc/systemd/system/multi-user.target.wants/acpid.service

}


install_backup () {

	choice=$1

	if [[ $choice = '' ]]; then
		echo -e "What backup system would you like to install?\n\n1. rsync \n2. snapper\n3. grub-btrfsd\n4. none\n"
			read -p "Choice: " -n 2 choice
	else
		echo "Installing $choice..."
	fi

			
	case $choice in
		1|rsync)					pacstrap_install rsync ;;

		2|snapper)				if [[ $fstype = btrfs ]]; then
										snapper_setup
										install_grub-btrfsd
									else
										echo -e "\nNot installing snapper as it is not compatablie with $fstype. Exiting.\n"
									fi
									;;
		
		3|grub-btrfsd)			[ $fstype = btrfs ] && install_grub-btrfsd 
									;;

		4|none|'')				echo "No backup installed." ;;

		*)							echo "Not an option. Exiting." ;;
	esac


}


install_grub-btrfsd () {

   pacstrap_install cronie grub-btrfs

   systemctl --root=$mnt enable cronie.service
   systemctl --root=$mnt enable grub-btrfsd.service

}



snapper_setup () {
	
	mount_disk

   if [[ $mnt = '' ]]; then

		pacstrap_install snapper
		
		if [[ "$(mount | grep $mnt$snapshot_dir)" ]]; then
			umount $mnt$snapshot_dir
		fi

		rm -rf $mnt$snapshot_dir

		if [[ ! -f $mnt/etc/snapper/configs/root ]]; then
			snapper -c root create-config /
		fi
		
		mount -a

		ID=$(btrfs su list / | grep -E "level 256 path .snapshots$" | awk '{print $2}')
   	echo -e "Deleting: ID $ID level 256 path .snaphots..."

		btrfs su delete -i $ID /

		mkdir -p $snapshot_dir
		mount -a

	else
		
		pacstrap_install snapper

		arch-chroot $mnt /bin/bash << EOF
umount $snapshot_dir
rm -r $snapshot_dir
sync
sleep 1

snapper --no-dbus -c root create-config /
sleep 1
btrfs subvolume delete $snapshot_dir
sleep 1
mkdir $snapshot_dir
mount -a
chmod 750 $snapshot_dir

btrfs su list /

sed -i 's/TIMELINE_CREATE="yes"/TIMELINE_CREATE="no"/; s/TIMELINE_CLEANUP="yes"/TIMELINE_CLEANUP="no"/' /etc/snapper/configs/root

EOF

	fi

}



do_backup () {

	[[ ! $backup_install = true ]] && return
	
	mount_disk
	
	case $backup_type in

		none|'')		echo ;;

		rsync)		if [[ $fstype = btrfs ]] || [[ $fstype = bcachefs ]]; then
							take_snapshot "$1"
						else
							echo "Backup not yet implimented for $fstype."
						fi
						;;

		snapper)		if [[ $fstype = btrfs ]]; then
							arch-chroot $mnt snapper --no-dbus create -d "$1"
							arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
						else
							echo "Will not do a backup with snapper on $fstype."
						fi
						;;
	esac

}


install_tweaks () {

	mount_disk
	
	#pacstrap_install terminus-font ncdu
	pacstrap_install ncdu

	echo 'vm.swappiness = 10' > $mnt/etc/sysctl.d/99-swappiness.conf
	echo 'vm.vfs_cache_pressure=50' > $mnt/etc/sysctl.d/99-cache-pressure.conf
	echo 'net.ipv4.tcp_fin_timeout = 30' > $mnt/etc/sysctl.d/99-net-timeout.conf
	echo 'net.ipv4.tcp_keepalive_time = 120' > $mnt/etc/sysctl.d/99-net-keepalive.conf

	echo 'vm.dirty_background_ratio=5' > $mnt/etc/sysctl.d/99-dirtyb-background.conf
	echo 'vm.dirty_ratio=10' > $mnt/etc/sysctl.d/99-dirty-ratio.conf

	systemctl --root=$mnt enable systemd-oomd

	echo 'kernel.core_pattern=/dev/null' > $mnt/etc/sysctl.d/50-coredump.conf

	mkdir -p $mnt/etc/systemd/coredump.conf.d/
	echo '[Coredump]
Storage=none
ProcessSizeMax=0' > $mnt/etc/systemd/coredump.conf.d/custom.conf

	echo '* hard core 0' > $mnt/etc/security/limits.conf


}



install_mksh () {

	mount_disk
	
	install_aur_packages mksh
	
	echo "HISTFILE=/root/.mksh_history
HISTSIZE=5000
export VISUAL=emacs
export EDITOR=/usr/bin/vim
set -o emacs" > $mnt/root/.mkshrc

   echo "HISTFILE=/home/$USER/.mksh_history
HISTSIZE=5000
export VISUAL=emacs
export EDITOR=/usr/bin/vim
set -o emacs" > $mnt/home/$user/.mkshrc
	
	chown $user:$user $mnt/home/$user/.mkshrc

	#echo -e 'export ENV="/home/$USER/.mkshrc"' >> $mnt/home/$user/.profile
	#chown $user:$user $mnt/home/$user/.profile 

	if [[ $mnt = '' ]]; then

		chsh -s /usr/bin/mksh
		echo $password | sudo -u $user chsh -s /bin/mksh

	else

		arch-chroot $mnt /bin/bash << EOF
chsh -s /usr/bin/mksh
echo $password | sudo -u $user chsh -s /bin/mksh
EOF

	fi

	cp $mnt/home/$user/.bash_profile $mnt/home/$user/.profile


}


add_hooks () {

	mount_disk
	echo -e "Adding $1 - $2..."

	if [ "$(grep ^$1= $mnt/etc/mkinitcpio.conf | grep -v $2)" ] && [ "$1" = "HOOKS" ]; then
		
		sed -i "s/filesystems/filesystems $2/" $mnt/etc/mkinitcpio.conf
	
	elif [ "$(grep ^$1= $mnt/etc/mkinitcpio.conf | grep -v $2)" ] && [ "$1" = "MODULES" ]; then
		sed -i "s/MODULES=(/MODULES=($2 /" $mnt/etc/mkinitcpio.conf

	fi

	sed -i "s/ )/)/" $mnt/etc/mkinitcpio.conf

return

#addThis='MODULES=liveroot|HOOKS=liveroot'
#addThis="$1"

	addThis="$1=$2"

	#$mnt/etc/mkinitcpio.conf


	awk -F'[=()]' -v add="${addThis}" '
  BEGIN {
    split(add, t, "[|=]")
    for(i=1; i in t;i+=2)
      addT[t[i]]=t[i+1]
  }
  $1 in addT {
	 sub( /\([^)]*/, ("& " addT[$1]) )
  }
  ' "$2" 
#> "/temp.txt"#  cat /temp.txt


  #cp /temp.txt "$2"
  #rm -rf /temp.txt

}


install_hooks () {

	mount_disk
	choice=$1

	if [[ $choice = '' ]]; then
		echo -e "What hooks would you like to install?\n\n1. liveroot \n2. overlayroot\n3. btrfs-rollback\n4. bcachefs-rollback\n5. exit\n"
			read -p "Choice: " -n 2 choice
	else
		echo "Installing $choice hook..."
	fi

			
	case $choice in
		1|liveroot)		[ ! "$fstype" = "btrfs" ] && echo "Not a btrfs file system." && return

							touch $mnt/etc/vconsole.conf
							pacstrap_install rsync squashfs-tools


							[ "$(grep 'add_binary btrfs' $mnt/usr/lib/initcpio/install/liveroot)" ] || sed -i 's/build() {/& \n        add_binary btrfs/g' $mnt/usr/lib/initcpio/install/liveroot
							
							add_hooks HOOKS liveroot


							# So systemd won't remount as 'rw'
							#systemctl --root=$mnt mask systemd-remount-fs.service

							# Don't remount /efi either
							#systemctl --root=$mnt mask efi.mount
							;;

		2|overlayroot)	
							#https://aur.archlinux.org/packages/overlayroot
							#https://github.com/hilderingt/archlinux-overlayroot
							
							#	pacman --noconfirm -U /home/user/.local/bin/overlayroot*.zst

							add_hooks MODULES overlay
                     add_hooks HOOKS overlayroot

	cp $mnt/etc/grub.d/10_linux $mnt/etc/grub.d/10_linux-overlay
	sed -i 's/\"\$title\"/\"\$title \(overlayroot\)\"/g' $mnt/etc/grub.d/10_linux-overlay
	sed -i 's/ rw / rw overlayroot /g' $mnt/etc/grub.d/10_linux-overlay
	

							if [[ $mnt = '' ]]; then
								grub-mkconfig -o /boot/grub/grub.cfg
							else
								arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
							fi

							echo -e "\nAdd 'overlayroot' to kernal parameters to run\n"
							;;

		3|btrfs-rollback)		pacstrap_install rsync squashfs-tools

										add_hooks MODULES btrfs 
										add_hooks MODULES squashfs 
                     			add_hooks HOOKS btrfs-rollback
	
							;;

		4|bcachefs-rollback)		pacstrap_install rsync squashfs-tools

										add_hooks MODULES bcachefs 
										add_hooks MODULES squashfs 
                     			add_hooks HOOKS bcachefs-rollback
	
							;;

		*) echo -e "\nInvalid choice: $choice\n"; return ;;
	esac

	if [[ $mnt = '' ]]; then
		mkinitcpio -p linux 
	else
		arch-chroot $mnt mkinitcpio -p linux 
	fi

}


reset_keys () {

	#rm -rf /etc/pacman.d/gnupg
	pacman-key --init
	pacman-key --populate

}



do_chroot () {

	check_on_root
	mount_disk

	mount -t proc /proc $mnt/proc/
	mount -t sysfs /sys $mnt/sys/
	mount -o bind /dev $mnt/dev/
	mount -o bind /run $mnt/run/
	mount -o bind /sys/firmware/efi/efivars $mnt/sys/firmware/efi/efivars/

	cp /etc/resolv.conf $mnt/etc/resolv.conf

	echo -e "\e[0;42m\n \nEntering chroot. Type 'exit' to leave.\n\e[0;29m\n"
		
	chroot $mnt /bin/bash -ic 'exec env PS1="(chroot) # " bash --norc'

	echo -e "\e[0;42m\n \nExiting chroot.\n\e[0;29m\n"

	unmount_disk

}


download_script () {

	echo -e "\nDowloading script from Github..."

	curl -sL https://raw.githubusercontent.com/bathtime/arch/main/arch.sh > $mnt/$arch_path/$arch_file
	chmod +x $mnt/$arch_path/$arch_file

}

clone () {

	check_on_root
	mount_disk

	check_pkg rsync

	echo -e "\n$1 $3 -> $4. Please be patient...\n"
	
	rsync_excludes=" --exclude=/run/timeshift/ --exclude=/etc/timeshift/timeshift.json --exclude=/etc/fstab --exclude=/etc/default/grub --exclude=/root.squashfs --exclude=/lost+found/ --exclude=$snapshot_dir/ --exclude=/dev/ --exclude=/proc/ --exclude=/sys/ --exclude=/tmp/ --exclude=/run/ --exclude=/var/tmp/ --exclude=/var/lib/dhcpcd/ --exclude=/var/log/ --exclude=/var/lib/systemd/random-seed --exclude=/root/.cache/ --exclude=/media/ --exclude=/mnt/ --exclude=/home/$user/.cache/ --exclude=/home/$user/.local/share/Trash/ --exclude=/snapshots/"


	rsync --dry-run -v $2 $rsync_excludes $3 $4 | less

	echo -e "\nType 'y' to proceed with rsync or any other key to exit..."
	read choice

	if [[ $choice = y ]]; then

		echo -e "\nRunning rsync...\n"
	
		rsync $2 --info=progress2 $rsync_excludes $3 $4
		#tar -C $3 -cf - . | tar -C $4 -xf -

		arch-chroot $mnt pacman -S --noconfirm linux	
  		install_grub
		choose_initramfs dracut
		choose_initramfs booster 
		setup_fstab
		#fs_packages
		
		pacstrap_install $(fs_packages)

		read -p "You must check that file system packages have been installed!"

	else
		echo "Exiting."
	fi

}


rsync_snapshot () {

	mount_disk	

	mkdir -p $mnt$snapshot_dir

	echo -e "\nList of snapshots:\n"
	ls -1N $mnt$snapshot_dir/

	if [[ $1 = '' ]]; then
		echo -e "\n\nWhat would you like to call this rsync snapshot?\n"
		read snapshotname
	else
		snapshotname="$1"
	fi


		echo -e "\nRunning dry run first..."
		sleep 1

		rsync_params="-axHAXSW --del --exclude=/run/timeshift/ --exclude=/etc/timeshift/timeshift.json --exclude=/lost+found/ --exclude=/dev/ --exclude=/proc/ --exclude=/sys/ --exclude=/tmp/ --exclude=/run/ --exclude=/var/tmp/ --exclude=/var/lib/dhcpcd/ --exclude=/var/log/ --exclude=/var/lib/systemd/random-seed --exclude=/root/.cache/ --exclude=/boot/ --exclude=/efi/ --exclude=/media/ --exclude=/mnt/ --exclude=/home/$user/.cache/ --exclude=/home/$user/.local/share/Trash/ --exclude=$mnt/ --exclude=$snapshot_dir/"
		
		rsync --dry-run -v $rsync_params / "$mnt$snapshot_dir/$snapshotname" | less

		echo -e "\nType 'y' to proceed with rsync or any other key to exit..."
		read choice

		if [[ $choice = y ]]; then

			echo -e "\nRunning rsync...\n"
			rsync $rsync_params --info=progress2 / "$snapshot_dir/$snapshotname"
	
		else
			echo "Exiting."
		fi

}


take_snapshot () {

	mount_disk
	
	filename=$(date +"%Y-%m-%d @ %H:%M:%S")
	
	echo -e "\nList of snapshots:\n"
	ls -1N $mnt$snapshot_dir/

	if [[ $1 = '' ]]; then
		echo -e "\n\nWhat would you like to call this snapshot (Name must not contain any spaces!)?\n"
		read snapshotname
	else
		snapshotname="$1"
	fi

	#[[ ! $snapshotname = '' ]] && snapshotname="$filename - $snapshotname"

	if [[ $fstype = bcachefs ]]; then
		
		if [ "$mnt" = '' ]; then
			echo -e "\nCreating snapshot: $mnt$snapshot_dir/$snapshotname...\n" 
			bcachefs subvolume snapshot -r $mnt/ "$mnt$snapshot_dir/$snapshotname"
		else
			# Saving as r/o is a bad idea. Too many errors at startup
			echo -e "\nCreating snapshot: $snapshot_dir/$snapshotname...\n" 
			arch-chroot $mnt bcachefs subvolume snapshot -r / "$snapshot_dir/$snapshotname"
		fi

	fi
	if [[ $fstype = btrfs ]]; then
		btrfs subvolume snapshot $mnt/ "$mnt$snapshot_dir/$snapshotname"
	fi

	echo -e "\nCreated snapshot: $mnt$snapshot_dir/$snapshotname\n"	
	
	ls -1N $mnt$snapshot_dir/
	echo

	# To update grub to include bootable snapshots	
	if [[ $fstype = btrfs ]]; then

	   if [[ $mnt = '' ]]; then
      	grub-mkconfig -o $mnt/boot/grub/grub.cfg
   	else
      	arch-chroot $mnt grub-mkconfig -o /boot/grub/grub.cfg
   	fi

	fi

}


restore_snapshot () {

	mount_disk	

	echo -e "\nChoose a host:\n"

	select host in $mnt$snapshot_dir/* $mnt$snapshot_dir/*/snapshot @ 'subvolid=256' / /mnt quit; do
      case host in
        	*) 		echo -e "\nYou chose: $host\n"; break ;;
     	esac
  	done

	[ $host = 'quit' ] && return
	
	echo -e "\nChoose a target:\n"

	select target in $mnt$snapshot_dir/* $mnt$snapshot_dir/*/snapshot @ 'subvolid=256' / /mnt quit; do
      case target in
        	*)			echo -e "\nYou chose: $target\n"; break ;;
     	esac
  	done
	
	[ $target = 'quit' ] && return


	echo -e "\nRestoring $host to $target...\n"
	sleep 1


	rsync_params="-axHAXSW --del --exclude=/var/lib/machines/ --exclude=/var/lib/portables/ --exclude=/etc/timeshift/timeshift.json --exclude=/run/timeshift/ --exclude=/lost+found/ --exclude=/dev/ --exclude=/proc/ --exclude=/sys/ --exclude=/tmp/ --exclude=/run/ --exclude=/var/tmp/ --exclude=/var/lib/dhcpcd/ --exclude=/var/log/ --exclude=/var/lib/systemd/random-seed --exclude=/root/.cache/ --exclude=/boot/ --exclude=/efi/ --exclude=/media/ --exclude=/mnt/ --exclude=/home/$user/.cache/ --exclude=/home/$user/.local/share/Trash/ --exclude=$mnt/ --exclude=$snapshot_dir/ --exclude=/@snapshots/ --exclude=/@var/tmp/ --exclude=$mnt2/ --exclude=$mnt3"

	
	if [[ $(echo $host | grep $snapshot_dir) ]]; then
			
		host=$host/

	elif [[ $host = @ ]]; then
			
		host=$mnt2
			
		[ $fstype = 'btrfs' ] && btrfs su list /
			
		#mount -t btrfs --mkdir -o subvol=@ $disk$rootPart $host
		mount -t $fstype --mkdir -o subvol=@ $disk$rootPart $host
	
	elif [[ $host = 'subvolid=256' ]]; then
			
		host=$mnt2
			
		subvolid=$(btrfs su list / | grep 'path @$' | awk '{print $2}')
		btrfs su list /
			
		mount -t btrfs --mkdir -o subvolid=$subvolid $disk$rootPart $host
			
	fi


	if [[ $(echo $target | grep $snapshot_dir) ]]; then
			
		target=$target/

	elif [[ $target = @ ]]; then
			
		target=$mnt3
			
		[ $fstype = 'btrfs' ] && btrfs su list /
			
		mount -t $fstype --mkdir -o subvol=@ $disk$rootPart $target
	
	elif [[ $target = 'subvolid=256' ]]; then
			
		target=$mnt3
			
		subvolid=$(btrfs su list / | grep 'path @$' | awk '{print $2}')
		btrfs su list /
			
		mount -t btrfs --mkdir -o subvolid=$subvolid $disk$rootPart $target
		
	fi

	
	rsync --dry-run -v $rsync_params $host $target | less

	
	read -p "Type 'y' to proceed with rsync or any other key to exit..." choice
		
	if [[ $choice = y ]]; then

		rsync --info=progress2 $rsync_params $host $target

	else
		echo "Exiting."
	fi

}


delete_snapshot () {

	mount_disk	

	echo -e "\nWhich snapshot would you like to delete?\n"

	cd $mnt$snapshot_dir
	
   select snapshot in *
	do
		case snapshot in
         *) echo -e "\nYou chose: $snapshot\n"; break ;;
      esac
   done

	if [ -d "$mnt$snapshot_dir/$snapshot" ] && [ ! "$snapshot" = '' ]; then
		
		if [ "$fstype" = "btrfs" ]; then
			btrfs subvolume delete "$mnt$snapshot_dir/$snapshot"
		elif [ "$fstype" = "bcachefs" ]; then
			
			bcachefs subvolume delete "$mnt$snapshot_dir/$snapshot"
			
			if [[ $(ls $snapshot_dir | grep "^$snapshot$") ]]; then
				
				echo -e "File $mnt$snapshot_dir/$snapshot still exists.\n
Press 'y' to run rm -rf.\n"
				read -sn1 key

				[ $key = 'y' ] && rm -rf --no-preserve-root "$mnt$snapshot_dir/$snapshot"
			
			fi
		fi

	else
		echo "Snapshot directory does not exist. Exiting."
	fi

}


delete_all_snapshots () {

	mount_disk

	if [ $fstype = 'bcachefs' ]; then
		
		if [ "$(ls $snapshot_dir)" ]; then

			bcachefs subvolume delete $snapshot_dir/*
			echo -e "\nSnapshots deleted.\n"

			if [ "$(ls $snapshot_dir)" ]; then

				echo -e "\nSnapshit directory not completely empty.\nPress 'y' to erase with rm -rf.\n"
				read -sn 1 choice
				[ $choice = 'y' ] && rm -rf $snapshot_dir/*

			fi

		else
			echo -e "\nNo snapshots to delete.\n"
		fi
	
	fi

}


bork_system () {

	echo -e "\nAre you sure you want to bork the system? Type 'yes' to proceed.\n"

	read choice
	
	[[ ! "$choice" = 'yes' ]] && return

	echo -e "\nCreating read only backup snapshot...\n"
	sleep 1
	
	if [ $fstype = 'btrfs' ]; then
		snapper create -d "Backup before bork"
	elif [ $fstype = 'bcachefs' ]; then
		bcachefs subvolume snapshot -r / "$snapshot_dir/Backup before bork"
	fi
	
	echo -e "\nActivating read only mode...\n"
	sleep 1

	readOnlyBootEfi true true

	sync_disk

	echo -e "\nPausing 3 seconds...\n"
	sleep 2

	cd /
	rm -rf --no-preserve-root /

}


readOnlyBootEfi () {

   mount_disk
	
	[ $2 = 'true' ] && [ "$(mount | grep /boot | grep rw)" ] && grub-mkconfig -o /boot/grub/grub.cfg

   if [ $2 = 'true' ]; then

		#sed -i 's#/efi.*vfat.*rw,#/efi            vfat            ro,#' $mnt/etc/fstab
		#sed -i 's#/boot/grub.*btrfs.*rw#/boot/grub      btrfs           ro#' $mnt/etc/fstab
		
		sed -i -E '/^#/b
			s#([[:space:]]/boot[[:space:]].*[[:space:],])rw([[:space:],])#\1ro\2#
			s#([[:space:]]/boot/grub[[:space:]].*[[:space:],])rw([[:space:],])#\1ro\2#
			s#([[:space:]]/efi[[:space:]].*[[:space:],])rw([[:space:],])#\1ro\2#' /etc/fstab
   
	else
	
		#sed -i 's#/efi.*vfat.*ro,#/efi            vfat            rw,#' $mnt/etc/fstab
		#sed -i 's#/boot/grub.*btrfs.*ro#/boot/grub      btrfs           rw#' $mnt/etc/fstab
   	
		sed -i -E '/^#/b
			s#([[:space:]]/boot[[:space:]].*[[:space:],])ro([[:space:],])#\1rw\2#
			s#([[:space:]]/boot/grub[[:space:]].*[[:space:],])ro([[:space:],])#\1rw\2#
			s#([[:space:]]/efi[[:space:]].*[[:space:],])ro([[:space:],])#\1rw\2#' /etc/fstab
  
	fi

   [ "$(mount | grep '/efi ')" ] && umount /efi
   [ "$(mount | grep '/boot/grub ')" ] && umount /boot/grub
   [ "$(mount | grep '/boot ')" ] && umount /boot

   systemctl daemon-reload
   mount -a

	[ $2 = 'false' ] && grub-mkconfig -o /boot/grub/grub.cfg

   cat $mnt/etc/fstab

}


snapper_status_undochange () {

	snapper list --columns number,date,cleanup,description,read-only

	echo -e "\nEnter first snapshot to compare from.\n
Press <ENTER> for current or 'q' to quit.\n"

	read from 
	[ "$from" = 'q' ] && return
	[ "$from" = '' ] && from=0

echo -e "\nEnter second snapshot to compare to.\n
Press <ENTER> for current or 'q' to quit.\n"

	read to
	[ "$to" = 'q' ] && return
	[ "$to" = '' ] && to=0

	snapper status $from..$to | less

	echo -e "\nEnter 'y' to proceed with change or any other key to exit.\n"

	read -sN1 choice

	if [ "$choice" = 'y' ]; then

				echo -e "\nRunning: snapper undochange $to..$from\n"

		snapper undochange $to..$from
		#pacman -Syyu --noconfirm

	fi

}


snapper-rollback () {

	snapper list --columns number,date,cleanup,description,read-only

	echo -e "\nWhich snapshot would you like to roll back to?\n
(Press <ENTER> to rollback to latest subvolume or 'q' to quit)\n"

	read choice
	
	[ "$choice" = 'q' ] && return
	
	snapper rollback $choice

	echo -e "\nPress 'r' to reboot or any other key to continue.\n"
	read -n 1 -s choice

	if [ $choice = 'r' ]; then
		sync_disk
		grub-mkconfig -o /boot/grub/grub.cfg
		reboot
	fi

}


btrfs-rollback () {

	snapper list --columns number,date,cleanup,description,read-only

	echo -e "\nWhich snapshot is your current default? ('q' to quit)\n"

	read default

	[ $default = 'q' ] && return

	echo -e "\nTarget for roll back? ('q' to quit)\n"

	read target
	
	[ $target = 'q' ] && return

	# You must not be in the directory you're about to move or it's a busy error
	cd /

	mv $snapshot_dir/$default/ $snapshot_dir/$default-rollback/

	btrfs su snapshot $snapshot_dir/$target/snapshot/ $snapshot_dir/$default/snapshot/

	btrfs su set-default $snapshot_dir/$default/snapshot


   echo -e "\nPress 'r' to reboot or any other key to continue.\n"
   read -n 1 -s choice

   if [ $choice = 'r' ]; then
      sync_disk
      reboot
   fi

}


set-default () {

	snapper list --columns number,date,cleanup,description,read-only

	echo -e "\nWhich snapshot would you like to set the default subvolume to? ('q' to quit)\n"

	read choice
	
	[ $choice = 'q' ] && return

	snapshot="$snapshot_dir/$choice/snapshot"

	if [ -d "$snapshot" ]; then 

		btrfs su set-default "$snapshot"
	

	   echo -e "\nPress 'r' to reboot or any other key to continue.\n"
   	read -n 1 -s choice

   	if [ $choice = 'r' ]; then
      	sync_disk
			grub-mkconfig -o /boot/grub/grub.cfg
			reboot
   	fi
	
	else
		echo -e "\n$snapshot not found.\n"
	fi

}


create_snapshot () {

	snapper list --columns number,date,cleanup,description,read-only
	
	echo -e "\nWhat would you like to name this snapshot?\n"
	read snapshot
	
	#touch "/root/snapper-$snapshot"
	
	if [ $1 = 'rw' ]; then
		snapper -c root create --read-write --description "$snapshot"
	else
		snapper -c root create --description "$snapshot"
	fi

	#sync_disk
	
	#sleep .1
	#rm -rf "/root/snapper-$snapshot"

}


btrfs_delete () {

	while true; do

		btrfs su list /

		echo -e "\nWhich subvolume would you like to delete? (q = quit)\n"

		read choice

		[ $choice = 'q' ] && return

		btrfs su delete -i $choice /

		echo

	done

}

snapper_delete () {

	while true; do

		snapper list --columns number,date,cleanup,description,read-only

		echo -e "\nWhich snapshot would you like to delete? (q = quit)\n"

		read choice

		[ $choice = 'q' ] && return

		# To free the space used by the snapshot(s) immediately, use --sync
		snapper -c root delete --sync $choice

		echo

	done

}


snapper_delete_by_date () {

	echo -e "\nHow many hours old to delete? (q = quit)\n"

	read hours

	[ $hours = 'q' ] && return

	mins=$(( hours * 60 ))

	snapshots="$(find $snapshot_dir/ -maxdepth 1 -type d -mmin +$mins | sed 's#/.snapshots/##')"
	for snapshot in $snapshots; do
		echo "Deleting snapshot #$snapshot..."
		snapper -c root delete --sync $snapshot
	done

}




snapper_delete_all () {


	cd $snapshot_dir
	list=*

	snapshots=$(snapper list | grep -v '┼\|  0 \|[0-9][-+\*] \| # ' | awk '{print $1}')

	if [ "$snapshots" ]; then

   	for snapshot in ${snapshots[@]}; do
			
			if [ ! "$snapshot" = 0 ] && [ ! "$snapshot" = '' ] && [ "$(ls "$snapshot" | grep 'snapshot')" ]; then
				echo "snapper -c root delete --sync $snapshot"
				snapper -c root delete --sync $snapshot
			fi

   	done
	
	else
		echo -e "\nNo snapshots to delete."
	fi


	snapshots=$(btrfs su list / | grep $snapshot_dir/ | awk '{ print $9 }' | sed 's/@//')
	default=$(btrfs su get-default / | awk '{ print $9 }' | sed 's/@//')
	current=$(mount | grep ' / ' | sed 's#.*.subvol=/@##; s/)//')

	if [ "$snapshots" ]; then

   	for snapshot in ${snapshots[@]}; do
			
			if [ ! $snapshot = 0 ] && [ ! $snapshot = '' ] && [ ! $snapshot = $default ] && [ ! $snapshot = $current ]; then
      		echo -e "btrfs su delete $snapshot"
				btrfs su delete "$snapshot"
			fi

   	done
	
	else
		echo -e "\nNo snapshots to delete."
	fi


	default=$(btrfs su get-default / | awk '{ print $9 }' | sed 's#@/.snapshots/##; s#/snapshot##')
	current=$(mount | grep ' / ' | sed 's#.*.subvol=/@/.snapshots/##; s/)//; s#/snapshot##')

	all_deletes=''

	echo

  	for snapshot in ${list[@]}; do

		if [ ! "$snapshot" = "$default" ] && [ ! "$snapshot" = "$current" ]; then
			delete="$snapshot_dir/$snapshot"
     		echo -e "Found $delete"
			all_deletes="$all_deletes $delete"
		fi

  	done

	if [ ! "$all_deletes" = '' ]; then
			
		echo -e "\nWould you like to delete these as well? (enter 'y' to delete):\n\n $all_deletes\n"
		read choice
		
		if [ "$choice" = 'y' ]; then
			
			echo

		  	for snapshot in ${list[@]}; do

				if [ ! "$snapshot" = "$default" ] && [ ! "$snapshot" = "$current" ]; then
					delete="$snapshot_dir/$snapshot"
     				echo -e "rm -rf $delete..."
					rm -rf "$delete"
				fi

  			done
  			
		else
			echo -e "Not deleting."
		fi

	else
		echo -e "Nothing else to delete."
	fi


	echo -e "\nDirectory $snapshot_dir:\n"

	ls $snapshot_dir*
	
	echo	
	sleep 4

}


btrfs-maintenance() {

	echo -e "\nRunning scrub...\n"
	btrfs scrub start /
	while ! btrfs scrub status / | grep finished; do
		sleep .5
	done

	echo -e "\nRunning balance...\n"
	btrfs balance start --full-balance / && echo "Finished without error." || echo "Finished with errors."

	echo -e "\nRunning defrag...\n"
	btrfs filesystem defragment -r / && echo "Finished without error." || echo "Finish
ed with errors."

}


extract_archive () {

	check_on_root
	mount_disk

	unsquashfs -d $mnt -f /root.squashfs

}


squashRecover () {

	ls $snapshot_dir

	echo -e "\nWhat would you like to call this snapshot?\n"
	read snapshotname

	rsync --dry-run --info=progress2 -axHAXSW --exclude='/lost+found/' --exclude=/dev/ --exclude=/proc/ --exclude=/sys/ --exclude=/tmp/ --exclude=/run/ --exclude=/var/tmp/ --exclude=/var/lib/dhcpcd/ --exclude=/var/log/ --exclude=/var/lib/systemd/random-seed --exclude=/root/.cache/ --exclude=/boot/ --exclude=/efi/ --exclude=/media/ --exclude=/mnt/ --exclude=/home/$user/.cache/ --exclude=/home/$user/.local/share/Trash/ --exclude=$snapshot_dir/ / "$snapshot_dir/$snapshotname" | less

	echo -e "\nType 'y' to proceed with rsync or any other key to exit..."
	read choice

	if [[ $choice = 'y' ]]; then

		echo -e "\nRunning rsync...\n"
		echo "Running: rsync --info=progress2 -axHAXSW --exclude=$snapshot_dir/ / $snapshot_dir/$snapshotname"

		rsync --info=progress2 -axHAXSW --exclude='/lost+found/' --exclude=/dev/ --exclude=/proc/ --exclude=/sys/ --exclude=/tmp/ --exclude=/run/ --exclude=/var/tmp/ --exclude=/var/lib/dhcpcd/ --exclude=/var/log/ --exclude=/var/lib/systemd/random-seed --exclude=/root/.cache/ --exclude=/boot/ --exclude=/efi/ --exclude=/media/ --exclude=/mnt/ --exclude=/home/$user/.cache/ --exclude=/home/$user/.local/share/Trash/ --exclude=$snapshot_dir/ / "$snapshot_dir/$snapshotname"
	
	else

		echo "Exiting."
	
	fi

}


create_archive () {

	check_pkg squashfs-tools

	echo "Creating archive file..."

	cd / 
	rm -rf /root.squashfs
	
	time mksquashfs / root.squashfs -comp lz4 -mem-percent 25 -no-recovery -noappend -e root.squashfs -e /dev/ -e /proc/ -e /sys -e /tmp/ -e /run -e /mnt -e $snapshot_dir/ -e /home/$user/.cache/ -e /root/.cache/ -e /var/tmp/ -e /var/log/ -e /home/$user/.local/share/Trash/

	ls -lah root.squashfs

	#lzma	2m21	2.3G
	#xz	2m34	2.3G
	#zstd	1m18	2.4G
	#gzip 50s	2.5G
	#lzo	1m13s	2.6G
	#lz4  5.5s	2.8G
	#none 6s		4.2G (-no-compression)

}



install_network () {

	echo -e "\nWhich network manager would you like to install?\n"
		
	net_choices=(quit dhcp iwd wpa_supplicant networkmanager) 
	select net_choice in "${net_choices[@]}"
	do
		case $net_choice in
			"quit")				break ;;
			"dhcp")				setup_dhcp; break ;;
			"iwd")				setup_iwd; break ;;
			"dhcp")				setup_dhcp; break ;;
			"wpa_supplicant")	setup_wpa; break ;;
			"networkmanager") setup_networkmanager; break ;;
			'')					echo -e "\nInvalid option!\n" ;;
		esac
	done

}



install_bootloader () {

	echo -e "\nWhich boot manager would you like to install?\n"

	choiceBoot=(grub rEFInd EFISTUB uki systemD limine quit) 
				
	select choiceBoot in "${choiceBoot[@]}"
	do
		case $choiceBoot in
			"grub")		install_grub; break ;;
			"rEFInd")	install_Refind; break ;;
			"EFISTUB")	install_EFISTUB; break ;;
			"uki")		install_uki; break ;;
			"systemD")	install_SYSTEMDBOOT; break ;;
			"limine")	install_limine; break ;;
			"quit")		break ;;
			'')			echo -e "\nInvalid option!\n" ;;
		esac
	done

}



copy_script () {

	check_on_root
	mount_disk

	echo "Copying $arch_path/$arch_file to $mnt$arch_path/..."

	mkdir -p $mnt$arch_path
	cp $arch_path/$arch_file $mnt$arch_path

}


install_aur_packages () {

	mount_disk

	apps="$1"
	packages=''
	cd $aur_apps_path

	if [ "$apps" = '' ]; then
		
		select choice in 'quit' *.*tar.zst; do
			case choice in
				*)		apps="$(echo $choice | sed 's/-[0-9].*//')"; break ;;
			esac
		done
	
		[ "$apps" = 'quit' ] && return

	fi


	for app in $apps; do
		
		package="$(ls $app-[0-9]*.*tar.zst)"

		if [ "$mnt" = '' ]; then
		
			pacman -U --noconfirm $package
		
		else
		
			echo -e "\nCopying $aur_apps_path$package to $mnt$aur_apps_path..."

			cp $aur_apps_path$package $mnt$aur_apps_path

			echo -e "\nInstalling $package...\n"

			arch-chroot $mnt pacman -U --noconfirm "$aur_apps_path$package"
	
		fi
	
	done

}


pacstrap_install () {


	if [ "$reinstall" -eq 1 ]; then

		packages="$@"

		[ "$copy_on_host" -eq 1 ] && pacman --noconfirm -S ${packages[@]}

	else

		packages=""

		for package in "$@"; do

			if [ ! "$(pacman --sysroot $mnt -Q $package 2>/dev/null)" ]; then
				packages="$package $packages"	
			else
				echo "Package: $package already installed."
			fi

		done

	fi

	if [[ "$packages" ]]; then

		if [[ "$mnt" = '' ]]; then
			pacman --noconfirm --needed -S ${packages[@]}
		else

			if [ "$copy_on_host" -eq 1 ] && [ ! "$(pacman -Q $package 2>/dev/null)" ]; then
				pacman --needed --noconfirm -S ${packages[@]}
			fi

			if [ "$offline" -eq 1 ]; then
				pacstrap -C /etc/pacman-offline.conf -c -K $mnt ${packages[@]}
			else

				# Ideal when using an arch install disk in ram
				if [ "$copy_on_host" -eq 0 ]; then
					pacstrap -K $mnt ${packages[@]}
				else
					pacstrap -c -K $mnt ${packages[@]}
				fi

			fi
		fi
	fi
}



custom_install () {

	#check_on_root
   mount_disk

	echo -e "Which package(s) would you like to install?\n"
	read packages

	if [ "$packages" ]; then
		pacstrap_install $packages
	else
		echo "No packages will be installed."
	fi

	copy_pkgs

}



copy_pkgs () {

	check_on_root
	mount_disk

	# Check which packages are installed on chroot system
	packages="$(pacman --sysroot $mnt -Q | sed 's/ /-/g; s/$/-/g')"

	# Copy only packages from host system that are installed on chroot system
	for package in ${packages}; do

		if [ "$(ls /var/cache/pacman/pkg/ | grep $package*)" ]; then

			echo -n "Copying $package... "

			# Only copy if the package is newer or nonexistant
			cp -u /var/cache/pacman/pkg/$package* $mnt/var/cache/pacman/pkg/ && echo "[done]"
		fi

	done

	echo -e "\nTotal packages: $(ls $mnt/var/cache/pacman/pkg/*.zst | wc -l)\n"


	#echo -e "\nUpdating package database. Please be patient...\n"

	#pacman-db-upgrade

	#repo-add -q -n /var/cache/pacman/pkg/./custom.db.tar.gz /var/cache/pacman/pkg/*.zst
	#repo-add -q -n $mnt/var/cache/pacman/pkg/./custom.db.tar.gz $mnt/var/cache/pacman/pkg/*.zst

	#pacman -U /mnt/var/cache/pacman/pkg/*.pkg.tar.zst
	#pacman -U /var/cache/pacman/pkg/*.pkg.tar.zst

	#echo -e "\nCopying AUR directory. Please be patient...\n"

}



update_mirrorlist () {

	#check_pkg reflector
	#echo -e "\nRunning reflector...\n"
	#reflector > /etc/pacman.d/mirrorlist
	
	echo -e "\nUpdating mirror list. Please be patient...\n"

	curl -s "https://archlinux.org/mirrorlist/?country=CA&country=US&protocol=https&use_mirror_status=on" | sed -e 's/^#Server/Server/' -e '/^#/d' | rankmirrors -n 10 - > /etc/pacman.d/mirrorlist

	cat /etc/pacman.d/mirrorlist
}



check_online () {

	curl -Is  http://www.google.com &>/dev/null && online=1 || online=0

	if [ $online -eq 0 ]; then
		echo -e "\nNo internet connection found. Offline mode enabled."
		offline=1 
	fi

}


window_manager () {

	#sed -i "s/^manager=.*$/manager=$1/" $mnt/home/$user/.bash_profile
	
	sed -i "s/^manager=.*/manager=\'$1\'/; s/^default_manager=.*$/default_manager=\'$2\'/ " $mnt/home/$user/.bash_profile

}


auto_install_root () {

	create_partitions
	install_base
	install_grub
	#install_Refind
	setup_fstab

	[ $autologin = true ] && auto_login root

	choose_initramfs $initramfs
	#choose_initramfs dracut 
	#choose_initramfs mkinitcpio
	#choose_initramfs booster
	

	general_setup

	#setup_iwd
	setup_networkmanager
	#setup_wpa
	
	install_tweaks
	
	#copy_pkgs
	copy_script
	
	#[ "$aur_apps_root" ] && install_aur_packages "$aur_apps_root"

	install_backup $backup_type
	do_backup "root-installed"
	
	sync_disk

}



auto_install_user () {

	auto_install_root
	
	pacstrap_install $user_install
	
	setup_user

	pacstrap_install sudo

	[ $autologin = true ] && auto_login $user

	copy_pkgs
	
	window_manager '' 'none'
	
	[ "$aur_apps_user" ] && install_aur_packages "$aur_apps_user"
	chaotic_aur

	do_backup "user-installed"
	sync_disk

}


auto_install_kde () {

	auto_install_user
 
	pacstrap_install $kde_install

	copy_pkgs


	install_config
	#install_mksh
	
	window_manager '' 'kde'		# 'kde' or 'kde-dbus'

	#install_hooks liveroot
	#install_hooks overlayroot

	if [ "$fstype" = 'btrfs' ]; then
		install_hooks btrfs-rollback
	elif [ "$fstype" = 'bcachefs' ]; then
		install_hooks bcachefs-rollback
	else
		install_hooks overlayroot
	fi

	[ "$aur_apps_kde" ] && install_aur_packages "$aur_apps_kde"

	do_backup "kde-installed"
	
	sync_disk

}


auto_install_gnome () {

	auto_install_user
	
	pacstrap_install $gnome_install
 
	copy_pkgs

	window_manager=gnome

	install_config
	do_backup "Gnome-installed"

}


auto_install_gnomekde () {

	auto_install_user

	pacstrap_install $gnome_install $kde_install
	
	copy_pkgs

	window_manager=choice

	install_config
	do_backup "Gnome-KDE-installed"

}


auto_install_cage () {

	auto_install_user
	
	pacstrap_install $cage_install

	copy_pkgs

	window_manager=cage

	install_config

	do_backup "Cage-installed"

}



auto_install_weston () {

	auto_install_user

	pacstrap_install $weston_install

	copy_pkgs
	
	window_manager=weston

	install_config
	do_backup "Weston-installed"

}


auto_install_phosh () {

	auto_install_user
	
	pacstrap_install $phosh_install

	copy_pkgs

	window_manager=phosh

	install_config
}


auto_install_all () {

	auto_install_user
	
	pacstrap_install $gnome_install $kde_install $weston_install

	copy_pkgs

	window_manager=choice

	install_config
}



clean_system () {

	#echo "Cleaning unused locales..."
	#ls /usr/share/locales/ | grep -xv "en_US" | xargs rm -r


	echo "Cleaning ~/.cache..."
	rm -rf /home/user/.cache/*
	

	echo "Cleaning brave..."

	rm -rf /home/$user/.config/BraveSoftware/Brave-Browser/component_crx_cache/* rm -rf /home/user/.config/BraveSoftware/Brave-Browser/Default/Service\ Worker/CacheStorage/*

	echo "Cleaning mozilla..."

	cd /home/$user/.mozilla/firefox
	rm -rf 'Crash Reports' 'Pending Pings'

	rm -rf /usr/lib/firefox/crashreporter /usr/lib/firefox/minidump-analyzer /usr/lib/firefox/pingsender

	profile=$(ls /home/user/.mozilla/firefox/ | grep .*.default-release)
	cd $profile

	rm -rf crashes minidumps datareporting sessionstore-backups saved-telemetry-pings storage browser-extension-data security_state gmp-gmpopenh264 synced-tabs.db-wal places.sqlite favicons.sqlite cert9.db places.sqlite-wal storage-sync-v2.sqlite-wal webappsstore.sqlite gmp-widevinecdm
	

	rm -rf storage/default 'Crash Reports/events' webappsstore.sqlite formhistory.sqlite sessionCheckpoints.json sessionstore.jsonlz4 sessionstore-backups content-prefs.sqlite places.sqlite favicons.sqlite storage.sqlite storage-sync-v2.sqlite bounce-tracking-protection.sqlite permissions.sqlite protections.sqlite cookies.sqlite cookies.sqlite-wal 


	#echo "Cleaning chromium..."

	#cd /home/$user/.config/chromium

	#rm -rf Safe\ Browsing component_crx_cache WidevineCdm IndexedDB GrShaderCache OnDeviceHeadSuggestModel hyphen-data ZxcvbnData ShaderCache Default/{Service\ Worker/,IndexedDB,History,GPUCache,Sessions,DawnCache,Extension\ State,Web\ Data,Visited\ Links}
	#rm -rf Safe\ Browsing component_crx_cache WidevineCdm GrShaderCache OnDeviceHeadSuggestModel hyphen-data ZxcvbnData ShaderCache Default/{Service\ Worker/,History,GPUCache,Sessions,DawnCache,Extension\ State,Web\ Data,Visited\ Links}

}


chaotic_aur () {

	mount_disk

	if [ "$mnt" = '' ]; then
		
		pacman-key --recv-key 3056513887B78AEB --keyserver keyserver.ubuntu.com
		pacman-key --lsign-key 3056513887B78AEB
		
		pacman -U --noconfirm 'https://cdn-mirror.chaotic.cx/chaotic-aur/chaotic-keyring.pkg.tar.zst'
		pacman -U --noconfirm 'https://cdn-mirror.chaotic.cx/chaotic-aur/chaotic-mirrorlist.pkg.tar.zst'

	else

		arch-chroot $mnt pacman-key --recv-key 3056513887B78AEB --keyserver keyserver.ubuntu.com
		arch-chroot $mnt pacman-key --lsign-key 3056513887B78AEB
		
		arch-chroot $mnt pacman -U --noconfirm 'https://cdn-mirror.chaotic.cx/chaotic-aur/chaotic-keyring.pkg.tar.zst'
		arch-chroot $mnt pacman -U --noconfirm 'https://cdn-mirror.chaotic.cx/chaotic-aur/chaotic-mirrorlist.pkg.tar.zst'

	fi

	if [ ! "$(cat $mnt/etc/pacman.conf | grep 'chaotic-aur')" ]; then

		echo '
[chaotic-aur]
Include = /etc/pacman.d/chaotic-mirrorlist' >> $mnt/etc/pacman.conf
	
	fi
	
	if [ "$mnt" = '' ]; then
		pacman -Sy
	else
		arch-chroot $mnt pacman -Sy
	fi

}


backup_config () {


	echo -e "\nCreating backup file. Please be patient...\n"
	
	#sudo -u $user tar -pcf $backup_file $CONFIG_FILES
	#tar -pcf $backup_file $CONFIG_FILES

	rm -rf /home/$user/.local/share/Trash/*
	rm -rf /home/$user/.cache/*

	#tar -pcf $backup_file $CONFIG_FILES
	time tar -pczf $backup_file $CONFIG_FILES
	
	#echo -e "\nVerifying file contents. Please be patient...\n"
	#tar xOf $backup_file &> /dev/null; echo $?
	#tar -tf $backup_file &> /dev/null; echo $?

	#sudo -u $user gpg --yes -c $backup_file

	#ls -lah $backup_file $backup_file.gpg
	
	ls -lah $backup_file

}



restore_config () {

	mount_disk
	
	cd $mnt/
	
	echo -e "\nVerifying file contents. Please be patient...\n"
	#tar xOf $backup_file &> /dev/null; echo $?
	tar -tf $mnt$backup_file &> /dev/null; echo $?

	echo "Extracting setup file..."
	time tar -xvf $backup_file
	
	chown -R $user:$user $mnt/home/$user/

}



install_config () {

	check_on_root
	mount_disk

	#read -p "Press any key when ready to enter password."
	#echo "Decrypting setup file..."
	#gpg --yes --output /home/$user/$backup_file --decrypt /home/$user/$backup_file.gpg


	#cp /home/$user/$backup_file{,.gpg} $mnt/home/$user/

	if [ $install_backup = true ]; then

		echo "Copying $backup_file to $mnt$backup_file..."
		cp $backup_file $mnt$backup_file

		if [ ! $copy_user_dir = true ]; then
			echo "Extracting setup files..."
			arch-chroot $mnt tar -xvf $backup_file --directory /
			arch-chroot $mnt chown -R $user:$user /home/$user/
		fi
	fi

	if [ $copy_user_dir = true ]; then
		echo "Copying config files to $mnt..."
		cp -rv --parents $CONFIG_FILES $mnt

		echo "Changing /home/$user permissions..."
		chmod -R $user:$user /home/$user/
				
		#echo "Copying /home/$user/* to $mnt/home/user..."
		#cp -rv /home/$user/{,.}* $mnt/home/$user/
	fi

	#count=ls -1 $mnt$aur_apps_path*.zst 2>/dev/null | wc -l
	count=0

	if [ $count != 0 ]; then
	
		echo -e "\nYou have $count packages in $aur_apps_path...\n"
	
		if [[ $mnt = '' ]]; then
			#sudo pacman -U --noconfirm /home/user/.local/bin/*.zst
			# Fix shared libraries error
			sudo ln -s /usr/lib/libalpm.so.15 /usr/lib/libalpm.so.13
		else
			#arch-chroot $mnt sudo pacman -U --noconfirm /home/user/.local/bin/*.zst
   		arch-chroot $mnt ln -s /usr/lib/libalpm.so.15 /usr/lib/libalpm.so.13
		fi

	fi

}



last_modified () {

	find / -cmin -1 -printf '%t %p\n' | sort -k 1 -n | cut -d' ' -f2- | grep -v /proc

}



edit_arch () {

	if [ "$(ls $arch_path | grep $arch_file)" ]; then
		$editor $arch_path/$arch_file && exit
	fi

}



wipe_disk () {

	check_on_root
	unmount_disk

	echo -e "\nDisk: $disk_info\n\nType 'yes' to wipe using $1 method.\n"
	read choiceWipe
	
	if [[ $choiceWipe = yes ]]; then

		echo -e "\nWiping $disk using $1 method. Please be patient...\n"

		error_bypass=1
		time dd if=/dev/$1 of=$disk bs=1M status=progress
		error_bypass=0

	else
		echo -e "\nNot wiping.\n"
	fi

}


clean () {

	echo -e "\nCleaning files...\n"
	rm -rfv		/var/log /var/tmp /tmp/{*,.*} /home/$user/.cache/{*,.*} \
					/root/.cache/{*,.*} /root/.bash_history 

	echo
	echo -e "\nClearing pagecache, dentries, and inodes...\n\nBefore:"
	free -h
	sudo sync; echo 3 > /proc/sys/vm/drop_caches
	echo -e "\nAfter:"
	free -h

}


wipe_freespace () {

	echo -e "\nWiping freespace on $disk using zero method. Please be patient...\n"
	echo -e "Run 'watch -n 1 -x df / --sync' in another terminal to see progress.\n"

	error_bypass=1

	cd /

	file=$(cat /dev/urandom | tr -cd 'a-f0-9' | head -c 32).tmp
	(dd if=/dev/zero of=$file bs=4M status=progress; dd if=/dev/zero of=$file.small bs=256 status=progress) &>/dev/null


	sync ; sleep 60 ; sync
	rm $file $file.small

	error_bypass=0

}



disk_info () {

	echo -ne "\nDisk: $(lsblk --output=PATH,SIZE,MODEL,TRAN -dn $disk) "

	if [[ "$mounted_on" = "" ]]; then
		echo "(unmounted)"
	elif [[ "$mounted_on" = "/" ]]; then
		echo "(host)"
	else
		echo "(mounted on $(lsblk -no MOUNTPOINT $disk$rootPart))"
	fi

   echo "File type: $fstype"

}



sync_disk () {

	echo -e "\nSyncing disk. Please be patient...\n"
	sync

	return

	dirty_threshold=0

	dirty=$(cat /proc/meminfo | awk '/Dirty:/ { print $2 }')
	initial_dirty=$dirty
	stall_count=0

	while [[ $dirty -gt $dirty_threshold ]]; do

		dirty=$(cat /proc/meminfo | awk '/Dirty:/ { print $2 }')
		[ $dirty -ne 0 ] && perc=$(( 100 - (dirty * 100 / initial_dirty) )) || perc=100 

		printf '\rSyncing: %i kB... (%i%%)   ' $dirty $perc

		if [[ $dirty -ge $last_dirty ]]; then
			stall_count=$(( stall_count + 1 ))
		else
			stall_count=0
		fi

		last_dirty=$dirty
		
		# If it's taking too long, it might be stuck and need to be resynced
		if [ $stall_count -gt 50 ]; then
			sync &
			stall_count=0
			echo "Resyncing..."
		fi

		sleep .1

	done

	echo
	sleep .5

}


aur_package_install () {

	check_on_root
	mount_disk

	#aur_app=brave-bin
	#aur_git='https://aur.archlinux.org/brave-bin.git'

	aur_app="$1"
	aur_git="$2"

	echo -e "Copying $aur_path/$aur_app..."

	mkdir -p $mnt$aur_path
	
	cp -r $aur_path/$aur_app $mnt$aur_path	
	chown -R user:user $mnt$aur_path
	ls -la $mnt$aur_path

	echo -e "Finished!"


		arch-chroot $mnt /bin/bash << EOF

	cd $aur_path

	sudo -u $user git clone $aur_git

	cd $aur_app
	sudo -u $user makepkg -sf

	echo "Installing $aur_path/$aur_app/$aur_app-*.zst..."
	pacman -U $aur_path/$aur_app/$aur_app-*.zst

EOF

}


benchmark () {
		
	mount_disk

	tempfile="$mnt/bench.tmp"
   
	choice=("1. Quit
2. dd
3. iozone
10. All")

   echo
   echo "${choice[@]}" | column
   echo  

   read -p "Which benchmark to run? " choice

   case $choice in
      quit|1)        ;;

		dd|2)

	cd $mnt/
	rm -rf $tempfile
	

	echo -e "\nEnter how many gigs to run:\n"
	read gigs
	
	echo -e "\nRunning dd to measure write speed...\n"
	dd if=/dev/zero of=$tempfile bs=$gigs'M' count=1024 conv=fdatasync,notrunc status=progress

	echo 3 > /proc/sys/vm/drop_caches
	echo -e "\nRunning dd to measure read speed...\n"
	dd if=$tempfile of=/dev/null bs=$gigs'M' count=1024 status=progress

	echo -e "\nRunning dd to measure buffer-cache speed...\n"
	dd if=$tempfile of=/dev/null bs=$gigs'M' count=1024 status=progress

	echo -e "\nRunning bash open/close test to measure buffer-cache speed..."
	time for (( i=1; i<=1000; i++ )); do bash -c 'exit' ;done

	

	echo -e "\nPress any key to continue."
	read -s -N 1
	rm $tempfile
	echo

	;;
	iozone|3)	echo -e "\nSequential 1M read/write:\n"
					#iozone -e -I -s 1g -r 1m -i 0 -i 1
					iozone -e -I -s 1g -r 1m -i 0 -i 1

					echo -e "\nSequential 128k read/write:\n"
					iozone -e -I -s 1g -r 128k -i 0 -i 1

					echo -e "\nRandom 4k read/write:\n"
					iozone -e -I -s 1g -r 4k -i 0 -i 2 -i 1

					;;
	esac

}


install_group () {

	config_choices=("1. Quit
2. kde
3. gnome")			

	config_choice=0

	while [ ! "$config_choice" = "1" ]; do

		echo
		echo "${config_choices[@]}" | column
		echo  

		read -p "Which option? " config_choice

   	case $config_choice in
   		quit|1)		echo "Quitting!"; break; ;;
   		kde|2)		pacstrap_install $kde_install ;;
   		gnome|3)		pacstrap_install $gnome_install ;;
   		'')			last_modified ;;
   		*)				echo -e "\nInvalid option ($config_choice)!\n" ;;
		esac

	done

}



snapshots_menu () {
	
	config_choices=("
1. Quit to main menu
2. Snapper snapshot (ro)
3. Snapper snapshot (rw)
4. Snapper status/undochange
5. Snapper set default
6. Snapper rollback
7. Snapper delete
8. Snapper delete by date
9. Snapper delete recovery
10. Snapper delete all
11. Rsync snapshot
12. Take btrfs/bcachefs snapshot
13. Restore btrfs/bcachefs snapshot
14. Delete btrfs/bcachefs snapshot
15. Delete all bcachefs snapshots
16. Btrfs delete subvolume
17. Bork system
18. btrfs rollback
19. btrfs scrub/balance/defrag")
	
	config_choice=0
	while [ ! "$config_choice" = "1" ]; do
		if [ $fstype = 'btrfs' ]; then
			snapper list --columns number,description,date,read-only
			echo
			btrfs su list $mnt/
		elif [ $fstype = 'bcachefs' ]; then

			echo -e "\nSnapshot directory:\n"
			ls /.snapshots/
			echo

		fi

		echo
		echo "${config_choices[@]}" | column
		echo  

		echo -e "Which option?\n"
		read config_choice

		case $config_choice in
			quit|1)					echo "Quitting!"; break ;;
			snapper|2)				create_snapshot ro ;;
			snapper|3)				create_snapshot rw ;;
			status|4)				snapper_status_undochange ;;
			default|5)				set-default ;;
			roll|6)					snapper-rollback ;;
			snapper-del|7)			snapper_delete ;;
			snapper-date|8) 		snapper_delete_by_date ;;
			delete-rec|9) 			snapper_delete_recovery ;;
			delete-all|10)			snapper_delete_all ;;
			rsync|11)				rsync_snapshot ;;
			snapshot|12)			take_snapshot ;;
			restore|13)				restore_snapshot ;;
			delete|14)				delete_snapshot ;;
			del-all|15)				delete_all_snapshots ;;
			btrfs-del|16)			btrfs_delete ;;
			bork|17)					bork_system ;;
			btrfs-rollback|18)	btrfs-rollback ;;
			btrfs-maitenance|19)	btrfs-maintenance ;;
	      '')						;;
      	*)							echo -e "\nInvalid option ($config_choice)!\n" ;;
		esac

	done 

}

clone_menu () {
	
	config_choices=("1. Quit to main menu
2. Unsquash to target
3. Part + Clone / -> $disk
4. Clone / -> $disk
5. Clone $disk -> /
6. Copy / -> $disk
7. Copy $disk -> /
8. Copy /home -> $disk$rootPart/
9. Copy $disk$rootPart/home -> /
10. Update / <-> $disk
11. Wipe (zero)
12. Wipe (urandom)
13. Wipe freespace     
14. Shred $disk
15. Create squashfs image
16. Squashfs recover to @root (bcachefs)")

	config_choice=0
	while [ ! "$config_choice" = "1" ]; do

		if [ $fstype = btrfs ]; then
			echo
			snapper list --columns number,description,date
			echo
			btrfs su list /
		fi

		echo
		echo "${config_choices[@]}" | column
		echo  

		echo -e "Which option?\n"
		read config_choice

		case $config_choice in
			quit|1)			echo "Quitting!"; break ;;
			unsquash|2)		extract_archive ;;
			clone|3)			create_partitions
								time clone Cloning '-aSW --del' / $mnt/ ;;
			clone|4)			time clone Cloning '-aSW --del' / $mnt/ ;; 
			clone|5)			clone Cloning '-aSW --del' $mnt/ / ;;
			copy|6)			clone Copying -aSW / $mnt/ ;;
			copy|7)			clone Copying -aSW $mnt/ / ;;
			copy|8)			clone Copying -aSW /home $mnt/ ;;
			copy|9)			clone Copying -aSW $mnt/home / ;;
			update|10)		clone Updating -auSW / $mnt/
								clone Updating -auSW $mnt/ / ;;
			wipe|11)			wipe_disk zero ;;
			wipe|12)			wipe_disk urandom ;;
			wipe-free|13)	wipe_freespace ;;
			shred|14)		unmount_disk; shred -n 1 -v -z $disk ;;
			squashfs|15)	create_archive ;;
			squashrec|16)	squashRecover ;;
      	'')				;;
      	*)					echo -e "\nInvalid option ($config_choice)!\n" ;;
		esac

	done 

}


setup_menu () {

	config_choices=("1. Quit
2. Backup config
3. Restore config
4. Install config
5. Cleanup system
6. Last modified
7. Copy config
8. Copy packages")			

	config_choice=0

	while [ ! "$config_choice" = "1" ]; do

		echo
		echo "${config_choices[@]}" | column
		echo  

		read -p "Which option? " config_choice

  		case $config_choice in
  			quit|1)		echo "Quitting!"; break; ;;
  			backup|2)	backup_config ;;
  			restore|3)	restore_config ;;
  			install|4)	install_config ;;
   		clean|5)		clean_system ;;
   		last|6)		last_modified ;;
			copy|7)		echo "Copying packages from $backup_file to $mnt/..."
							cp $backup_file $mnt/ ;;
			pkgs|8)		copy_pkgs ;;
         '')			last_modified ;;
         *)				echo -e "\nInvalid option ($config_choice)!\n" ;;
		esac

	done 

}


packages_menu () {

	config_choices=("1. Quit to main menu
2. Reset pacman keys
3. Update mirror list
4. Copy packages
5. Download script
6. Copy script
7. DL bcache script
8. Add chaotic aur")

	config_choice=0
	while [ ! "$config_choice" = "1" ]; do

		echo
		echo "${config_choices[@]}" | column
		echo  

		read -p "Which option? " config_choice

   	case $config_choice in
   		quit|1)				echo "Quitting!"; break ;;
			reset|keys|2)		reset_keys ;;
			mirrorlist|3)		update_mirrorlist ;;
			pkgs|4)				copy_pkgs ;;
			script|5)			download_script ;;
			copy_script|6)		copy_script ;;
			rollback|7)			echo -e "\nDowloading script from Github..."
            
	curl -sL https://raw.githubusercontent.com/bathtime/arch/refs/heads/main/hooks/btrfs-rollback > $mnt/lib/initcpio/hooks/btrfs-rollback           
   chmod +x $mnt/lib/initcpio/hooks/btrfs-rollback

	curl -sL https://raw.githubusercontent.com/bathtime/arch/refs/heads/main/install/btrfs-rollback > $mnt/lib/initcpio/install/btrfs-rollback
	chmod +x $mnt/lib/initcpio/install/btrfs-rollback
  
	#curl -sL https://raw.githubusercontent.com/bathtime/bcachefs-rollback/refs/heads/main/hooks/bcachefs-rollback > $mnt/lib/initcpio/hooks/bcachefs-rollback           
   #chmod +x $mnt/lib/initcpio/hooks/bcachefs-rollback

	#curl -sL https://raw.githubusercontent.com/bathtime/bcachefs-rollback/refs/heads/main/install/bcachefs-rollback > $mnt/lib/initcpio/install/bcachefs-rollback
	#chmod +x $mnt/lib/initcpio/install/bcachefs-rollback
  
 
									mkinitcpio -P
									;;
			8|chaotic)			chaotic_aur ;;
   		'')					last_modified ;;
			*)						echo -e "\nInvalid option ($config_choice)!\n" ;;
		esac

	done

}


auto_install_menu () {
	
	config_os=("1. Quit
2. Root
3. User
4. Weston
5. KDE
6. Gnome
7. Phosh
8. Cage
9. Gnome/Kde
10. All")

	echo
	echo "${config_os[@]}" | column
	echo  

	read -p "Which option? " config_os

   case $config_os in
   	quit|1)			;;
   	root|2)			time auto_install_root ;;
   	user|3)			time auto_install_user ;;
   	weston|4)		time auto_install_weston ;;
   	kde|5)			time auto_install_kde ;;
   	gnome|6)			time auto_install_gnome ;;
   	phosh|7)			time auto_install_phosh ;;
   	cage|8)			time auto_install_cage ;;
		gnomekde|9)		time auto_install_gnomekde ;;
      all|10)			time auto_install_all ;;
     	'')				;;
	  	*)					echo -e "\nInvalid option ($config_os)!\n" ;;
	esac

}



if [ "$1" ]; then
	disk="$1"
else
	choose_disk
fi

check_viable_disk

disk_info

check_online &


while :; do

	choices=("1. Back to main menu 
2. Edit $arch_file in $editor
3. Chroot
4. Auto-install
5. Partition disk
6. Install base
7. Install boot manager
8. Setup fstab
9. General setup
10. Setup user
11. Setup network
12. Install aur
13. Install tweaks
14. Install mksh
15. Install hooks
16. Setup acpid
17. Choose initramfs
18. Install aur packages
19. Mount $mnt
20. Unmount $mnt
21. Update grub
22. Connect wireless
23. Install backup
24. Packages/script
26. Auto-login root
27. Auto-login user
28. Hypervisor setup
29. Test
30. Custom install
31. Setup ~ files
32. Snapshot ->
33. clone/sync/wipe ->
34. Install aur packages ->
35. Install group packages ->
36. Benchmark
37. r/o boot/efi on
38. r/o boot/efi off")


echo
echo "${choices[@]}" | column   

echo -ne "\nEnter an option: "

read choice

echo

	case $choice in
		Quit|quit|q|exit|1)	choose_disk ;;
		arch|2)					edit_arch;;
		Chroot|chroot|3)		do_chroot ;;
		auto|4)					auto_install_menu ;;
		partition|5)			create_partitions ;;
		base|6)					install_base ;;
		boot|7)					install_bootloader ;;
		fstab|8)					setup_fstab ;;
		setup|9)					general_setup ;;
		user|10)					setup_user ;;
      network|11)				install_network ;;
		aur|12)					install_aur ;;
		tweaks|13)				install_tweaks ;;
      mksh|14)					install_mksh ;;
		hooks|15)				install_hooks ;;
		acpid|16)				setup_acpid ;;
		initramfs|17)			choose_initramfs ;;
      backup|18)				install_aur_packages ;;
		mount|19)				mount_disk  ;;
      unmount|20)				unmount_disk  ;;
		grub|21)					grub-mkconfig -o $mnt/boot/grub/grub.cfg ;;
		connect|iwd|22)		connect_wireless ;;
		backup|23)				install_backup ;;
		packages|24)			packages_menu ;;
		loginroot|26)			auto_login root ;;
		loginuser|27)			auto_login user ;;
		hypervisor|28)			hypervisor_setup ;;

		test|29)					
									install_aur_packages 'snapper mksh'

									;;
		custom|30)				custom_install ;;
		setup|31)				setup_menu ;;
		snapshot|32)			snapshots_menu ;;
		copy|33)					clone_menu ;;
		setup|34)      		aur_package_install ;; 
		packages|35)			install_group ;;
		benchmark|36)			benchmark ;;
      readOnlyTrue|37)     readOnlyBootEfi true true ;;
      readOnlyFalse|38)    readOnlyBootEfi false false ;;

		'')						;;
		*)							echo -e "\nInvalid option ($choice)!\n"; ;;
	esac

	sync_disk
	disk_info

done