115 lines
2.4 KiB
Go
115 lines
2.4 KiB
Go
// 24 may 2014
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"go/token"
|
|
"go/ast"
|
|
"go/parser"
|
|
"sort"
|
|
)
|
|
|
|
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 preamble(pkg string) string {
|
|
return "// autogenerated by windowsconstgen; do not edit\n" +
|
|
"package " + pkg + "\n"
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
// keep sorted for git
|
|
consts := make([]string, 0, len(known))
|
|
|
|
for ident, kind := range known {
|
|
if kind == "const" || kind == "var" {
|
|
consts = append(consts, ident)
|
|
}
|
|
}
|
|
sort.Strings(consts)
|
|
|
|
fmt.Printf("%s" +
|
|
"import \"fmt\"\n" +
|
|
"// #include <windows.h>\n" +
|
|
"import \"C\"\n" +
|
|
"func main() {\n" +
|
|
" fmt.Println(%q)\n",
|
|
preamble("main"), preamble("ui"))
|
|
for _, ident := range consts {
|
|
fmt.Printf(" fmt.Println(\"const %s =\", C.%s)\n", ident, ident[1:])
|
|
}
|
|
fmt.Printf("}\n")
|
|
}
|