123 lines
2.6 KiB
JavaScript
Raw Normal View History

2019-07-16 21:24:09 +02:00
const _ = require('lodash');
const { convertRestQueryParams, buildQuery } = require('strapi-utils');
module.exports = ({ model }) => ({
find(params, populate) {
const filters = convertRestQueryParams(params);
return buildQuery({
model,
filters,
populate: populate || model.associations.map(x => x.alias),
}).lean();
},
count(params) {
const filters = convertRestQueryParams(params);
return buildQuery({
model,
filters: { where: filters.where },
}).count();
},
findOne(params, populate) {
const primaryKey = params[model.primaryKey] || params.id;
if (primaryKey) {
params = {
[model.primaryKey]: primaryKey,
};
}
return model
.findOne(params)
.populate(populate || model.associations.map(x => x.alias).join(' '))
.lean();
},
create(params) {
return model
.create(
Object.keys(params).reduce((acc, current) => {
if (
_.get(model._attributes, [current, 'type']) ||
_.get(model._attributes, [current, 'model'])
) {
acc[current] = params[current];
}
return acc;
}, {})
)
.catch(err => {
if (err.message.indexOf('index:') !== -1) {
const message = err.message.split('index:');
const field = _.words(_.last(message).split('_')[0]);
const error = { message: `This ${field} is already taken`, field };
throw error;
}
throw err;
});
},
update(search, params = {}) {
if (_.isEmpty(params)) {
params = search;
}
const primaryKey = search[model.primaryKey] || search.id;
if (primaryKey) {
search = {
[model.primaryKey]: primaryKey,
};
}
return model
.updateOne(search, params, {
strict: false,
})
.catch(error => {
const field = _.last(_.words(error.message.split('_')[0]));
const err = { message: `This ${field} is already taken`, field };
throw err;
});
},
delete(params) {
// Delete entry.
return model.deleteOne({
[model.primaryKey]: params[model.primaryKey] || params.id,
});
},
deleteMany(params) {
// Delete entry.
return model.deleteMany({
[model.primaryKey]: {
$in: params[model.primaryKey] || params.id,
},
});
},
search(params) {
const re = new RegExp(params.id);
return model.find({
$or: [{ username: re }, { email: re }],
});
},
addPermission(params) {
return model.create(params);
},
removePermission(params) {
return model.remove(params);
},
});