-
Notifications
You must be signed in to change notification settings - Fork 532
/
query.go
85 lines (74 loc) · 2.03 KB
/
query.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
package scaffolder
import (
"context"
"github.com/gobuffalo/genny/v2"
"github.com/ignite/cli/v29/ignite/pkg/errors"
"github.com/ignite/cli/v29/ignite/pkg/multiformatname"
"github.com/ignite/cli/v29/ignite/templates/field"
"github.com/ignite/cli/v29/ignite/templates/query"
)
// AddQuery adds a new query to scaffolded app.
func (s Scaffolder) AddQuery(
ctx context.Context,
moduleName,
queryName,
description string,
reqFields,
resFields []string,
paginated bool,
) error {
// If no module is provided, we add the type to the app's module
if moduleName == "" {
moduleName = s.modpath.Package
}
mfName, err := multiformatname.NewName(moduleName, multiformatname.NoNumber)
if err != nil {
return err
}
moduleName = mfName.LowerCase
name, err := multiformatname.NewName(queryName)
if err != nil {
return err
}
if err := checkComponentValidity(s.appPath, moduleName, name, true); err != nil {
return err
}
// Check and parse provided request fields
if ok := containsCustomTypes(reqFields); ok {
return errors.New("query request params can't contain custom type")
}
parsedReqFields, err := field.ParseFields(reqFields, checkGoReservedWord)
if err != nil {
return err
}
// Check and parse provided response fields
if err := checkCustomTypes(ctx, s.appPath, s.modpath.Package, s.protoDir, moduleName, resFields); err != nil {
return err
}
parsedResFields, err := field.ParseFields(resFields, checkGoReservedWord)
if err != nil {
return err
}
var (
g *genny.Generator
opts = &query.Options{
AppName: s.modpath.Package,
AppPath: s.appPath,
ProtoDir: s.protoDir,
ProtoVer: "v1", // TODO(@julienrbrt): possibly in the future add flag to specify custom proto version.
ModulePath: s.modpath.RawPath,
ModuleName: moduleName,
QueryName: name,
ReqFields: parsedReqFields,
ResFields: parsedResFields,
Description: description,
Paginated: paginated,
}
)
// Scaffold
g, err = query.NewGenerator(s.Tracer(), opts)
if err != nil {
return err
}
return s.Run(g)
}