Convert to functions
This commit is contained in:
+166
-140
@@ -1,78 +1,181 @@
|
||||
#!/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")
|
||||
MOUNT_POINTS=("/dev" "/dev/pts" "/proc" "/sys" "/sys/firmware/efi/efivars" "/run")
|
||||
NO_WRITE=false
|
||||
NO_RETABLE=false
|
||||
NO_USB=false
|
||||
|
||||
|
||||
# arguments
|
||||
for arg in "$@"; do
|
||||
case $arg in
|
||||
--nowrite)
|
||||
NO_WRITE=true
|
||||
shift
|
||||
;;
|
||||
--noretable)
|
||||
NO_RETABLE=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
|
||||
case $arg in
|
||||
--nowrite)
|
||||
NO_WRITE=true
|
||||
shift
|
||||
;;
|
||||
--noretable)
|
||||
NO_RETABLE=true
|
||||
shift
|
||||
;;
|
||||
--nousb)
|
||||
NO_USB=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
|
||||
|
||||
main() {
|
||||
root_check
|
||||
|
||||
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
|
||||
|
||||
trap 'echo "Status Code: $?"; cleanup;' EXIT
|
||||
|
||||
if [[ $NO_RETABLE == true ]]; then
|
||||
echo -e "${YELLOW}--noretable is specified. Tables will not be erased and no partitions will be created${RESET}"
|
||||
else
|
||||
rewrite_gpt_table
|
||||
make_filesystem
|
||||
fi
|
||||
|
||||
if [[ $NO_WRITE == true ]]; then
|
||||
echo -e "${YELLOW}--nowrite is specified. No need for Img no disks will be written to${RESET}"
|
||||
fi
|
||||
|
||||
# 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
|
||||
|
||||
echo "Generating fstab..."
|
||||
boot_uuid=$()
|
||||
|
||||
cp ./fstab_template ${mount_path}/etc/fstab
|
||||
|
||||
sed --in-place "s/boot_uuid/$(get_partuuid $disk_name "2")/g" ${mount_path}/etc/fstab
|
||||
sed --in-place "s/efi_uuid/$(get_partuuid $disk_name "1")/g" ${mount_path}/etc/fstab
|
||||
|
||||
# Chroot and reinstall GRUB
|
||||
echo "Entering chroot to reinstall GRUB..."
|
||||
chroot $mount_path /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}"
|
||||
}
|
||||
|
||||
write_image() {
|
||||
if [[ $NO_USB == true ]]; then
|
||||
echo "Writing image to $(get_partition $disk_name "3")..."
|
||||
lz4 -dc ./Image/PBX.img3.lz4 | dd of=$(get_partition $disk_name "3") status=progress
|
||||
|
||||
partprobe || true
|
||||
|
||||
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 $(get_partition $disk_name "3")..."
|
||||
lz4 -dc PBX.img.lz4 | dd of=$(get_partition $disk_name "3") status=progress
|
||||
|
||||
partprobe || true
|
||||
fi
|
||||
|
||||
}
|
||||
|
||||
rewrite_gpt_table() {
|
||||
sgdisk -Z $disk_name
|
||||
|
||||
sgdisk -c $disk_name
|
||||
|
||||
sgdisk -n 1:1MiB:+512M -t 1:ef00 -c 1:"EFI System" $disk_name
|
||||
sgdisk -n 2:0:+1G -t 2:8300 -c 2:"boot" $disk_name
|
||||
sgdisk -n 3:0:0 -t 3:8e00 -c 3:"Linux LVM" $disk_name
|
||||
|
||||
partprobe || true
|
||||
}
|
||||
|
||||
make_filesystem() {
|
||||
mkfs.vfat -F 32 $(get_partition $disk_name "1")
|
||||
mkfs.ext4 $(get_partition $disk_name "2")
|
||||
}
|
||||
|
||||
root_check() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
echo -e "${RED}You must be root to do this." 1>&2
|
||||
exit 100
|
||||
fi
|
||||
}
|
||||
|
||||
cleanup() {
|
||||
echo "Cleaning Mounts"
|
||||
|
||||
umount -lf /mnt/ventoy 2>/dev/null || true
|
||||
echo "Cleaning Mounts"
|
||||
|
||||
for i in "${mount_points[@]}"; do umount -lf "/mnt/debian$i" 2>/dev/null || true; done
|
||||
umount -lf /mnt/ventoy 2>/dev/null || true
|
||||
|
||||
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
|
||||
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 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"
|
||||
)"
|
||||
local part_path="$(
|
||||
lsblk -lnp -o NAME,TYPE "$disk_name" |
|
||||
awk '$2=="part"{print $1}' |
|
||||
sed -n "${part_num}p"
|
||||
)"
|
||||
|
||||
printf "$part_path"
|
||||
printf "$part_path"
|
||||
}
|
||||
|
||||
|
||||
get_partuuid() {
|
||||
local disk_name="$1"
|
||||
local part_num="$2"
|
||||
@@ -84,99 +187,22 @@ get_partuuid() {
|
||||
printf "$part_uuid"
|
||||
}
|
||||
|
||||
mount_chroot() {
|
||||
vgchange -ay PBX-vg
|
||||
|
||||
# Set your target disk here
|
||||
udevadm trigger
|
||||
mount_path="/mnt/debian"
|
||||
|
||||
lsblk
|
||||
echo
|
||||
echo "Mounting restored system..."
|
||||
|
||||
read -rp "Enter the target disk (e.g., sda): " disk_suffix
|
||||
disk_name="/dev/${disk_suffix}"
|
||||
mkdir -p $mount_path/boot
|
||||
mount /dev/mapper/PBX--vg-root $mount_path
|
||||
mount $(get_partition $disk_name "2") ${mount_path}/boot
|
||||
|
||||
echo "Chose disk $disk_name"
|
||||
echo
|
||||
mkdir -p $mount_path/boot/efi
|
||||
mount $(get_partition $disk_name "1") ${mount_path}/boot/efi
|
||||
|
||||
for i in "${MOUNT_POINTS[@]}"; do mount -B $i ${mount_path}${i}; done
|
||||
}
|
||||
|
||||
if [[ $NO_RETABLE == true ]]; then
|
||||
echo -e "${YELLOW}--noretable is specified. Tables will not be erased and no partitions will be created${RESET}"
|
||||
else
|
||||
sgdisk -Z $disk_name
|
||||
|
||||
sgdisk -c $disk_name
|
||||
|
||||
sgdisk -n 1:1MiB:+512M -t 1:ef00 -c 1:"EFI System" $disk_name
|
||||
sgdisk -n 2:0:+1G -t 2:8300 -c 2:"boot" $disk_name
|
||||
sgdisk -n 3:0:0 -t 3:8e00 -c 3:"Linux LVM" $disk_name
|
||||
|
||||
partprobe || true
|
||||
fi
|
||||
|
||||
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 $(get_partition $disk_name "3")..."
|
||||
lz4 -dc PBX.img.lz4 | dd of=$(get_partition $disk_name "3") status=progress
|
||||
|
||||
partprobe || true
|
||||
fi
|
||||
|
||||
|
||||
# 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
|
||||
mount_path="/mnt/debian"
|
||||
echo "Mounting restored system..."
|
||||
mkdir -p $mount_path
|
||||
mount /dev/mapper/PBX--vg-root $mount_path
|
||||
mount $(get_partition $disk_name "2") ${mount_path}/boot
|
||||
mount $(get_partition $disk_name "1") ${mount_path}/boot/efi
|
||||
|
||||
for i in "${mount_points[@]}"; do mount -B $i ${mount_path}${i}; done
|
||||
|
||||
echo "Generating fstab..."
|
||||
boot_uuid=$()
|
||||
|
||||
cp ./fstab.template ${mount_path}/etc/fstab
|
||||
|
||||
sed --in-place "s/boot_uuid/$(get_partuuid $disk_name "2")/g" ${mount_path}/etc/fstab
|
||||
sed --in-place "s/efi_uuid/$(get_partuuid $disk_name "1")/g" ${mount_path}/etc/fstab
|
||||
|
||||
|
||||
# Chroot and reinstall GRUB
|
||||
echo "Entering chroot to reinstall GRUB..."
|
||||
chroot $mount_path /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}"
|
||||
|
||||
main
|
||||
cleanup
|
||||
|
||||
Reference in New Issue
Block a user