fstab changes

CloneToDisk.sh

@@ -11,7 +11,7 @@ NO_RETABLE=false
 NO_USB=false
 NO_CLEANUP=false
 
-IMG_FILE="PBX-root.img.lz4"
+IMG_FILE="PBX-Image-*.tar.zst"
 
 MOUNT_PATH="/mnt/debian"
 MOUNT_POINTS=("/dev" "/dev/pts" "/proc" "/sys" "/run")
@@ -74,8 +74,10 @@ main() {
     make_filesystem
   fi
 
-  mount_chroot
+  mount_partitions
   write_img
+  mount_chroot
+
   gen_fstab
   install_grub
 
@@ -121,9 +123,24 @@ dialogue() {
 
   echo "Chose disk $disk_name"
   echo
+
+  echo -e "${RED}WARNING: ALL DATA ON ${disk_name} WILL BE DESTROYED${RESET}"
+  lsblk $disk_name 2>/dev/null || true
+  echo
+  read -rp "Are you sure you want to continue? (yes/no): " confirm
+  if [[ "$confirm" != "yes" ]]; then
+    echo "Aborted."
+    exit 0
+  fi
+  echo "Chose disk $disk_name"
+  echo
 }
 
 rewrite_gpt_table() {
+  vgchange -an $VGROUP 2>/dev/null || true
+  vgremove -f $VGROUP 2>/dev/null || true
+  pvremove -ff $ROOT_PART 2>/dev/null || true
+
   sgdisk -Z $disk_name
 
   sgdisk -c $disk_name
@@ -159,10 +176,10 @@ lvm_init() {
 }
 
 make_filesystem() {
-  mkfs.vfat -F 32 $EFI_PART
-  mkfs.ext4 $BOOT_PART
-  mkfs.ext4 "${VG_PATH}${ROOT_LV}"
-  mkswap "${VG_PATH}${SWAP_LV}"
+  mkfs.vfat -I -F 32 $EFI_PART
+  mkfs.ext4 -F $BOOT_PART
+  mkfs.ext4 -F "${VG_PATH}${ROOT_LV}"
+  mkswap -f "${VG_PATH}${SWAP_LV}"
   echo -e "${GREEN}Make filesystems finished.${RESET}"
 }
 
@@ -172,22 +189,19 @@ write_img() {
     return 0
   fi
 
-  img_file_dir="null"
+  local img_file_dir=$(ls -1 ./PBX-Image-*.tar.zst 2>/dev/null | head -n1)
 
-  if [[ -f "./${IMG_FILE}" ]]; then
-    img_file_dir="./${IMG_FILE}"
-  elif [[ -f "./Image/${IMG_FILE}" ]]; then
-    img_file_dir="./Image/${IMG_FILE}"
-  elif [[ ! -f $img_file_dir ]]; then
-    echo -e "${RED}NO IMG FILE FOUND${RESET}"
+  if [[ -z "$img_file_dir" ]]; then
+    img_file_dir=$(ls -1 ./Image/PBX-Image-*.tar.zst 2>/dev/null | head -n1)
+  fi
+  if [[ -z "$img_file_dir" ]]; then
+    echo -e "${RED}NO TAR.ZST FILE FOUND${RESET}"
     exit 100
   fi
 
   echo "Writing image to ${VG_PATH}${ROOT_LV}..."
 
-  lz4 -dc ${img_file_dir} | partclone.ext4 -r -o "${VG_PATH}${ROOT_LV}"
-
-  tar -I zstd -xvf "${img_file_dir}" -C "${MOUNT_PATH}"
+  pv "${img_file_dir}" | tar -I zstd -x -C "${MOUNT_PATH}"
 
   echo -e "${GREEN}Finished writing image to disk${RESET}"
 }
@@ -208,25 +222,27 @@ get_partition() {
 get_partuuid() {
   local target_part="$1"
 
-  local part_uuid="$(
-    lsblk -ln -o PATH,PARTUUID $disk_name | awk -v target_part="$target_part" '$1==target_part {print $2}'
-  )"
+  blkid -s UUID -o value "$target_part"
+}
 
-  printf "$part_uuid"
+mount_partitions() {
+  echo "Mounting partitions..."
+
+  vgchange -ay $VGROUP
+
+  mkdir -p "${MOUNT_PATH}"
+  mount "${VG_PATH}${ROOT_LV}" $MOUNT_PATH
+
+  mkdir -p "${MOUNT_PATH}/boot"
+  mount $BOOT_PART "${MOUNT_PATH}/boot"
+
+  mkdir -p "${MOUNT_PATH}/boot/efi"
+  mount $EFI_PART "${MOUNT_PATH}/boot/efi"
 }
 
 mount_chroot() {
-  vgchange -ay $VGROUP
-
   echo "Mounting restored system..."
 
-  mkdir -p "${MOUNT_PATH}/boot"
-  mkdir -p "${MOUNT_PATH}/boot/efi"
-
-  mount "${VG_PATH}${ROOT_LV}" $MOUNT_PATH
-  mount $BOOT_PART "${MOUNT_PATH}/boot"
-  mount $EFI_PART "${MOUNT_PATH}/boot/efi"
-
   for i in "${MOUNT_POINTS[@]}"; do mount --bind -m "$i" "${MOUNT_PATH}${i}"; done
 
   echo -e "${GREEN}Mounts prepared for chroot.${RESET}"