Updated 'ref' parameter to allow population by string (#154)

Updated 'ref' parameter to allow population by string
szokodiakos · Aug 20, 2018 · 252c740 · 252c740
1 parent 7d5a021
commit 252c740
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/prop.ts b/src/prop.ts
@@ -92,13 +92,20 @@ const baseProp = (rawOptions, Type, target, key, isArray = false) => {
   }
 
   const ref = rawOptions.ref;
-  if (ref) {
+  if (typeof ref === 'string') {
     schema[name][key] = {
       ...schema[name][key],
       type: mongoose.Schema.Types.ObjectId,
-      ref: ref.name,
+      ref,
     };
     return;
+  } else if (ref) {
+     schema[name][key] = {
+      ...schema[name][key],
+      type: mongoose.Schema.Types.ObjectId,
+      ref: ref.name,
+    };
+     return;
   }
 
   const itemsRef = rawOptions.itemsRef;