@graphitation/supermassive 3.5.1 → 3.5.3

package/lib/executeWithoutSchema.mjs

@@ -185,6 +185,10 @@ function executeOperation(exeContext) {
     );
     const path = void 0;
     let result;
+    const hooks = exeContext.fieldExecutionHooks;
+    if (hooks == null ? void 0 : hooks.beforeOperationExecute) {
+      invokeBeforeOperationExecuteHook(exeContext);
+    }
     switch (operation.operation) {
       case "query":
         result = executeFields(
@@ -242,6 +246,7 @@ function executeOperation(exeContext) {
   }
 }
 function buildResponse(exeContext, data) {
+  var _a, _b;
   if (isPromise(data)) {
     return data.then(
       (resolved) => buildResponse(exeContext, resolved),
@@ -251,6 +256,7 @@ function buildResponse(exeContext, data) {
       }
     );
   }
+  const hooks = exeContext.fieldExecutionHooks;
   try {
     const initialResult = exeContext.errors.length === 0 ? { data } : { errors: exeContext.errors, data };
     if (exeContext.subsequentPayloads.size > 0) {
@@ -261,6 +267,12 @@ function buildResponse(exeContext, data) {
         subsequentResults: yieldSubsequentPayloads(exeContext)
       };
     } else {
+      if (hooks == null ? void 0 : hooks.afterBuildResponse) {
+        invokeAfterBuildResponseHook(exeContext, initialResult);
+        if (exeContext.errors.length > ((_b = (_a = initialResult.errors) == null ? void 0 : _a.length) != null ? _b : 0)) {
+          initialResult.errors = exeContext.errors;
+        }
+      }
       return initialResult;
     }
   } catch (error) {
@@ -404,6 +416,9 @@ function executeSubscriptionImpl(exeContext) {
   const rootTypeName = getOperationRootTypeName(operation);
   const { groupedFieldSet } = collectFields(exeContext, rootTypeName);
   const firstRootField = groupedFieldSet.entries().next().value;
+  if (firstRootField === void 0) {
+    throw locatedError("Must have at least one root field.", []);
+  }
   const [responseName, fieldGroup] = firstRootField;
   const fieldName = fieldGroup[0].name.value;
   const fieldDef = Definitions.getField(
@@ -432,17 +447,63 @@ function executeSubscriptionImpl(exeContext) {
     typeNameFromReference(returnTypeRef),
     path
   );
+  const isDefaultResolverUsed = resolveFn === exeContext.subscribeFieldResolver;
+  const hooks = exeContext.fieldExecutionHooks;
+  let hookContext = void 0;
   try {
     const args = getArgumentValues(exeContext, fieldDef, fieldGroup[0]);
     const contextValue = exeContext.contextValue;
+    if (!isDefaultResolverUsed && (hooks == null ? void 0 : hooks.beforeFieldResolve)) {
+      hookContext = invokeBeforeFieldResolveHook(info, exeContext);
+    }
     const result = resolveFn(rootValue, args, contextValue, info);
     if (isPromise(result)) {
-      return result.then(assertEventStream).then(void 0, (error) => {
-        throw locatedError(error, fieldGroup, pathToArray(path));
-      });
+      return result.then(assertEventStream).then(
+        (resolved) => {
+          if (!isDefaultResolverUsed && (hooks == null ? void 0 : hooks.afterFieldResolve)) {
+            hookContext = invokeAfterFieldResolveHook(
+              info,
+              exeContext,
+              hookContext,
+              resolved
+            );
+          }
+          return resolved;
+        },
+        (error) => {
+          if (!isDefaultResolverUsed && (hooks == null ? void 0 : hooks.afterFieldResolve)) {
+            hookContext = invokeAfterFieldResolveHook(
+              info,
+              exeContext,
+              hookContext,
+              void 0,
+              error
+            );
+          }
+          throw locatedError(error, fieldGroup, pathToArray(path));
+        }
+      );
+    }
+    const stream = assertEventStream(result);
+    if (!isDefaultResolverUsed && (hooks == null ? void 0 : hooks.afterFieldResolve)) {
+      hookContext = invokeAfterFieldResolveHook(
+        info,
+        exeContext,
+        hookContext,
+        stream
+      );
     }
-    return assertEventStream(result);
+    return stream;
   } catch (error) {
+    if (!isDefaultResolverUsed && (hooks == null ? void 0 : hooks.afterFieldResolve)) {
+      hookContext = invokeAfterFieldResolveHook(
+        info,
+        exeContext,
+        hookContext,
+        void 0,
+        error
+      );
+    }
     throw locatedError(error, fieldGroup, pathToArray(path));
   }
 }
@@ -478,6 +539,10 @@ function mapResultOrEventStreamOrPromise(resultOrStreamOrPromise, exeContext, pa
           exeContext,
           payload
         );
+        const hooks = exeContext == null ? void 0 : exeContext.fieldExecutionHooks;
+        if (hooks == null ? void 0 : hooks.beforeSubscriptionEventEmit) {
+          invokeBeforeSubscriptionEventEmitHook(perEventContext, payload);
+        }
         try {
           const data = executeFields(
             exeContext,
@@ -1207,6 +1272,77 @@ function invokeAfterFieldCompleteHook(resolveInfo, exeContext, hookContext, resu
     }
   );
 }
+function invokeBeforeOperationExecuteHook(exeContext) {
+  var _a;
+  const hook = (_a = exeContext.fieldExecutionHooks) == null ? void 0 : _a.beforeOperationExecute;
+  if (!hook) {
+    return;
+  }
+  return executeSafe(
+    () => hook({
+      context: exeContext.contextValue,
+      operation: exeContext.operation
+    }),
+    (_, rawError) => {
+      if (rawError) {
+        const error = toGraphQLError(
+          rawError,
+          void 0,
+          "Unexpected error in beforeOperationExecute hook"
+        );
+        exeContext.errors.push(error);
+      }
+    }
+  );
+}
+function invokeBeforeSubscriptionEventEmitHook(exeContext, eventPayload) {
+  var _a;
+  const hook = (_a = exeContext.fieldExecutionHooks) == null ? void 0 : _a.beforeSubscriptionEventEmit;
+  if (!hook) {
+    return;
+  }
+  return executeSafe(
+    () => hook({
+      context: exeContext.contextValue,
+      operation: exeContext.operation,
+      eventPayload
+    }),
+    (_, rawError) => {
+      if (rawError) {
+        const error = toGraphQLError(
+          rawError,
+          void 0,
+          "Unexpected error in beforeSubscriptionEventEmit hook"
+        );
+        exeContext.errors.push(error);
+      }
+    }
+  );
+}
+function invokeAfterBuildResponseHook(exeContext, result) {
+  var _a;
+  const hook = (_a = exeContext.fieldExecutionHooks) == null ? void 0 : _a.afterBuildResponse;
+  if (!hook) {
+    return;
+  }
+  return executeSafe(
+    () => hook({
+      context: exeContext.contextValue,
+      operation: exeContext.operation,
+      result
+    }),
+    (_, rawError) => {
+      if (rawError) {
+        const error = toGraphQLError(
+          rawError,
+          void 0,
+          "Unexpected error in afterBuildResponse hook"
+        );
+        exeContext.errors.push(error);
+      }
+    }
+  );
+}
 function executeSafe(execute, onComplete) {
   let error;
   let result;
@@ -1222,7 +1358,7 @@ function executeSafe(execute, onComplete) {
 function toGraphQLError(originalError, path, prependMessage) {
   const originalMessage = originalError instanceof Error ? originalError.message : inspect(originalError);
   const error = new Error(`${prependMessage}: ${originalMessage}`);
-  return locatedError(error, void 0, pathToArray(path));
+  return locatedError(error, void 0, path ? pathToArray(path) : void 0);
 }
 var defaultTypeResolver = function(value) {
   if (isObjectLike(value) && typeof value.__typename === "string") {