From 39167dbc303014930f120b719fd3cdcf13774497 Mon Sep 17 00:00:00 2001
From: Johannes Schauer Marin Rodrigues <josch@mister-muffin.de>
Date: Sat, 6 Feb 2021 09:18:05 +0100
Subject: [PATCH] expose hook name to hooks via MMDEBSTRAP_HOOK environment
 variable

---
 coverage.sh | 28 +++++++++++++---------------
 mmdebstrap  |  4 ++++
 2 files changed, 17 insertions(+), 15 deletions(-)

diff --git a/coverage.sh b/coverage.sh
index cc4d28e..63b652c 100755
--- a/coverage.sh
+++ b/coverage.sh
@@ -2295,7 +2295,7 @@ else
 	runtests=$((runtests+1))
 fi
 
-print_header "mode=root,variant=apt: test special hooks using helpers"
+print_header "mode=root,variant=apt: test special hooks using helpers and env vars"
 cat << END > shared/test.sh
 #!/bin/sh
 set -eu
@@ -2304,31 +2304,29 @@ cat << 'SCRIPT' > /tmp/script.sh
 #!/bin/sh
 set -eu
 echo "MMDEBSTRAP_APT_CONFIG \$MMDEBSTRAP_APT_CONFIG"
+echo "\$MMDEBSTRAP_HOOK" >> /tmp/hooks
 [ "\$MMDEBSTRAP_MODE" = "root" ]
-echo test-content \$2 > test
-$CMD --hook-helper "\$1" "\$MMDEBSTRAP_MODE" "\$2" env 1 upload test /test <&\$MMDEBSTRAP_HOOKSOCK >&\$MMDEBSTRAP_HOOKSOCK
+echo test-content \$MMDEBSTRAP_HOOK > test
+$CMD --hook-helper "\$1" "\$MMDEBSTRAP_MODE" "\$MMDEBSTRAP_HOOK" env 1 upload test /test <&\$MMDEBSTRAP_HOOKSOCK >&\$MMDEBSTRAP_HOOKSOCK
 rm test
 echo "content inside chroot:"
 cat "\$1/test"
-[ "test-content \$2" = "\$(cat "\$1/test")" ]
-$CMD --hook-helper "\$1" "\$MMDEBSTRAP_MODE" "\$2" env 1 download /test test <&\$MMDEBSTRAP_HOOKSOCK >&\$MMDEBSTRAP_HOOKSOCK
+[ "test-content \$MMDEBSTRAP_HOOK" = "\$(cat "\$1/test")" ]
+$CMD --hook-helper "\$1" "\$MMDEBSTRAP_MODE" "\$MMDEBSTRAP_HOOK" env 1 download /test test <&\$MMDEBSTRAP_HOOKSOCK >&\$MMDEBSTRAP_HOOKSOCK
 echo "content outside chroot:"
 cat test
-[ "test-content \$2" = "\$(cat test)" ]
+[ "test-content \$MMDEBSTRAP_HOOK" = "\$(cat test)" ]
 rm test
 SCRIPT
 chmod +x /tmp/script.sh
-for h in setup extract essential customize; do
-	printf '#!/bin/sh\nset -eu\nexec /tmp/script.sh "\$1" "'"\$h"'"' > "/tmp/\$h.sh"
-	chmod +x "/tmp/\$h.sh"
-done
 $CMD --mode=root --variant=apt \
-	--setup-hook=/tmp/setup.sh \
-	--extract-hook=/tmp/extract.sh \
-	--essential-hook=/tmp/essential.sh \
-	--customize-hook=/tmp/customize.sh \
+	--setup-hook=/tmp/script.sh \
+	--extract-hook=/tmp/script.sh \
+	--essential-hook=/tmp/script.sh \
+	--customize-hook=/tmp/script.sh \
 	$DEFAULT_DIST /tmp/debian-chroot $mirror
-rm /tmp/script.sh /tmp/setup.sh /tmp/extract.sh /tmp/essential.sh /tmp/customize.sh
+printf "setup\nextract\nessential\ncustomize\n" | diff -u - /tmp/hooks
+rm /tmp/script.sh /tmp/hooks
 rm -r /tmp/debian-chroot
 END
 if [ "$HAVE_QEMU" = "yes" ]; then
diff --git a/mmdebstrap b/mmdebstrap
index 91df29c..c58bf72 100755
--- a/mmdebstrap
+++ b/mmdebstrap
@@ -1235,6 +1235,10 @@ sub run_hooks {
     # their behavior depending on the mode. It's also important for when the
     # hook wants to use the mmdebstrap --hook-helper.
     push @env_opts, "MMDEBSTRAP_MODE=$options->{mode}";
+    # Storing the hook name is important for hook scripts to potentially change
+    # their behavior depending on the hook. It's also important for when the
+    # hook wants to use the mmdebstrap --hook-helper.
+    push @env_opts, "MMDEBSTRAP_HOOK=$name";
     # This is the file descriptor of the socket that the mmdebstrap
     # --hook-helper can write to and read from to communicate with the outside.
     push @env_opts, ("MMDEBSTRAP_HOOKSOCK=" . fileno($options->{hooksock}));