From 60fe8e890ac48bfb9f0c48bed8fa3b59018bb9d3 Mon Sep 17 00:00:00 2001
From: Alvar Penning <post@0x21.biz>
Date: Sun, 12 Dec 2021 16:12:39 +0100
Subject: [PATCH] modules/pifile.sh: remove duplicate stage code

---
 modules/pifile.sh | 45 +++++++++++----------------------------------
 1 file changed, 11 insertions(+), 34 deletions(-)

diff --git a/modules/pifile.sh b/modules/pifile.sh
index 55e966f..d7b7bb1 100644
--- a/modules/pifile.sh
+++ b/modules/pifile.sh
@@ -31,41 +31,18 @@ execute_pifile() {
 
   bash -n "$1"
 
-  pushd "$(dirname "$0")/modules" > /dev/null || exit 2
-  . "../stages/00-commands.sh"
-  . "../stages/10-setup.sh"
-  popd > /dev/null || exit 2
-  pre_stage
-  # shellcheck disable=SC1090
-  . "$1"
-  post_stage
+  declare -a stages=( "10-setup" "20-prepare" "30-chroot" "40-postprocess" )
+  for stage in "${stages[@]}"; do
+    pushd "$(dirname "$0")/modules" > /dev/null || exit 2
+    . "../stages/00-commands.sh"
+    . "../stages/${stage}.sh"
+    popd > /dev/null || exit 2
 
-  pushd "$(dirname "$0")/modules" > /dev/null || exit 2
-  . "../stages/00-commands.sh"
-  . "../stages/20-prepare.sh"
-  popd > /dev/null || exit 2
-  pre_stage
-  # shellcheck disable=SC1090
-  . "$1"
-  post_stage
+    pre_stage
 
-  pushd "$(dirname "$0")/modules" > /dev/null || exit 2
-  . "../stages/00-commands.sh"
-  . "../stages/30-chroot.sh"
-  popd > /dev/null || exit 2
-  pre_stage
-  # shellcheck disable=SC1090
-  . "$1"
-  post_stage
+    # shellcheck disable=SC1090
+    . "$1"
 
-  pushd "$(dirname "$0")/modules" > /dev/null || exit 2
-  . "../stages/00-commands.sh"
-  . "../stages/40-postprocess.sh"
-  popd > /dev/null || exit 2
-  pre_stage
-  # shellcheck disable=SC1090
-  . "$1"
-  post_stage
+    post_stage
+  done
 }
-
-