把获取模块路径的逻辑从lint包里搬过来

Author: yangyile

go.mod

@@ -6,13 +6,13 @@ require (
 	github.com/emirpasic/gods/v2 v2.0.0-alpha
 	github.com/spf13/cobra v1.9.1
 	github.com/stretchr/testify v1.10.0
-	github.com/yyle88/erero v1.0.21
+	github.com/yyle88/erero v1.0.23
 	github.com/yyle88/must v0.0.25
 	github.com/yyle88/neatjson v0.0.12
 	github.com/yyle88/osexec v0.0.31
 	github.com/yyle88/osexistpath v0.0.17
 	github.com/yyle88/runpath v1.0.23
-	github.com/yyle88/zaplog v0.0.24
+	github.com/yyle88/zaplog v0.0.26
 )
 
 require (
@@ -23,14 +23,13 @@ require (
 	github.com/spf13/pflag v1.0.6 // indirect
 	github.com/yyle88/done v1.0.26 // indirect
 	github.com/yyle88/eroticgo v0.0.4 // indirect
-	github.com/yyle88/mutexmap v1.0.13 // indirect
+	github.com/yyle88/mutexmap v1.0.14 // indirect
 	github.com/yyle88/printgo v1.0.5 // indirect
 	github.com/yyle88/rese v0.0.10 // indirect
-	github.com/yyle88/sure v0.0.38 // indirect
-	github.com/yyle88/syntaxgo v0.0.52 // indirect
+	github.com/yyle88/sure v0.0.39 // indirect
+	github.com/yyle88/syntaxgo v0.0.53 // indirect
 	github.com/yyle88/tern v0.0.8 // indirect
 	go.uber.org/multierr v1.11.0 // indirect
 	go.uber.org/zap v1.27.0 // indirect
-	gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

go.sum

@@ -24,16 +24,16 @@ github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOf
 github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/yyle88/done v1.0.26 h1:lncTfGVFZWDV+ni9kgJWXjwl9qryVrX5xBYKW1Os1yg=
 github.com/yyle88/done v1.0.26/go.mod h1:Y/QHJS7guYtGMnbar4SHXGUtwUqckomTmRBaCcwW/y4=
-github.com/yyle88/erero v1.0.21 h1:SlVxmw1PsIXKkx23MojbrkYT4LtLEvSx+ZbDCDJ6ijE=
-github.com/yyle88/erero v1.0.21/go.mod h1:JH9yodnM7L9/0mCIkwwSXinLG1KY9xaaPGZRShJNTfw=
+github.com/yyle88/erero v1.0.23 h1:AY5grHGm+CgwCzPjn60LT7APCLSCuaHhWu2uOle6Nk0=
+github.com/yyle88/erero v1.0.23/go.mod h1:ZbUp//iNppPtn4yxilmlJHbpZ5U2ycipje6/SkMYHo4=
 github.com/yyle88/eroticgo v0.0.4 h1:ovmaRGWM8iN6P2uGlGNXaP8dVoWloLvYmC5Ta9rBNnM=
 github.com/yyle88/eroticgo v0.0.4/go.mod h1:e748J+JnGHnPrHVExVb14VzmYjaJelyhnAzzi5JF/GQ=
-github.com/yyle88/formatgo v1.0.24 h1:jdH+LkQ8vDozh/1YM5TnIUKN3okts90E/UgNBaqpY/w=
-github.com/yyle88/formatgo v1.0.24/go.mod h1:0Iaj84Kw5slGwGMXK46EyHpsiutpA6e2Lct+J5MOooY=
+github.com/yyle88/formatgo v1.0.25 h1:KbrrFte9ngOO5nxnEAl/APyGBKwmMdMbc9b+isRGMfQ=
+github.com/yyle88/formatgo v1.0.25/go.mod h1:LICth0xTd1MjDHrKYHxrghqyHDp7QMoiKJlr2Mqihdw=
 github.com/yyle88/must v0.0.25 h1:SF1wAYNTfupFNfnXJcnizLZAJtCUCJf2dUVY/9nvtUQ=
 github.com/yyle88/must v0.0.25/go.mod h1:SE9pZwrZLt8+wnWJJEAve5QKOm8rz3lSxokUxGGgemo=
-github.com/yyle88/mutexmap v1.0.13 h1:wp/6mv/gsC2DSqNbhMYiil4gtnb0sDCIhHHIHK2h0cQ=
-github.com/yyle88/mutexmap v1.0.13/go.mod h1:QUYDuARLPlGj414kHewQ5tt8jkDxQXoai8H3C4Gg+yc=
+github.com/yyle88/mutexmap v1.0.14 h1:aBdhtKR0XmFAJFoyswfjAEg9dzBvdaUBXU3Iw50AlB0=
+github.com/yyle88/mutexmap v1.0.14/go.mod h1:QUYDuARLPlGj414kHewQ5tt8jkDxQXoai8H3C4Gg+yc=
 github.com/yyle88/neatjson v0.0.12 h1:M6y4IsHbe2/3drF/kDl3zBpaN29lmfW4MetEQcoG04A=
 github.com/yyle88/neatjson v0.0.12/go.mod h1:LT3nIhKyB3lkD3INiIXCN3FejNu+g+qvzCJ+fZSZaRc=
 github.com/yyle88/osexec v0.0.31 h1:nMJLAv7r0mB1wKrD7njA+Kf2qlujtDDPu+SXy80Z4a0=
@@ -46,14 +46,14 @@ github.com/yyle88/rese v0.0.10 h1:rMEcybaIe/CNVMmilgor3nMQWB4TcJgXRttvDjKtvLg=
 github.com/yyle88/rese v0.0.10/go.mod h1:2jCdwu7qpJDdZqVulM4+zz/qiSxJWbOK/ndzQhg+5G0=
 github.com/yyle88/runpath v1.0.23 h1:VEdeaMXfvFY7CuF9Nceuxb6HSNDcmiSSGEf+zT7O900=
 github.com/yyle88/runpath v1.0.23/go.mod h1:JRGxn/0Ytg6CvGoE2VrO74oX8Lu4jbOPZDKxr8tzPEg=
-github.com/yyle88/sure v0.0.38 h1:+yw3++UKizkh7JsBtoZKH6jC2K1Jr4Z2U088zYBLJ/w=
-github.com/yyle88/sure v0.0.38/go.mod h1:zkSgOlFeW4xETRZNoR0GZ2v/q7wxnlHl3dxI4f4tftI=
-github.com/yyle88/syntaxgo v0.0.52 h1:G3A9MyTHp82j1Y2LPYqtD3lv/slutJ9/nVW/yi4XG4E=
-github.com/yyle88/syntaxgo v0.0.52/go.mod h1:wODs8eDwW75ARUJ9PBDBxOvot/Gj01gXPmsFQouTWsU=
+github.com/yyle88/sure v0.0.39 h1:uYPNFZekzJtsF0ZlULK8I/MYfFktCqG7/+9BeHzyN7E=
+github.com/yyle88/sure v0.0.39/go.mod h1:GVXwOj6Iw6+wdPg/H2dl0/jeAuhGYO2UTzWANztafdQ=
+github.com/yyle88/syntaxgo v0.0.53 h1:3W4S5ncRdq3hUp3Qjw4GqB+mAxypJCycMo/mMJP+1vc=
+github.com/yyle88/syntaxgo v0.0.53/go.mod h1:68EidTlDxVi/iaCJeg0menpA4v/xbq+ITxa1aKdmjLo=
 github.com/yyle88/tern v0.0.8 h1:Qn6dpYNbYA8bhPX/SjQHnML0MIoKScGNRgGH+VFox1U=
 github.com/yyle88/tern v0.0.8/go.mod h1:g9weyOMLtXYyt37EAshZPUTTHdH7WFauuxDi4oPr7/8=
-github.com/yyle88/zaplog v0.0.24 h1:8/G2Id1H2wSiQKav4e4vipB8LFeFlIYicTsv4sQO6wQ=
-github.com/yyle88/zaplog v0.0.24/go.mod h1:3eaCIi66BgCHh5IIbHlvuRWJPXz89m/EDPfqLe7p6Us=
+github.com/yyle88/zaplog v0.0.26 h1:NXLMlLLXF88CGfJiCFSirjiXFTNcHr5FgsPBwrr96bw=
+github.com/yyle88/zaplog v0.0.26/go.mod h1:0QHBqSYOkJ1giM4CDELhMGuMum3pbs29wG1LI+lQ8UE=
 go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
 go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
 go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
@@ -71,7 +71,5 @@ golang.org/x/tools v0.30.0/go.mod h1:c347cR/OJfw5TI+GfX7RUPNMdDRRbjvYTS0jPyvsVtY
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
-gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc=
-gopkg.in/natefinch/lumberjack.v2 v2.2.1/go.mod h1:YD8tP3GAjkrDg1eZH7EGmyESg/lsYskCTPBJVb9jqSc=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

goworkcmd/goworkcmd_test.go

@@ -1,9 +1,9 @@
 package main
 
 import (
-	"github.com/go-mate/go-work/goworkcmd"
 	"github.com/go-mate/go-work/worksexec"
 	"github.com/go-mate/go-work/workspace"
+	"github.com/go-mate/go-work/worksubcmd"
 	"github.com/spf13/cobra"
 	"github.com/yyle88/must"
 	"github.com/yyle88/osexec"
@@ -19,13 +19,13 @@ func main() {
 	projectPath := runpath.PARENT.Up(3)
 	zaplog.SUG.Debugln(projectPath)
 
-	wsp := workspace.NewWorkspace("", []string{projectPath})
+	workSpace := workspace.NewWorkSpace([]string{projectPath})
 
 	commandConfig := osexec.NewCommandConfig()
 	commandConfig.WithBash()
 	commandConfig.WithDebug()
 
-	wse := worksexec.NewWorksExec(commandConfig, []*workspace.Workspace{wsp})
+	worksExec := worksexec.NewWorksExec(commandConfig, []*workspace.Workspace{workSpace})
 
 	// 定义根命令
 	var rootCmd = &cobra.Command{
@@ -36,8 +36,8 @@ func main() {
 			zaplog.LOG.Info("run")
 		},
 	}
-	rootCmd.AddCommand(goworkcmd.NewWorkCmd(wse))
-	rootCmd.AddCommand(goworkcmd.NewModCmd(wse))
+	rootCmd.AddCommand(worksubcmd.NewWorkCmd(worksExec))
+	rootCmd.AddCommand(worksubcmd.NewModCmd(worksExec))
 
 	must.Done(rootCmd.Execute())
 }

internal/demos/demo1x/main.go

@@ -0,0 +1,23 @@
+package utils
+
+import (
+	"path/filepath"
+
+	"github.com/yyle88/osexistpath/osomitexist"
+)
+
+func GetProjectPath(currentPath string) (string, string, bool) {
+	projectPath := currentPath
+	shortMiddle := ""
+	for !osomitexist.IsFile(filepath.Join(projectPath, "go.mod")) {
+		subName := filepath.Base(projectPath)
+
+		prePath := filepath.Dir(projectPath)
+		if prePath == projectPath {
+			return "", "", false
+		}
+		projectPath = prePath
+		shortMiddle = filepath.Join(subName, shortMiddle)
+	}
+	return projectPath, shortMiddle, true
+}

internal/utils/utils.go

@@ -0,0 +1,20 @@
+package utils
+
+import (
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/yyle88/runpath"
+)
+
+func TestGetProjectPath(t *testing.T) {
+	path := runpath.PARENT.Path()
+	t.Log(path)
+	projectPath, shortMiddle, isGoModule := GetProjectPath(path)
+	require.True(t, isGoModule)
+	t.Log(projectPath)
+	t.Log(shortMiddle)
+
+	require.Equal(t, path, filepath.Join(projectPath, shortMiddle))
+}

internal/utils/utils_test.go

@@ -0,0 +1,130 @@
+package modulepath
+
+import (
+	"io/fs"
+	"path/filepath"
+	"strings"
+
+	"github.com/emirpasic/gods/v2/sets/linkedhashset"
+	"github.com/go-mate/go-work/internal/utils"
+	"github.com/yyle88/must"
+	"github.com/yyle88/neatjson/neatjsons"
+	"github.com/yyle88/osexistpath/osmustexist"
+	"github.com/yyle88/zaplog"
+)
+
+func GetProjectPath(currentPath string) (string, string, bool) {
+	return utils.GetProjectPath(currentPath)
+}
+
+func GetModulePaths(currentPath string, options *Options) []string {
+	set := linkedhashset.New[string]()
+
+	if options.IncludeHere {
+		projectPath, shortMiddle, isGoModule := GetProjectPath(currentPath)
+		if !isGoModule {
+			must.None(projectPath)
+			must.None(shortMiddle)
+		} else {
+			set.Add(currentPath) //把当前目录添加到需要lint的目录里
+		}
+	}
+
+	//这里很有可能，当前目录下就是 go.mod 文件，就是把当前目录设置两次，因此使用 hash-set 去重复
+	must.Done(filepath.Walk(currentPath, func(path string, info fs.FileInfo, err error) error {
+		if exSkip, isHide := isHidePath(info); isHide {
+			return exSkip
+		}
+		if !info.IsDir() && info.Name() == "go.mod" {
+			if subRoot := filepath.Dir(path); osmustexist.IsRoot(subRoot) {
+				set.Add(subRoot)
+			}
+			return nil
+		}
+		return nil
+	}))
+	if options.debugMode {
+		zaplog.SUG.Debugln(neatjsons.S(set))
+	}
+
+	if options.ExcludeNoGo {
+		//但是有些项目里是没有go文件的，比如空项目，或者大项目里只有子项目，而没有逻辑，因此需要去除
+		set = set.Select(func(index int, value string) bool {
+			if options.debugMode {
+				zaplog.SUG.Debugln(index, value)
+			}
+			return hasGoFiles(value)
+		})
+	}
+
+	if options.debugMode {
+		zaplog.SUG.Debugln(neatjsons.S(set))
+	}
+	roots := set.Values()
+	return roots
+}
+
+func hasGoFiles(root string) bool {
+	existGo := false
+	must.Done(filepath.Walk(root, func(path string, info fs.FileInfo, err error) error {
+		if exSkip, isHide := isHidePath(info); isHide {
+			return exSkip
+		}
+
+		if info.IsDir() {
+			//当遇到其它项目的 go.mod 时结束（由于传进来的就是项目根目录，因此要排除当前目录）
+			if path != root && osmustexist.IsFile(filepath.Join(path, "go.mod")) {
+				return filepath.SkipDir
+			}
+		} else {
+			if filepath.Ext(info.Name()) == ".go" {
+				existGo = true
+				return filepath.SkipAll
+			}
+		}
+		return nil
+	}))
+	return existGo
+}
+
+func isHidePath(info fs.FileInfo) (error, bool) {
+	if info.IsDir() {
+		if strings.HasPrefix(info.Name(), ".") {
+			return filepath.SkipDir, true
+		}
+	} else {
+		if strings.HasPrefix(info.Name(), ".") {
+			return nil, true
+		}
+	}
+	return nil, false
+}
+
+type Options struct {
+	IncludeHere bool // 假如当前项目是go项目时，是否当前项目
+	ExcludeNoGo bool
+	debugMode   bool
+}
+
+func NewOptions() *Options {
+	return &Options{
+		IncludeHere: true, //要包含当前目录
+		ExcludeNoGo: true, //不包含没有 go 文件的目录
+		debugMode:   false,
+	}
+}
+
+func (c *Options) WithIncludeHere(includeHere bool) *Options {
+	c.IncludeHere = includeHere
+	return c
+}
+
+func (c *Options) WithExcludeNoGo(excludeNoGo bool) *Options {
+	c.ExcludeNoGo = excludeNoGo
+	return c
+}
+
+func (c *Options) WithDebugMode(debugMode bool) *Options {
+	c.debugMode = debugMode
+	return c
+}

modulepath/module_path.go

@@ -0,0 +1,12 @@
+package modulepath
+
+import (
+	"testing"
+
+	"github.com/yyle88/runpath"
+)
+
+func TestGetModulePaths(t *testing.T) {
+	paths := GetModulePaths(runpath.PARENT.Path(), NewOptions().WithDebugMode(true))
+	t.Log(paths)
+}