Convert to functions
This commit is contained in:
+165
-139
@@ -1,78 +1,181 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
|
|
||||||
# THE RAINBOW HAHAHAH
|
|
||||||
RED='\e[31m'
|
RED='\e[31m'
|
||||||
GREEN='\e[32m'
|
GREEN='\e[32m'
|
||||||
YELLOW='\e[33m'
|
YELLOW='\e[33m'
|
||||||
RESET='\e[0m'
|
RESET='\e[0m'
|
||||||
|
|
||||||
|
MOUNT_POINTS=("/dev" "/dev/pts" "/proc" "/sys" "/sys/firmware/efi/efivars" "/run")
|
||||||
# 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
|
NO_WRITE=false
|
||||||
|
NO_RETABLE=false
|
||||||
|
NO_USB=false
|
||||||
|
|
||||||
|
|
||||||
# arguments
|
|
||||||
for arg in "$@"; do
|
for arg in "$@"; do
|
||||||
case $arg in
|
case $arg in
|
||||||
--nowrite)
|
--nowrite)
|
||||||
NO_WRITE=true
|
NO_WRITE=true
|
||||||
shift
|
shift
|
||||||
;;
|
;;
|
||||||
--noretable)
|
--noretable)
|
||||||
NO_RETABLE=true
|
NO_RETABLE=true
|
||||||
shift
|
shift
|
||||||
;;
|
;;
|
||||||
-h)
|
--nousb)
|
||||||
printf "%-20s %s\n" "Flags:" "Description:"
|
NO_USB=true
|
||||||
echo
|
shift
|
||||||
printf "%-20s %s\n" "--nowrite" "Disables Writing to Disk only modifys Existing Partitions"
|
;;
|
||||||
exit 1
|
-h)
|
||||||
;;
|
printf "%-20s %s\n" "Flags:" "Description:"
|
||||||
*)
|
echo
|
||||||
echo "Unknown option: $arg"
|
printf "%-20s %s\n" "--nowrite" "Disables Writing to Disk only modifys Existing Partitions"
|
||||||
echo "Try -h"
|
exit 1
|
||||||
exit 1
|
;;
|
||||||
;;
|
*)
|
||||||
esac
|
echo "Unknown option: $arg"
|
||||||
|
echo "Try -h"
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
done
|
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() {
|
cleanup() {
|
||||||
echo "Cleaning Mounts"
|
echo "Cleaning Mounts"
|
||||||
|
|
||||||
umount -lf /mnt/ventoy 2>/dev/null || true
|
umount -lf /mnt/ventoy 2>/dev/null || true
|
||||||
|
|
||||||
for i in "${mount_points[@]}"; do umount -lf "/mnt/debian$i" 2>/dev/null || true; done
|
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/efi 2>/dev/null || true
|
||||||
umount -lf /mnt/debian/boot 2>/dev/null || true
|
umount -lf /mnt/debian/boot 2>/dev/null || true
|
||||||
umount -lf /mnt/debian 2>/dev/null || true
|
umount -lf /mnt/debian 2>/dev/null || true
|
||||||
}
|
}
|
||||||
trap 'echo "Status Code: $?"; cleanup;' EXIT
|
|
||||||
|
|
||||||
|
|
||||||
get_partition() {
|
get_partition() {
|
||||||
local disk_name="$1"
|
local disk_name="$1"
|
||||||
local part_num="$2"
|
local part_num="$2"
|
||||||
|
|
||||||
local part_path="$(
|
local part_path="$(
|
||||||
lsblk -lnp -o NAME,TYPE "$disk_name" \
|
lsblk -lnp -o NAME,TYPE "$disk_name" |
|
||||||
| awk '$2=="part"{print $1}' \
|
awk '$2=="part"{print $1}' |
|
||||||
| sed -n "${part_num}p"
|
sed -n "${part_num}p"
|
||||||
)"
|
)"
|
||||||
|
|
||||||
printf "$part_path"
|
printf "$part_path"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
get_partuuid() {
|
get_partuuid() {
|
||||||
local disk_name="$1"
|
local disk_name="$1"
|
||||||
local part_num="$2"
|
local part_num="$2"
|
||||||
@@ -84,99 +187,22 @@ get_partuuid() {
|
|||||||
printf "$part_uuid"
|
printf "$part_uuid"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
mount_chroot() {
|
||||||
|
vgchange -ay PBX-vg
|
||||||
|
|
||||||
# Set your target disk here
|
mount_path="/mnt/debian"
|
||||||
udevadm trigger
|
|
||||||
|
|
||||||
lsblk
|
echo "Mounting restored system..."
|
||||||
echo
|
|
||||||
|
|
||||||
read -rp "Enter the target disk (e.g., sda): " disk_suffix
|
mkdir -p $mount_path/boot
|
||||||
disk_name="/dev/${disk_suffix}"
|
mount /dev/mapper/PBX--vg-root $mount_path
|
||||||
|
mount $(get_partition $disk_name "2") ${mount_path}/boot
|
||||||
|
|
||||||
echo "Chose disk $disk_name"
|
mkdir -p $mount_path/boot/efi
|
||||||
echo
|
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
|
main
|
||||||
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}"
|
|
||||||
|
|
||||||
cleanup
|
cleanup
|
||||||
|
|||||||
Reference in New Issue
Block a user