-
Notifications
You must be signed in to change notification settings - Fork 7
/
plugin.go
155 lines (116 loc) · 3.5 KB
/
plugin.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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
package dataloader
import (
"path/filepath"
"github.com/mitchellh/mapstructure"
"github.com/pkg/errors"
"github.com/EGT-Ukraine/go2gql/generator"
"github.com/EGT-Ukraine/go2gql/generator/plugins/graphql"
)
const (
PluginName = "dataloader"
DataLoadersConfigsKey = "data_loaders"
)
type Plugin struct {
gqlPlugin *graphql.Plugin
generateCfg *generator.GenerateConfig
dataLoader *DataLoader
dataLoaderConfigs *DataLoadersConfig
loaders map[string]LoaderModel
}
func (p *Plugin) AddLoader(loader LoaderModel) {
p.loaders[loader.Name] = loader
}
func (p *Plugin) Prepare() error {
return nil
}
func (p *Plugin) Init(config *generator.GenerateConfig, plugins []generator.Plugin) error {
p.generateCfg = config
for _, plugin := range plugins {
if g, ok := plugin.(*graphql.Plugin); ok {
p.gqlPlugin = g
break
}
}
if p.gqlPlugin == nil {
return errors.New("graphql plugin was not found")
}
var dataLoadersConfig DataLoadersConfig
if config.PluginsConfigs[DataLoadersConfigsKey] != nil {
if err := mapstructure.Decode(config.PluginsConfigs[DataLoadersConfigsKey], &dataLoadersConfig); err != nil {
return errors.Wrap(err, "failed to decode dataloaders config")
}
outPath, err := filepath.Abs(dataLoadersConfig.OutputPath)
if err != nil {
return errors.Wrapf(err, "Failed to normalize path")
}
dataLoadersConfig.OutputPath = outPath
p.dataLoaderConfigs = &dataLoadersConfig
}
p.loaders = make(map[string]LoaderModel)
return nil
}
func (p Plugin) Name() string {
return PluginName
}
func (p *Plugin) validateOutputObjects(gqlFiles map[string]*graphql.TypesFile) error {
for _, gqlFile := range gqlFiles {
for _, outputObject := range gqlFile.OutputObjects {
for _, dataLoaderField := range outputObject.DataLoaderFields {
dataLoader, ok := p.loaders[dataLoaderField.DataLoaderName]
if !ok {
return errors.Errorf(
"Failed to found dataloader with name %s in object %s",
dataLoaderField.DataLoaderName,
outputObject.GraphQLName,
)
}
outputArgument := outputObject.FindFieldByName(dataLoaderField.ParentKeyFieldName)
if outputArgument == nil {
return errors.Errorf(
"Field `%s` not found in `%s`",
dataLoaderField.ParentKeyFieldName,
outputObject.GraphQLName,
)
}
if !outputArgument.GoType.Scalar {
return errors.Errorf(
"Field `%s` in `%s` must be scalar",
dataLoaderField.ParentKeyFieldName,
outputObject.GraphQLName,
)
}
if dataLoader.InputGoType.ElemType.Kind != outputArgument.GoType.Kind {
// TODO: use type casting if possible.
return errors.New("input argument must be same type as output")
}
}
}
}
return nil
}
func (p *Plugin) PrintInfo(info generator.Infos) {
}
func (p *Plugin) Infos() map[string]string {
return nil
}
func (p *Plugin) Generate() error {
if p.dataLoaderConfigs == nil {
return nil
}
dataLoader, err := p.createDataLoader(p.dataLoaderConfigs, p.generateCfg.VendorPath)
if err != nil {
return errors.Wrap(err, "failed to process dataloader config")
}
p.dataLoader = dataLoader
p.gqlPlugin.AddOutputObjectFieldRenderer(&fieldsRenderer{
dataLoader: p.dataLoader,
})
if err := p.validateOutputObjects(p.gqlPlugin.Types()); err != nil {
return errors.Wrap(err, "failed to validate graphql files")
}
loaderGen := NewLoaderGenerator(dataLoader)
if err := loaderGen.GenerateDataLoaders(); err != nil {
return errors.Wrap(err, "failed to generate data loader files")
}
return nil
}