From f00fdafc100bb862d1990847cf9d7256e515ebc4 Mon Sep 17 00:00:00 2001
From: Sambo Chea <sombochea@cubetiqs.com>
Date: Mon, 14 Aug 2023 10:32:39 +0700
Subject: [PATCH] Update go-update

---
 go-update | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/go-update b/go-update
index 2b52bdb..ec38c9a 100644
--- a/go-update
+++ b/go-update
@@ -4,7 +4,12 @@ CURRENT_GO_PATH=$(which go)
 
 if [ -z "$CURRENT_GO_PATH" ]; then
     echo "Go is not installed!"
-    CURRENT_GO_PATH="/usr/local/go"
+    CURRENT_GO_PATH=$(echo $GOROOT)
+    if [ -z "$CURRENT_GO_PATH" ]; then
+        CURRENT_GO_PATH="/usr/local/go"
+    else
+        CURRENT_GO_PATH="$CURRENT_GO_PATH"
+    fi
     echo "Installing Go in $CURRENT_GO_PATH"
 else
     echo "Found Go in $CURRENT_GO_PATH"