yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-10-31 09:56:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle relationships except many-to-many

Browse Source
This commit is contained in:
Aurélien Georget 2016-08-05 15:51:55 +02:00
parent 4e6ca004b0
commit c79f55b0d9
1 changed files with 43 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
packages/strapi-mongoose/lib/index.js
View File

@ -7,6 +7,7 @@
// Public node modules.
const _ = require('lodash');
const mongoose = require('mongoose');
const mongooseUtils = require('mongoose/lib/utils');
const pluralize = require('pluralize');
// Local helpers.
@ -58,6 +59,7 @@ module.exports = function (strapi) {
_.set(strapi, 'mongoose.collections', {});
const loadedAttributes = _.after(_.size(strapi.models), function () {
console.log(strapi.mongoose.collections);
_.forEach(strapi.models, function (definition, model) {
try {
// Initialize lifecycle callbacks.
@ -82,9 +84,13 @@ module.exports = function (strapi) {
// }
// });
// };
console.log(model);
// Generate schema without virtual populate
const schema = mongoose.Schema(_.omitBy(definition.loadedModel, model => {
console.log("NO VIRTUAL", _.omitBy(definition.loadedModel, model => {
return model.type === 'virtual';
}));
console.log("VIRTUAL", _.pickBy(definition.loadedModel, model => {
return model.type === 'virtual';
}));
@ -92,22 +98,25 @@ module.exports = function (strapi) {
_.forEach(_.pickBy(definition.loadedModel, model => {
return model.type === 'virtual'
}), (value, key) => {
schema.virtual(key, {
strapi.mongoose.collections[mongooseUtils.toCollectionName(definition.globalName)].schema.virtual(key.replace('_v', ''), {
ref: value.ref,
localField: '_id',
foreignField: value.via
foreignField: value.via,
justOne: value.justOne || false
});
});
schema.set('toObject', {
console.log('------------------');
strapi.mongoose.collections[mongooseUtils.toCollectionName(definition.globalName)].schema.set('toObject', {
virtuals: true
});
schema.set('toJSON', {
strapi.mongoose.collections[mongooseUtils.toCollectionName(definition.globalName)].schema.set('toJSON', {
virtuals: true
});
global[definition.globalName] = mongoose.model(definition.globalName, schema);;
global[definition.globalName] = mongoose.model(definition.globalName, strapi.mongoose.collections[mongooseUtils.toCollectionName(definition.globalName)].schema);
// Push model to strapi global variables.
strapi.mongoose.collections[mongooseUtils.toCollectionName(definition.globalName)] = global[definition.globalName];
@ -164,6 +173,12 @@ module.exports = function (strapi) {
// Call this callback function after we are done parsing
// all attributes for relationships-- see below.
const done = _.after(_.size(definition.attributes), function () {
// Generate schema without virtual populate
_.set(strapi.mongoose.collections, mongooseUtils.toCollectionName(definition.globalName) + '.schema', mongoose.Schema(_.omitBy(definition.loadedModel, model => {
return model.type === 'virtual';
})));
console.log('##################');
loadedAttributes();
});
@ -181,6 +196,8 @@ module.exports = function (strapi) {
let FK;
console.log(model, verbose)
switch (verbose) {
case 'hasOne':
definition.loadedModel[name] = {
@ -207,22 +224,33 @@ module.exports = function (strapi) {
break;
case 'belongsTo':
definition.loadedModel[name] = {
type: mongoose.Schema.Types.ObjectId,
ref: _.capitalize(details.model)
};
FK = _.find(definition.associations, { alias : name});
if (FK && FK.nature === 'oneToOne') {
definition.loadedModel[name] = {
type: 'virtual',
ref: _.capitalize(details.model),
via: FK.via,
justOne: true
};
} else {
definition.loadedModel[name] = {
type: mongoose.Schema.Types.ObjectId,
ref: _.capitalize(details.model)
};
}
break;
case 'belongsToMany':
FK = _.find(definition.associations, { alias : name});
console.log(FK);
if (FK) {
definition.loadedModel[name] = {
definition.loadedModel[name + '_v'] = {
type: 'virtual',
ref: _.capitalize(details.collection),
via: FK
via: FK.via
};
} else {
definition.loadedModel[name] = [{
type: mongoose.Schema.Types.ObjectId,
ref: _.capitalize(details.collection)