feat(): allow registering global field middleware functions

nestjs · Jan 22, 2021 · 773d5d8 · 773d5d8
1 parent c451529
commit 773d5d8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/lib/interfaces/build-schema-options.interface.ts b/lib/interfaces/build-schema-options.interface.ts
@@ -1,4 +1,5 @@
 import { GraphQLDirective, GraphQLScalarType } from 'graphql';
+import { FieldMiddleware } from './field-middleware.interface';
 
 export type DateScalarMode = 'isoDate' | 'timestamp';
 export type NumberScalarMode = 'float' | 'integer';
@@ -45,4 +46,9 @@ export interface BuildSchemaOptions {
    * GraphQL schema directives mapping
    */
   schemaDirectives?: Record<string, any>;
+
+  /**
+   * Array of global field middleware functions
+   */
+  fieldMiddleware?: FieldMiddleware[];
 }
diff --git a/lib/schema-builder/factories/object-type-definition.factory.ts b/lib/schema-builder/factories/object-type-definition.factory.ts
@@ -129,7 +129,7 @@ export class ObjectTypeDefinitionFactory {
           options,
           field.options,
         );
-        const resolve = this.createFieldResolver(field);
+        const resolve = this.createFieldResolver(field, options);
 
         fields[field.schemaName] = {
           type,
@@ -174,12 +174,15 @@ export class ObjectTypeDefinitionFactory {
     TContext = {},
     TArgs = { [argName: string]: any },
     TOutput = any
-  >(field: PropertyMetadata) {
+  >(field: PropertyMetadata, options: BuildSchemaOptions) {
     const rootFieldResolver = (root: object) => {
       const value = root[field.name];
       return typeof value === 'undefined' ? field.options.defaultValue : value;
     };
-    if (!field.middleware || field.middleware?.length === 0) {
+    const middlewareFunctions = (options.fieldMiddleware || []).concat(
+      field.middleware || [],
+    );
+    if (middlewareFunctions?.length === 0) {
       return rootFieldResolver;
     }
 
@@ -199,10 +202,10 @@ export class ObjectTypeDefinitionFactory {
         index = currentIndex;
         let middlewareFn: FieldMiddleware;
 
-        if (currentIndex === field.middleware.length) {
+        if (currentIndex === middlewareFunctions.length) {
           middlewareFn = () => rootFieldResolver(root);
         } else {
-          middlewareFn = field.middleware[currentIndex];
+          middlewareFn = middlewareFunctions[currentIndex];
         }
 
         let tempResult: TOutput = undefined;