91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
// 24 may 2014
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"go/token"
|
|
"go/ast"
|
|
"go/parser"
|
|
)
|
|
|
|
func getPackage(path string) (pkg *ast.Package) {
|
|
fileset := token.NewFileSet() // parser.ParseDir() actually writes to this; not sure why it doesn't return one instead
|
|
filter := func(i os.FileInfo) bool {
|
|
return strings.HasSuffix(i.Name(), "_windows.go")
|
|
}
|
|
pkgs, err := parser.ParseDir(fileset, path, filter, parser.AllErrors)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if len(pkgs) != 1 {
|
|
panic("more than one package found")
|
|
}
|
|
for k, _ := range pkgs { // get the sole key
|
|
pkg = pkgs[k]
|
|
}
|
|
return pkg
|
|
}
|
|
|
|
type walker struct {
|
|
desired func(string) bool
|
|
}
|
|
|
|
var known = map[string]string{}
|
|
var unknown = map[string]struct{}{}
|
|
|
|
func (w *walker) Visit(node ast.Node) ast.Visitor {
|
|
if n, ok := node.(*ast.Ident); ok {
|
|
if w.desired(n.Name) {
|
|
if n.Obj != nil {
|
|
delete(unknown, n.Name)
|
|
kind := n.Obj.Kind.String()
|
|
if known[n.Name] != "" && known[n.Name] != kind {
|
|
panic(n.Name + "(" + kind + ") already known to be a " + known[n.Name])
|
|
}
|
|
known[n.Name] = kind
|
|
} else if _, ok := known[n.Name]; !ok { // only if not known
|
|
unknown[n.Name] = struct{}{}
|
|
}
|
|
}
|
|
}
|
|
return w
|
|
}
|
|
|
|
func gatherNames(pkg *ast.Package) {
|
|
desired := func(name string) bool {
|
|
if strings.HasPrefix(name, "_") && len(name) > 1 {
|
|
return !strings.ContainsAny(name,
|
|
"abcdefghijklmnopqrstuvwxyz")
|
|
}
|
|
return false
|
|
}
|
|
for _, f := range pkg.Files {
|
|
for _, d := range f.Decls {
|
|
ast.Walk(&walker{desired}, d)
|
|
}
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
if len(os.Args) != 2 {
|
|
panic("usage: " + os.Args[0] + " path")
|
|
}
|
|
|
|
pkg := getPackage(os.Args[1])
|
|
gatherNames(pkg)
|
|
|
|
if len(unknown) > 0 {
|
|
s := "error: the following are still unknown!"
|
|
for k, _ := range unknown {
|
|
s += "\n" + k
|
|
}
|
|
panic(s)
|
|
}
|
|
|
|
for ident, kind := range known {
|
|
fmt.Printf("%-30s %s\n", ident, kind)
|
|
}
|
|
}
|