From 367337b559c7f8fa5a8682f41487df41d1e1910e Mon Sep 17 00:00:00 2001
From: Achim Gsell <achim.gsell@psi.ch>
Date: Fri, 16 Aug 2024 13:29:34 +0200
Subject: [PATCH] modulecmd: set _LMFILES_ and LOADEDMODULES if removed by Lmod

---
 Pmodules/modulecmd.bash.in | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/Pmodules/modulecmd.bash.in b/Pmodules/modulecmd.bash.in
index 2fa4932..82a7e45 100644
--- a/Pmodules/modulecmd.bash.in
+++ b/Pmodules/modulecmd.bash.in
@@ -985,9 +985,10 @@ subcommand_unload() {
 	# the loop again, if it has been unset.
 	local saved_home="${PMODULES_HOME}"
 
-	# nothing to do, if _LMFILES_ is not set. This can happen
-	# if only Lua modules are loaded.
-	[[ -v _LMFILES_ ]] || return 0
+	if [[ ! -v _LMFILES_ ]]; then
+		declare -x _LMFILES_=''
+		declare -x LOADEDMODULES=''
+	fi
 
 	IFS=':' read -r -a _lmfiles_ <<< "${_LMFILES_}"
         local arg
@@ -1026,6 +1027,10 @@ subcommand_unload() {
 		PMODULES_HOME="${saved_home}"
 		export_env 'PMODULES_HOME'
 	fi
+	if [[ ! -v _LMFILES_ ]]; then
+		declare -x _LMFILES_=''
+		declare -x LOADEDMODULES=''
+	fi
 	set_lmfiles
 	export_env 'LOADEDMODULES' '_LMFILES_'
 	EnvMustBeSaved='yes'
@@ -3675,6 +3680,12 @@ if [[ ! -v Version ]] || \
                 _LMFILES_
 fi
 
+# re-define these variables. Lmod muught have unset them!
+if [[ ! -v _LMFILES_ ]]; then
+	declare -x _LMFILES_=''
+	declare -x LOADEDMODULES=''
+fi
+
 # we need to handle help text and options for sub-cmd aliases
 SubCommand=${Subcommands[${SubCommand}]}