From 9f5ecdb814a8b5f67ada94bc9ca6aee8923c9928 Mon Sep 17 00:00:00 2001
From: Dennis ten Hoove <dennistenhoove@protonmail.com>
Date: Tue, 27 Aug 2024 09:10:34 +0200
Subject: [PATCH] [arkdep] Trap after update.sh

---
 arkdep | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/arkdep b/arkdep
index f8c372e..257d94e 100755
--- a/arkdep
+++ b/arkdep
@@ -1146,9 +1146,6 @@ deploy () {
 		printf '\e[1;33m<!>\e[0m\e[1m Not running bootctl set-default because overwritten with ARKDEP_NO_BOOTCTL\e[0m\n'
 	fi
 
-	# Image deployment finished, allow for interupts again
-	trap 'echo "User interupt received, canceling cleanup step"; unlock_and_quit 4' INT TERM
-
 	# Check if there is an update script available
 	if tar -xf $arkdep_dir/cache/${data[0]}.tar.${data[1]} -C $arkdep_dir/cache/ ./${data[0]}-update.sh 2> /dev/null && [[ ! -n $ARKDEP_ROOT ]]; then
 		printf '\e[1;34m-->\e[0m\e[1m Running update script\e[0m\n'
@@ -1159,6 +1156,9 @@ deploy () {
 		rm $arkdep_dir/cache/${data[0]}-update.sh
 	fi
 
+	# Image deployment finished, allow for interupts again
+	trap 'echo "User interupt received, canceling cleanup step"; unlock_and_quit 4' INT TERM
+
 	# Get list of all tarballs in cache
 	declare -r tarball_hits=($(ls $arkdep_dir/cache/ | grep -E '.*.tar..*'))