150 lines
3.4 KiB
Bash
Executable File
150 lines
3.4 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
|
|
# THE RAINBOW HAHAHAH
|
|
RED='\e[31m'
|
|
GREEN='\e[32m'
|
|
YELLOW='\e[33m'
|
|
RESET='\e[0m'
|
|
|
|
# root check
|
|
if [[ $EUID -ne 0 ]]; then
|
|
echo -e "${RED}You must be root to do this." 1>&2
|
|
exit 100
|
|
fi
|
|
|
|
mount_points=("/dev" "/dev/pts" "/proc" "/sys" "/sys/firmware/efi/efivars" "/run")
|
|
NO_WRITE=false
|
|
|
|
# arguments
|
|
for arg in "$@"; do
|
|
case $arg in
|
|
--nowrite)
|
|
NO_WRITE=true
|
|
shift
|
|
;;
|
|
-h)
|
|
printf "%-20s %s\n" "Flags:" "Description:"
|
|
echo
|
|
printf "%-20s %s\n" "--nowrite" "Disables Writing to Disk only modifys Existing Partitions"
|
|
exit 1
|
|
;;
|
|
*)
|
|
echo "Unknown option: $arg"
|
|
echo "Try -h"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
|
|
cleanup() {
|
|
echo "Cleaning Mounts"
|
|
|
|
umount -lf /mnt/ventoy 2>/dev/null || true
|
|
|
|
for i in "${mount_points[@]}"; do umount -lf "/mnt/debian$i" 2>/dev/null || true; done
|
|
|
|
umount -lf /mnt/debian/boot/efi 2>/dev/null || true
|
|
umount -lf /mnt/debian/boot 2>/dev/null || true
|
|
umount -lf /mnt/debian 2>/dev/null || true
|
|
}
|
|
trap 'echo "Status Code: $?"; cleanup;' EXIT
|
|
|
|
get_partition() {
|
|
local disk_name="$1"
|
|
local part_num="$2"
|
|
|
|
local part_path="$(
|
|
lsblk -lnp -o NAME,TYPE "$disk_name" \
|
|
| awk '$2=="part"{print $1}' \
|
|
| sed -n "${part_num}p"
|
|
)"
|
|
printf "$part_path"
|
|
}
|
|
|
|
# Set your target disk here
|
|
udevadm trigger
|
|
|
|
lsblk
|
|
echo
|
|
|
|
|
|
|
|
read -rp "Enter the target disk (e.g., sda): " disk_suffix
|
|
disk_name="/dev/${disk_suffix}"
|
|
|
|
echo "Chose disk $disk_name"
|
|
echo
|
|
|
|
if [[ $NO_WRITE == true ]]; then
|
|
echo -e "${YELLOW}--nowrite is specified no need for Img no disks will be written to${RESET}"
|
|
else
|
|
read -rp "Enter the name of the usb partition in /dev/mapper (e.g., sda1): " disk_suffix
|
|
|
|
usb_name="/dev/mapper/${disk_suffix}"
|
|
|
|
user_response="n"
|
|
|
|
read -rp "$(echo -e "${RED}You are about to WRITE and ERASE disk ${disk_name} | Are you sure? y/N: ${RESET}")" user_response
|
|
|
|
if [[ $user_response != y ]]; then
|
|
echo "Exiting script"
|
|
exit
|
|
fi
|
|
|
|
echo "Mounting Ventoy..."
|
|
mkdir -p /mnt/ventoy
|
|
mount $usb_name /mnt/ventoy
|
|
cd /mnt/ventoy/Images
|
|
|
|
echo "Writing image to $disk_name..."
|
|
lz4 -dc PBX.img.lz4 | dd of=$disk_name status=progress
|
|
fi
|
|
|
|
|
|
# Repartition disk using sgdisk
|
|
echo "Repairing partition tables"
|
|
sgdisk -e $disk_name || true
|
|
sgdisk -v $disk_name
|
|
|
|
partprobe || true
|
|
|
|
echo "Expanding Partition 3 on $disk_name..."
|
|
sgdisk --delete=3 $disk_name
|
|
sgdisk --new=3:0:0 --typecode=3:8e00 --change-name=3:"Linux LVM" $disk_name
|
|
|
|
partprobe || true
|
|
|
|
# Resize physical volume
|
|
echo "Resizing physical volume..."
|
|
pvresize "$(get_partition $disk_name "3")" || true
|
|
|
|
# Resize swap and root logical volumes
|
|
echo "Resizing swap and root volumes..."
|
|
lvresize -L +7G PBX-vg/swap_1
|
|
lvresize -l +100%FREE --resizefs PBX-vg/root
|
|
|
|
|
|
# Mount restored system for chroot
|
|
echo "Mounting restored system..."
|
|
mkdir -p /mnt/debian
|
|
mount /dev/mapper/PBX--vg-root /mnt/debian/
|
|
mount "$(get_partition $disk_name "2")" /mnt/debian/boot
|
|
mount "$(get_partition $disk_name "3")" /mnt/debian/boot/efi
|
|
|
|
|
|
for i in "${mount_points[@]}"; do mount -B $i /mnt/debian$i; done
|
|
|
|
# Chroot and reinstall GRUB
|
|
echo "Entering chroot to reinstall GRUB..."
|
|
chroot /mnt/debian /bin/bash -c "
|
|
grub-install --target=x86_64-efi --efi-directory=/boot/efi --bootloader-id=debian
|
|
update-grub
|
|
"
|
|
|
|
echo -e "${GREEN}Restore and resize complete!${RESET}"
|
|
|
|
cleanup
|