update: safename function to parse environment variable.

Author: Esonhugh

.foam/templates/user.md

@@ -19,7 +19,7 @@ foam_template:
 ```
 
 ```zsh env-invoked
-set_current_user ${FOAM_TITLE}
+set_current_user ${FOAM_TITLE/@(\S*)$//}
 ```
 
 #### information

.vscode/env.zsh

@@ -51,6 +51,15 @@ function cut_lines_from_markdown_codes() {
         done
 }
 
+function safe_name() {
+        local name=$1
+        if [[ -z $name ]]; then
+                echo "Usage: safe_name <name>"
+                return 1
+        fi
+        echo "$name" | tr '@$.-' '____' # replace . and - with _ to avoid env var issues
+}
+
 function update_host_to_env() {
         if [[ -x "$(command -v yq)" && -d "${PROJECT_FOLDER}/hosts" ]]; then
                 for ur in $(ls -1 ${PROJECT_FOLDER}/hosts); do
@@ -59,7 +68,7 @@ function update_host_to_env() {
                                 local host_data=$(cut_lines_from_markdown_codes "$file" "yaml host")
 
                                 local hostname=$(echo "$host_data" | yq '.[0].hostname' -r)
-                                local _var=$(echo "$hostname" | sed -e "s/\./_/g" | sed -e "s/-/_/g" | sed -e 's/\$/_/g') # replace . and - with _ to avoid env var issues
+                                local _var=$(safe_name "$hostname" ) # replace . and - with _ to avoid env var issues
 
                                 local ip=$(echo "$host_data" | yq '.[0].ip' -r)
                                 local is_dc=$(echo "$host_data" | yq '.[0].is_dc' -r)
@@ -94,7 +103,7 @@ function set_current_host() {
                 return 1
         fi
 
-        export CURRENT_HOST=$(echo "$1" | sed -e "s/\./_/g" | sed -e "s/-/_/g" | sed -e 's/\$/_/g') # replace . and - with _ to avoid env var issues
+        export CURRENT_HOST=$(safe_name "$1") # replace . and - with _ to avoid env var issues
         export CURRENT_IP=$(eval echo '$IP_'$CURRENT_HOST)         # alias for IP_dc01 or IP_dc02
         export CURRENT_HOSTNAME=$(eval echo '$HOST_'$CURRENT_HOST) # alias for HOST_dc01 or HOST_dc02
 
@@ -119,9 +128,11 @@ function update_user_cred_to_env() {
                                 local usercred=$(cut_lines_from_markdown_codes "$file" "yaml credentials")
 
                                 local user=$(echo "$usercred" | yq '.[0].user' -r)
-                                local _var=$(echo "$user" | sed -e "s/\./_/g" | sed -e "s/-/_/g" | sed -e 's/\$/_/g') # replace . and - with _ to avoid env var issues
+                                local _var=$(safe_name "$user") # replace . and - with _ to avoid env var issues
                                 local pass=$(echo "$usercred" | yq '.[0].password' -r)
                                 local nt_hash=$(echo "$usercred" | yq '.[0].nt_hash' -r)
+                                local login=$(echo "$usercred" | yq '.[0].login' -r)
+                                local LOGIN_${_var}=$login
                                 export USER_${_var}=$user
                                 export PASS_${_var}=$pass
                                 export NT_HASH_${_var}=$nt_hash
@@ -141,17 +152,19 @@ function set_current_user() {
                 env | egrep '^USER_' | sed -e 's/USER_//g' | awk '{printf "- " $1 "\n"}' | sed -e 's/=/: /g' | sort
                 return 1
         fi
-        export CURRENT=$(echo "$1" | sed -e "s/\./_/g" | sed -e "s/-/_/g" | sed -e 's/\$/_/g') # replace . and - with _ to avoid env var issues
+        export CURRENT=$(safe_name "$1" ) # replace . and - with _ to avoid env var issues
         export CURRENT_USER=$(eval echo '$USER_'$CURRENT)       # alias for USER_A or USER_B
         export CURRENT_PASS=$(eval echo '$PASS_'$CURRENT)       # alias for PASS_A or PASS_B
         export CURRENT_NT_HASH=$(eval echo '$NT_HASH_'$CURRENT) # alias for NT_HASH_A or NT_HASH_B
-
+        export CURRENT_LOGIN=$(eval echo '$LOGIN_'$CURRENT) # alias for LOGIN_A or LOGIN_B
+        
         # defined variables if u need
         export USER=${CURRENT_USER}
         export USERNAME=${CURRENT_USER}
         export PASS=${CURRENT_PASS}
         export PASSWORD=${CURRENT_PASS}   # alias for PASS
         export NT_HASH=${CURRENT_NT_HASH} # alias for NT_HASH_A
+        export LOGIN=${CURRENT_LOGIN}     # alias for LOGIN_A
 }
 # set_current_user