From 7c949970235c6b84c9ec397ca8f33c7db306acc8 Mon Sep 17 00:00:00 2001
From: Rodrigo Arias Mallo <rodrigo.arias@bsc.es>
Date: Fri, 16 Apr 2021 09:16:03 +0200
Subject: [PATCH] control: add trap for bad exit

---
 garlic/stages/control.nix | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/garlic/stages/control.nix b/garlic/stages/control.nix
index 123c560..3e04485 100644
--- a/garlic/stages/control.nix
+++ b/garlic/stages/control.nix
@@ -18,9 +18,25 @@ stdenv.mkDerivation {
   installPhase = ''
     cat > $out <<"EOF"
     #!/bin/sh -e
-    for n in $(seq 1 ${toString loops}); do
+
+    function badexit() {
+      errcode=$?
+      if [ $errcode != 0 ]; then
+        printf "exit %d\n" $errcode > "$basedir/status"
+        echo "exiting with $errcode"
+      fi
+
+      echo 1 > "$basedir/done"
+      exit $errcode
+    }
+
+    trap badexit EXIT
+
+    basedir=$(pwd)
+    loops=${toString loops}
+    for n in $(seq 1 $loops); do
       export GARLIC_RUN="$n"
-      echo "running $n of ${toString loops}" > status
+      echo "run $n/$loops" > status
       mkdir "$n"
       cd "$n"
       mkdir .garlic
@@ -29,7 +45,7 @@ stdenv.mkDerivation {
       date +%s > .garlic/total_time_end
       cd ..
     done
-    echo "completed" > status
+    echo "ok" > status
     EOF
     chmod +x $out
   '';