Refactor getting flattened GQL input

This will make it easier for other input exceptions to use this info

Author: CarsonF

src/common/exceptions/duplicate.exception.ts

@@ -1,8 +1,4 @@
 import { ArgumentsHost } from '@nestjs/common';
-import {
-  GqlContextType as ContextKey,
-  GqlExecutionContext,
-} from '@nestjs/graphql';
 import { lowerCase, upperFirst } from 'lodash';
 import type { ExclusivityViolationError } from '~/core/edgedb';
 import { InputException } from './input.exception';
@@ -27,38 +23,14 @@ export class DuplicateException extends InputException {
 
     // Attempt to add path prefix automatically to the property name, based
     // on given GQL input.
-    if (context && context.getType<ContextKey>() === 'graphql') {
-      let gqlArgs = GqlExecutionContext.create(context as any).getArgs();
-
-      // unwind single `input` argument, based on our own conventions
-      if (Object.keys(gqlArgs).length === 1 && 'input' in gqlArgs) {
-        gqlArgs = gqlArgs.input;
-      }
-
-      const flattened = flattenObject(gqlArgs);
-      // Guess the correct path based on property name.
-      // This kinda assumes the property name will be unique amongst all the input.
-      const guessedPath = Object.keys(flattened).find(
-        (path) => property === path || path.endsWith('.' + property),
-      );
-      property = guessedPath ?? property;
-    }
+    // This kinda assumes the property name will be unique amongst all the input.
+    const guessedPath = Object.keys(
+      InputException.getFlattenInput(context),
+    ).find((path) => property === path || path.endsWith('.' + property));
+    property = guessedPath ?? property;
 
     const ex = new DuplicateException(property, message, exception);
     ex.stack = exception.stack;
     return ex;
   }
 }
-
-const flattenObject = (obj: object, prefix = '') => {
-  const result: Record<string, any> = {};
-  for (const [key, value] of Object.entries(obj)) {
-    if (value && typeof value === 'object' && !Array.isArray(value)) {
-      const nestedObj = flattenObject(value, prefix + key + '.');
-      Object.assign(result, nestedObj);
-    } else {
-      result[prefix + key] = value;
-    }
-  }
-  return result;
-};

src/common/exceptions/input.exception.ts

@@ -1,3 +1,8 @@
+import { ArgumentsHost } from '@nestjs/common';
+import {
+  GqlContextType as ContextKey,
+  GqlExecutionContext,
+} from '@nestjs/graphql';
 import { ClientException } from './exception';
 
 export type InputExceptionArgs =
@@ -104,4 +109,35 @@ export class InputException extends ClientException {
     }
     return [message, field, previous] as const;
   }
+
+  static getFlattenInput(context?: ArgumentsHost) {
+    if (!context || context.getType<ContextKey>() !== 'graphql') {
+      return {};
+    }
+    const gqlContext =
+      context instanceof GqlExecutionContext
+        ? context
+        : GqlExecutionContext.create(context as any);
+    let gqlArgs = gqlContext.getArgs();
+
+    // unwind single `input` argument, based on our own conventions
+    if (Object.keys(gqlArgs).length === 1 && 'input' in gqlArgs) {
+      gqlArgs = gqlArgs.input;
+    }
+
+    return flattenObject(gqlArgs);
+  }
 }
+
+const flattenObject = (obj: object, prefix = '') => {
+  const result: Record<string, any> = {};
+  for (const [key, value] of Object.entries(obj)) {
+    if (value && typeof value === 'object' && !Array.isArray(value)) {
+      const nestedObj = flattenObject(value, prefix + key + '.');
+      Object.assign(result, nestedObj);
+    } else {
+      result[prefix + key] = value;
+    }
+  }
+  return result;
+};

src/core/exception/exception.normalizer.ts

@@ -114,13 +114,17 @@ export class ExceptionNormalizer {
       };
     }
 
+    const gqlContext =
+      context && context.getType<ContextKey>() === 'graphql'
+        ? GqlExecutionContext.create(context as any)
+        : undefined;
+
     if (ex instanceof ExclusivityViolationError) {
-      ex = DuplicateException.fromDB(ex, context);
+      ex = DuplicateException.fromDB(ex, gqlContext);
     } else if (ex instanceof Edge.EdgeDBError) {
       // Mask actual DB error with a nicer user error message.
       let message = 'Failed';
-      if (context && context.getType<ContextKey>() === 'graphql') {
-        const gqlContext = GqlExecutionContext.create(context as any);
+      if (gqlContext) {
         const info = gqlContext.getInfo<GraphQLResolveInfo>();
         if (info.operation.operation === 'mutation') {
           message += ` to ${lowerCase(info.fieldName)}`;