Merge pull request #19 from Code-Hex/feature/update-wrapper

supported isRepeatable schema

Author: Code-Hex

internal/gqlgen/gqlgen.go

@@ -420,6 +420,8 @@ func (ec *executionContext) ___Directive(ctx context.Context, sel ast.SelectionS
 			if out.Values[i] == gql.Null {
 				invalids++
 			}
+		case "isRepeatable":
+			out.Values[i] = gql.MarshalBoolean(obj.IsRepeatable)
 		default:
 			panic("unknown field " + strconv.Quote(field.Name))
 		}

internal/gqlgen/gqlgen_test.yml

@@ -245,4 +245,62 @@ tests:
           }
         }
       json: |
-        {"__schema":{"description":"This is a simple schema"}}
+        {"__schema":{"description":"This is a simple schema"}}
+
+    - name: querying schema operation types
+      schema: |
+        schema {
+          query: Query
+          mutation: Mutation
+          subscription: Subscription
+        }
+        type Query {
+          queryField: String
+        }
+        type Mutation {
+          mutationField: String
+        }
+        type Subscription {
+          subscriptionField: String
+        }
+      query: |
+        {
+          __schema {
+            queryType { name }
+            mutationType { name }
+            subscriptionType { name }
+          }
+        }
+      json: |
+        {"__schema":{"mutationType":{"name":"Mutation"},"queryType":{"name":"Query"},"subscriptionType":{"name":"Subscription"}}}
+
+    - name: querying schema operation types
+      schema: |
+        directive @foo repeatable on FIELD_DEFINITION
+
+        schema {
+          query: Query
+        }
+        type Query {
+          queryField: String @foo @foo
+        }
+      query: |
+        {
+          __schema {
+            queryType { name }
+            directives {
+              name
+              isRepeatable
+            }
+          }
+        }
+      json: |
+        {"__schema":
+          {"queryType":{"name":"Query"},
+            "directives":[
+              {"isRepeatable":false,"name":"deprecated"},
+              {"isRepeatable":true,"name":"foo"},
+              {"isRepeatable":false,"name":"include"},
+              {"isRepeatable":false,"name":"skip"}
+            ]}
+        }

internal/wrapper/schema.go

@@ -58,6 +58,9 @@ func (s *Schema) Directives() []Directive {
 	for _, d := range s.schema.Directives {
 		res = append(res, s.directiveFromDef(d))
 	}
+	sort.Slice(res, func(i, j int) bool {
+		return strings.Compare(res[i].Name, res[j].Name) < 0
+	})
 
 	return res
 }
@@ -79,9 +82,10 @@ func (s *Schema) directiveFromDef(d *ast.DirectiveDefinition) Directive {
 	}
 
 	return Directive{
-		Name:        d.Name,
-		Description: d.Description,
-		Locations:   locs,
-		Args:        args,
+		Name:         d.Name,
+		Description:  d.Description,
+		Locations:    locs,
+		Args:         args,
+		IsRepeatable: d.IsRepeatable,
 	}
 }

internal/wrapper/wrapper.go

@@ -4,10 +4,11 @@ import "github.com/gqlgo/gqlparser/v2/ast"
 
 type (
 	Directive struct {
-		Name        string
-		Description string
-		Locations   []string
-		Args        []InputValue
+		Name         string
+		Description  string
+		Locations    []string
+		Args         []InputValue
+		IsRepeatable bool
 	}
 
 	EnumValue struct {