TypeDefinitionRegistry support (#3)

Author: bbakerman

src/main/java/graphql/validation/constraints/DirectiveConstraints.java

@@ -4,12 +4,13 @@
 import graphql.schema.GraphQLArgument;
 import graphql.schema.GraphQLFieldDefinition;
 import graphql.schema.GraphQLFieldsContainer;
-import graphql.validation.constraints.standard.ExpressionConstraint;
+import graphql.schema.idl.TypeDefinitionRegistry;
 import graphql.validation.constraints.standard.AssertFalseConstraint;
 import graphql.validation.constraints.standard.AssertTrueConstraint;
 import graphql.validation.constraints.standard.DecimalMaxConstraint;
 import graphql.validation.constraints.standard.DecimalMinConstraint;
 import graphql.validation.constraints.standard.DigitsConstraint;
+import graphql.validation.constraints.standard.ExpressionConstraint;
 import graphql.validation.constraints.standard.MaxConstraint;
 import graphql.validation.constraints.standard.MinConstraint;
 import graphql.validation.constraints.standard.NegativeConstraint;
@@ -85,6 +86,14 @@ public String getDirectivesSDL() {
         return sb.toString();
     }
 
+    public TypeDefinitionRegistry getDirectivesDeclaration() {
+        TypeDefinitionRegistry registry = new TypeDefinitionRegistry();
+        for (DirectiveConstraint directiveConstraint : constraints.values()) {
+            registry.merge(directiveConstraint.getDocumentation().getDirectiveDeclaration());
+        }
+        return registry;
+    }
+
     public List<DirectiveConstraint> whichApplyTo(GraphQLFieldDefinition fieldDefinition, GraphQLFieldsContainer fieldsContainer) {
         return constraints.values()
                 .stream()

src/main/java/graphql/validation/constraints/Documentation.java

@@ -1,5 +1,8 @@
 package graphql.validation.constraints;
 
+import graphql.schema.idl.SchemaParser;
+import graphql.schema.idl.TypeDefinitionRegistry;
+
 import java.util.Arrays;
 import java.util.List;
 
@@ -8,13 +11,15 @@ public class Documentation {
     private final String description;
     private final String example;
     private final String directiveSDL;
+    private final TypeDefinitionRegistry directiveDeclaration;
     private final String messageTemplate;
     private final List<String> applicableTypeNames;
 
     private Documentation(Builder builder) {
         this.description = builder.description;
         this.example = builder.example;
         this.directiveSDL = builder.directiveSDL;
+        this.directiveDeclaration = new SchemaParser().parse(builder.directiveSDL);
         this.messageTemplate = builder.messageTemplate;
         this.applicableTypeNames = builder.applicableTypeNames;
     }
@@ -35,6 +40,10 @@ public String getDirectiveSDL() {
         return directiveSDL;
     }
 
+    public TypeDefinitionRegistry getDirectiveDeclaration() {
+        return directiveDeclaration;
+    }
+
     public String getMessageTemplate() {
         return messageTemplate;
     }

src/test/groovy/graphql/validation/constraints/DirectiveConstraintsTest.groovy

@@ -107,4 +107,16 @@ class DirectiveConstraintsTest extends BaseConstraintTestSupport {
         "field( testArg : Int! @Size(max : 2) ) : ID" | _ // not allowed for Size
     }
 
+    def "can combine as type registry"() {
+
+        def directiveValidationRules = DirectiveConstraints.newDirectiveConstraints()
+                .build()
+
+        when:
+        def declaration = directiveValidationRules.getDirectivesDeclaration()
+        then:
+        declaration != null
+        declaration.getDirectiveDefinitions().size() == 17
+
+    }
 }