feat: visually show if plugins need to be updated

install.zsh

@@ -8,6 +8,19 @@ main() {
     local ZAP_DIR="$HOME/.local/share/zap"
     local ZSHRC="${ZDOTDIR:-$HOME}/.zshrc"
 
+    # check if ZAP_DIR already exists
+    [[ -d "$ZAP_DIR" ]] && {
+        echo "Zap is already installed in '$ZAP_DIR'!"
+        read -q "res?Reinstall Zap? [y/N] "
+        echo ""
+        [[ $res == "n" ]] && {
+            echo "❕ skipped!"
+            return
+        }
+        echo "Reinstalling Zap..."
+        rm -rf "$ZAP_DIR"
+    }
+
     # Check if the current .zshrc file exists
     if [ -f "$ZSHRC" ]; then
         # Move the current .zshrc file to the new filename
@@ -31,19 +44,6 @@ main() {
 
     [[ $1 == "--branch" || $1 == "-b" && -n $2 ]] && local BRANCH="$2"
 
-    # check if ZAP_DIR already exists
-    [[ -d "$ZAP_DIR" ]] && {
-        echo "Zap is already installed in '$ZAP_DIR'!"
-        read -q "res?Reinstall Zap? [y/N] "
-        echo ""
-        [[ $res == "n" ]] && {
-            echo "❕ skipped!"
-            return
-        }
-        echo "Reinstalling Zap..."
-        rm -rf "$ZAP_DIR"
-    }
-
     git clone --depth 1 -b "${BRANCH:-master}" https://github.com/zap-zsh/zap.git "$ZAP_DIR" > /dev/null 2>&1 || { echo "❌ Failed to install Zap" && return 2 }
 
     echo " Zapped"
@@ -54,6 +54,6 @@ main() {
 
 main $@
 
-[[ $? -eq 0 ]] && exec zsh || return
+[[ $? -eq 0 ]] && source "$ZSHRC" || return
 
 # vim: ft=zsh ts=4 et

zap.zsh

@@ -72,19 +72,39 @@ function _zap_list() {
 }
 
 function _zap_update() {
-    local _plugin _plug
-    echo "⚡ Zap - Update\n\n   0  ⚡ Zap"
+
+    local _plugin _plug _status
+
+    function _check() {
+        git -C "$1" remote update &> /dev/null
+        case $(LANG=en_US git -C "$1" status -uno | grep -Eo '(ahead|behind|up to date)') in
+            ahead)
+                _status='\033[1;34mLocal ahead remote\033[0m' ;;
+            behind)
+                _status='\033[1;33mOut of date\033[0m' ;;
+            'up to date')
+                _status='\033[1;32mUp to date\033[0m' ;;
+            *)
+                _status='\033[1;31mDiverged state\033[0m' ;;
+        esac
+    }
+
+    echo "⚡ Zap - Update\n"
+    _check "$ZAP_DIR"
+    printf '   0 ⚡ Zap (%b)\n' "$_status"
     for _plugin in ${ZAP_INSTALLED_PLUGINS[@]}; do
-        printf '%4s  🔌 %s\n' $ZAP_INSTALLED_PLUGINS[(Ie)$_plugin] $_plugin
+        _check "$ZAP_PLUGIN_DIR/$_plugin"
+        printf '%4s 🔌 %s (%b)\n' $ZAP_INSTALLED_PLUGINS[(Ie)$_plugin] $_plugin $_status
     done
-    echo -n "\n🔌 Plugin Number | (a) All Plugins | (0) ⚡ Zap Itself: " && read _plugin
+    echo -n "\n  🔌 Plugin Number | (0) ⚡ Zap Itself | (a) All Plugins | (⏎) Abort: " && read _plugin
     case $_plugin in
         [[:digit:]]*)
             [[ $_plugin -gt ${#ZAP_INSTALLED_PLUGINS[@]} ]] && { echo "❌ Invalid option" && return 1 }
             [[ $_plugin -eq 0 ]] && {
                 git -C "$ZAP_DIR" pull &> /dev/null && { echo -e "\e[1A\e[K⚡ Zap updated!"; return 0 } || { echo -e "\e[1A\e[K❌ Failed to pull"; return 14 }
             } || { _pull "$ZAP_PLUGIN_DIR/$ZAP_INSTALLED_PLUGINS[$_plugin]" } ;;
         'a'|'A')
+            echo "\nUpdating All Plugins\n"
             for _plug in ${ZAP_INSTALLED_PLUGINS[@]}; do
                 _pull "$ZAP_PLUGIN_DIR/$_plug"
             done ;;