171 lines
4.4 KiB
JavaScript
Raw Normal View History

const _ = require('lodash');
2019-03-13 19:27:18 +01:00
const { convertRestQueryParams, buildQuery } = require('strapi-utils');
module.exports = ({ model }) => ({
find(params, populate, raw = false) {
2019-03-13 19:27:18 +01:00
const filters = convertRestQueryParams(params);
2019-03-13 19:27:18 +01:00
const query = buildQuery({
model,
filters,
populate: populate || model.associations.map(x => x.alias),
});
return raw ? query.lean() : query;
},
count(params) {
2019-03-13 19:27:18 +01:00
const filters = convertRestQueryParams(params);
return buildQuery({
model,
filters: { where: filters.where },
}).count();
},
search(params, populate) {
2019-03-13 19:27:18 +01:00
// eslint-disable-line no-unused-vars
const $or = Object.keys(model.attributes).reduce((acc, curr) => {
switch (model.attributes[curr].type) {
2018-06-07 18:50:43 +02:00
case 'integer':
2019-01-21 05:11:48 -07:00
case 'biginteger':
2018-06-07 18:50:43 +02:00
case 'float':
case 'decimal':
if (!_.isNaN(_.toNumber(params.search))) {
return acc.concat({ [curr]: params.search });
}
return acc;
case 'string':
case 'text':
case 'password':
return acc.concat({
[curr]: { $regex: params.search, $options: 'i' },
});
2018-06-07 18:50:43 +02:00
case 'boolean':
if (params.search === 'true' || params.search === 'false') {
return acc.concat({ [curr]: params.search === 'true' });
}
return acc;
default:
return acc;
}
}, []);
return model
.find({ $or })
2018-06-07 18:50:43 +02:00
.limit(Number(params.limit))
.sort(params.sort)
.skip(Number(params.skip))
.populate(populate || model.associations.map(x => x.alias).join(' '))
.lean();
2018-06-07 14:35:09 +02:00
},
countSearch(params = {}) {
2019-03-13 19:27:18 +01:00
// eslint-disable-line no-unused-vars
const $or = Object.keys(model.attributes).reduce((acc, curr) => {
switch (model.attributes[curr].type) {
2018-06-07 18:50:43 +02:00
case 'integer':
2019-01-21 05:11:48 -07:00
case 'biginteger':
2018-06-07 18:50:43 +02:00
case 'float':
case 'decimal':
if (!_.isNaN(_.toNumber(params.search))) {
return acc.concat({ [curr]: params.search });
}
return acc;
case 'string':
case 'text':
case 'password':
return acc.concat({
[curr]: { $regex: params.search, $options: 'i' },
});
2018-06-07 18:50:43 +02:00
case 'boolean':
if (params.search === 'true' || params.search === 'false') {
return acc.concat({ [curr]: params.search === 'true' });
}
return acc;
default:
return acc;
}
}, []);
return model.find({ $or }).countDocuments();
2018-06-07 14:35:09 +02:00
},
findOne(params, populate, raw = true) {
const query = model
.findOne({
[model.primaryKey]: params[model.primaryKey] || params.id,
})
.populate(populate || model.associations.map(x => x.alias).join(' '));
return raw ? query.lean() : query;
},
async create(params) {
// Exclude relationships.
const values = Object.keys(params.values).reduce((acc, current) => {
if (model._attributes[current] && model._attributes[current].type) {
acc[current] = params.values[current];
}
return acc;
}, {});
2017-12-06 15:58:20 +01:00
const request = await model.create(values).catch(err => {
2019-03-13 19:27:18 +01:00
if (err.message) {
const message = err.message.split('index:');
const field = _.words(_.last(message).split('_')[0]);
err = { message: `This ${field} is already taken`, field };
}
throw err;
});
// Transform to JSON object.
2018-04-25 12:23:53 +02:00
const entry = request.toJSON ? request.toJSON() : request;
// Extract relations.
const relations = model.associations.reduce((acc, association) => {
2018-05-02 15:39:12 +02:00
if (params.values[association.alias]) {
acc[association.alias] = params.values[association.alias];
}
return acc;
}, {});
return this.update({
[model.primaryKey]: entry[model.primaryKey],
2019-03-13 19:27:18 +01:00
values: _.assign(
{
id: entry[model.primaryKey],
2019-03-13 19:27:18 +01:00
},
relations
),
});
},
update(params) {
// Call the business logic located in the hook.
// This function updates no-relational and relational data.
return model.updateRelations(params);
},
delete(params) {
2017-09-20 13:19:36 +02:00
// Delete entry.
return model.findOneAndDelete({
[model.primaryKey]: params.id,
2019-03-13 19:27:18 +01:00
});
},
deleteMany(params) {
return model.deleteMany({
[model.primaryKey]: {
$in: params[model.primaryKey] || params.id,
2019-03-13 19:27:18 +01:00
},
});
},
});