Automatically lock Zellij when a certain command is executed

zellij-lock.zsh

# Define an array of commands in ENV that will lock Zellij when ran
# export ZELLIJ_LOCK=("micro" "nano" "vim")

# Store the last command we ran, this is to be used in precmd() after it exists
command=''

# Run before a command is executed
function preexec() {
	# Extract the command from the input string
	local cmd=${1%% *}
	command=$cmd
	if [[ " ${ZELLIJ_LOCK[@]} " =~ " $cmd " ]]; then
		zellij ac switch-mode locked
	fi
}

# Run after a command exits
function precmd() {
	if [[ " ${ZELLIJ_LOCK[@]} " =~ " $command " ]]; then
		zellij ac switch-mode normal
	fi
}

Made it use env variable rather than local variable to allow more flexibility, for example defining ZELLIJ_LOCK inside another file like ~/.zshenv