Refactored to set_partition_paths function

scripts/image/base.sh

@@ -18,6 +18,12 @@ echo_partition_name(){
   fi
 }
 
+set_partition_paths(){
+  info "Setting partition paths..."
+  root_partition_path=$(echo_partition_name "2")
+  boot_partition_path=$(echo_partition_name "1")
+}
+
 # Routine to echo the full sd-card-path
 set_device_path(){
   info "Available devices:"

scripts/image/chroot.sh

@@ -14,8 +14,7 @@ error
 
 root_mount_path="/mnt/raspbian"
 boot_mount_path="/mnt/raspbian/boot"
-root_partition_path=$(echo_partition_name "2")
+set_partition_paths
-boot_partition_path=$(echo_partition_name "1")
 
 info "Mount partitions..."
 mount -o rw "$boot_partition_path" "$boot_mount_path" ||

scripts/image/setup.sh

@@ -157,8 +157,7 @@ root_mount_path="$working_folder""root/"
 mkdir -v "$boot_mount_path"
 mkdir -v "$root_mount_path"
 
-boot_partition_path=$(echo_partition_name "1")
+set_partition_paths
-root_partition_path=$(echo_partition_name "2")
 
 mount_partitions(){
   info "Mount boot and root partition..."