aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/github.com/vektah/gqlgen/codegen/enum_build.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/vektah/gqlgen/codegen/enum_build.go')
-rw-r--r--vendor/github.com/vektah/gqlgen/codegen/enum_build.go39
1 files changed, 39 insertions, 0 deletions
diff --git a/vendor/github.com/vektah/gqlgen/codegen/enum_build.go b/vendor/github.com/vektah/gqlgen/codegen/enum_build.go
new file mode 100644
index 00000000..f2e6f63c
--- /dev/null
+++ b/vendor/github.com/vektah/gqlgen/codegen/enum_build.go
@@ -0,0 +1,39 @@
+package codegen
+
+import (
+ "sort"
+ "strings"
+
+ "github.com/vektah/gqlgen/codegen/templates"
+ "github.com/vektah/gqlgen/neelance/schema"
+)
+
+func (cfg *Config) buildEnums(types NamedTypes) []Enum {
+ var enums []Enum
+
+ for _, typ := range cfg.schema.Types {
+ namedType := types[typ.TypeName()]
+ e, isEnum := typ.(*schema.Enum)
+ if !isEnum || strings.HasPrefix(typ.TypeName(), "__") || namedType.IsUserDefined {
+ continue
+ }
+
+ var values []EnumValue
+ for _, v := range e.Values {
+ values = append(values, EnumValue{v.Name, v.Desc})
+ }
+
+ enum := Enum{
+ NamedType: namedType,
+ Values: values,
+ }
+ enum.GoType = templates.ToCamel(enum.GQLType)
+ enums = append(enums, enum)
+ }
+
+ sort.Slice(enums, func(i, j int) bool {
+ return strings.Compare(enums[i].GQLType, enums[j].GQLType) == -1
+ })
+
+ return enums
+}