113 lines
2.6 KiB
JavaScript
Raw Normal View History

2017-11-15 16:59:12 +01:00
const _ = require('lodash');
module.exports = {
find: async function (params) {
return this.query(function(qb) {
2017-11-28 15:32:25 +01:00
_.forEach(params.where, (where, key) => {
qb.where(key, where[0].symbol, where[0].value);
2017-11-15 16:59:12 +01:00
});
2017-11-28 15:32:25 +01:00
if (params.sort) {
qb.orderBy(params.sort);
}
qb.offset(params.start);
qb.limit(params.limit);
}).fetchAll({
withRelated: _.keys(_.groupBy(_.reject(this.associations, {autoPopulate: false}), 'alias'))
});
2017-11-15 16:59:12 +01:00
},
count: async function (params) {
return await this
.forge()
.count();
},
findOne: async function (params) {
2017-12-07 16:53:30 +01:00
if (_.get(params, '_id')) {
params.id = params._id;
delete params._id;
2017-11-28 15:32:25 +01:00
}
2017-11-15 16:59:12 +01:00
const record = await this
2017-12-07 16:53:30 +01:00
.forge(params)
2017-11-15 16:59:12 +01:00
.fetch({
withRelated: this.associations.map(x => x.alias)
});
return record ? record.toJSON() : record;
},
create: async function (params) {
2017-11-29 17:09:19 +01:00
return this
2017-11-15 16:59:12 +01:00
.forge()
2017-11-29 17:09:19 +01:00
.save(Object.keys(params).reduce((acc, current) => {
2017-11-20 16:28:50 +01:00
if (_.get(this, ['_attributes', current, 'type'])) {
2017-11-29 17:09:19 +01:00
acc[current] = params[current];
2017-11-15 16:59:12 +01:00
}
return acc;
}, {}))
.catch((err) => {
2017-12-07 14:56:25 +01:00
if (err.detail) {
const field = _.last(_.words(err.detail.split('=')[0]));
err = { message: `This ${field} is already taken`, field };
}
2017-12-06 15:11:55 +01:00
2017-12-07 14:56:25 +01:00
throw err;
});
2017-11-15 16:59:12 +01:00
},
update: async function (params) {
2017-11-29 18:45:51 +01:00
if (_.get(params, '_id')) {
params.id = params._id;
delete params._id;
2017-11-15 16:59:12 +01:00
}
2017-11-29 18:45:51 +01:00
return this.forge({
[this.primaryKey]: params[this.primaryKey]
})
.save(params, {
patch: true
2017-12-06 15:11:55 +01:00
})
.catch((err) => {
const field = _.last(_.words(err.detail.split('=')[0]));
const error = { message: `This ${field} is already taken`, field };
throw error;
2017-11-29 18:45:51 +01:00
});
2017-11-15 16:59:12 +01:00
},
delete: async function (params) {
return await this
.forge({
2017-11-29 18:45:51 +01:00
[this.primaryKey]: params._id
2017-11-15 16:59:12 +01:00
})
.destroy();
},
2017-11-29 16:24:11 +01:00
search: async function (params) {
return this
.query(function(qb) {
qb
.where('username', 'LIKE', `%${params.id}%`)
.orWhere('email', 'LIKE', `%${params.id}%`);
})
.fetchAll();
2017-11-30 12:27:04 +01:00
},
countByRoles: async function () {
2017-12-07 18:16:15 +01:00
const result = await strapi.connections[this.connection].raw(`SELECT COUNT("id") AS total, "role" FROM "${strapi.plugins['users-permissions'].models.user.collectionName}" GROUP BY "role";`);
2017-11-30 12:27:04 +01:00
return result.rows.reduce((acc, current) => {
acc.push({
_id: parseFloat(current.role),
total: parseFloat(current.total)
});
return acc;
}, []);
2017-11-15 16:59:12 +01:00
}
};