mirror of
https://github.com/strapi/strapi.git
synced 2025-07-21 07:57:45 +00:00
123 lines
2.6 KiB
JavaScript
123 lines
2.6 KiB
JavaScript
![]() |
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);
|
||
|
},
|
||
|
});
|