Adding support for dynamic refs

Fixes #26 https://mongoosejs.com/docs/populate.html#dynamic-ref
CampbellSoftwareSolutions · Apr 24, 2020 · 49fa236 · 49fa236
1 parent 36e63b3
commit 49fa236
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/lib/id-validator.js b/lib/id-validator.js
@@ -41,6 +41,7 @@ IdValidator.prototype.validateSchema = function (
 
         var validateFunction = null
         var refModelName = null
+        var refModelPath = null
         var conditions = {}
         var optionsSource = null
 
@@ -49,6 +50,11 @@ IdValidator.prototype.validateSchema = function (
             if (schemaType.options.refConditions) {
                 conditions = schemaType.options.refConditions
             }
+        } else if (schemaType.options && schemaType.options.refPath) {
+            refModelPath = schemaType.options.refPath
+            if (schemaType.options.refConditions) {
+                conditions = schemaType.options.refConditions
+            }
         } else if (schemaType.caster && schemaType.caster.instance &&
             schemaType.caster.options && schemaType.caster.options.ref) {
             refModelName = schemaType.caster.options.ref
@@ -63,7 +69,7 @@ IdValidator.prototype.validateSchema = function (
             (schemaType['$isMongooseArray'] === true)
         validateFunction = isArraySchemaType ? validateIdArray : validateId
 
-        if (refModelName) {
+        if (refModelName || refModelPath) {
             schema.path(path).validate({
                 validator: function (value) {
                     return new Promise(function (resolve, reject) {
@@ -84,8 +90,12 @@ IdValidator.prototype.validateSchema = function (
                                     }
                                 })
                             }
+                            var localRefModelName = refModelName
+                            if (refModelPath) {
+                                localRefModelName = this[refModelPath]
+                            }
 
-                            return validateFunction(this, connection, refModelName,
+                            return validateFunction(this, connection, localRefModelName,
                                 value, conditionsCopy, resolve, reject, allowDuplicates)
                         }
                         resolve(true)