Implement http directive for join__Graph enum

apollographql · Mar 19, 2021 · f9f3784 · f9f3784 · glasser · Mar 19, 2021
1 parent f1e0a8a
commit f9f3784
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/federation-js/src/service/__tests__/printCoreSchema.test.ts b/federation-js/src/service/__tests__/printCoreSchema.test.ts
@@ -134,12 +134,12 @@ describe('printCoreSchema', () => {
       scalar join__FieldSet
 
       enum join__Graph {
-        ACCOUNTS
-        BOOKS
-        DOCUMENTS
-        INVENTORY
-        PRODUCT
-        REVIEWS
+        ACCOUNTS @http(url: \\"https://accounts.api.com\\")
+        BOOKS @http(url: \\"https://books.api.com\\")
+        DOCUMENTS @http(url: \\"https://documents.api.com\\")
+        INVENTORY @http(url: \\"https://inventory.api.com\\")
+        PRODUCT @http(url: \\"https://product.api.com\\")
+        REVIEWS @http(url: \\"https://reviews.api.com\\")
       }
 
       type KeyValue {

diff --git a/federation-js/src/service/printCoreSchema.ts b/federation-js/src/service/printCoreSchema.ts
@@ -278,14 +278,22 @@ function printEnum(type: GraphQLEnumType, options?: Options): string {
         printDescription(options, value, '  ', !i) +
         '  ' +
         value.name +
-        printDeprecated(value),
+        printDeprecated(value) +
+        printHttpDirective(type, value),
     );
 
   return (
     printDescription(options, type) + `enum ${type.name}` + printBlock(values)
   );
 }
 
+function printHttpDirective(type: GraphQLEnumType, value: GraphQLEnumValue) {
+  if (type.name === "join__Graph") {
+    return ` @http(url: "${value.value}")`
+  }
+  return '';
+}
+
 function printInputObject(
   type: GraphQLInputObjectType,
   options?: Options,