78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package common
|
|
|
|
import (
|
|
"github.com/fatih/structtag"
|
|
"go/ast"
|
|
"golang.org/x/tools/go/analysis"
|
|
"strings"
|
|
)
|
|
|
|
func ParseModels(pass *analysis.Pass, models *map[string]Model) {
|
|
for _, file := range pass.Files {
|
|
ast.Inspect(file, func(node ast.Node) bool {
|
|
typeSpec, ok := node.(*ast.TypeSpec)
|
|
if !ok {
|
|
return true
|
|
}
|
|
structure, ok := typeSpec.Type.(*ast.StructType)
|
|
if !ok {
|
|
return true
|
|
}
|
|
|
|
if err := CheckUnnamedModel(*typeSpec); err != nil {
|
|
pass.Reportf(structure.Pos(), err.Error())
|
|
return false
|
|
}
|
|
|
|
var model Model
|
|
model.Name = typeSpec.Name.Name
|
|
model.Comment = typeSpec.Comment.Text()
|
|
model.Pos = structure.Pos()
|
|
model.Fields = make(map[string]Field)
|
|
|
|
for _, field := range structure.Fields.List {
|
|
var structField Field
|
|
if err := CheckUnnamedField(typeSpec.Name.Name, *field); err != nil {
|
|
pass.Reportf(field.Pos(), err.Error())
|
|
return false
|
|
}
|
|
|
|
structField.Name = field.Names[0].Name
|
|
structField.Pos = field.Pos()
|
|
structField.Comment = field.Comment.Text()
|
|
structField.Type = field.Type
|
|
|
|
if field.Tag != nil {
|
|
structField.Tags = &field.Tag.Value
|
|
|
|
tags, err := structtag.Parse(NormalizeStructTags(field.Tag.Value))
|
|
if err != nil {
|
|
pass.Reportf(field.Pos(), "Invalid structure tag: %s\n", err)
|
|
return false
|
|
}
|
|
if tags != nil {
|
|
gormTag, parseErr := tags.Get("gorm")
|
|
if gormTag != nil && parseErr == nil {
|
|
gormTag.Options = append([]string{gormTag.Name}, gormTag.Options...)
|
|
for _, opt := range gormTag.Options {
|
|
if strings.Contains(opt, ":") {
|
|
structField.Params = append(structField.Params, opt)
|
|
} else {
|
|
structField.Options = append(structField.Options, opt)
|
|
}
|
|
}
|
|
}
|
|
if parseErr != nil {
|
|
pass.Reportf(field.Pos(), "Invalid structure tag: %s\n", parseErr)
|
|
return false
|
|
}
|
|
}
|
|
}
|
|
model.Fields[structField.Name] = structField
|
|
(*models)[model.Name] = model
|
|
}
|
|
return false
|
|
})
|
|
}
|
|
}
|