feat: supports installation of target versions via wildcards. #103 #108

Author: voidint

cli/install.go

@@ -22,12 +22,6 @@ func install(ctx *cli.Context) (err error) {
 	if vname == "" {
 		return cli.ShowSubcommandHelp(ctx)
 	}
-	targetV := filepath.Join(versionsDir, vname)
-
-	// 检查版本是否已经安装
-	if finfo, err := os.Stat(targetV); err == nil && finfo.IsDir() {
-		return cli.Exit(fmt.Sprintf("[g] %q version has been installed.", vname), 1)
-	}
 
 	// 查找版本
 	c, err := collector.NewCollector(strings.Split(os.Getenv(mirrorEnv), ",")...)
@@ -38,10 +32,20 @@ func install(ctx *cli.Context) (err error) {
 	if err != nil {
 		return cli.Exit(errstring(err), 1)
 	}
-	v, err := version.FindVersion(items, vname)
+
+	v, err := version.NewFinder(items).Find(vname)
 	if err != nil {
 		return cli.Exit(errstring(err), 1)
 	}
+
+	vname = v.Name()
+	targetV := filepath.Join(versionsDir, vname)
+
+	// 检查版本是否已经安装
+	if finfo, err := os.Stat(targetV); err == nil && finfo.IsDir() {
+		return cli.Exit(fmt.Sprintf("[g] %q version has been installed.", vname), 1)
+	}
+
 	// 查找版本下当前平台的安装包
 	pkgs, err := v.FindPackages(version.ArchiveKind, runtime.GOOS, runtime.GOARCH)
 	if err != nil {

pkg/errs/error.go

@@ -23,6 +23,8 @@ var (
 // VersionNotFoundError 版本不存在错误
 type VersionNotFoundError struct {
 	version string
+	goos    string
+	goarch  string
 }
 
 func IsVersionNotFound(err error) bool {
@@ -33,14 +35,16 @@ func IsVersionNotFound(err error) bool {
 	return ok
 }
 
-func NewVersionNotFoundError(version string) error {
+func NewVersionNotFoundError(version, goos, goarch string) error {
 	return &VersionNotFoundError{
 		version: version,
+		goos:    goos,
+		goarch:  goarch,
 	}
 }
 
 func (e VersionNotFoundError) Error() string {
-	return fmt.Sprintf("Version not found %q", e.version)
+	return fmt.Sprintf("Version not found %q [%s,%s]", e.version, e.goos, e.goarch)
 }
 
 func (e VersionNotFoundError) Version() string {

version/finder.go

@@ -0,0 +1,70 @@
+package version
+
+import (
+	"runtime"
+	"sort"
+
+	"github.com/Masterminds/semver/v3"
+	"github.com/voidint/g/pkg/errs"
+)
+
+type Finder struct {
+	goos   string
+	goarch string
+	items  []*Version
+}
+
+func NewFinder(items []*Version) *Finder {
+	sort.Sort(Collection(items)) // 升序
+
+	return &Finder{
+		goos:   runtime.GOOS,
+		goarch: runtime.GOARCH,
+		items:  items,
+	}
+}
+
+// Find 返回满足条件的语义化版本号。版本格式：主版本号.次版本号.修订号。
+// vname 支持以下几类版本标识：
+// 1、具体版本号：如'1.21.4'
+// 2、最新版本：latest
+// 3、通配符：如'1.21.x'、'1.x'、'1.18.*'等
+// 4、兼容某个主版本号：如'^1'、'^1.18'、'^1.18.10'等，在主版本号保持一致的前提下，次版本号和修订号均保持最新。
+// 5、匹配某个主次版本号：如'~1.18'，在主次版本号保持一致的前提下，修订号保持最新。
+// 6、大于某个版本：如'>1.18'，大于该版本的前提下，匹配最大的版本号。
+// 7、小于某个版本：如'<1.16'，小于该版本的前提下，匹配最大的版本号。
+// 8、版本区间：如'1.18 - 1.20'，匹配该区间范围内的最大版本。
+func (fdr *Finder) Find(vname string) (*Version, error) {
+	if vname == latest {
+		return fdr.findLatest()
+	}
+
+	for i := len(fdr.items) - 1; i > 0; i-- {
+		if fdr.items[i].name == vname && fdr.items[i].match(fdr.goos, fdr.goarch) {
+			return fdr.items[i], nil
+		}
+	}
+
+	cs, err := semver.NewConstraint(vname)
+	if err != nil {
+		return nil, errs.NewVersionNotFoundError(vname, fdr.goos, fdr.goarch)
+	}
+
+	for i := len(fdr.items) - 1; i > 0; i-- { // 优先匹配高版本
+		if fdr.items[i].match(fdr.goos, fdr.goarch) && cs.Check(fdr.items[i].sv) {
+			return fdr.items[i], nil
+		}
+	}
+	return nil, errs.NewVersionNotFoundError(vname, fdr.goos, fdr.goarch)
+}
+
+const latest = "latest"
+
+func (fdr *Finder) findLatest() (*Version, error) {
+	for i := len(fdr.items) - 1; i > 0; i-- {
+		if fdr.items[i].match(fdr.goos, fdr.goarch) {
+			return fdr.items[i], nil
+		}
+	}
+	return nil, errs.NewVersionNotFoundError(latest, fdr.goos, fdr.goarch)
+}

version/version.go

@@ -34,16 +34,6 @@ func Semantify(vname string) (*semver.Version, error) {
 	return sv, nil
 }
 
-// FindVersion 返回指定名称的版本
-func FindVersion(all []*Version, name string) (*Version, error) {
-	for i := range all {
-		if all[i].name == name {
-			return all[i], nil
-		}
-	}
-	return nil, errs.NewVersionNotFoundError(name)
-}
-
 // Version go版本
 type Version struct {
 	name string // 源版本名，如'1.12.4'，不完全等于sv.Original()
@@ -106,17 +96,13 @@ func (v *Version) SemanticVersion() semver.Version {
 	return *v.sv
 }
 
-// FindPackage 返回指定操作系统和硬件架构的版本包
-func (v *Version) FindPackage(kind, goos, goarch string) (*Package, error) {
-	prefix := fmt.Sprintf("go%s.%s-%s", v.name, goos, goarch)
-	for i := range v.pkgs {
-		if v.pkgs[i] == nil || !strings.EqualFold(v.pkgs[i].Kind, kind) || !strings.HasPrefix(v.pkgs[i].FileName, prefix) {
-			continue
+func (v *Version) match(goos, goarch string) bool {
+	for _, pkg := range v.pkgs {
+		if strings.Contains(pkg.FileName, goos) && strings.Contains(pkg.FileName, goarch) { // TODO 不够严谨
+			return true
 		}
-		return v.pkgs[i], nil
 	}
-
-	return nil, errs.ErrPackageNotFound
+	return false
 }
 
 // FindPackages 返回指定操作系统和硬件架构的版本包

version/version_test.go

@@ -13,70 +13,6 @@ import (
 	"github.com/voidint/g/pkg/errs"
 )
 
-func TestFindVersion(t *testing.T) {
-	t.Run("查找指定名称的版本", func(t *testing.T) {
-		v0 := MustNew("1.12.5")
-		v1 := MustNew("1.11.10")
-		v2 := MustNew("1.9.7")
-		v3 := MustNew("1.20")
-
-		items := []*Version{v0, v1, v2, v3}
-
-		v, err := FindVersion(items, "1.11.10")
-		assert.Nil(t, err)
-		assert.NotNil(t, v)
-		assert.Equal(t, "1.11.10", v.Name())
-
-		v, err = FindVersion(items, "1.11.11")
-		assert.True(t, errs.IsVersionNotFound(err))
-		assert.Nil(t, v)
-
-		v, err = FindVersion(items, "1.20")
-		assert.Nil(t, err)
-		assert.NotNil(t, v)
-		assert.Equal(t, "1.20", v.Name())
-	})
-}
-
-func TestFindPackage(t *testing.T) {
-	t.Run("查询版本下的安装包", func(t *testing.T) {
-		pkgs := []*Package{
-			{
-				FileName: "go1.12.4.src.tar.gz",
-				Kind:     SourceKind,
-				Size:     "21MB",
-			},
-			{
-				FileName: "go1.12.4.darwin-amd64.tar.gz",
-				Kind:     ArchiveKind,
-				OS:       "macOS",
-				Arch:     "x86-64",
-				Size:     "122MB",
-			},
-			{
-				FileName: "go1.12.4.windows-386.msi",
-				Kind:     InstallerKind,
-				OS:       "Windows",
-				Arch:     "x86",
-				Size:     "102MB",
-			},
-		}
-		v := MustNew("1.12.4", WithPackages(pkgs))
-
-		pkg, err := v.FindPackage(ArchiveKind, "darwin", "amd64")
-		assert.Nil(t, err)
-		assert.NotNil(t, pkg)
-		assert.Equal(t, "go1.12.4.darwin-amd64.tar.gz", pkg.FileName)
-		assert.Equal(t, ArchiveKind, pkg.Kind)
-		assert.Equal(t, "macOS", pkg.OS)
-		assert.Equal(t, "x86-64", pkg.Arch)
-
-		pkg, err = v.FindPackage(ArchiveKind, "darwin", "386")
-		assert.Equal(t, errs.ErrPackageNotFound, err)
-		assert.Nil(t, pkg)
-	})
-}
-
 func TestVerifyChecksum(t *testing.T) {
 	t.Run("检查安装包校验和", func(t *testing.T) {
 		filename := fmt.Sprintf("%d.txt", time.Now().Unix())