@theguild/federation-composition 0.12.1 → 0.13.0-alpha-20241002090328-16d32e8

package/cjs/subgraph/state.js

@@ -361,6 +361,11 @@ function createSubgraphStateBuilder(graph, typeDefs, version, links) {
                                 }
                                 break;
                             }
+                            case graphql_1.Kind.ENUM_TYPE_DEFINITION:
+                            case graphql_1.Kind.ENUM_TYPE_EXTENSION: {
+                                enumTypeBuilder.setDirective(typeDef.name.value, node);
+                                break;
+                            }
                             default:
                                 throw new Error(`Directives on "${typeDef.kind}" types are not supported yet`);
                         }
@@ -1126,6 +1131,9 @@ function enumTypeFactory(state) {
             getOrCreateEnumType(state, typeName).referencedByOutputType = true;
             getOrCreateEnumType(state, typeName).outputTypeReferences.add(schemaCoordinate);
         },
+        setDirective(typeName, directive) {
+            getOrCreateEnumType(state, typeName).ast.directives.push(directive);
+        },
         value: {
             setValue(typeName, valueName) {
                 getOrCreateEnumValue(state, typeName, valueName);
@@ -1317,6 +1325,9 @@ function getOrCreateEnumType(state, typeName) {
         referencedByOutputType: false,
         inputTypeReferences: new Set(),
         outputTypeReferences: new Set(),
+        ast: {
+            directives: [],
+        },
     };
     state.types.set(typeName, enumType);
     return enumType;

package/cjs/supergraph/composition/enum-type.js

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.enumTypeBuilder = void 0;
 const ast_js_1 = require("./ast.js");
+const common_js_1 = require("./common.js");
 function enumTypeBuilder() {
     return {
         visitSubgraphState(graph, state, typeName, type) {
@@ -37,6 +38,9 @@ function enumTypeBuilder() {
                     enumTypeState.outputTypeReferences.add(ref);
                 });
             }
+            type.ast.directives.forEach(directive => {
+                enumTypeState.ast.directives.push(directive);
+            });
             enumTypeState.byGraph.set(graph.id, {
                 inaccessible: type.inaccessible,
                 version: graph.version,
@@ -89,6 +93,9 @@ function enumTypeBuilder() {
                         graph: graphName.toUpperCase(),
                     })),
                 },
+                ast: {
+                    directives: (0, common_js_1.convertToConst)(enumType.ast.directives),
+                },
             });
         },
     };
@@ -130,6 +137,9 @@ function getOrCreateEnumType(state, typeName) {
         inputTypeReferences: new Set(),
         outputTypeReferences: new Set(),
         byGraph: new Map(),
+        ast: {
+            directives: [],
+        },
     };
     state.set(typeName, def);
     return def;

package/esm/subgraph/state.js

@@ -358,6 +358,11 @@ export function createSubgraphStateBuilder(graph, typeDefs, version, links) {
                                 }
                                 break;
                             }
+                            case Kind.ENUM_TYPE_DEFINITION:
+                            case Kind.ENUM_TYPE_EXTENSION: {
+                                enumTypeBuilder.setDirective(typeDef.name.value, node);
+                                break;
+                            }
                             default:
                                 throw new Error(`Directives on "${typeDef.kind}" types are not supported yet`);
                         }
@@ -1120,6 +1125,9 @@ function enumTypeFactory(state) {
             getOrCreateEnumType(state, typeName).referencedByOutputType = true;
             getOrCreateEnumType(state, typeName).outputTypeReferences.add(schemaCoordinate);
         },
+        setDirective(typeName, directive) {
+            getOrCreateEnumType(state, typeName).ast.directives.push(directive);
+        },
         value: {
             setValue(typeName, valueName) {
                 getOrCreateEnumValue(state, typeName, valueName);
@@ -1311,6 +1319,9 @@ function getOrCreateEnumType(state, typeName) {
         referencedByOutputType: false,
         inputTypeReferences: new Set(),
         outputTypeReferences: new Set(),
+        ast: {
+            directives: [],
+        },
     };
     state.types.set(typeName, enumType);
     return enumType;

package/esm/supergraph/composition/enum-type.js

@@ -1,4 +1,5 @@
 import { createEnumTypeNode } from './ast.js';
+import { convertToConst } from './common.js';
 export function enumTypeBuilder() {
     return {
         visitSubgraphState(graph, state, typeName, type) {
@@ -34,6 +35,9 @@ export function enumTypeBuilder() {
                     enumTypeState.outputTypeReferences.add(ref);
                 });
             }
+            type.ast.directives.forEach(directive => {
+                enumTypeState.ast.directives.push(directive);
+            });
             enumTypeState.byGraph.set(graph.id, {
                 inaccessible: type.inaccessible,
                 version: graph.version,
@@ -86,6 +90,9 @@ export function enumTypeBuilder() {
                         graph: graphName.toUpperCase(),
                     })),
                 },
+                ast: {
+                    directives: convertToConst(enumType.ast.directives),
+                },
             });
         },
     };
