/*! * Dependencies */ var assert = require('assert'); var util = require('util'); var i8n = require('inflection'); var defineScope = require('./scope.js').defineScope; var mergeQuery = require('./scope.js').mergeQuery; var ModelBaseClass = require('./model.js'); exports.Relation = Relation; exports.RelationDefinition = RelationDefinition; var RelationTypes = { belongsTo: 'belongsTo', hasMany: 'hasMany', hasOne: 'hasOne', hasAndBelongsToMany: 'hasAndBelongsToMany' }; exports.RelationTypes = RelationTypes; exports.HasMany = HasMany; exports.HasManyThrough = HasManyThrough; exports.HasOne = HasOne; exports.HasAndBelongsToMany = HasAndBelongsToMany; exports.BelongsTo = BelongsTo; var RelationClasses = { belongsTo: BelongsTo, hasMany: HasMany, hasManyThrough: HasManyThrough, hasOne: HasOne, hasAndBelongsToMany: HasAndBelongsToMany }; function normalizeType(type) { if (!type) { return type; } var t1 = type.toLowerCase(); for (var t2 in RelationTypes) { if (t2.toLowerCase() === t1) { return t2; } } return null; } /** * Relation definition class. Use to define relationships between models. * @param {Object} definition * @class RelationDefinition */ function RelationDefinition(definition) { if (!(this instanceof RelationDefinition)) { return new RelationDefinition(definition); } definition = definition || {}; this.name = definition.name; assert(this.name, 'Relation name is missing'); this.type = normalizeType(definition.type); assert(this.type, 'Invalid relation type: ' + definition.type); this.modelFrom = definition.modelFrom; assert(this.modelFrom, 'Source model is required'); this.keyFrom = definition.keyFrom; this.modelTo = definition.modelTo; this.keyTo = definition.keyTo; this.typeTo = definition.typeTo; if (!this.typeTo) { assert(this.modelTo, 'Target model is required'); } this.modelThrough = definition.modelThrough; this.keyThrough = definition.keyThrough; this.multiple = (this.type !== 'belongsTo' && this.type !== 'hasOne'); this.properties = definition.properties || {}; this.options = definition.options || {}; this.scope = definition.scope; } RelationDefinition.prototype.toJSON = function () { var json = { name: this.name, type: this.type, modelFrom: this.modelFrom.modelName, keyFrom: this.keyFrom, modelTo: this.modelTo.modelName, keyTo: this.keyTo, multiple: this.multiple }; if (this.modelThrough) { json.modelThrough = this.modelThrough.modelName; json.keyThrough = this.keyThrough; } return json; }; /** * Apply the configured scope to the filter/query object. * @param {Object} modelInstance * @param {Object} filter (where, order, limit, fields, ...) */ RelationDefinition.prototype.applyScope = function(modelInstance, filter) { filter.where = filter.where || {}; if ((this.type !== 'belongsTo' || this.type === 'hasOne') && typeof this.typeTo === 'string') { // polymorphic filter.where[this.typeTo] = this.modelFrom.modelName; } if (typeof this.scope === 'function') { var scope = this.scope.call(this, modelInstance, filter); } else { var scope = this.scope; } if (typeof scope === 'object') { mergeQuery(filter, scope); } }; /** * Apply the configured properties to the target object. * @param {Object} modelInstance * @param {Object} target */ RelationDefinition.prototype.applyProperties = function(modelInstance, target) { if (typeof this.properties === 'function') { var data = this.properties.call(this, modelInstance); for(var k in data) { target[k] = data[k]; } } else if (typeof this.properties === 'object') { for(var k in this.properties) { var key = this.properties[k]; target[key] = modelInstance[k]; } } if ((this.type !== 'belongsTo' || this.type === 'hasOne') && typeof this.typeTo === 'string') { // polymorphic target[this.typeTo] = this.modelFrom.modelName; } }; /** * A relation attaching to a given model instance * @param {RelationDefinition|Object} definition * @param {Object} modelInstance * @returns {Relation} * @constructor * @class Relation */ function Relation(definition, modelInstance) { if (!(this instanceof Relation)) { return new Relation(definition, modelInstance); } if (!(definition instanceof RelationDefinition)) { definition = new RelationDefinition(definition); } this.definition = definition; this.modelInstance = modelInstance; } Relation.prototype.resetCache = function (cache) { cache = cache || undefined; this.modelInstance.__cachedRelations[this.definition.name] = cache; }; Relation.prototype.getCache = function () { return this.modelInstance.__cachedRelations[this.definition.name]; }; /** * HasMany subclass * @param {RelationDefinition|Object} definition * @param {Object} modelInstance * @returns {HasMany} * @constructor * @class HasMany */ function HasMany(definition, modelInstance) { if (!(this instanceof HasMany)) { return new HasMany(definition, modelInstance); } assert(definition.type === RelationTypes.hasMany); Relation.apply(this, arguments); } util.inherits(HasMany, Relation); HasMany.prototype.removeFromCache = function (id) { var cache = this.modelInstance.__cachedRelations[this.definition.name]; var idName = this.definition.modelTo.definition.idName(); if (Array.isArray(cache)) { for (var i = 0, n = cache.length; i < n; i++) { if (cache[i][idName] === id) { return cache.splice(i, 1); } } } return null; }; HasMany.prototype.addToCache = function (inst) { if (!inst) { return; } var cache = this.modelInstance.__cachedRelations[this.definition.name]; if (cache === undefined) { cache = this.modelInstance.__cachedRelations[this.definition.name] = []; } var idName = this.definition.modelTo.definition.idName(); if (Array.isArray(cache)) { for (var i = 0, n = cache.length; i < n; i++) { if (cache[i][idName] === inst[idName]) { cache[i] = inst; return; } } cache.push(inst); } }; /** * HasManyThrough subclass * @param {RelationDefinition|Object} definition * @param {Object} modelInstance * @returns {HasManyThrough} * @constructor * @class HasManyThrough */ function HasManyThrough(definition, modelInstance) { if (!(this instanceof HasManyThrough)) { return new HasManyThrough(definition, modelInstance); } assert(definition.type === RelationTypes.hasMany); assert(definition.modelThrough); HasMany.apply(this, arguments); } util.inherits(HasManyThrough, HasMany); /** * BelongsTo subclass * @param {RelationDefinition|Object} definition * @param {Object} modelInstance * @returns {BelongsTo} * @constructor * @class BelongsTo */ function BelongsTo(definition, modelInstance) { if (!(this instanceof BelongsTo)) { return new BelongsTo(definition, modelInstance); } assert(definition.type === RelationTypes.belongsTo); Relation.apply(this, arguments); } util.inherits(BelongsTo, Relation); /** * HasAndBelongsToMany subclass * @param {RelationDefinition|Object} definition * @param {Object} modelInstance * @returns {HasAndBelongsToMany} * @constructor * @class HasAndBelongsToMany */ function HasAndBelongsToMany(definition, modelInstance) { if (!(this instanceof HasAndBelongsToMany)) { return new HasAndBelongsToMany(definition, modelInstance); } assert(definition.type === RelationTypes.hasAndBelongsToMany); Relation.apply(this, arguments); } util.inherits(HasAndBelongsToMany, Relation); /** * HasOne subclass * @param {RelationDefinition|Object} definition * @param {Object} modelInstance * @returns {HasOne} * @constructor * @class HasOne */ function HasOne(definition, modelInstance) { if (!(this instanceof HasOne)) { return new HasOne(definition, modelInstance); } assert(definition.type === RelationTypes.hasOne); Relation.apply(this, arguments); } util.inherits(HasOne, Relation); /*! * Find the relation by foreign key * @param {*} foreignKey The foreign key * @returns {Object} The relation object */ function findBelongsTo(modelFrom, modelTo, keyTo) { var relations = modelFrom.relations; var keys = Object.keys(relations); for (var k = 0; k < keys.length; k++) { var rel = relations[keys[k]]; if (rel.type === RelationTypes.belongsTo && rel.modelTo === modelTo && (keyTo === undefined || rel.keyTo === keyTo)) { return rel.keyFrom; } } return null; } /*! * Look up a model by name from the list of given models * @param {Object} models Models keyed by name * @param {String} modelName The model name * @returns {*} The matching model class */ function lookupModel(models, modelName) { if(models[modelName]) { return models[modelName]; } var lookupClassName = modelName.toLowerCase(); for (var name in models) { if (name.toLowerCase() === lookupClassName) { return models[name]; } } } /** * Define a "one to many" relationship by specifying the model name * * Examples: * ``` * User.hasMany(Post, {as: 'posts', foreignKey: 'authorId'}); * ``` * * ``` * Book.hasMany(Chapter); * ``` * Or, equivalently: * ``` * Book.hasMany('chapters', {model: Chapter}); * ``` * @param {Model} modelFrom Source model class * @param {Object|String} modelTo Model object (or String name of model) to which you are creating the relationship. * @options {Object} params Configuration parameters; see below. * @property {String} as Name of the property in the referring model that corresponds to the foreign key field in the related model. * @property {String} foreignKey Property name of foreign key field. * @property {Object} model Model object */ RelationDefinition.hasMany = function hasMany(modelFrom, modelTo, params) { var thisClassName = modelFrom.modelName; params = params || {}; if (typeof modelTo === 'string') { params.as = modelTo; if (params.model) { modelTo = params.model; } else { var modelToName = i8n.singularize(modelTo).toLowerCase(); modelTo = lookupModel(modelFrom.dataSource.modelBuilder.models, modelToName); } } var relationName = params.as || i8n.camelize(modelTo.pluralModelName, true); var fk = params.foreignKey || i8n.camelize(thisClassName + '_id', true); var idName = modelFrom.dataSource.idName(modelFrom.modelName) || 'id'; var typeTo; if (typeof params.polymorphic === 'string') { var polymorphic = params.polymorphic; typeTo = i8n.camelize(polymorphic + '_type', true); if (!params.invert) { fk = i8n.camelize(polymorphic + '_id', true); } if (!params.through) { modelTo.dataSource.defineProperty(modelTo.modelName, typeTo, { type: 'string', index: true }); } } var definition = new RelationDefinition({ name: relationName, type: RelationTypes.hasMany, modelFrom: modelFrom, keyFrom: idName, keyTo: fk, typeTo: typeTo, modelTo: modelTo, multiple: true, properties: params.properties, scope: params.scope, options: params.options }); definition.modelThrough = params.through; var keyThrough = definition.throughKey || i8n.camelize(modelTo.modelName + '_id', true); definition.keyThrough = keyThrough; modelFrom.relations[relationName] = definition; if (!params.through) { // obviously, modelTo should have attribute called `fk` // for polymorphic relations, it is assumed to share the same fk type for all // polymorphic models modelTo.dataSource.defineForeignKey(modelTo.modelName, fk, modelFrom.modelName); } var scopeMethods = { findById: scopeMethod(definition, 'findById'), destroy: scopeMethod(definition, 'destroyById'), updateById: scopeMethod(definition, 'updateById'), exists: scopeMethod(definition, 'exists') }; var findByIdFunc = scopeMethods.findById; modelFrom.prototype['__findById__' + relationName] = findByIdFunc; var destroyByIdFunc = scopeMethods.destroy; modelFrom.prototype['__destroyById__' + relationName] = destroyByIdFunc; var updateByIdFunc = scopeMethods.updateById; modelFrom.prototype['__updateById__' + relationName] = updateByIdFunc; if(definition.modelThrough) { scopeMethods.create = scopeMethod(definition, 'create'); scopeMethods.add = scopeMethod(definition, 'add'); scopeMethods.remove = scopeMethod(definition, 'remove'); var addFunc = scopeMethods.add; modelFrom.prototype['__link__' + relationName] = addFunc; var removeFunc = scopeMethods.remove; modelFrom.prototype['__unlink__' + relationName] = removeFunc; } else { scopeMethods.create = scopeMethod(definition, 'create'); scopeMethods.build = scopeMethod(definition, 'build'); } // Mix the property and scoped methods into the prototype class defineScope(modelFrom.prototype, params.through || modelTo, relationName, function () { var filter = {}; filter.where = {}; filter.where[fk] = this[idName]; definition.applyScope(this, filter); if (params.through && params.polymorphic && params.invert) { filter.where[typeTo] = modelTo.modelName; // overwrite filter.collect = params.polymorphic; filter.include = filter.collect; } else if (params.through) { filter.collect = i8n.camelize(modelTo.modelName, true); filter.include = filter.collect; } return filter; }, scopeMethods); }; function scopeMethod(definition, methodName) { var relationClass = RelationClasses[definition.type]; if (definition.type === RelationTypes.hasMany && definition.modelThrough) { relationClass = RelationClasses.hasManyThrough; } var method = function () { var relation = new relationClass(definition, this); return relation[methodName].apply(relation, arguments); }; var relationMethod = relationClass.prototype[methodName]; if (relationMethod.shared) { method.shared = true; method.accepts = relationMethod.accepts; method.returns = relationMethod.returns; method.http = relationMethod.http; method.description = relationMethod.description; } return method; } /** * Find a related item by foreign key * @param {*} fkId The foreign key * @param {Function} cb The callback function */ HasMany.prototype.findById = function (fkId, cb) { var modelTo = this.definition.modelTo; var fk = this.definition.keyTo; var pk = this.definition.keyFrom; var modelInstance = this.modelInstance; var idName = this.definition.modelTo.definition.idName(); var filter = {}; filter.where = {}; filter.where[idName] = fkId; filter.where[fk] = modelInstance[pk]; this.definition.applyScope(modelInstance, filter); modelTo.findOne(filter, function (err, inst) { if (err) { return cb(err); } if (!inst) { err = new Error('No instance with id ' + fkId + ' found for ' + modelTo.modelName); err.statusCode = 404; return cb(err); } // Check if the foreign key matches the primary key if (inst[fk] && inst[fk].toString() === modelInstance[pk].toString()) { cb(null, inst); } else { err = new Error('Key mismatch: ' + this.definition.modelFrom.modelName + '.' + pk + ': ' + modelInstance[pk] + ', ' + modelTo.modelName + '.' + fk + ': ' + inst[fk]); err.statusCode = 400; cb(err); } }); }; /** * Find a related item by foreign key * @param {*} fkId The foreign key * @param {Function} cb The callback function */ HasMany.prototype.exists = function (fkId, cb) { var modelTo = this.definition.modelTo; var fk = this.definition.keyTo; var pk = this.definition.keyFrom; var modelInstance = this.modelInstance; modelTo.findById(fkId, function (err, inst) { if (err) { return cb(err); } if (!inst) { return cb(null, false); } // Check if the foreign key matches the primary key if (inst[fk] && inst[fk].toString() === modelInstance[pk].toString()) { cb(null, true); } else { cb(null, false); } }); }; /** * Update a related item by foreign key * @param {*} fkId The foreign key * @param {Function} cb The callback function */ HasMany.prototype.updateById = function (fkId, data, cb) { this.findById(fkId, function (err, inst) { if (err) { return cb && cb(err); } inst.updateAttributes(data, cb); }); }; /** * Delete a related item by foreign key * @param {*} fkId The foreign key * @param {Function} cb The callback function */ HasMany.prototype.destroyById = function (fkId, cb) { var self = this; this.findById(fkId, function(err, inst) { if (err) { return cb(err); } self.removeFromCache(inst[fkId]); inst.destroy(cb); }); }; var throughKeys = function(definition) { var modelThrough = definition.modelThrough; var pk2 = definition.modelTo.definition.idName(); if (definition.typeTo) { // polymorphic var fk1 = definition.keyTo; var fk2 = definition.keyThrough; } else { var fk1 = findBelongsTo(modelThrough, definition.modelFrom, definition.keyFrom); var fk2 = findBelongsTo(modelThrough, definition.modelTo, pk2); } return [fk1, fk2]; } /** * Find a related item by foreign key * @param {*} fkId The foreign key value * @param {Function} cb The callback function */ HasManyThrough.prototype.findById = function (fkId, cb) { var self = this; var modelTo = this.definition.modelTo; var pk = this.definition.keyFrom; var modelInstance = this.modelInstance; var modelThrough = this.definition.modelThrough; self.exists(fkId, function (err, exists) { if (err || !exists) { if (!err) { err = new Error('No relation found in ' + modelThrough.modelName + ' for (' + self.definition.modelFrom.modelName + '.' + modelInstance[pk] + ',' + modelTo.modelName + '.' + fkId + ')'); err.statusCode = 404; } return cb(err); } modelTo.findById(fkId, function (err, inst) { if (err) { return cb(err); } if (!inst) { err = new Error('No instance with id ' + fkId + ' found for ' + modelTo.modelName); err.statusCode = 404; return cb(err); } cb(err, inst); }); }); }; /** * Delete a related item by foreign key * @param {*} fkId The foreign key * @param {Function} cb The callback function */ HasManyThrough.prototype.destroyById = function (fkId, cb) { var self = this; var modelTo = this.definition.modelTo; var pk = this.definition.keyFrom; var modelInstance = this.modelInstance; var modelThrough = this.definition.modelThrough; self.exists(fkId, function (err, exists) { if (err || !exists) { if (!err) { err = new Error('No record found in ' + modelThrough.modelName + ' for (' + self.definition.modelFrom.modelName + '.' + modelInstance[pk] + ' ,' + modelTo.modelName + '.' + fkId + ')'); err.statusCode = 404; } return cb(err); } self.remove(fkId, function(err) { if(err) { return cb(err); } modelTo.deleteById(fkId, cb); }); }); }; // Create an instance of the target model and connect it to the instance of // the source model by creating an instance of the through model HasManyThrough.prototype.create = function create(data, done) { var self = this; var definition = this.definition; var modelTo = definition.modelTo; var modelThrough = definition.modelThrough; if (typeof data === 'function' && !done) { done = data; data = {}; } var modelInstance = this.modelInstance; // First create the target model modelTo.create(data, function (err, to) { if (err) { return done && done(err, to); } // The primary key for the target model var pk2 = definition.modelTo.definition.idName(); var keys = throughKeys(definition); var fk1 = keys[0]; var fk2 = keys[1]; var d = {}; d[fk1] = modelInstance[definition.keyFrom]; d[fk2] = to[pk2]; definition.applyProperties(modelInstance, d); // Then create the through model modelThrough.create(d, function (e, through) { if (e) { // Undo creation of the target model to.destroy(function () { done && done(e); }); } else { self.addToCache(to); done && done(err, to); } }); }); }; /** * Add the target model instance to the 'hasMany' relation * @param {Object|ID} acInst The actual instance or id value */ HasManyThrough.prototype.add = function (acInst, done) { var self = this; var definition = this.definition; var modelThrough = definition.modelThrough; var pk1 = definition.keyFrom; var data = {}; var query = {}; // The primary key for the target model var pk2 = definition.modelTo.definition.idName(); var keys = throughKeys(definition); var fk1 = keys[0]; var fk2 = keys[1]; query[fk1] = this.modelInstance[pk1]; query[fk2] = (acInst instanceof definition.modelTo) ? acInst[pk2] : acInst; var filter = { where: query }; definition.applyScope(this.modelInstance, filter); data[fk1] = this.modelInstance[pk1]; data[fk2] = (acInst instanceof definition.modelTo) ? acInst[pk2] : acInst; definition.applyProperties(this.modelInstance, data); // Create an instance of the through model modelThrough.findOrCreate(filter, data, function(err, ac) { if(!err) { if (acInst instanceof definition.modelTo) { self.addToCache(acInst); } } done(err, ac); }); }; /** * Check if the target model instance is related to the 'hasMany' relation * @param {Object|ID} acInst The actual instance or id value */ HasManyThrough.prototype.exists = function (acInst, done) { var definition = this.definition; var modelThrough = definition.modelThrough; var pk1 = definition.keyFrom; var query = {}; // The primary key for the target model var pk2 = definition.modelTo.definition.idName(); var keys = throughKeys(definition); var fk1 = keys[0]; var fk2 = keys[1]; query[fk1] = this.modelInstance[pk1]; query[fk2] = (acInst instanceof definition.modelTo) ? acInst[pk2] : acInst; var filter = { where: query }; definition.applyScope(this.modelInstance, filter); modelThrough.count(filter.where, function(err, ac) { done(err, ac > 0); }); }; /** * Remove the target model instance from the 'hasMany' relation * @param {Object|ID) acInst The actual instance or id value */ HasManyThrough.prototype.remove = function (acInst, done) { var self = this; var definition = this.definition; var modelThrough = definition.modelThrough; var pk1 = definition.keyFrom; var query = {}; // The primary key for the target model var pk2 = definition.modelTo.definition.idName(); var keys = throughKeys(definition); var fk1 = keys[0]; var fk2 = keys[1]; query[fk1] = this.modelInstance[pk1]; query[fk2] = (acInst instanceof definition.modelTo) ? acInst[pk2] : acInst; var filter = { where: query }; definition.applyScope(this.modelInstance, filter); modelThrough.deleteAll(filter.where, function (err) { if (!err) { self.removeFromCache(query[fk2]); } done(err); }); }; /** * Declare "belongsTo" relation that sets up a one-to-one connection with * another model, such that each instance of the declaring model "belongs to" * one instance of the other model. * * For example, if an application includes users and posts, and each post can * be written by exactly one user. The following code specifies that `Post` has * a reference called `author` to the `User` model via the `userId` property of * `Post` as the foreign key. * ``` * Post.belongsTo(User, {as: 'author', foreignKey: 'userId'}); * ``` * * This optional parameter default value is false, so the related object will * be loaded from cache if available. * * @param {Class|String} modelTo Model object (or String name of model) to * which you are creating the relationship. * @options {Object} params Configuration parameters; see below. * @property {String} as Name of the property in the referring model that * corresponds to the foreign key field in the related model. * @property {String} foreignKey Name of foreign key property. * */ RelationDefinition.belongsTo = function (modelFrom, modelTo, params) { var typeTo, params = params || {}; if ('string' === typeof modelTo && !params.polymorphic) { params.as = modelTo; if (params.model) { modelTo = params.model; } else { var modelToName = modelTo.toLowerCase(); modelTo = lookupModel(modelFrom.dataSource.modelBuilder.models, modelToName); } } if (params.polymorphic) { var polymorphic = modelTo; modelTo = null; // will lookup dynamically var idName = params.idName || 'id'; var relationName = params.as || i8n.camelize(polymorphic, true); var fk = i8n.camelize(polymorphic + '_id', true); var typeTo = i8n.camelize(polymorphic + '_type', true); if (typeof params.idType === 'string') { // explicit key type modelFrom.dataSource.defineProperty(modelFrom.modelName, fk, { type: params.idType, index: true }); } else { // try to use the same foreign key type as modelFrom modelFrom.dataSource.defineForeignKey(modelFrom.modelName, fk, modelFrom.modelName); } modelFrom.dataSource.defineProperty(modelFrom.modelName, typeTo, { type: 'string', index: true }); } else { var idName = modelFrom.dataSource.idName(modelTo.modelName) || 'id'; var relationName = params.as || i8n.camelize(modelTo.modelName, true); var fk = params.foreignKey || relationName + 'Id'; modelFrom.dataSource.defineForeignKey(modelFrom.modelName, fk, modelTo.modelName); } var relationDef = modelFrom.relations[relationName] = new RelationDefinition({ name: relationName, type: RelationTypes.belongsTo, modelFrom: modelFrom, keyFrom: fk, keyTo: idName, typeTo: typeTo, modelTo: modelTo, properties: params.properties, scope: params.scope, options: params.options }); // Define a property for the scope so that we have 'this' for the scoped methods Object.defineProperty(modelFrom.prototype, relationName, { enumerable: true, configurable: true, get: function() { var relation = new BelongsTo(relationDef, this); var relationMethod = relation.related.bind(relation); relationMethod.create = relation.create.bind(relation); relationMethod.build = relation.build.bind(relation); if (relationDef.modelTo) { relationMethod._targetClass = relationDef.modelTo.modelName; } return relationMethod; } }); // FIXME: [rfeng] Wrap the property into a function for remoting // so that it can be accessed as /api/// // For example, /api/orders/1/customer var fn = function() { var f = this[relationName]; f.apply(this, arguments); }; modelFrom.prototype['__get__' + relationName] = fn; }; BelongsTo.prototype.create = function(targetModelData, cb) { var self = this; var modelTo = this.definition.modelTo; var fk = this.definition.keyFrom; var pk = this.definition.keyTo; var modelInstance = this.modelInstance; if (typeof targetModelData === 'function' && !cb) { cb = targetModelData; targetModelData = {}; } this.definition.applyProperties(modelInstance, targetModelData || {}); modelTo.create(targetModelData, function(err, targetModel) { if(!err) { modelInstance[fk] = targetModel[pk]; self.resetCache(targetModel); cb && cb(err, targetModel); } else { cb && cb(err); } }); }; BelongsTo.prototype.build = function(targetModelData) { var modelTo = this.definition.modelTo; this.definition.applyProperties(this.modelInstance, targetModelData || {}); return new modelTo(targetModelData); }; /** * Define the method for the belongsTo relation itself * It will support one of the following styles: * - order.customer(refresh, callback): Load the target model instance asynchronously * - order.customer(customer): Synchronous setter of the target model instance * - order.customer(): Synchronous getter of the target model instance * * @param refresh * @param params * @returns {*} */ BelongsTo.prototype.related = function (refresh, params) { var self = this; var modelFrom = this.definition.modelFrom; var modelTo = this.definition.modelTo; var typeTo = this.definition.typeTo; var pk = this.definition.keyTo; var fk = this.definition.keyFrom; var modelInstance = this.modelInstance; if (arguments.length === 1) { params = refresh; refresh = false; } else if (arguments.length > 2) { throw new Error('Method can\'t be called with more than two arguments'); } var cachedValue; if (!refresh) { cachedValue = self.getCache(); } if (params instanceof ModelBaseClass) { // acts as setter modelTo = params.constructor; modelInstance[fk] = params[pk]; if (typeTo) modelInstance[typeTo] = params.constructor.modelName; self.resetCache(params); } else if (typeof params === 'function') { // acts as async getter if (typeTo && !modelTo) { var modelToName = modelInstance[typeTo]; if (typeof modelToName !== 'string') { throw new Error('Polymorphic model not found: `' + typeTo + '` not set'); } modelToName = modelToName.toLowerCase(); modelTo = lookupModel(modelFrom.dataSource.modelBuilder.models, modelToName); if (!modelTo) { throw new Error('Polymorphic model not found: `' + modelToName + '`'); } } var cb = params; if (cachedValue === undefined) { var query = {where: {}}; query.where[pk] = modelInstance[fk]; this.definition.applyScope(modelInstance, query); modelTo.findOne(query, function (err, inst) { if (err) { return cb(err); } if (!inst) { return cb(null, null); } // Check if the foreign key matches the primary key if (inst[pk] && modelInstance[fk] && inst[pk].toString() === modelInstance[fk].toString()) { self.resetCache(inst); cb(null, inst); } else { err = new Error('Key mismatch: ' + self.definition.modelFrom.modelName + '.' + fk + ': ' + modelInstance[fk] + ', ' + modelTo.modelName + '.' + pk + ': ' + inst[pk]); err.statusCode = 400; cb(err); } }); return modelInstance[fk]; } else { cb(null, cachedValue); return cachedValue; } } else if (params === undefined) { // acts as sync getter return cachedValue; } else { // setter modelInstance[fk] = params; self.resetCache(); } }; /** * A hasAndBelongsToMany relation creates a direct many-to-many connection with * another model, with no intervening model. For example, if your application * includes users and groups, with each group having many users and each user * appearing in many groups, you could declare the models this way: * ``` * User.hasAndBelongsToMany('groups', {model: Group, foreignKey: 'groupId'}); * ``` * * @param {String|Object} modelTo Model object (or String name of model) to * which you are creating the relationship. * @options {Object} params Configuration parameters; see below. * @property {String} as Name of the property in the referring model that * corresponds to the foreign key field in the related model. * @property {String} foreignKey Property name of foreign key field. * @property {Object} model Model object */ RelationDefinition.hasAndBelongsToMany = function hasAndBelongsToMany(modelFrom, modelTo, params) { params = params || {}; var models = modelFrom.dataSource.modelBuilder.models; if ('string' === typeof modelTo) { params.as = modelTo; if (params.model) { modelTo = params.model; } else { modelTo = lookupModel(models, i8n.singularize(modelTo)) || modelTo; } if (typeof modelTo === 'string') { throw new Error('Could not find "' + modelTo + '" relation for ' + modelFrom.modelName); } } var isPolymorphic = (typeof params.polymorphic === 'string'); if (!params.through) { if (isPolymorphic) throw new Error('Polymorphic relations need a through model'); var name1 = modelFrom.modelName + modelTo.modelName; var name2 = modelTo.modelName + modelFrom.modelName; params.through = lookupModel(models, name1) || lookupModel(models, name2) || modelFrom.dataSource.define(name1); } var options = {as: params.as, through: params.through}; if (isPolymorphic) { options.polymorphic = params.polymorphic; var accessor = params.through.prototype[params.polymorphic]; if (typeof accessor !== 'function') { // declare once params.through.belongsTo(params.polymorphic, { polymorphic: true }); } } else { params.through.belongsTo(modelFrom); } params.through.belongsTo(modelTo); this.hasMany(modelFrom, modelTo, options); }; /** * A HasOne relation creates a one-to-one connection from modelFrom to modelTo. * This relation indicates that each instance of a model contains or possesses * one instance of another model. For example, each supplier in your application * has only one account. * * @param {Function} modelFrom The declaring model class * @param {String|Function} modelTo Model object (or String name of model) to * which you are creating the relationship. * @options {Object} params Configuration parameters; see below. * @property {String} as Name of the property in the referring model that * corresponds to the foreign key field in the related model. * @property {String} foreignKey Property name of foreign key field. * @property {Object} model Model object */ RelationDefinition.hasOne = function (modelFrom, modelTo, params) { params = params || {}; if ('string' === typeof modelTo) { params.as = modelTo; if (params.model) { modelTo = params.model; } else { var modelToName = modelTo.toLowerCase(); modelTo = lookupModel(modelFrom.dataSource.modelBuilder.models, modelToName); } } var pk = modelFrom.dataSource.idName(modelTo.modelName) || 'id'; var relationName = params.as || i8n.camelize(modelTo.modelName, true); var fk = params.foreignKey || i8n.camelize(modelFrom.modelName + '_id', true); var typeTo; if (typeof params.polymorphic === 'string') { var polymorphic = params.polymorphic; fk = i8n.camelize(polymorphic + '_id', true); typeTo = i8n.camelize(polymorphic + '_type', true); if (!params.through) { modelTo.dataSource.defineProperty(modelTo.modelName, typeTo, { type: 'string', index: true }); } } var relationDef = modelFrom.relations[relationName] = new RelationDefinition({ name: relationName, type: RelationTypes.hasOne, modelFrom: modelFrom, keyFrom: pk, keyTo: fk, typeTo: typeTo, modelTo: modelTo, properties: params.properties, options: params.options }); modelFrom.dataSource.defineForeignKey(modelTo.modelName, fk, modelFrom.modelName); // Define a property for the scope so that we have 'this' for the scoped methods Object.defineProperty(modelFrom.prototype, relationName, { enumerable: true, configurable: true, get: function() { var relation = new HasOne(relationDef, this); var relationMethod = relation.related.bind(relation) relationMethod.create = relation.create.bind(relation); relationMethod.build = relation.build.bind(relation); relationMethod._targetClass = relationDef.modelTo.modelName; return relationMethod; } }); }; /** * Create a target model instance * @param {Object} targetModelData The target model data * @callback {Function} [cb] Callback function * @param {String|Object} err Error string or object * @param {Object} The newly created target model instance */ HasOne.prototype.create = function (targetModelData, cb) { var self = this; var modelTo = this.definition.modelTo; var fk = this.definition.keyTo; var pk = this.definition.keyFrom; var modelInstance = this.modelInstance; if (typeof targetModelData === 'function' && !cb) { cb = targetModelData; targetModelData = {}; } targetModelData = targetModelData || {}; targetModelData[fk] = modelInstance[pk]; var query = {where: {}}; query.where[fk] = targetModelData[fk]; this.definition.applyScope(modelInstance, query); this.definition.applyProperties(modelInstance, targetModelData); modelTo.findOne(query, function(err, result) { if(err) { cb(err); } else if(result) { cb(new Error('HasOne relation cannot create more than one instance of ' + modelTo.modelName)); } else { modelTo.create(targetModelData, function (err, targetModel) { if (!err) { // Refresh the cache self.resetCache(targetModel); cb && cb(err, targetModel); } else { cb && cb(err); } }); } }); }; /** * Create a target model instance * @param {Object} targetModelData The target model data * @callback {Function} [cb] Callback function * @param {String|Object} err Error string or object * @param {Object} The newly created target model instance */ HasMany.prototype.create = function (targetModelData, cb) { var self = this; var modelTo = this.definition.modelTo; var fk = this.definition.keyTo; var pk = this.definition.keyFrom; var modelInstance = this.modelInstance; if (typeof targetModelData === 'function' && !cb) { cb = targetModelData; targetModelData = {}; } targetModelData = targetModelData || {}; targetModelData[fk] = modelInstance[pk]; this.definition.applyProperties(modelInstance, targetModelData); modelTo.create(targetModelData, function(err, targetModel) { if(!err) { // Refresh the cache self.addToCache(targetModel); cb && cb(err, targetModel); } else { cb && cb(err); } }); }; /** * Build a target model instance * @param {Object} targetModelData The target model data * @returns {Object} The newly built target model instance */ HasMany.prototype.build = HasOne.prototype.build = function(targetModelData) { var modelTo = this.definition.modelTo; var pk = this.definition.keyFrom; var fk = this.definition.keyTo; targetModelData = targetModelData || {}; targetModelData[fk] = this.modelInstance[pk]; this.definition.applyProperties(this.modelInstance, targetModelData); return new modelTo(targetModelData); }; /** * Define the method for the hasOne relation itself * It will support one of the following styles: * - order.customer(refresh, callback): Load the target model instance asynchronously * - order.customer(customer): Synchronous setter of the target model instance * - order.customer(): Synchronous getter of the target model instance * * @param {Boolean} refresh Reload from the data source * @param {Object|Function} params Query parameters * @returns {Object} */ HasOne.prototype.related = function (refresh, params) { var self = this; var modelTo = this.definition.modelTo; var fk = this.definition.keyTo; var pk = this.definition.keyFrom; var definition = this.definition; var modelInstance = this.modelInstance; if (arguments.length === 1) { params = refresh; refresh = false; } else if (arguments.length > 2) { throw new Error('Method can\'t be called with more than two arguments'); } var cachedValue; if (!refresh) { cachedValue = self.getCache(); } if (params instanceof ModelBaseClass) { // acts as setter params[fk] = modelInstance[pk]; self.resetCache(params); } else if (typeof params === 'function') { // acts as async getter var cb = params; if (cachedValue === undefined) { var query = {where: {}}; query.where[fk] = modelInstance[pk]; definition.applyScope(modelInstance, query); modelTo.findOne(query, function (err, inst) { if (err) { return cb(err); } if (!inst) { return cb(null, null); } // Check if the foreign key matches the primary key if (inst[fk] && modelInstance[pk] && inst[fk].toString() === modelInstance[pk].toString()) { self.resetCache(inst); cb(null, inst); } else { err = new Error('Key mismatch: ' + self.definition.modelFrom.modelName + '.' + pk + ': ' + modelInstance[pk] + ', ' + modelTo.modelName + '.' + fk + ': ' + inst[fk]); err.statusCode = 400; cb(err); } }); return modelInstance[pk]; } else { cb(null, cachedValue); return cachedValue; } } else if (params === undefined) { // acts as sync getter return cachedValue; } else { // setter params[fk] = modelInstance[pk]; self.resetCache(); } };