From ed1ea53194add463d4dc1fd27cddcb654ef3efb1 Mon Sep 17 00:00:00 2001
From: Martin Kourim <martin.kourim@iohk.io>
Date: Thu, 5 Sep 2024 16:28:15 +0200
Subject: [PATCH] feat: add cleanup function and trap for SIGINT

- Introduced _cleanup function to stop all running cluster instances
  and PostgreSQL.
- Added a trap to call _cleanup on SIGINT (Ctrl+C).
- Replaced direct calls to stop_instances and stop_postgres with
  _cleanup function.
---
 .github/regression.sh | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/.github/regression.sh b/.github/regression.sh
index 98d45dc54..8b11066f2 100755
--- a/.github/regression.sh
+++ b/.github/regression.sh
@@ -131,6 +131,17 @@ fi
 # shellcheck disable=SC1090,SC1091
 . .github/nix_override_cardano_node.sh
 
+_cleanup() {
+  # stop all running cluster instances
+  stop_instances "$WORKDIR"
+
+  # stop postgres if running
+  stop_postgres || true
+}
+
+# cleanup on Ctrl+C
+trap 'set +e; _cleanup; exit 130' SIGINT
+
 echo "::group::Nix env setup"
 printf "start: %(%H:%M:%S)T\n" -1
 
@@ -183,11 +194,7 @@ if [ "${KEEP_CLUSTERS_RUNNING:-""}" = 1 ]; then
   set -"$sets"
 fi
 
-# stop all running cluster instances
-stop_instances "$WORKDIR"
-
-# stop postgres if running
-stop_postgres || true
+_cleanup
 
 # prepare artifacts for upload in Github Actions
 if [ -n "${GITHUB_ACTIONS:-""}" ]; then