forked from github/multus-cni
79 lines
1.3 KiB
Go
79 lines
1.3 KiB
Go
package parser
|
|
|
|
import (
|
|
"go/ast"
|
|
"go/parser"
|
|
"go/token"
|
|
"strings"
|
|
)
|
|
|
|
const structComment = "easyjson:json"
|
|
|
|
type Parser struct {
|
|
PkgPath string
|
|
PkgName string
|
|
StructNames []string
|
|
AllStructs bool
|
|
}
|
|
|
|
type visitor struct {
|
|
*Parser
|
|
|
|
name string
|
|
explicit bool
|
|
}
|
|
|
|
func (p *Parser) needType(comments string) bool {
|
|
for _, v := range strings.Split(comments, "\n") {
|
|
if strings.HasPrefix(v, structComment) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (v *visitor) Visit(n ast.Node) (w ast.Visitor) {
|
|
switch n := n.(type) {
|
|
case *ast.File:
|
|
v.PkgName = n.Name.String()
|
|
return v
|
|
|
|
case *ast.GenDecl:
|
|
v.explicit = v.needType(n.Doc.Text())
|
|
|
|
if !v.explicit && !v.AllStructs {
|
|
return nil
|
|
}
|
|
return v
|
|
case *ast.TypeSpec:
|
|
v.name = n.Name.String()
|
|
|
|
// Allow to specify non-structs explicitly independent of '-all' flag.
|
|
if v.explicit {
|
|
v.StructNames = append(v.StructNames, v.name)
|
|
return nil
|
|
}
|
|
return v
|
|
case *ast.StructType:
|
|
v.StructNames = append(v.StructNames, v.name)
|
|
return nil
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Parser) Parse(fname string) error {
|
|
var err error
|
|
if p.PkgPath, err = getPkgPath(fname); err != nil {
|
|
return err
|
|
}
|
|
|
|
fset := token.NewFileSet()
|
|
f, err := parser.ParseFile(fset, fname, nil, parser.ParseComments)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ast.Walk(&visitor{Parser: p}, f)
|
|
return nil
|
|
}
|