1package codegen
 2
 3import (
 4	"sort"
 5
 6	"github.com/vektah/gqlparser/ast"
 7	"golang.org/x/tools/go/loader"
 8)
 9
10func (cfg *Config) buildModels(types NamedTypes, prog *loader.Program, imports *Imports) ([]Model, error) {
11	var models []Model
12
13	for _, typ := range cfg.schema.Types {
14		var model Model
15		switch typ.Kind {
16		case ast.Object:
17			obj, err := cfg.buildObject(types, typ, imports)
18			if err != nil {
19				return nil, err
20			}
21			if obj.Root || obj.IsUserDefined {
22				continue
23			}
24			model = cfg.obj2Model(obj)
25		case ast.InputObject:
26			obj, err := cfg.buildInput(types, typ)
27			if err != nil {
28				return nil, err
29			}
30			if obj.IsUserDefined {
31				continue
32			}
33			model = cfg.obj2Model(obj)
34		case ast.Interface, ast.Union:
35			intf := cfg.buildInterface(types, typ, prog)
36			if intf.IsUserDefined {
37				continue
38			}
39			model = int2Model(intf)
40		default:
41			continue
42		}
43		model.Description = typ.Description // It's this or change both obj2Model and buildObject
44
45		models = append(models, model)
46	}
47
48	sort.Slice(models, func(i, j int) bool {
49		return models[i].GQLType < models[j].GQLType
50	})
51
52	return models, nil
53}
54
55func (cfg *Config) obj2Model(obj *Object) Model {
56	model := Model{
57		NamedType: obj.NamedType,
58		Fields:    []ModelField{},
59	}
60
61	model.GoType = ucFirst(obj.GQLType)
62	model.Marshaler = &Ref{GoType: obj.GoType}
63
64	for i := range obj.Fields {
65		field := &obj.Fields[i]
66		mf := ModelField{Type: field.Type, GQLName: field.GQLName}
67
68		if field.GoFieldName != "" {
69			mf.GoFieldName = field.GoFieldName
70		} else {
71			mf.GoFieldName = field.GoNameExported()
72		}
73
74		model.Fields = append(model.Fields, mf)
75	}
76
77	return model
78}
79
80func int2Model(obj *Interface) Model {
81	model := Model{
82		NamedType: obj.NamedType,
83		Fields:    []ModelField{},
84	}
85
86	model.GoType = ucFirst(obj.GQLType)
87	model.Marshaler = &Ref{GoType: obj.GoType}
88
89	return model
90}