Merge pull request #183 from sangria-graphql/fed2-add-@tag

feat(fed2): add @tag
sangria-graphql · Oct 13, 2022 · 5b7b748 · 5b7b748
2 parents 6760326 + f639301
commit 5b7b748
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/core/src/main/scala/sangria/federation/v2/Directives.scala b/core/src/main/scala/sangria/federation/v2/Directives.scala
@@ -75,6 +75,25 @@ object Directives {
         Argument("from", StringType)
       ),
       locations = Set(DirectiveLocation.FieldDefinition)
+    ),
+    Directive(
+      name = "tag",
+      arguments = List(
+        Argument("name", StringType)
+      ),
+      locations = Set(
+        DirectiveLocation.FieldDefinition,
+        DirectiveLocation.Object,
+        DirectiveLocation.Interface,
+        DirectiveLocation.Union,
+        DirectiveLocation.ArgumentDefinition,
+        DirectiveLocation.Scalar,
+        DirectiveLocation.Enum,
+        DirectiveLocation.EnumValue,
+        DirectiveLocation.InputObject,
+        DirectiveLocation.InputFieldDefinition
+      ),
+      repeatable = true
     )
   )
 

diff --git a/core/src/test/scala/sangria/federation/v2/FederationSpec.scala b/core/src/test/scala/sangria/federation/v2/FederationSpec.scala
@@ -79,6 +79,8 @@ class FederationSpec extends AsyncFreeSpec {
             directive @inaccessible on FIELD_DEFINITION | OBJECT | INTERFACE | UNION | ARGUMENT_DEFINITION | SCALAR | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
 
             directive @override(from: String!) on FIELD_DEFINITION
+            
+            directive @tag(name: String!) repeatable on FIELD_DEFINITION | INTERFACE | OBJECT | UNION | ARGUMENT_DEFINITION | SCALAR | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
           """)
           .extend(Document(
             Vector(_FieldSet.Type.toAst, Link__Import.Type.toAst, Link__Purpose.Type.toAst)))
@@ -159,6 +161,8 @@ class FederationSpec extends AsyncFreeSpec {
             directive @inaccessible on FIELD_DEFINITION | OBJECT | INTERFACE | UNION | ARGUMENT_DEFINITION | SCALAR | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
 
             directive @override(from: String!) on FIELD_DEFINITION
+            
+            directive @tag(name: String!) repeatable on FIELD_DEFINITION | INTERFACE | OBJECT | UNION | ARGUMENT_DEFINITION | SCALAR | ENUM | ENUM_VALUE | INPUT_OBJECT | INPUT_FIELD_DEFINITION
           """)
           .extend(Document(
             Vector(_FieldSet.Type.toAst, Link__Import.Type.toAst, Link__Purpose.Type.toAst)))