mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 08:26:00 +01:00
3dcc23c001
* Downgrade gqlgen to v0.17.2 The current version of gqlgen (v0.17.24) gives SIGILL crashes on armv7. * Update gqlparser
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package codegen
|
|
|
|
import (
|
|
"fmt"
|
|
"go/types"
|
|
"strings"
|
|
|
|
"github.com/99designs/gqlgen/codegen/config"
|
|
"github.com/99designs/gqlgen/codegen/templates"
|
|
"github.com/vektah/gqlparser/v2/ast"
|
|
)
|
|
|
|
type ArgSet struct {
|
|
Args []*FieldArgument
|
|
FuncDecl string
|
|
}
|
|
|
|
type FieldArgument struct {
|
|
*ast.ArgumentDefinition
|
|
TypeReference *config.TypeReference
|
|
VarName string // The name of the var in go
|
|
Object *Object // A link back to the parent object
|
|
Default interface{} // The default value
|
|
Directives []*Directive
|
|
Value interface{} // value set in Data
|
|
}
|
|
|
|
// ImplDirectives get not Builtin and location ARGUMENT_DEFINITION directive
|
|
func (f *FieldArgument) ImplDirectives() []*Directive {
|
|
d := make([]*Directive, 0)
|
|
for i := range f.Directives {
|
|
if !f.Directives[i].Builtin && f.Directives[i].IsLocation(ast.LocationArgumentDefinition) {
|
|
d = append(d, f.Directives[i])
|
|
}
|
|
}
|
|
|
|
return d
|
|
}
|
|
|
|
func (f *FieldArgument) DirectiveObjName() string {
|
|
return "rawArgs"
|
|
}
|
|
|
|
func (f *FieldArgument) Stream() bool {
|
|
return f.Object != nil && f.Object.Stream
|
|
}
|
|
|
|
func (b *builder) buildArg(obj *Object, arg *ast.ArgumentDefinition) (*FieldArgument, error) {
|
|
tr, err := b.Binder.TypeReference(arg.Type, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
argDirs, err := b.getDirectives(arg.Directives)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
newArg := FieldArgument{
|
|
ArgumentDefinition: arg,
|
|
TypeReference: tr,
|
|
Object: obj,
|
|
VarName: templates.ToGoPrivate(arg.Name),
|
|
Directives: argDirs,
|
|
}
|
|
|
|
if arg.DefaultValue != nil {
|
|
newArg.Default, err = arg.DefaultValue.Value(nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("default value is not valid: %w", err)
|
|
}
|
|
}
|
|
|
|
return &newArg, nil
|
|
}
|
|
|
|
func (b *builder) bindArgs(field *Field, params *types.Tuple) ([]*FieldArgument, error) {
|
|
var newArgs []*FieldArgument
|
|
|
|
nextArg:
|
|
for j := 0; j < params.Len(); j++ {
|
|
param := params.At(j)
|
|
for _, oldArg := range field.Args {
|
|
if strings.EqualFold(oldArg.Name, param.Name()) {
|
|
tr, err := b.Binder.TypeReference(oldArg.Type, param.Type())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
oldArg.TypeReference = tr
|
|
|
|
newArgs = append(newArgs, oldArg)
|
|
continue nextArg
|
|
}
|
|
}
|
|
|
|
// no matching arg found, abort
|
|
return nil, fmt.Errorf("arg %s not in schema", param.Name())
|
|
}
|
|
|
|
return newArgs, nil
|
|
}
|
|
|
|
func (a *Data) Args() map[string][]*FieldArgument {
|
|
ret := map[string][]*FieldArgument{}
|
|
for _, o := range a.Objects {
|
|
for _, f := range o.Fields {
|
|
if len(f.Args) > 0 {
|
|
ret[f.ArgsFunc()] = f.Args
|
|
}
|
|
}
|
|
}
|
|
|
|
for _, d := range a.Directives() {
|
|
if len(d.Args) > 0 {
|
|
ret[d.ArgsFunc()] = d.Args
|
|
}
|
|
}
|
|
return ret
|
|
}
|