Protected against non-Model generation requests

This commit is contained in:
Shelby Sanders 2014-10-07 16:52:27 -07:00
parent 7f4e8b3f87
commit fd30b2d151
1 changed files with 12 additions and 16 deletions

View File

@ -24,10 +24,8 @@ var modelHelper = module.exports = {
modelName = modelName.replace(/[\[\]]/g, '');
}
var model = app.models[modelName];
if (model) {
if (referencedModels.indexOf(model) === -1) {
referencedModels.push(model);
}
if (model && referencedModels.indexOf(model) === -1) {
referencedModels.push(model);
}
}
}
@ -49,7 +47,7 @@ var modelHelper = module.exports = {
for (var m in def.settings.models) {
var model = modelClass[m];
if (typeof model === 'function' && model.modelName) {
if (referencedModels.indexOf(model) === -1) {
if (model && referencedModels.indexOf(model) === -1) {
referencedModels.push(model);
}
}
@ -113,15 +111,11 @@ var modelHelper = module.exports = {
// Generate model definitions for related models
for (var r in modelClass.relations) {
var rel = modelClass.relations[r];
if (rel.modelTo){
if (referencedModels.indexOf(rel.modelTo) === -1) {
referencedModels.push(rel.modelTo);
}
if (rel.modelTo && referencedModels.indexOf(rel.modelTo) === -1) {
referencedModels.push(rel.modelTo);
}
if (rel.modelThrough) {
if (referencedModels.indexOf(rel.modelThrough) === -1) {
referencedModels.push(rel.modelThrough);
}
if (rel.modelThrough && referencedModels.indexOf(rel.modelThrough) === -1) {
referencedModels.push(rel.modelThrough);
}
}
@ -151,7 +145,9 @@ var modelHelper = module.exports = {
}
for (var i = 0, n = referencedModels.length; i < n; i++) {
generateModelDefinition(referencedModels[i], out);
if (referencedModels[i].definition) {
generateModelDefinition(referencedModels[i], out);
}
}
return out;
},