From c18716e844483d1feacfc6682bdcb898956b6cbd Mon Sep 17 00:00:00 2001
From: Pulkit Kathuria <kevincobain2000@gmail.com>
Date: Wed, 1 Nov 2023 10:32:05 +0900
Subject: [PATCH] Support when no current version

---
 gobrew.go  | 23 ++++++++++++++++-------
 helpers.go |  3 +++
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/gobrew.go b/gobrew.go
index 52d8a34..033a05d 100644
--- a/gobrew.go
+++ b/gobrew.go
@@ -104,7 +104,7 @@ func (gb *GoBrew) Interactive(ask bool) {
 		color.Warnln("GO Installed Version", ".......", currentVersion)
 	} else {
 		labels := []string{}
-		if currentMajorVersion != modVersion && modVersion != "None" {
+		if modVersion != "None" && currentMajorVersion != modVersion {
 			labels = append(labels, "not same as go.mod")
 		}
 		if currentVersion != latestVersion {
@@ -130,11 +130,18 @@ func (gb *GoBrew) Interactive(ask bool) {
 	color.Successln("GO Latest Version", "   .......", latestVersion)
 	fmt.Println()
 
-	// do not implement this yet. gobrew should not tinker with someone's project and edit their go.mod files
-	// if latestMajorVersion != modVersion {
-	// 	color.Yellowf("GO Installed Major Version (%s) and GO Latest Version (%s) are different.\n", currentMajorVersion, latestMajorVersion)
-	// // do you want to install latest and update go.mod file
-	// }
+	if currentVersion == "None" {
+		color.Warnln("GO is not installed.")
+		c := true
+		if ask {
+			c = AskForConfirmation("Do you want to use latest GO version (" + latestVersion + ")?")
+		}
+		if c {
+			gb.Install(latestVersion)
+			gb.Use(latestVersion)
+		}
+		return
+	}
 
 	if currentMajorVersion != modVersion {
 		color.Warnf("GO Installed Version (%s) and go.mod Version (%s) are different.\n", currentMajorVersion, modVersion)
@@ -143,6 +150,7 @@ func (gb *GoBrew) Interactive(ask bool) {
 			c = AskForConfirmation("Do you want to use GO version same as go.mod version (" + modVersion + "@latest)?")
 		}
 		if c {
+			gb.Install(modVersion + "@latest")
 			gb.Use(modVersion + "@latest")
 		}
 		return
@@ -155,6 +163,7 @@ func (gb *GoBrew) Interactive(ask bool) {
 			c = AskForConfirmation("Do you want to update GO to latest version (" + latestVersion + ")?")
 		}
 		if c {
+			gb.Install(latestVersion)
 			gb.Use(latestVersion)
 		}
 		return
@@ -621,7 +630,7 @@ func (gb *GoBrew) Upgrade(currentVersion string) {
 		"[Error] Download GoBrew failed")
 
 	source, err := os.Open(tmpFile)
-	utils.CheckError(err, "==> [Error] Cannot open file")
+	utils.CheckError(err, "[Error] Cannot open file")
 	defer func(source *os.File) {
 		_ = source.Close()
 		utils.CheckError(os.Remove(source.Name()), "==> [Error] Cannot remove tmp file:")
diff --git a/helpers.go b/helpers.go
index 7a5f18a..d7dbfac 100644
--- a/helpers.go
+++ b/helpers.go
@@ -12,6 +12,9 @@ import (
 
 func ExtractMajorVersion(version string) string {
 	parts := strings.Split(version, ".")
+	if len(parts) < 2 {
+		return ""
+	}
 
 	// Take the first two parts and join them back with a period to create the new version.
 	majorVersion := strings.Join(parts[:2], ".")