Added fieldReference support

Fixes omar-dulaimi#57
Brettm12345 · Dec 3, 2022 · 8b1d50b · 8b1d50b
1 parent d0cbfef
commit 8b1d50b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/transformer.ts b/src/transformer.ts
@@ -3,7 +3,7 @@ import path from 'path';
 import {
   checkModelHasModelRelation,
   findModelByName,
-  isMongodbRawOp
+  isMongodbRawOp,
 } from './helpers';
 import { isAggregateInputType } from './helpers/aggregate-helpers';
 import { AggregateOperationSupport, TransformerParams } from './types';
@@ -145,6 +145,8 @@ export default class Transformer {
         result.push(
           this.wrapWithZodValidators('z.literal(true)', field, inputType),
         );
+      } else if (inputType.type.toString().includes('Ref')) {
+        return result;
       } else {
         const isEnum = inputType.location === 'enumTypes';
 
@@ -235,8 +237,8 @@ export default class Transformer {
       inputType.type === this.name
         ? objectSchemaLine
         : isEnum
-          ? enumSchemaLine
-          : objectSchemaLine;
+        ? enumSchemaLine
+        : objectSchemaLine;
 
     const arr = inputType.isList ? '.array()' : '';