Merge pull request #21 from lirown/lirown-tree-iteration-support-old-version

fixes

Author: lirown

package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphql-custom-directives",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Create custom graphql directives with the ability to hook the execution of graphql",
   "main": "dist/index.js",
   "scripts": {

src/custom.js

@@ -141,11 +141,10 @@ function resolveMiddlewareWrapper(resolve = defaultResolveFn, directives = {}) {
  * Scanning the shema and wrapping the resolve of each field with the support
  * of the graphql custom directives resolve execution
  */
-function wrapFieldsWithMiddleware(type, deepWrap = true, typeMet = {}) {
+function wrapFieldsWithMiddleware(type, deepWrap = true, typeMet = {}, level = 1) {
   if (!type) {
     return;
   }
-
   let fields = type._fields;
   typeMet[type.name] = true;
   for (let label in fields) {
@@ -157,21 +156,22 @@ function wrapFieldsWithMiddleware(type, deepWrap = true, typeMet = {}) {
           field.directives,
         );
         if (field.type._fields && deepWrap) {
-          wrapFieldsWithMiddleware(field.type, deepWrap, typeMet);
+          wrapFieldsWithMiddleware(field.type, deepWrap, typeMet, level);
         } else if (field.type.ofType && field.type.ofType._fields && deepWrap) {
-          let child = field.type;
-          while (child.ofType) {
-            child = child.ofType;
-          }
-          if (child._fields) {
-            wrapFieldsWithMiddleware(child._fields);
+          if (level >= 6) {
+            continue;
           }
+          wrapFieldsWithMiddleware(
+            field.type.ofType,
+            deepWrap,
+            typeMet,
+            ++level,
+          );
         }
       }
     }
   }
 }
-
 /**
  * create a new graphql custom directive which contain a resolve
  * function for altering the execution of the graphql