-
Notifications
You must be signed in to change notification settings - Fork 14
/
main.go
75 lines (64 loc) · 1.28 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
68
69
70
71
72
73
74
75
package main
import (
"flag"
"fmt"
"go/ast"
"go/parser"
"go/token"
"log"
"os"
"path/filepath"
_ "github.com/mattn/go-sqlite3"
)
var verbose bool
var debug bool
func main() {
verbosep := flag.Bool("v", false, "Verbose mode.")
debugp := flag.Bool("d", false, "Debug mode.")
flag.Parse()
verbose = *verbosep
debug = *debugp
if debug {
verbose = true
}
basePath, err := filepath.Abs(".")
if err != nil {
log.Fatalf("failed to get absolute path: %s", err)
}
err = do(basePath)
if err != nil {
log.Fatal(err)
}
}
func do(basePath string) error {
fset := token.NewFileSet() // positions are relative to fset
g, err := newGlobal(fset, basePath)
if err != nil {
return err
}
_ = filepath.Walk(basePath, func(path string, info os.FileInfo, err error) error {
if info.IsDir() {
// if hidden directory - skip the entire dir
if info.Name()[0] == '.' {
if verbose {
fmt.Println("Hidden folder, skipping: ", path)
}
return filepath.SkipDir
}
pkgs, err := parser.ParseDir(fset, path, nil, 0)
if err != nil {
log.Println("Error in parsing directory, skipping: ", path, err)
return nil
}
for _, p := range pkgs {
ast.Inspect(p, g.parse)
}
}
return nil
})
err = g.finalize()
if err != nil {
return err
}
return nil
}