👔 up: sysutil - SearchPath will limit ext search on windows

sysutil/sysenv.go

@@ -8,6 +8,7 @@ import (
 	"strings"
 	"syscall"
 
+	"github.com/gookit/goutil/internal/checkfn"
 	"github.com/gookit/goutil/internal/comfunc"
 	"golang.org/x/term"
 )
@@ -144,6 +145,12 @@ func EnvPaths() []string {
 	return filepath.SplitList(os.Getenv("PATH"))
 }
 
+// SearchPathOption settings for SearchPath
+type SearchPathOption struct {
+	// 限制的扩展名
+	LimitExt []string
+}
+
 // SearchPath search executable files in the system $PATH
 //
 // Usage:
@@ -154,6 +161,9 @@ func SearchPath(keywords string, limit int) []string {
 	ptn := "*" + keywords + "*"
 	list := make([]string, 0)
 
+	// if windows, will limit with .exe, .bat, .cmd
+	isWindows := IsWindows()
+	winExts := []string{".exe", ".bat", ".cmd"}
 	checked := make(map[string]bool)
 	for _, dir := range filepath.SplitList(path) {
 		// Unix shell semantics: path element "" means "."
@@ -169,10 +179,21 @@ func SearchPath(keywords string, limit int) []string {
 		checked[dir] = true
 		matches, err := filepath.Glob(filepath.Join(dir, ptn))
 		if err == nil && len(matches) > 0 {
-			list = append(list, matches...)
-			size := len(list)
+			if isWindows {
+				// if windows, will limit with .exe, .bat, .cmd
+				for _, fPath := range matches {
+					fExt := filepath.Ext(fPath)
+					if checkfn.StringsContains(winExts, fExt) {
+						continue
+					}
+					list = append(list, fPath)
+				}
+			} else {
+				list = append(list, matches...)
+			}
 
 			// limit result size
+			size := len(list)
 			if limit > 0 && size >= limit {
 				list = list[:limit]
 				break