From 733be4c7c8e22d1605851e17b332bd7818b5c6b9 Mon Sep 17 00:00:00 2001
From: Menno van Leeuwen <menno@vleeuwen.me>
Date: Wed, 15 Jan 2025 13:44:34 +0100
Subject: [PATCH] refactor setup script to improve argument handling for
 continuation flag

---
 setup.sh | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/setup.sh b/setup.sh
index 2222cb5..97376b2 100755
--- a/setup.sh
+++ b/setup.sh
@@ -454,10 +454,17 @@ main() {
         log_info "Cloning dotfiles repo..."
         git clone "$GIT_REPO" "$DOTFILES_DIR" || die "Failed to clone dotfiles repository"
     fi
-
-    CONTINUE=$1
-    if [ "$CONTINUE" = "--continue" ]; then
-        log_info "Continuing setup..."
+    
+    if [ -n "${1:-}" ]; then
+        CONTINUE=$1
+        if [ "$CONTINUE" = "--continue" ]; then
+            log_info "Continuing setup..."
+        else
+            warning_prompt
+            prepare_hostname
+            check_selinux
+            install_nix
+        fi
     else
         warning_prompt
         prepare_hostname
@@ -465,6 +472,7 @@ main() {
         install_nix
     fi
 
+
     install_home_manager
     setup_symlinks