-
Notifications
You must be signed in to change notification settings - Fork 0
/
makeenum.go
73 lines (66 loc) · 1.59 KB
/
makeenum.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
package geco
import (
"go/ast"
"go/token"
"io"
)
type MakeEnum struct {
fileSet *token.FileSet
file *ast.File
typeName string
output io.Writer
Values []string
}
func (m *MakeEnum) Run() error {
// find fields
var typ string
ast.Inspect(m.file, func(n ast.Node) bool {
decl, ok := n.(*ast.GenDecl)
if !ok || decl.Tok != token.CONST {
// We only care about const declarations.
return true
}
for _, spec := range decl.Specs {
vspec := spec.(*ast.ValueSpec) // Guaranteed to succeed as this is CONST.
if vspec.Type == nil && len(vspec.Values) > 0 {
// "X = 1". With no type but a value. If the constant is untyped,
// skip this vspec and reset the remembered type.
typ = ""
// If this is a simple type conversion, remember the type.
// We don't mind if this is actually a call; a qualified call won't
// be matched (that will be SelectorExpr, not Ident), and only unusual
// situations will result in a function call that appears to be
// a type conversion.
ce, ok := vspec.Values[0].(*ast.CallExpr)
if !ok {
continue
}
id, ok := ce.Fun.(*ast.Ident)
if !ok {
continue
}
typ = id.Name
}
if vspec.Type != nil {
// "X T". We have a type. Remember it.
ident, ok := vspec.Type.(*ast.Ident)
if !ok {
continue
}
typ = ident.Name
}
if typ != m.typeName {
// This is not the type we're looking for.
continue
}
for _, name := range vspec.Names {
if name.Name == "_" {
continue
}
m.Values = append(m.Values, name.Name)
}
}
return false
})
return nil
}