From 68b0d8f2c467361b7a2229dcd12930ade59db82d Mon Sep 17 00:00:00 2001
From: AtomicCoding <atomicgaming4life@gmail.com>
Date: Mon, 16 Apr 2018 17:39:48 -0700
Subject: [PATCH] Add module updater

---
 modules/utility/README.md               |  8 ++++
 modules/utility/functions/module_update | 62 +++++++++++++++++++++++++
 templates/zimrc                         |  8 ++++
 3 files changed, 78 insertions(+)
 create mode 100644 modules/utility/functions/module_update

diff --git a/modules/utility/README.md b/modules/utility/README.md
index e5dda67..7aec67a 100644
--- a/modules/utility/README.md
+++ b/modules/utility/README.md
@@ -46,3 +46,11 @@ Aliases `get` to ( `aria2c` || `axel` || `wget` || `curl` ).
 | alias | description |
 | ----- | ----------- |
 | `mkcd` | `mkdir -p` and `cd` |
+
+Commands
+--------
+
+| command | description |
+| ----- | ----------- |
+| `module_update` | `Updates git-installed modules and themes specified in the zimrc` |
+
diff --git a/modules/utility/functions/module_update b/modules/utility/functions/module_update
new file mode 100644
index 0000000..c0bb512
--- /dev/null
+++ b/modules/utility/functions/module_update
@@ -0,0 +1,62 @@
+local currentdir=$(pwd)
+local commandgood="0"
+local updated=0
+local failed=0
+local module
+local mlocation
+for module in $umodules; do
+  if [ -d ${ZIM_HOME}/modules/${module} ]; then
+    mlocation=${ZIM_HOME}/modules/${module}
+  elif [ -d ${ZIM_HOME}/modules/prompt/external-themes/${module} ]; then
+    mlocation=${ZIM_HOME}/modules/prompt/external-themes/${module}
+  else 
+    echo "$fg[red]Cant find $module module...$reset_color"
+    break
+  fi
+  echo -ne "$fg[blue]Updating $module... $reset_color"\\r
+  output=$(cd $mlocation && git remote update -p && git merge --ff-only @\{u\})
+  if [ $? -ne 0 ]; then
+  	echo -ne "$fg[red]Failed to update $module...\n $output $reset_color"\\r
+  	failed=$((failed+1))
+  fi
+  if echo "$output" | grep -i "up to date" > /dev/null; then
+  	echo -ne "$fg[blue]$module at latest version.      $reset_color"\\r
+  else
+  	echo -ne "$fg[green]Updated $module!      $reset_color"\\r
+      updated=$((updated+1))
+  fi
+  printf '\n'
+done
+for pmodule in $pmodules; do
+    echo -ne "$fg[blue]Updating $pmodule... $reset_color"\\r
+    output=$(cd ${ZIM_HOME}/modules/prompt/external-themes/$pmodule && git remote update -p && git merge --ff-only @\{u\})
+    if [ $? -ne 0 ]; then
+        echo -ne "$fg[red]Failed to update $pmodule...\n $output $reset_color"\\r
+        failed=$((failed+1))
+    fi
+    if echo "$output" | grep -i "up to date" > /dev/null; then
+        echo -ne "$fg[blue]$pmodule at latest version.      $reset_color"\\r
+    else
+        echo -ne "$fg[green]Updated $pmodule!      $reset_color"\\r
+        updated=$((updated+1))
+    fi
+    printf '\n'
+done
+if [ $updated -gt 0 ]; then
+	if [ $failed -eq 0 ]; then
+		echo "$fg[green]\033[1mSucessfully updated $updated modules!$reset_color"
+	fi
+else
+	if [ $failed -eq 0 ]; then
+		echo "$fg[blue]\033[1mNo module updates avaliable.$reset_color"
+	fi
+fi
+if [ $failed -gt 0 ]; then
+	if [ $updated -gt 0 ]; then
+		echo "$fg[blue]\033[1mSucessfully updated $updated modules and failed to update $failed modules.$reset_color"
+	else
+        echo "$fg[red]\033[1mFailed to update $failed modules, no module updates available.$rest_color"
+	fi
+
+fi
+cd $currentdir
\ No newline at end of file
diff --git a/templates/zimrc b/templates/zimrc
index 203a7a6..db36679 100644
--- a/templates/zimrc
+++ b/templates/zimrc
@@ -53,6 +53,14 @@ zprompt_theme='steeef'
 # The example below uses the following format: 'username@host:/current/directory'
 ztermtitle='%n@%m:%~'
 
+#
+# Updater
+#
+
+# Set modules or themes to update (they must be a valid git repo)
+#umodules=()
+
+
 #
 # Input
 #