Merge pull request #2 from Code-Hex/add/repeatable-directive

supported repeatable

ast/definition.go

@@ -85,9 +85,10 @@ type EnumValueDefinition struct {
 }
 
 type DirectiveDefinition struct {
-	Description string
-	Name        string
-	Arguments   ArgumentDefinitionList
-	Locations   []DirectiveLocation
-	Position    *Position `dump:"-"`
+	Description  string
+	Name         string
+	Arguments    ArgumentDefinitionList
+	Locations    []DirectiveLocation
+	IsRepeatable bool
+	Position     *Position `dump:"-"`
 }

parser/schema.go

@@ -463,6 +463,11 @@ func (p *parser) parseDirectiveDefinition(description string) *DirectiveDefiniti
 	def.Name = p.parseName()
 	def.Arguments = p.parseArgumentDefs()
 
+	if peek := p.peek(); peek.Kind == lexer.Name && peek.Value == "repeatable" {
+		def.IsRepeatable = true
+		p.skip(lexer.Name)
+	}
+
 	p.expectKeyword("on")
 	def.Locations = p.parseDirectiveLocations()
 	return &def

parser/schema_test.yml

@@ -520,6 +520,18 @@ directives:
             Name: "foo"
             Locations: [DirectiveLocation]
             - DirectiveLocation("FIELD")
+            IsRepeatable: false
+
+  - name: repeatable
+    input: directive @foo repeatable on FIELD
+    ast: |
+      <SchemaDocument>
+        Directives: [DirectiveDefinition]
+        - <DirectiveDefinition>
+            Name: "foo"
+            Locations: [DirectiveLocation]
+            - DirectiveLocation("FIELD")
+            IsRepeatable: true
 
   - name: invalid location
     input: "directive @foo on FIELD | INCORRECT_LOCATION"

validator/prelude.graphql

@@ -96,6 +96,7 @@ type __Directive {
     description: String
     locations: [__DirectiveLocation!]!
     args: [__InputValue!]!
+    isRepeatable: Boolean!
 }
 
 enum __DirectiveLocation {

validator/schema_test.go

@@ -54,6 +54,17 @@ func TestLoadSchema(t *testing.T) {
 		require.Equal(t, "dogEvents", s.Subscription.Fields[0].Name)
 
 		require.Equal(t, "owner", s.Types["Dog"].Fields[1].Name)
+
+		directives := s.Types["Person"].Directives
+		require.Len(t, directives, 2)
+		wantArgs := []string{"sushi", "tempura"}
+		for i, directive := range directives {
+			require.Equal(t, "favorite", directive.Name)
+			require.True(t, directive.Definition.IsRepeatable)
+			for _, arg := range directive.Arguments {
+				require.Equal(t, wantArgs[i], arg.Value.Raw)
+			}
+		}
 	})
 
 	testrunner.Test(t, "./schema_test.yml", func(t *testing.T, input string) testrunner.Spec {

validator/testdata/extensions.graphql

@@ -18,10 +18,12 @@ type Dog {
     name: String!
 }
 
-type Person {
+type Person @favorite(name: "sushi") @favorite(name: "tempura") {
     name: String!
 }
 
+directive @favorite(name: String!) repeatable on OBJECT
+
 extend type Dog {
     owner: Person! @permission(permission: "admin")
 }