support gomod

.gitignore

@@ -1,2 +1,3 @@
 dist/
 gopkgs
+vendor/

cmd/gopkgs/main.go

@@ -4,19 +4,19 @@ import (
 	"bufio"
 	"flag"
 	"fmt"
+	"github.com/3Xpl0it3r/gopkgs"
 	"os"
 	"sort"
 	"strings"
 	"text/template"
-
-	"github.com/haya14busa/gopkgs"
 )
 
 var (
 	fullpath    = flag.Bool("fullpath", false, `output absolute file path to package directory. ("/usr/lib/go/src/net/http")`)
 	short       = flag.Bool("short", true, `output vendorless import path ("net/http", "foo/bar/vendor/a/b")`)
 	f           = flag.String("f", "", "alternate format for the output using the syntax of template package. e.g. {{.Name}};{{ImportPathShort}}")
 	includeName = flag.Bool("include-name", false, "fill Pkg.Name which can be used with -f flag")
+	noVendor    = flag.Bool("no-vendor", false, "exclude vendor dependencies except under workDir ")
 )
 
 var usageInfo = `
@@ -122,11 +122,9 @@ func main() {
 
 	w := bufio.NewWriter(os.Stdout)
 	defer w.Flush()
-
 	opt := gopkgs.DefaultOption()
 	opt.IncludeName = *includeName
 	pkgs := gopkgs.Packages(opt)
-
 	if *fullpath {
 		sort.Sort(ByFullPath(pkgs))
 		// Fullpaths is already unique

go.mod

@@ -0,0 +1,5 @@
+module github.com/3Xpl0it3r/gopkgs
+
+go 1.18
+
+require golang.org/x/tools v0.1.12 // indirect

go.sum

@@ -0,0 +1,2 @@
+golang.org/x/tools v0.1.12 h1:VveCTK38A2rkS8ZqFY25HIDFscX5X9OoEhJd3quQmXU=
+golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=

gopkgs.go

@@ -7,7 +7,7 @@ import (
 	"os"
 	"strings"
 
-	"github.com/haya14busa/gopkgs/x/tools/imports"
+	"github.com/3Xpl0it3r/gopkgs/x/tools/imports"
 )
 
 // Pkg represents exported go packages.

gopkgs_test.go

@@ -1,6 +1,8 @@
 package gopkgs
 
-import "testing"
+import (
+	"testing"
+)
 
 func TestPackages(t *testing.T) {
 	pkgs := Packages(&Option{IncludeName: false})

x/tools/imports/export.go

@@ -19,6 +19,7 @@ func GoPath() map[string]*Pkg {
 	populateIgnoreOnce.Do(populateIgnore)
 	scanGoRootOnce.Do(scanGoRoot) // async
 	scanGoPathOnce.Do(scanGoPath)
+	scanGoModPathOnce.Do(scanGoMod)
 	<-scanGoRootDone
 	dirScanMu.Lock()
 	defer dirScanMu.Unlock()

x/tools/imports/fastwalk.go

@@ -16,9 +16,11 @@ package imports
 
 import (
 	"errors"
+	"go/build"
 	"os"
 	"path/filepath"
 	"runtime"
+	"strings"
 )
 
 // traverseLink is a sentinel error for fastWalk, similar to filepath.SkipDir.
@@ -168,5 +170,15 @@ func (w *walker) walk(root string, runUserCallback bool) error {
 		}
 	}
 
+	if strings.Contains(root, build.Default.GOROOT+"/src/internal") {
+		return nil
+	}
+	if strings.Contains(root, build.Default.GOROOT+"/src/runtime") {
+		return nil
+	}
+	if strings.Contains(root, build.Default.GOROOT+"/src/cmd") {
+		return nil
+	}
+
 	return readDir(root, w.onDirEnt)
 }

x/tools/imports/fix.go

@@ -352,6 +352,8 @@ var (
 	// scanGoPathOnce guards calling scanGoPath (for $GOPATH)
 	scanGoPathOnce sync.Once
 
+	scanGoModPathOnce sync.Once
+
 	// populateIgnoreOnce guards calling populateIgnore
 	populateIgnoreOnce sync.Once
 	ignoredDirs        []os.FileInfo
@@ -490,14 +492,28 @@ var scanGoRootDone = make(chan struct{}) // closed when scanGoRoot is done
 
 func scanGoRoot() {
 	go func() {
-		scanGoDirs(true)
+		scanGoDirs(true, nil)
 		close(scanGoRootDone)
 	}()
 }
 
-func scanGoPath() { scanGoDirs(false) }
+func scanGoPath() { scanGoDirs(false, nil) }
+
+func scanGoMod() {
+	curDir, err := os.Getwd()
+	if err != nil {
+		return
+	}
+	if rootDir, ok := findGoModPath(curDir); ok {
+		scanGoDirs(false, []string{rootDir})
+	}
+}
 
-func scanGoDirs(goRoot bool) {
+func scanGoWorkspace() {
+
+}
+
+func scanGoDirs(goRoot bool, extPath []string) {
 	if Debug {
 		which := "$GOROOT"
 		if !goRoot {
@@ -512,8 +528,15 @@ func scanGoDirs(goRoot bool) {
 	}
 	dirScanMu.Unlock()
 
-	for _, srcDir := range build.Default.SrcDirs() {
+	var srcDirs []string
+	if len(extPath) == 0 || extPath == nil {
+		srcDirs = build.Default.SrcDirs()
+	} else {
+		srcDirs = extPath
+	}
+	for _, srcDir := range srcDirs {
 		isGoroot := srcDir == filepath.Join(build.Default.GOROOT, "src")
+
 		if isGoroot != goRoot {
 			continue
 		}
@@ -976,3 +999,35 @@ func fileInDir(file, dir string) bool {
 	// Check for boundary: either nothing (file == dir), or a slash.
 	return len(rest) == 0 || rest[0] == '/' || rest[0] == '\\'
 }
+
+func findGoModPath(path string) (string, bool) {
+	if strings.Compare(path, "/") == 0 {
+		return "", false
+	}
+	rd, err := ioutil.ReadDir(path)
+	if err != nil {
+		return "", false
+	}
+	for _, fi := range rd {
+		if strings.Compare(fi.Name(), "go.mod") == 0 {
+			return path, true
+		}
+		if strings.Compare(fi.Name(), ".git") == 0 {
+			return path, true
+		}
+	}
+	return findGoModPath(getParentDirectory(path))
+}
+
+func substr(s string, pos, length int) string {
+	runes := []rune(s)
+	l := pos + length
+	if l > len(runes) {
+		l = len(runes)
+	}
+	return string(runes[pos:l])
+}
+
+func getParentDirectory(dirctory string) string {
+	return substr(dirctory, 0, strings.LastIndex(dirctory, "/"))
+}