fixes bug with not using formatted vals in buildQuery

This commit is contained in:
James
2020-11-10 09:17:33 -05:00
parent 10a657e9d7
commit d5ecbc96a3

View File

@@ -3,24 +3,19 @@
const mongoose = require('mongoose');
const validOperators = ['like', 'in', 'all', 'not_in', 'greater_than_equal', 'greater_than', 'less_than_equal', 'less_than', 'not_equals', 'equals', 'exists'];
function addSearchParam(key, value, searchParams) {
return {
...searchParams,
[key]: value,
};
}
function convertArrayFromCommaDelineated(input) {
if (Array.isArray(input)) return input;
if (input.indexOf(',') > -1) {
return input.split(',');
}
return [input];
}
class ParamParser {
constructor(model, rawParams, locale) {
this.parse = this.parse.bind(this);
@@ -47,16 +42,13 @@ class ParamParser {
this.query.sort = this.rawParams[key];
}
}
return this.query;
}
return {};
}
async parsePathOrRelation(object) {
let result = {};
// We need to determine if the whereKey is an AND, OR, or a schema path
for (const relationOrPath of Object.keys(object)) {
if (relationOrPath.toLowerCase() === 'and') {
@@ -72,12 +64,10 @@ class ParamParser {
// For example - title like 'test' and title not equal to 'tester'
// So we need to loop on keys again here to handle each operator independently
const pathOperators = object[relationOrPath];
if (typeof pathOperators === 'object') {
for (const operator of Object.keys(pathOperators)) {
if (validOperators.includes(operator)) {
const searchParam = await this.buildSearchParam(this.model.schema, relationOrPath, pathOperators[operator], operator);
if (Array.isArray(searchParam)) {
const [key, value] = searchParam;
result = addSearchParam(key, value, result);
@@ -87,7 +77,6 @@ class ParamParser {
}
}
}
return result;
}
@@ -102,46 +91,36 @@ class ParamParser {
completedConditions.push(result);
}
}
return completedConditions;
}
// Checks to see
async buildSearchParam(schema, key, val, operator) {
let schemaObject = schema.obj[key];
const sanitizedKey = key.replace(/__/gi, '.');
let localizedKey = this.getLocalizedKey(sanitizedKey, schemaObject);
if (key === '_id' || key === 'id') {
localizedKey = '_id';
if (!mongoose.Types.ObjectId.isValid(val)) {
return null;
}
}
if (key.includes('.') || key.includes('__')) {
const paths = key.split('.');
schemaObject = schema.obj[paths[0]];
const localizedPath = this.getLocalizedKey(paths[0], schemaObject);
const path = schema.paths[localizedPath];
// If the schema object has a dot, split on the dot
// Check the path of the first index of the newly split array
// If it's an array OR an ObjectID, we need to recurse
if (path) {
// If the path is an ObjectId with a direct ref,
// Grab it
let { ref } = path.options;
// If the path is an Array, grab the ref of the first index type
if (path.instance === 'Array') {
ref = path.options && path.options.type && path.options.type[0].ref;
}
// //////////////////////////////////////////////////////////////////////////
// TODO:
//
@@ -151,101 +130,75 @@ class ParamParser {
// In that case, we would need to get the allowed referenced models
// and run the subModel query on each - building up a list of $in IDs.
// //////////////////////////////////////////////////////////////////////////
if (ref) {
const subModel = mongoose.model(ref);
let subQuery = {};
const localizedSubKey = this.getLocalizedKey(paths[1], subModel.schema.obj[paths[1]]);
const [searchParamKey, searchParamValue] = await this.buildSearchParam(subModel.schema, localizedSubKey, val, operator);
subQuery = addSearchParam(searchParamKey, searchParamValue, subQuery, subModel.schema);
const matchingSubDocuments = await subModel.find(subQuery);
return [localizedPath, {
$in: matchingSubDocuments.map((subDoc) => subDoc.id),
}];
}
}
}
let formattedValue = val;
if (schemaObject && schemaObject.type === Boolean && typeof val === 'string') {
if (val.toLowerCase() === 'true') formattedValue = true;
if (val.toLowerCase() === 'false') formattedValue = false;
}
if (schemaObject && schemaObject.ref && val === 'null') {
formattedValue = null;
}
if (operator && validOperators.includes(operator)) {
switch (operator) {
case 'greater_than_equal':
formattedValue = { $gte: val };
formattedValue = { $gte: formattedValue };
break;
case 'less_than_equal':
formattedValue = { $lte: val };
formattedValue = { $lte: formattedValue };
break;
case 'less_than':
formattedValue = { $lt: val };
formattedValue = { $lt: formattedValue };
break;
case 'greater_than':
formattedValue = { $gt: val };
formattedValue = { $gt: formattedValue };
break;
case 'in':
case 'all':
formattedValue = { [`$${operator}`]: convertArrayFromCommaDelineated(val) };
formattedValue = { [`$${operator}`]: convertArrayFromCommaDelineated(formattedValue) };
break;
case 'not_in':
formattedValue = { $nin: convertArrayFromCommaDelineated(val) };
formattedValue = { $nin: convertArrayFromCommaDelineated(formattedValue) };
break;
case 'not_equals':
formattedValue = { $ne: val };
formattedValue = { $ne: formattedValue };
break;
case 'like':
if (localizedKey === '_id') {
formattedValue = val;
} else {
formattedValue = { $regex: val, $options: '-i' };
if (localizedKey !== '_id') {
formattedValue = { $regex: formattedValue, $options: '-i' };
}
break;
case 'exists':
formattedValue = { $exists: (val === 'true' || val === true) };
formattedValue = { $exists: (formattedValue === 'true' || formattedValue === true) };
break;
default:
formattedValue = val;
break;
}
}
return [localizedKey, formattedValue];
}
}
// This plugin asynchronously builds a list of Mongoose query constraints
// which can then be used in subsequent Mongoose queries.
function buildQueryPlugin(schema) {
const modifiedSchema = schema;
async function buildQuery(rawParams, locale) {
const paramParser = new ParamParser(this, rawParams, locale);
const params = await paramParser.parse();
return params.searchParams;
}
modifiedSchema.statics.buildQuery = buildQuery;
}
module.exports = buildQueryPlugin;