From 28e72a8ff2b00d01e83145f6f14b89dc2883f15b Mon Sep 17 00:00:00 2001
From: mannem srinivas <m.srinivas365@gmail.com>
Date: Mon, 16 Sep 2024 14:40:01 +0530
Subject: [PATCH] Update install.sh

adding node version to command prompt
---
 install.sh | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/install.sh b/install.sh
index 459d3b5..7ef9120 100755
--- a/install.sh
+++ b/install.sh
@@ -366,6 +366,15 @@ nvm_check_global_modules() {
   fi
 }
 
+#!/bin/bash
+
+# Function to get current node version
+get_node_version() {
+  if command -v node >/dev/null 2>&1; then
+    echo "(node $(node -v))"
+  fi
+}
+
 nvm_do_install() {
   if [ -n "${NVM_DIR-}" ] && ! [ -d "${NVM_DIR}" ]; then
     if [ -e "${NVM_DIR}" ]; then
@@ -477,6 +486,30 @@ nvm_do_install() {
   if ${BASH_OR_ZSH} ; then
     command printf "${COMPLETION_STR}"
   fi
+
+  # Update prompt based on the shell (bash or zsh)
+  if [ "$SHELL" = "/bin/bash" ] || [ "$SHELL" = "/usr/bin/bash" ]; then
+    # For bash
+    echo "Setting up for bash..."
+  
+    # Add dynamic prompt setup for bash
+    PROMPT_COMMAND='PS1="\u@\h \w \$(get_node_version)$ "'
+    export PROMPT_COMMAND
+    
+  elif [ "$SHELL" = "/bin/zsh" ] || [ "$SHELL" = "/usr/bin/zsh" ]; then
+    # For zsh
+    echo "Setting up for zsh..."
+  
+    # Set the prompt dynamically in zsh
+    precmd() {
+      PS1='%n@%m %~ $(get_node_version)$ '
+    }
+    export -f precmd
+  
+    # Enable prompt substitution in zsh
+    setopt promptsubst
+  fi
+  
 }
 
 #