/
main.go
67 lines (55 loc) · 1.48 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package main
import (
"flag"
"github.com/kataras/golog"
gobuild "go/build"
"golang.org/x/tools/godoc/vfs"
"golang.org/x/tools/godoc/vfs/mapfs"
"os"
"path"
"path/filepath"
)
var (
fs = vfs.NewNameSpace()
)
var (
flagSkipCodeGen bool
flagModule string
flagVerbose bool
flagOutputFile string
flagOutputPkg string
)
func main() {
flag.BoolVar(&flagSkipCodeGen, "no-codegen", true, "skip code gen")
flag.StringVar(&flagModule, "m", "", "module to leech comments")
flag.BoolVar(&flagVerbose, "v", false, "verbose")
flag.StringVar(&flagOutputPkg, "pkg", "main", "Package name to use in the generated code. (default \"main\")")
flag.StringVar(&flagOutputFile, "o", "./leech_gen.go", "Optional name of the output file to be generated. (default \"./leech_gen.go\")")
flag.Parse()
if flagModule == "" {
flag.Usage()
return
}
if flagVerbose {
golog.SetLevel(golog.Levels[golog.DebugLevel].Name)
}
fs.Bind("/", vfs.OS(gobuild.Default.GOROOT), "/", vfs.BindReplace)
for _, p := range filepath.SplitList(gobuild.Default.GOPATH) {
mp := filepath.Join(p, "src", flagModule)
if _, err := os.ReadDir(mp); err == nil {
fs.Bind(path.Join("/src", flagModule), vfs.OS(mp), "/", vfs.BindReplace)
fs.Bind(path.Join("/src", flagModule, "vendor"), mapfs.New(nil), "/", vfs.BindReplace)
break
}
}
leech, err := NewLeech(fs)
if err != nil {
golog.Fatal(err)
}
leech.Walk(flagModule)
err = leech.save()
if err != nil {
golog.Fatal(err)
}
leech.PrintCommentRate()
}