prune

cmd/gobrew/main.go

@@ -23,6 +23,7 @@ var allowedArgs = []string{
 	"install",
 	"use",
 	"uninstall",
+	"prune",
 	"version",
 	"self-update",
 }
@@ -72,6 +73,8 @@ func main() {
 		gb.Use(versionArg)
 	case "uninstall":
 		gb.Uninstall(versionArg)
+	case "prune":
+		gb.Prune()
 	case "version":
 		gb.Version(version)
 	case "self-update":
@@ -123,6 +126,7 @@ Usage:
     gobrew uninstall <version>     Uninstall <version>
     gobrew list                    List installed versions
     gobrew self-update             Self update this tool
+    gobrew prune                   Uninstall all go versions except current version
     gobrew version                 Show gobrew version
     gobrew help                    Show this message
 

gobrew.go

@@ -34,8 +34,9 @@ type Command interface {
 	Uninstall(version string)
 	Install(version string)
 	Use(version string)
-	Version(version string)
-	Upgrade(version string)
+	Prune()
+	Version(currentVersion string)
+	Upgrade(currentVersion string)
 	Helper
 }
 
@@ -85,6 +86,31 @@ func (gb *GoBrew) getArch() string {
 	return runtime.GOOS + "-" + runtime.GOARCH
 }
 
+// Prune removes all installed versions of go execept current version
+func (gb *GoBrew) Prune() error {
+	currentVersion := gb.CurrentVersion()
+	utils.Infof("[Info] Current version: %s \n", currentVersion)
+
+	entries, err := os.ReadDir(gb.versionsDir)
+	utils.CheckError(err, "[Error]: List versions failed")
+	files := make([]fs.FileInfo, 0, len(entries))
+	for _, entry := range entries {
+		info, err := entry.Info()
+		utils.CheckError(err, "[Error]: List versions failed")
+		files = append(files, info)
+	}
+
+	for _, f := range files {
+		if f.Name() != currentVersion {
+			version := f.Name()
+			utils.Infof("[Info] Uninstalling version: %s \n", version)
+			gb.Uninstall(version)
+			utils.Infof("[Info] Uninstalled version: %s \n", version)
+		}
+	}
+	return nil
+}
+
 // ListVersions that are installed by dir ls
 // highlight the version that is currently symbolic linked
 func (gb *GoBrew) ListVersions() error {