Added SSH_SERVER and PROMPT customizations

Author: satishweb

docker-compose-template.yaml

@@ -26,6 +26,9 @@ services:
       - OPENAI_TEMPERATURE=0.2
       - USER_DEFAULT_PROMPT_MODE=exec
       - USER_PREFERENCES="Never execute any commands without my confirmation!"
+      - PROMPT_KUBECTL_ENABLED=1
+      - PROMPT_PREFIX="[ TOOLS ]"
+      - SSH_SERVER_ENABLED="0"
       # - DEBUG=1
     # ports:
     #   - "18000-19000:18000-19000"

entrypoint.sh

@@ -57,7 +57,6 @@ if [[ "$HOME" != "$HOME_TEMPLATE" && $HOME ]]; then
         "${HOME_TEMPLATE}/.kubectl_aliases"
         "${HOME_TEMPLATE}/.tmux.conf"
         "${HOME_TEMPLATE}/.vimrc"
-        "${HOME_TEMPLATE}/.wget-hsts"
         "${HOME_TEMPLATE}/.zshrc"
     )
     for item in "${items_to_copy[@]}"; do
@@ -110,6 +109,13 @@ for env_key in "${env_keys[@]}"; do
     fi
 done
 
+# Start SSH server if SSH_SERVER is enabled
+SSH_SERVER_ENABLED="${SSH_SERVER_ENABLED:-0}"
+if [ "${SSH_SERVER_ENABLED}" = "1" ]; then
+    printf "| Starting SSH server ... "
+    (sudo service ssh start >/dev/null 2>&1 && echo "[ OK ]") || echo "[ FAILED ]"
+fi
+
 printf "| Initialization complete! Container ready for use \n"
 print_divider
 

files/.zshrc

@@ -140,5 +140,8 @@ export PATH="${HOME}/.krew/bin:${HOME}/bin:${HOME}/.local/bin:${PATH}"
 [[ $(which kubectl) ]] && source <(kubectl completion zsh)
 export KUBE_PS1_SYMBOL_ENABLE="false"
 source "${ZSH}/custom/plugins/kube-ps1/kube-ps1.sh"
-PROMPT='[ TOOLS ] ''$(kube_ps1)'$PROMPT
+PROMPT_KUBECTL_ENABLED="${PROMPT_KUBECTL_ENABLED:-1}"
+PROMPT_PREFIX="${PROMPT_PREFIX:-[ TOOLS ]}"
+[ "$PROMPT_KUBECTL_ENABLED" = "1" ] && PROMPT="$(kube_ps1)${PROMPT}"
+PROMPT="${PROMPT_PREFIX}${PROMPT}"
 if which pyenv-virtualenv-init > /dev/null; then eval "$(pyenv virtualenv-init -)"; fi