mirror of
https://github.com/strapi/strapi.git
synced 2025-09-28 18:01:26 +00:00
181 lines
4.3 KiB
JavaScript
181 lines
4.3 KiB
JavaScript
![]() |
const _ = require('lodash');
|
||
|
|
||
|
module.exports = {
|
||
|
find: async function(params = {}, populate) {
|
||
|
const records = await this.query(function(qb) {
|
||
|
_.forEach(params.where, (where, key) => {
|
||
|
if (
|
||
|
_.isArray(where.value) &&
|
||
|
where.symbol !== 'IN' &&
|
||
|
where.symbol !== 'NOT IN'
|
||
|
) {
|
||
|
for (const value in where.value) {
|
||
|
qb[value ? 'where' : 'orWhere'](
|
||
|
key,
|
||
|
where.symbol,
|
||
|
where.value[value],
|
||
|
);
|
||
|
}
|
||
|
} else {
|
||
|
qb.where(key, where.symbol, where.value);
|
||
|
}
|
||
|
});
|
||
|
if (params.start) {
|
||
|
qb.offset(params.start);
|
||
|
}
|
||
|
if (params.limit) {
|
||
|
qb.limit(params.limit);
|
||
|
}
|
||
|
if (params.sort) {
|
||
|
if (params.sort.key) {
|
||
|
qb.orderBy(params.sort.key, params.sort.order);
|
||
|
} else {
|
||
|
qb.orderBy(params.sort);
|
||
|
}
|
||
|
}
|
||
|
}).fetchAll({
|
||
|
withRelated:
|
||
|
populate ||
|
||
|
_.keys(
|
||
|
_.groupBy(
|
||
|
_.reject(this.associations, { autoPopulate: false }),
|
||
|
'alias',
|
||
|
),
|
||
|
),
|
||
|
});
|
||
|
|
||
|
return records ? records.toJSON() : records;
|
||
|
},
|
||
|
|
||
|
count: async function(params = {}) {
|
||
|
return await this.forge() // Instanciate the model
|
||
|
.query(qb => {
|
||
|
_.forEach(params.where, (where, key) => {
|
||
|
if (_.isArray(where.value)) {
|
||
|
for (const value in where.value) {
|
||
|
qb[value ? 'where' : 'orWhere'](
|
||
|
key,
|
||
|
where.symbol,
|
||
|
where.value[value],
|
||
|
);
|
||
|
}
|
||
|
} else {
|
||
|
qb.where(key, where.symbol, where.value);
|
||
|
}
|
||
|
});
|
||
|
})
|
||
|
.count();
|
||
|
},
|
||
|
|
||
|
findOne: async function(params, populate) {
|
||
|
const primaryKey = params[this.primaryKey] || params._id;
|
||
|
|
||
|
if (primaryKey) {
|
||
|
params = {
|
||
|
[this.primaryKey]: primaryKey,
|
||
|
};
|
||
|
}
|
||
|
|
||
|
const record = await this.forge(params).fetch({
|
||
|
withRelated: populate || this.associations.map(x => x.alias),
|
||
|
});
|
||
|
|
||
|
return record ? record.toJSON() : record;
|
||
|
},
|
||
|
|
||
|
create: async function(params) {
|
||
|
return this.forge()
|
||
|
.save(
|
||
|
Object.keys(params).reduce((acc, current) => {
|
||
|
if (
|
||
|
_.get(this._attributes, [current, 'type']) ||
|
||
|
_.get(this._attributes, [current, 'model'])
|
||
|
) {
|
||
|
acc[current] = params[current];
|
||
|
}
|
||
|
|
||
|
return acc;
|
||
|
}, {}),
|
||
|
)
|
||
|
.catch(err => {
|
||
|
if (err.detail) {
|
||
|
const field = _.last(_.words(err.detail.split('=')[0]));
|
||
|
err = { message: `This ${field} is already taken`, field };
|
||
|
}
|
||
|
|
||
|
throw err;
|
||
|
});
|
||
|
},
|
||
|
|
||
|
update: async function(search, params = {}) {
|
||
|
if (_.isEmpty(params)) {
|
||
|
params = search;
|
||
|
}
|
||
|
|
||
|
const primaryKey = search[this.primaryKey] || search.id;
|
||
|
|
||
|
if (primaryKey) {
|
||
|
search = {
|
||
|
[this.primaryKey]: primaryKey,
|
||
|
};
|
||
|
} else {
|
||
|
const entry = await module.exports.findOne.call(this, search);
|
||
|
|
||
|
search = {
|
||
|
[this.primaryKey]: entry[this.primaryKey] || entry.id,
|
||
|
};
|
||
|
}
|
||
|
|
||
|
return this.forge(search)
|
||
|
.save(params, {
|
||
|
patch: true,
|
||
|
})
|
||
|
.catch(err => {
|
||
|
const field = _.last(_.words(err.detail.split('=')[0]));
|
||
|
const error = { message: `This ${field} is already taken`, field };
|
||
|
|
||
|
throw error;
|
||
|
});
|
||
|
},
|
||
|
|
||
|
delete: async function(params) {
|
||
|
return await this.forge({
|
||
|
[this.primaryKey]: params[this.primaryKey] || params.id,
|
||
|
}).destroy();
|
||
|
},
|
||
|
|
||
|
deleteMany: async function(params) {
|
||
|
return await this.query(qb => {
|
||
|
qb.whereIn(this.primaryKey, params[this.primaryKey] || params.id);
|
||
|
}).destroy();
|
||
|
},
|
||
|
|
||
|
search: async function(params) {
|
||
|
return this.query(function(qb) {
|
||
|
qb.where('username', 'LIKE', `%${params.id}%`).orWhere(
|
||
|
'email',
|
||
|
'LIKE',
|
||
|
`%${params.id}%`,
|
||
|
);
|
||
|
}).fetchAll();
|
||
|
},
|
||
|
|
||
|
addPermission: async function(params) {
|
||
|
return this.forge(params).save();
|
||
|
},
|
||
|
|
||
|
removePermission: async function(params) {
|
||
|
/* eslint-disable indent */
|
||
|
const value = params[this.primaryKey]
|
||
|
? {
|
||
|
[this.primaryKey]: params[this.primaryKey] || params.id,
|
||
|
}
|
||
|
: params;
|
||
|
|
||
|
/* eslint-enable indent */
|
||
|
return this.forge()
|
||
|
.where(value)
|
||
|
.destroy();
|
||
|
},
|
||
|
};
|