auto use first install version

PWZER · Nov 19, 2024 · fc31ca8 · fc31ca8
1 parent 2b1c1c2
commit fc31ca8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/cmd/install.go b/cmd/install.go
@@ -29,6 +29,10 @@ var installCmd = &cobra.Command{
 				if err := internal.Install(version); err != nil {
 					return err
 				}
+
+				if err := internal.UseIfNotLinkGoBin(version); err != nil {
+					return err
+				}
 			}
 			return nil
 		}

diff --git a/internal/version.go b/internal/version.go
@@ -56,6 +56,17 @@ func UseGoVersion(version string) (err error) {
 	return nil
 }
 
+func UseIfNotLinkGoBin(version string) (err error) {
+	if !localVersionAlreadyInstalled(version) {
+		return fmt.Errorf("Local version %s not found", version)
+	}
+	goBin := getGoBinaryLinkFile()
+	if _, err = os.Stat(goBin); os.IsNotExist(err) {
+		return UseGoVersion(version)
+	}
+	return nil
+}
+
 func GetVersions(remote bool, all bool) (versions []*Version, err error) {
 	if remote {
 		return GetRemoteVersions(all)