zsh: PROMPT HOSTINFO

Author: YodaEmbedding

zsh/.zshrc

@@ -124,7 +124,8 @@ setup_theme() {
   PROMPT_BGCOLOR="${PROMPT_BGCOLOR:-96}"
   PROMPT_FGCOLOR="${PROMPT_FGCOLOR:-217}"
 
-  HOSTINFO="${SLURM_JOB_ID:+SLURM_JOB_ID=$SLURM_JOB_ID}"
+  PROMPT_SLURM_JOB="${SLURM_ARRAY_JOB_ID:-$SLURM_JOB_ID}${SLURM_ARRAY_TASK_ID:+_${SLURM_ARRAY_TASK_ID}}"
+  HOSTINFO="${PROMPT_SLURM_JOB:+slurm_job=$PROMPT_SLURM_JOB}"
   HOSTNAME="${HOSTNAME:-$(hostname)}"
   WHOAMI="${WHOAMI:-$(whoami)}"