@@ -126,6 +133,9 @@ function getOrCreateEnumType(state, typeName) {
         inputTypeReferences: new Set(),
         outputTypeReferences: new Set(),
         byGraph: new Map(),
+        ast: {
+            directives: [],
+        },
     };
     state.set(typeName, def);
     return def;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@theguild/federation-composition",
-  "version": "0.12.1",
+  "version": "0.13.0-alpha-20241002090328-16d32e8",
   "description": "Open Source Composition library for Apollo Federation",
   "peerDependencies": {
     "graphql": "^16.0.0"

package/typings/subgraph/state.d.cts

@@ -120,6 +120,9 @@ export interface EnumType {
     referencedByOutputType: boolean;
     inputTypeReferences: Set<string>;
     outputTypeReferences: Set<string>;
+    ast: {
+        directives: DirectiveNode[];
+    };
 }
 export interface Field {
     name: string;
@@ -369,6 +372,7 @@ export declare function createSubgraphStateBuilder(graph: {
         setTag(typeName: string, tag: string): void;
         setReferencedByInputType(typeName: string, schemaCoordinate: string): void;
         setReferencedByOutputType(typeName: string, schemaCoordinate: string): void;
+        setDirective(typeName: string, directive: DirectiveNode): void;
         value: {
             setValue(typeName: string, valueName: string): void;
             setDescription(typeName: string, valueName: string, description: Description): void;

package/typings/subgraph/state.d.ts

@@ -120,6 +120,9 @@ export interface EnumType {
     referencedByOutputType: boolean;
     inputTypeReferences: Set<string>;
     outputTypeReferences: Set<string>;
+    ast: {
+        directives: DirectiveNode[];
+    };
 }
 export interface Field {
     name: string;
@@ -369,6 +372,7 @@ export declare function createSubgraphStateBuilder(graph: {
         setTag(typeName: string, tag: string): void;
         setReferencedByInputType(typeName: string, schemaCoordinate: string): void;
         setReferencedByOutputType(typeName: string, schemaCoordinate: string): void;
+        setDirective(typeName: string, directive: DirectiveNode): void;
         value: {
             setValue(typeName: string, valueName: string): void;
             setDescription(typeName: string, valueName: string, description: Description): void;

package/typings/subgraph/validation/validation-context.d.cts

@@ -169,6 +169,7 @@ export declare function createSubgraphValidationContext(subgraph: {
             setTag(typeName: string, tag: string): void;
             setReferencedByInputType(typeName: string, schemaCoordinate: string): void;
             setReferencedByOutputType(typeName: string, schemaCoordinate: string): void;
+            setDirective(typeName: string, directive: import("graphql").DirectiveNode): void;
             value: {
                 setValue(typeName: string, valueName: string): void;
                 setDescription(typeName: string, valueName: string, description: import("../state.js").Description): void;

package/typings/subgraph/validation/validation-context.d.ts

@@ -169,6 +169,7 @@ export declare function createSubgraphValidationContext(subgraph: {
             setTag(typeName: string, tag: string): void;
             setReferencedByInputType(typeName: string, schemaCoordinate: string): void;
             setReferencedByOutputType(typeName: string, schemaCoordinate: string): void;
+            setDirective(typeName: string, directive: import("graphql").DirectiveNode): void;
             value: {
                 setValue(typeName: string, valueName: string): void;
                 setDescription(typeName: string, valueName: string, description: import("../state.js").Description): void;

package/typings/supergraph/composition/enum-type.d.cts

@@ -1,6 +1,7 @@
+import type { DirectiveNode } from 'graphql';
 import { FederationVersion } from '../../specifications/federation.cjs';
 import { Deprecated, Description, EnumType } from '../../subgraph/state.cjs';
-import type { MapByGraph, TypeBuilder } from './common.cjs';
+import { type MapByGraph, type TypeBuilder } from './common.cjs';
 export declare function enumTypeBuilder(): TypeBuilder<EnumType, EnumTypeState>;
 export type EnumTypeState = {
     kind: 'enum';
@@ -18,6 +19,9 @@ export type EnumTypeState = {
     inputTypeReferences: Set<string>;
     outputTypeReferences: Set<string>;
     values: Map<string, EnumValueState>;
+    ast: {
+        directives: DirectiveNode[];
+    };
 };
 type EnumValueState = {
     name: string;

package/typings/supergraph/composition/enum-type.d.ts

@@ -1,6 +1,7 @@
+import type { DirectiveNode } from 'graphql';
 import { FederationVersion } from '../../specifications/federation.js';
 import { Deprecated, Description, EnumType } from '../../subgraph/state.js';
-import type { MapByGraph, TypeBuilder } from './common.js';
+import { type MapByGraph, type TypeBuilder } from './common.js';
 export declare function enumTypeBuilder(): TypeBuilder<EnumType, EnumTypeState>;
 export type EnumTypeState = {
     kind: 'enum';
@@ -18,6 +19,9 @@ export type EnumTypeState = {
     inputTypeReferences: Set<string>;
     outputTypeReferences: Set<string>;
     values: Map<string, EnumValueState>;
+    ast: {
+        directives: DirectiveNode[];
+    };
 };
 type EnumValueState = {
     name: string;