332 lines
8.1 KiB
JavaScript
332 lines
8.1 KiB
JavaScript
exports.safeRequire = safeRequire;
|
|
exports.fieldsToArray = fieldsToArray;
|
|
exports.selectFields = selectFields;
|
|
exports.removeUndefined = removeUndefined;
|
|
exports.parseSettings = parseSettings;
|
|
exports.mergeSettings = mergeSettings;
|
|
exports.isPlainObject = isPlainObject;
|
|
exports.defineCachedRelations = defineCachedRelations;
|
|
exports.sortObjectsByIds = sortObjectsByIds;
|
|
exports.setScopeValuesFromWhere = setScopeValuesFromWhere;
|
|
exports.mergeQuery = mergeQuery;
|
|
|
|
var traverse = require('traverse');
|
|
|
|
function safeRequire(module) {
|
|
try {
|
|
return require(module);
|
|
} catch (e) {
|
|
console.log('Run "npm install loopback-datasource-juggler ' + module
|
|
+ '" command to use loopback-datasource-juggler using ' + module
|
|
+ ' database engine');
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Extracting fixed property values for the scope from the where clause into
|
|
* the data object
|
|
*
|
|
* @param {Object} The data object
|
|
* @param {Object} The where clause
|
|
*/
|
|
function setScopeValuesFromWhere(data, where, targetModel) {
|
|
for (var i in where) {
|
|
if (i === 'and') {
|
|
// Find fixed property values from each subclauses
|
|
for (var w = 0, n = where[i].length; w < n; w++) {
|
|
setScopeValuesFromWhere(data, where[i][w], targetModel);
|
|
}
|
|
continue;
|
|
}
|
|
var prop = targetModel.definition.properties[i];
|
|
if (prop) {
|
|
var val = where[i];
|
|
if (typeof val !== 'object' || val instanceof prop.type
|
|
|| prop.type.name === 'ObjectID') // MongoDB key
|
|
{
|
|
// Only pick the {propertyName: propertyValue}
|
|
data[i] = where[i];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/*!
|
|
* Merge query parameters
|
|
* @param {Object} base The base object to contain the merged results
|
|
* @param {Object} update The object containing updates to be merged
|
|
* @returns {*|Object} The base object
|
|
* @private
|
|
*/
|
|
function mergeQuery(base, update) {
|
|
if (!update) {
|
|
return;
|
|
}
|
|
base = base || {};
|
|
if (update.where && Object.keys(update.where).length > 0) {
|
|
if (base.where && Object.keys(base.where).length > 0) {
|
|
base.where = {and: [base.where, update.where]};
|
|
} else {
|
|
base.where = update.where;
|
|
}
|
|
}
|
|
|
|
// Merge inclusion
|
|
if (update.include) {
|
|
if (!base.include) {
|
|
base.include = update.include;
|
|
} else {
|
|
var saved = base.include;
|
|
base.include = {};
|
|
base.include[update.include] = saved;
|
|
}
|
|
}
|
|
if (update.collect) {
|
|
base.collect = update.collect;
|
|
}
|
|
|
|
// set order
|
|
if (!base.order && update.order) {
|
|
base.order = update.order;
|
|
}
|
|
|
|
// overwrite pagination
|
|
if (update.limit !== undefined) {
|
|
base.limit = update.limit;
|
|
}
|
|
if (update.skip !== undefined) {
|
|
base.skip = update.skip;
|
|
}
|
|
if (update.offset !== undefined) {
|
|
base.offset = update.offset;
|
|
}
|
|
|
|
// Overwrite fields
|
|
if (update.fields !== undefined) {
|
|
base.fields = update.fields;
|
|
}
|
|
return base;
|
|
}
|
|
|
|
function fieldsToArray(fields, properties) {
|
|
if (!fields) return;
|
|
|
|
// include all properties by default
|
|
var result = properties;
|
|
|
|
if (typeof fields === 'string') {
|
|
return [fields];
|
|
}
|
|
|
|
if (Array.isArray(fields) && fields.length > 0) {
|
|
// No empty array, including all the fields
|
|
return fields;
|
|
}
|
|
|
|
if ('object' === typeof fields) {
|
|
// { field1: boolean, field2: boolean ... }
|
|
var included = [];
|
|
var excluded = [];
|
|
var keys = Object.keys(fields);
|
|
if (!keys.length) return;
|
|
|
|
keys.forEach(function (k) {
|
|
if (fields[k]) {
|
|
included.push(k);
|
|
} else if ((k in fields) && !fields[k]) {
|
|
excluded.push(k);
|
|
}
|
|
});
|
|
if (included.length > 0) {
|
|
result = included;
|
|
} else if (excluded.length > 0) {
|
|
excluded.forEach(function (e) {
|
|
var index = result.indexOf(e);
|
|
result.splice(index, 1);
|
|
});
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
function selectFields(fields) {
|
|
// map function
|
|
return function (obj) {
|
|
var result = {};
|
|
var key;
|
|
|
|
for (var i = 0; i < fields.length; i++) {
|
|
key = fields[i];
|
|
|
|
result[key] = obj[key];
|
|
}
|
|
return result;
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Remove undefined values from the queury object
|
|
* @param query
|
|
* @returns {exports.map|*}
|
|
*/
|
|
function removeUndefined(query) {
|
|
if (typeof query !== 'object' || query === null) {
|
|
return query;
|
|
}
|
|
// WARNING: [rfeng] Use map() will cause mongodb to produce invalid BSON
|
|
// as traverse doesn't transform the ObjectId correctly
|
|
return traverse(query).forEach(function (x) {
|
|
if (x === undefined) {
|
|
this.remove();
|
|
}
|
|
|
|
if (!Array.isArray(x) && (typeof x === 'object' && x !== null
|
|
&& x.constructor !== Object)) {
|
|
// This object is not a plain object
|
|
this.update(x, true); // Stop navigating into this object
|
|
return x;
|
|
}
|
|
|
|
return x;
|
|
});
|
|
}
|
|
|
|
var url = require('url');
|
|
var qs = require('qs');
|
|
|
|
/**
|
|
* Parse a URL into a settings object
|
|
* @param {String} urlStr The URL for connector settings
|
|
* @returns {Object} The settings object
|
|
*/
|
|
function parseSettings(urlStr) {
|
|
if (!urlStr) {
|
|
return {};
|
|
}
|
|
var uri = url.parse(urlStr, false);
|
|
var settings = {};
|
|
settings.connector = uri.protocol && uri.protocol.split(':')[0]; // Remove the trailing :
|
|
settings.host = settings.hostname = uri.hostname;
|
|
settings.port = uri.port && Number(uri.port); // port is a string
|
|
settings.user = settings.username = uri.auth && uri.auth.split(':')[0]; // <username>:<password>
|
|
settings.password = uri.auth && uri.auth.split(':')[1];
|
|
settings.database = uri.pathname && uri.pathname.split('/')[1]; // remove the leading /
|
|
settings.url = urlStr;
|
|
if (uri.query) {
|
|
var params = qs.parse(uri.query);
|
|
for (var p in params) {
|
|
settings[p] = params[p];
|
|
}
|
|
}
|
|
return settings;
|
|
}
|
|
|
|
/**
|
|
* Merge model settings
|
|
*
|
|
* Folked from https://github.com/nrf110/deepmerge/blob/master/index.js
|
|
*
|
|
* The original function tries to merge array items if they are objects
|
|
*
|
|
* @param {Object} target The target settings object
|
|
* @param {Object} src The source settings object
|
|
* @returns {Object} The merged settings object
|
|
*/
|
|
function mergeSettings(target, src) {
|
|
var array = Array.isArray(src);
|
|
var dst = array && [] || {};
|
|
|
|
if (array) {
|
|
target = target || [];
|
|
dst = dst.concat(target);
|
|
src.forEach(function (e) {
|
|
if (dst.indexOf(e) === -1) {
|
|
dst.push(e);
|
|
}
|
|
});
|
|
} else {
|
|
if (target && typeof target === 'object') {
|
|
Object.keys(target).forEach(function (key) {
|
|
dst[key] = target[key];
|
|
});
|
|
}
|
|
Object.keys(src).forEach(function (key) {
|
|
if (typeof src[key] !== 'object' || !src[key]) {
|
|
dst[key] = src[key];
|
|
}
|
|
else {
|
|
if (!target[key]) {
|
|
dst[key] = src[key]
|
|
} else {
|
|
dst[key] = mergeSettings(target[key], src[key]);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
return dst;
|
|
}
|
|
|
|
/**
|
|
* Define an non-enumerable __cachedRelations property
|
|
* @param {Object} obj The obj to receive the __cachedRelations
|
|
*/
|
|
function defineCachedRelations(obj) {
|
|
if (!obj.__cachedRelations) {
|
|
Object.defineProperty(obj, '__cachedRelations', {
|
|
writable: true,
|
|
enumerable: false,
|
|
configurable: true,
|
|
value: {}
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Check if the argument is plain object
|
|
* @param {*) obj The obj value
|
|
* @returns {boolean}
|
|
*/
|
|
function isPlainObject(obj) {
|
|
return (typeof obj === 'object') && (obj !== null)
|
|
&& (obj.constructor === Object);
|
|
}
|
|
|
|
|
|
|
|
function sortObjectsByIds(idName, ids, objects, strict) {
|
|
ids = ids.map(function(id) {
|
|
return (typeof id === 'object') ? String(id) : id;
|
|
});
|
|
|
|
var indexOf = function(x) {
|
|
var isObj = (typeof x[idName] === 'object'); // ObjectID
|
|
var id = isObj ? String(x[idName]) : x[idName];
|
|
return ids.indexOf(id);
|
|
};
|
|
|
|
var heading = [];
|
|
var tailing = [];
|
|
|
|
objects.forEach(function(x) {
|
|
if (typeof x === 'object') {
|
|
var idx = indexOf(x);
|
|
if (strict && idx === -1) return;
|
|
idx === -1 ? tailing.push(x) : heading.push(x);
|
|
}
|
|
});
|
|
|
|
heading.sort(function(x, y) {
|
|
var a = indexOf(x);
|
|
var b = indexOf(y);
|
|
if (a === -1 || b === -1) return 1; // last
|
|
if (a === b) return 0;
|
|
if (a > b) return 1;
|
|
if (a < b) return -1;
|
|
});
|
|
|
|
return heading.concat(tailing);
|
|
};
|