feat: add gen_provider.go and gomod package

pkg/utils/gomod/mod.go

@@ -0,0 +1,142 @@
+package gomod
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"regexp"
+	"strings"
+
+	"golang.org/x/mod/modfile"
+)
+
+var goMod *GoMod
+
+type GoMod struct {
+	file    *modfile.File
+	modules map[string]ModuleInfo
+}
+
+type ModuleInfo struct {
+	Name    string
+	Version string
+	Path    string
+}
+
+// ParseGoMod 解析当前目录下的go.mod文件
+func Parse(modPath string) error {
+	// 查找当前目录下的go.mod文件
+	// 读取文件内容
+	content, err := os.ReadFile(modPath)
+	if err != nil {
+		return err
+	}
+
+	// 使用官方包解析go.mod
+	f, err := modfile.Parse(modPath, content, nil)
+	if err != nil {
+		return err
+	}
+	goMod = &GoMod{file: f, modules: make(map[string]ModuleInfo)}
+
+	for _, require := range f.Require {
+		if !require.Indirect {
+			continue
+		}
+
+		name, err := getPackageName(require.Mod.Path, require.Mod.Version)
+		if err != nil {
+			continue
+		}
+
+		goMod.modules[require.Mod.Path] = ModuleInfo{
+			Name:    name,
+			Version: require.Mod.Version,
+			Path:    require.Mod.Path,
+		}
+	}
+
+	return nil
+}
+
+// GetModuleName 获取模块名
+func GetModuleName() string {
+	return goMod.file.Module.Mod.Path
+}
+
+// GetModuleVersion 获取模块版本
+func GetModuleVersion() string {
+	return goMod.file.Module.Mod.Version
+}
+
+func GetPackageModuleName(pkg string) string {
+	if module, ok := goMod.modules[pkg]; ok {
+		return module.Name
+	}
+
+	return filepath.Base(pkg)
+}
+
+// GetPackageModuleName 获取包的真实包名
+func getPackageName(pkg, version string) (string, error) {
+	gopath := os.Getenv("GOPATH")
+	if gopath == "" {
+		gopath = filepath.Join(os.Getenv("HOME"), "go")
+	}
+
+	pkgPath := fmt.Sprintf("%s@%s", pkg, version)
+	// 构建包的本地路径
+	pkgLocalPath := filepath.Join(gopath, "pkg", "mod", pkgPath)
+
+	// 获取目录下任意一个非_test.go文件，读取他的package name
+	files, err := filepath.Glob(filepath.Join(pkgLocalPath, "*.go"))
+	if err != nil {
+		return "", err
+	}
+
+	packagePattern := regexp.MustCompile(`package\s+(\w+)`)
+	if len(files) > 0 {
+		for _, file := range files {
+			if strings.HasSuffix(file, "_test.go") {
+				continue
+			}
+			// 读取文件内容
+
+			content, err := os.ReadFile(file)
+			if err != nil {
+				return "", err
+			}
+
+			packageName := packagePattern.FindStringSubmatch(string(content))
+			if len(packageName) == 2 {
+				return packageName[1], nil
+			}
+		}
+	}
+	// 读取go.mod 文件内容
+	modFile := filepath.Join(pkgLocalPath, "go.mod")
+	content, err := os.ReadFile(modFile)
+	if err != nil {
+		return "", err
+	}
+
+	f, err := modfile.Parse(modFile, content, nil)
+	if err != nil {
+		return "", err
+	}
+
+	path := f.Module.Mod.Path
+
+	// 获取包名
+	path, name := filepath.Split(path)
+	versionPattern := regexp.MustCompile(`^v\d+$`)
+	if versionPattern.MatchString(name) {
+		_, name = filepath.Split(strings.TrimSuffix(path, "/"))
+	}
+
+	if strings.Contains(name, "-") {
+		name = strings.ReplaceAll(name, "-", "")
+	}
+
+	return name, nil
+}

pkg/utils/gomod/mod_test.go

@@ -0,0 +1,32 @@
+package gomod
+
+import (
+	"testing"
+
+	"github.com/rogeecn/fabfile"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+func Test_ParseGoMod(t *testing.T) {
+	Convey("Test ParseGoMod", t, func() {
+		Convey("parse go.mod", func() {
+			modFile := fabfile.MustFind("go.mod")
+			err := Parse(modFile)
+			So(err, ShouldBeNil)
+
+			t.Logf("%+v", goMod)
+		})
+	})
+}
+
+func Test_getPackageName(t *testing.T) {
+	Convey("Test getPackageName", t, func() {
+		Convey("", func() {
+			Convey("github.com/redis/go-redis/v9@v9.7.0", func() {
+				name, err := getPackageName("github.com/redis/go-redis/v9", "v9.7.0")
+				So(err, ShouldBeNil)
+				So(name, ShouldEqual, "redis")
+			})
+		})
+	})
+}