123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252 |
- 'use strict';
- var slice = require('sliced');
- var assert = require('assert');
- var util = require('util');
- var utils = require('./utils');
- var debug = require('debug')('mquery');
- function Query(criteria, options) {
- if (!(this instanceof Query))
- return new Query(criteria, options);
- var proto = this.constructor.prototype;
- this.op = proto.op || undefined;
- this.options = Object.assign({}, proto.options);
- this._conditions = proto._conditions
- ? utils.clone(proto._conditions)
- : {};
- this._fields = proto._fields
- ? utils.clone(proto._fields)
- : undefined;
- this._update = proto._update
- ? utils.clone(proto._update)
- : undefined;
- this._path = proto._path || undefined;
- this._distinct = proto._distinct || undefined;
- this._collection = proto._collection || undefined;
- this._traceFunction = proto._traceFunction || undefined;
- if (options) {
- this.setOptions(options);
- }
- if (criteria) {
- if (criteria.find && criteria.remove && criteria.update) {
-
- this.collection(criteria);
- } else {
- this.find(criteria);
- }
- }
- }
- var $withinCmd = '$geoWithin';
- Object.defineProperty(Query, 'use$geoWithin', {
- get: function( ) { return $withinCmd == '$geoWithin'; },
- set: function(v) {
- if (true === v) {
-
- $withinCmd = '$geoWithin';
- } else {
- $withinCmd = '$within';
- }
- }
- });
- Query.prototype.toConstructor = function toConstructor() {
- function CustomQuery(criteria, options) {
- if (!(this instanceof CustomQuery))
- return new CustomQuery(criteria, options);
- Query.call(this, criteria, options);
- }
- utils.inherits(CustomQuery, Query);
-
- var p = CustomQuery.prototype;
- p.options = {};
- p.setOptions(this.options);
- p.op = this.op;
- p._conditions = utils.clone(this._conditions);
- p._fields = utils.clone(this._fields);
- p._update = utils.clone(this._update);
- p._path = this._path;
- p._distinct = this._distinct;
- p._collection = this._collection;
- p._traceFunction = this._traceFunction;
- return CustomQuery;
- };
- Query.prototype.setOptions = function(options) {
- if (!(options && utils.isObject(options)))
- return this;
-
- var methods = utils.keys(options),
- method;
- for (var i = 0; i < methods.length; ++i) {
- method = methods[i];
-
- if ('function' == typeof this[method]) {
- var args = utils.isArray(options[method])
- ? options[method]
- : [options[method]];
- this[method].apply(this, args);
- } else {
- this.options[method] = options[method];
- }
- }
- return this;
- };
- Query.prototype.collection = function collection(coll) {
- this._collection = new Query.Collection(coll);
- return this;
- };
- Query.prototype.collation = function(value) {
- this.options.collation = value;
- return this;
- };
- Query.prototype.$where = function(js) {
- this._conditions.$where = js;
- return this;
- };
- Query.prototype.where = function() {
- if (!arguments.length) return this;
- if (!this.op) this.op = 'find';
- var type = typeof arguments[0];
- if ('string' == type) {
- this._path = arguments[0];
- if (2 === arguments.length) {
- this._conditions[this._path] = arguments[1];
- }
- return this;
- }
- if ('object' == type && !Array.isArray(arguments[0])) {
- return this.merge(arguments[0]);
- }
- throw new TypeError('path must be a string or object');
- };
- Query.prototype.equals = function equals(val) {
- this._ensurePath('equals');
- var path = this._path;
- this._conditions[path] = val;
- return this;
- };
- Query.prototype.eq = function eq(val) {
- this._ensurePath('eq');
- var path = this._path;
- this._conditions[path] = val;
- return this;
- };
- Query.prototype.or = function or(array) {
- var or = this._conditions.$or || (this._conditions.$or = []);
- if (!utils.isArray(array)) array = [array];
- or.push.apply(or, array);
- return this;
- };
- Query.prototype.nor = function nor(array) {
- var nor = this._conditions.$nor || (this._conditions.$nor = []);
- if (!utils.isArray(array)) array = [array];
- nor.push.apply(nor, array);
- return this;
- };
- Query.prototype.and = function and(array) {
- var and = this._conditions.$and || (this._conditions.$and = []);
- if (!Array.isArray(array)) array = [array];
- and.push.apply(and, array);
- return this;
- };
- 'gt gte lt lte ne in nin all regex size maxDistance minDistance'.split(' ').forEach(function($conditional) {
- Query.prototype[$conditional] = function() {
- var path, val;
- if (1 === arguments.length) {
- this._ensurePath($conditional);
- val = arguments[0];
- path = this._path;
- } else {
- val = arguments[1];
- path = arguments[0];
- }
- var conds = this._conditions[path] === null || typeof this._conditions[path] === 'object' ?
- this._conditions[path] :
- (this._conditions[path] = {});
- conds['$' + $conditional] = val;
- return this;
- };
- });
- Query.prototype.mod = function() {
- var val, path;
- if (1 === arguments.length) {
- this._ensurePath('mod');
- val = arguments[0];
- path = this._path;
- } else if (2 === arguments.length && !utils.isArray(arguments[1])) {
- this._ensurePath('mod');
- val = slice(arguments);
- path = this._path;
- } else if (3 === arguments.length) {
- val = slice(arguments, 1);
- path = arguments[0];
- } else {
- val = arguments[1];
- path = arguments[0];
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- conds.$mod = val;
- return this;
- };
- Query.prototype.exists = function() {
- var path, val;
- if (0 === arguments.length) {
- this._ensurePath('exists');
- path = this._path;
- val = true;
- } else if (1 === arguments.length) {
- if ('boolean' === typeof arguments[0]) {
- this._ensurePath('exists');
- path = this._path;
- val = arguments[0];
- } else {
- path = arguments[0];
- val = true;
- }
- } else if (2 === arguments.length) {
- path = arguments[0];
- val = arguments[1];
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- conds.$exists = val;
- return this;
- };
- Query.prototype.elemMatch = function() {
- if (null == arguments[0])
- throw new TypeError('Invalid argument');
- var fn, path, criteria;
- if ('function' === typeof arguments[0]) {
- this._ensurePath('elemMatch');
- path = this._path;
- fn = arguments[0];
- } else if (utils.isObject(arguments[0])) {
- this._ensurePath('elemMatch');
- path = this._path;
- criteria = arguments[0];
- } else if ('function' === typeof arguments[1]) {
- path = arguments[0];
- fn = arguments[1];
- } else if (arguments[1] && utils.isObject(arguments[1])) {
- path = arguments[0];
- criteria = arguments[1];
- } else {
- throw new TypeError('Invalid argument');
- }
- if (fn) {
- criteria = new Query;
- fn(criteria);
- criteria = criteria._conditions;
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- conds.$elemMatch = criteria;
- return this;
- };
- Query.prototype.within = function within() {
-
- this._ensurePath('within');
- this._geoComparison = $withinCmd;
- if (0 === arguments.length) {
- return this;
- }
- if (2 === arguments.length) {
- return this.box.apply(this, arguments);
- } else if (2 < arguments.length) {
- return this.polygon.apply(this, arguments);
- }
- var area = arguments[0];
- if (!area)
- throw new TypeError('Invalid argument');
- if (area.center)
- return this.circle(area);
- if (area.box)
- return this.box.apply(this, area.box);
- if (area.polygon)
- return this.polygon.apply(this, area.polygon);
- if (area.type && area.coordinates)
- return this.geometry(area);
- throw new TypeError('Invalid argument');
- };
- Query.prototype.box = function() {
- var path, box;
- if (3 === arguments.length) {
-
- path = arguments[0];
- box = [arguments[1], arguments[2]];
- } else if (2 === arguments.length) {
-
- this._ensurePath('box');
- path = this._path;
- box = [arguments[0], arguments[1]];
- } else {
- throw new TypeError('Invalid argument');
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- conds[this._geoComparison || $withinCmd] = { '$box': box };
- return this;
- };
- Query.prototype.polygon = function() {
- var val, path;
- if ('string' == typeof arguments[0]) {
-
- path = arguments[0];
- val = slice(arguments, 1);
- } else {
-
- this._ensurePath('polygon');
- path = this._path;
- val = slice(arguments);
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- conds[this._geoComparison || $withinCmd] = { '$polygon': val };
- return this;
- };
- Query.prototype.circle = function() {
- var path, val;
- if (1 === arguments.length) {
- this._ensurePath('circle');
- path = this._path;
- val = arguments[0];
- } else if (2 === arguments.length) {
- path = arguments[0];
- val = arguments[1];
- } else {
- throw new TypeError('Invalid argument');
- }
- if (!('radius' in val && val.center))
- throw new Error('center and radius are required');
- var conds = this._conditions[path] || (this._conditions[path] = {});
- var type = val.spherical
- ? '$centerSphere'
- : '$center';
- var wKey = this._geoComparison || $withinCmd;
- conds[wKey] = {};
- conds[wKey][type] = [val.center, val.radius];
- if ('unique' in val)
- conds[wKey].$uniqueDocs = !!val.unique;
- return this;
- };
- Query.prototype.near = function near() {
- var path, val;
- this._geoComparison = '$near';
- if (0 === arguments.length) {
- return this;
- } else if (1 === arguments.length) {
- this._ensurePath('near');
- path = this._path;
- val = arguments[0];
- } else if (2 === arguments.length) {
- path = arguments[0];
- val = arguments[1];
- } else {
- throw new TypeError('Invalid argument');
- }
- if (!val.center) {
- throw new Error('center is required');
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- var type = val.spherical
- ? '$nearSphere'
- : '$near';
-
- if (Array.isArray(val.center)) {
- conds[type] = val.center;
- var radius = 'maxDistance' in val
- ? val.maxDistance
- : null;
- if (null != radius) {
- conds.$maxDistance = radius;
- }
- if (null != val.minDistance) {
- conds.$minDistance = val.minDistance;
- }
- } else {
-
- if (val.center.type != 'Point' || !Array.isArray(val.center.coordinates)) {
- throw new Error(util.format('Invalid GeoJSON specified for %s', type));
- }
- conds[type] = { $geometry : val.center };
-
- if ('maxDistance' in val) {
- conds[type]['$maxDistance'] = val.maxDistance;
- }
- if ('minDistance' in val) {
- conds[type]['$minDistance'] = val.minDistance;
- }
- }
- return this;
- };
- Query.prototype.intersects = function intersects() {
-
- this._ensurePath('intersects');
- this._geoComparison = '$geoIntersects';
- if (0 === arguments.length) {
- return this;
- }
- var area = arguments[0];
- if (null != area && area.type && area.coordinates)
- return this.geometry(area);
- throw new TypeError('Invalid argument');
- };
- Query.prototype.geometry = function geometry() {
- if (!('$within' == this._geoComparison ||
- '$geoWithin' == this._geoComparison ||
- '$near' == this._geoComparison ||
- '$geoIntersects' == this._geoComparison)) {
- throw new Error('geometry() must come after `within()`, `intersects()`, or `near()');
- }
- var val, path;
- if (1 === arguments.length) {
- this._ensurePath('geometry');
- path = this._path;
- val = arguments[0];
- } else {
- throw new TypeError('Invalid argument');
- }
- if (!(val.type && Array.isArray(val.coordinates))) {
- throw new TypeError('Invalid argument');
- }
- var conds = this._conditions[path] || (this._conditions[path] = {});
- conds[this._geoComparison] = { $geometry: val };
- return this;
- };
- Query.prototype.select = function select() {
- var arg = arguments[0];
- if (!arg) return this;
- if (arguments.length !== 1) {
- throw new Error('Invalid select: select only takes 1 argument');
- }
- this._validate('select');
- var fields = this._fields || (this._fields = {});
- var type = typeof arg;
- var i, len;
- if (('string' == type || utils.isArgumentsObject(arg)) &&
- 'number' == typeof arg.length || Array.isArray(arg)) {
- if ('string' == type)
- arg = arg.split(/\s+/);
- for (i = 0, len = arg.length; i < len; ++i) {
- var field = arg[i];
- if (!field) continue;
- var include = '-' == field[0] ? 0 : 1;
- if (include === 0) field = field.substring(1);
- fields[field] = include;
- }
- return this;
- }
- if (utils.isObject(arg)) {
- var keys = utils.keys(arg);
- for (i = 0; i < keys.length; ++i) {
- fields[keys[i]] = arg[keys[i]];
- }
- return this;
- }
- throw new TypeError('Invalid select() argument. Must be string or object.');
- };
- Query.prototype.slice = function() {
- if (0 === arguments.length)
- return this;
- this._validate('slice');
- var path, val;
- if (1 === arguments.length) {
- var arg = arguments[0];
- if (typeof arg === 'object' && !Array.isArray(arg)) {
- var keys = Object.keys(arg);
- var numKeys = keys.length;
- for (var i = 0; i < numKeys; ++i) {
- this.slice(keys[i], arg[keys[i]]);
- }
- return this;
- }
- this._ensurePath('slice');
- path = this._path;
- val = arguments[0];
- } else if (2 === arguments.length) {
- if ('number' === typeof arguments[0]) {
- this._ensurePath('slice');
- path = this._path;
- val = slice(arguments);
- } else {
- path = arguments[0];
- val = arguments[1];
- }
- } else if (3 === arguments.length) {
- path = arguments[0];
- val = slice(arguments, 1);
- }
- var myFields = this._fields || (this._fields = {});
- myFields[path] = { '$slice': val };
- return this;
- };
- Query.prototype.sort = function(arg) {
- if (!arg) return this;
- var i, len, field;
- this._validate('sort');
- var type = typeof arg;
-
- if (Array.isArray(arg)) {
- len = arg.length;
- for (i = 0; i < arg.length; ++i) {
- if (!Array.isArray(arg[i])) {
- throw new Error('Invalid sort() argument, must be array of arrays');
- }
- _pushArr(this.options, arg[i][0], arg[i][1]);
- }
- return this;
- }
-
- if (1 === arguments.length && 'string' == type) {
- arg = arg.split(/\s+/);
- len = arg.length;
- for (i = 0; i < len; ++i) {
- field = arg[i];
- if (!field) continue;
- var ascend = '-' == field[0] ? -1 : 1;
- if (ascend === -1) field = field.substring(1);
- push(this.options, field, ascend);
- }
- return this;
- }
-
- if (utils.isObject(arg)) {
- var keys = utils.keys(arg);
- for (i = 0; i < keys.length; ++i) {
- field = keys[i];
- push(this.options, field, arg[field]);
- }
- return this;
- }
- if (typeof Map !== 'undefined' && arg instanceof Map) {
- _pushMap(this.options, arg);
- return this;
- }
- throw new TypeError('Invalid sort() argument. Must be a string, object, or array.');
- };
- var _validSortValue = {
- '1': 1,
- '-1': -1,
- 'asc': 1,
- 'ascending': 1,
- 'desc': -1,
- 'descending': -1
- };
- function push(opts, field, value) {
- if (Array.isArray(opts.sort)) {
- throw new TypeError('Can\'t mix sort syntaxes. Use either array or object:' +
- '\n- `.sort([[\'field\', 1], [\'test\', -1]])`' +
- '\n- `.sort({ field: 1, test: -1 })`');
- }
- var s;
- if (value && value.$meta) {
- s = opts.sort || (opts.sort = {});
- s[field] = { $meta : value.$meta };
- return;
- }
- s = opts.sort || (opts.sort = {});
- var val = String(value || 1).toLowerCase();
- val = _validSortValue[val];
- if (!val) throw new TypeError('Invalid sort value: { ' + field + ': ' + value + ' }');
- s[field] = val;
- }
- function _pushArr(opts, field, value) {
- opts.sort = opts.sort || [];
- if (!Array.isArray(opts.sort)) {
- throw new TypeError('Can\'t mix sort syntaxes. Use either array or object:' +
- '\n- `.sort([[\'field\', 1], [\'test\', -1]])`' +
- '\n- `.sort({ field: 1, test: -1 })`');
- }
- var val = String(value || 1).toLowerCase();
- val = _validSortValue[val];
- if (!val) throw new TypeError('Invalid sort value: [ ' + field + ', ' + value + ' ]');
- opts.sort.push([field, val]);
- }
- function _pushMap(opts, map) {
- opts.sort = opts.sort || new Map();
- if (!(opts.sort instanceof Map)) {
- throw new TypeError('Can\'t mix sort syntaxes. Use either array or ' +
- 'object or map consistently');
- }
- map.forEach(function(value, key) {
- var val = String(value || 1).toLowerCase();
- val = _validSortValue[val];
- if (!val) throw new TypeError('Invalid sort value: < ' + key + ': ' + value + ' >');
- opts.sort.set(key, val);
- });
- }
- ['limit', 'skip', 'maxScan', 'batchSize', 'comment'].forEach(function(method) {
- Query.prototype[method] = function(v) {
- this._validate(method);
- this.options[method] = v;
- return this;
- };
- });
- Query.prototype.maxTime = Query.prototype.maxTimeMS = function(ms) {
- this._validate('maxTime');
- this.options.maxTimeMS = ms;
- return this;
- };
- Query.prototype.snapshot = function() {
- this._validate('snapshot');
- this.options.snapshot = arguments.length
- ? !!arguments[0]
- : true;
- return this;
- };
- Query.prototype.hint = function() {
- if (0 === arguments.length) return this;
- this._validate('hint');
- var arg = arguments[0];
- if (utils.isObject(arg)) {
- var hint = this.options.hint || (this.options.hint = {});
-
- for (var k in arg) {
- hint[k] = arg[k];
- }
- return this;
- }
- if (typeof arg === 'string') {
- this.options.hint = arg;
- return this;
- }
- throw new TypeError('Invalid hint. ' + arg);
- };
- Query.prototype.j = function j(val) {
- this.options.j = val;
- return this;
- };
- Query.prototype.slaveOk = function(v) {
- this.options.slaveOk = arguments.length ? !!v : true;
- return this;
- };
- Query.prototype.read = Query.prototype.setReadPreference = function(pref) {
- if (arguments.length > 1 && !Query.prototype.read.deprecationWarningIssued) {
- console.error('Deprecation warning: \'tags\' argument is not supported anymore in Query.read() method. Please use mongodb.ReadPreference object instead.');
- Query.prototype.read.deprecationWarningIssued = true;
- }
- this.options.readPreference = utils.readPref(pref);
- return this;
- };
- Query.prototype.readConcern = Query.prototype.r = function(level) {
- this.options.readConcern = utils.readConcern(level);
- return this;
- };
- Query.prototype.tailable = function() {
- this._validate('tailable');
- this.options.tailable = arguments.length
- ? !!arguments[0]
- : true;
- return this;
- };
- Query.prototype.writeConcern = Query.prototype.w = function writeConcern(concern) {
- if ('object' === typeof concern) {
- if ('undefined' !== typeof concern.j) this.options.j = concern.j;
- if ('undefined' !== typeof concern.w) this.options.w = concern.w;
- if ('undefined' !== typeof concern.wtimeout) this.options.wtimeout = concern.wtimeout;
- } else {
- this.options.w = 'm' === concern ? 'majority' : concern;
- }
- return this;
- };
- Query.prototype.wtimeout = Query.prototype.wTimeout = function wtimeout(ms) {
- this.options.wtimeout = ms;
- return this;
- };
- Query.prototype.merge = function(source) {
- if (!source)
- return this;
- if (!Query.canMerge(source))
- throw new TypeError('Invalid argument. Expected instanceof mquery or plain object');
- if (source instanceof Query) {
-
- if (source._conditions) {
- utils.merge(this._conditions, source._conditions);
- }
- if (source._fields) {
- this._fields || (this._fields = {});
- utils.merge(this._fields, source._fields);
- }
- if (source.options) {
- this.options || (this.options = {});
- utils.merge(this.options, source.options);
- }
- if (source._update) {
- this._update || (this._update = {});
- utils.mergeClone(this._update, source._update);
- }
- if (source._distinct) {
- this._distinct = source._distinct;
- }
- return this;
- }
-
- utils.merge(this._conditions, source);
- return this;
- };
- Query.prototype.find = function(criteria, callback) {
- this.op = 'find';
- if ('function' === typeof criteria) {
- callback = criteria;
- criteria = undefined;
- } else if (Query.canMerge(criteria)) {
- this.merge(criteria);
- }
- if (!callback) return this;
- var conds = this._conditions;
- var options = this._optionsForExec();
- if (this.$useProjection) {
- options.projection = this._fieldsForExec();
- } else {
- options.fields = this._fieldsForExec();
- }
- debug('find', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('find', callback, {
- conditions: conds,
- options: options
- });
- this._collection.find(conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.cursor = function cursor(criteria) {
- if (this.op) {
- if (this.op !== 'find') {
- throw new TypeError('.cursor only support .find method');
- }
- } else {
- this.find(criteria);
- }
- var conds = this._conditions;
- var options = this._optionsForExec();
- if (this.$useProjection) {
- options.projection = this._fieldsForExec();
- } else {
- options.fields = this._fieldsForExec();
- }
- debug('findCursor', this._collection.collectionName, conds, options);
- return this._collection.findCursor(conds, options);
- };
- Query.prototype.findOne = function(criteria, callback) {
- this.op = 'findOne';
- if ('function' === typeof criteria) {
- callback = criteria;
- criteria = undefined;
- } else if (Query.canMerge(criteria)) {
- this.merge(criteria);
- }
- if (!callback) return this;
- var conds = this._conditions;
- var options = this._optionsForExec();
- if (this.$useProjection) {
- options.projection = this._fieldsForExec();
- } else {
- options.fields = this._fieldsForExec();
- }
- debug('findOne', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('findOne', callback, {
- conditions: conds,
- options: options
- });
- this._collection.findOne(conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.count = function(criteria, callback) {
- this.op = 'count';
- this._validate();
- if ('function' === typeof criteria) {
- callback = criteria;
- criteria = undefined;
- } else if (Query.canMerge(criteria)) {
- this.merge(criteria);
- }
- if (!callback) return this;
- var conds = this._conditions,
- options = this._optionsForExec();
- debug('count', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('count', callback, {
- conditions: conds,
- options: options
- });
- this._collection.count(conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.distinct = function(criteria, field, callback) {
- this.op = 'distinct';
- this._validate();
- if (!callback) {
- switch (typeof field) {
- case 'function':
- callback = field;
- if ('string' == typeof criteria) {
- field = criteria;
- criteria = undefined;
- }
- break;
- case 'undefined':
- case 'string':
- break;
- default:
- throw new TypeError('Invalid `field` argument. Must be string or function');
- }
- switch (typeof criteria) {
- case 'function':
- callback = criteria;
- criteria = field = undefined;
- break;
- case 'string':
- field = criteria;
- criteria = undefined;
- break;
- }
- }
- if ('string' == typeof field) {
- this._distinct = field;
- }
- if (Query.canMerge(criteria)) {
- this.merge(criteria);
- }
- if (!callback) {
- return this;
- }
- if (!this._distinct) {
- throw new Error('No value for `distinct` has been declared');
- }
- var conds = this._conditions,
- options = this._optionsForExec();
- debug('distinct', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('distinct', callback, {
- conditions: conds,
- options: options
- });
- this._collection.distinct(this._distinct, conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.update = function update(criteria, doc, options, callback) {
- var force;
- switch (arguments.length) {
- case 3:
- if ('function' == typeof options) {
- callback = options;
- options = undefined;
- }
- break;
- case 2:
- if ('function' == typeof doc) {
- callback = doc;
- doc = criteria;
- criteria = undefined;
- }
- break;
- case 1:
- switch (typeof criteria) {
- case 'function':
- callback = criteria;
- criteria = options = doc = undefined;
- break;
- case 'boolean':
-
- force = criteria;
- criteria = undefined;
- break;
- default:
- doc = criteria;
- criteria = options = undefined;
- break;
- }
- }
- return _update(this, 'update', criteria, doc, options, force, callback);
- };
- Query.prototype.updateMany = function updateMany(criteria, doc, options, callback) {
- var force;
- switch (arguments.length) {
- case 3:
- if ('function' == typeof options) {
- callback = options;
- options = undefined;
- }
- break;
- case 2:
- if ('function' == typeof doc) {
- callback = doc;
- doc = criteria;
- criteria = undefined;
- }
- break;
- case 1:
- switch (typeof criteria) {
- case 'function':
- callback = criteria;
- criteria = options = doc = undefined;
- break;
- case 'boolean':
-
- force = criteria;
- criteria = undefined;
- break;
- default:
- doc = criteria;
- criteria = options = undefined;
- break;
- }
- }
- return _update(this, 'updateMany', criteria, doc, options, force, callback);
- };
- Query.prototype.updateOne = function updateOne(criteria, doc, options, callback) {
- var force;
- switch (arguments.length) {
- case 3:
- if ('function' == typeof options) {
- callback = options;
- options = undefined;
- }
- break;
- case 2:
- if ('function' == typeof doc) {
- callback = doc;
- doc = criteria;
- criteria = undefined;
- }
- break;
- case 1:
- switch (typeof criteria) {
- case 'function':
- callback = criteria;
- criteria = options = doc = undefined;
- break;
- case 'boolean':
-
- force = criteria;
- criteria = undefined;
- break;
- default:
- doc = criteria;
- criteria = options = undefined;
- break;
- }
- }
- return _update(this, 'updateOne', criteria, doc, options, force, callback);
- };
- Query.prototype.replaceOne = function replaceOne(criteria, doc, options, callback) {
- var force;
- switch (arguments.length) {
- case 3:
- if ('function' == typeof options) {
- callback = options;
- options = undefined;
- }
- break;
- case 2:
- if ('function' == typeof doc) {
- callback = doc;
- doc = criteria;
- criteria = undefined;
- }
- break;
- case 1:
- switch (typeof criteria) {
- case 'function':
- callback = criteria;
- criteria = options = doc = undefined;
- break;
- case 'boolean':
-
- force = criteria;
- criteria = undefined;
- break;
- default:
- doc = criteria;
- criteria = options = undefined;
- break;
- }
- }
- this.setOptions({ overwrite: true });
- return _update(this, 'replaceOne', criteria, doc, options, force, callback);
- };
- function _update(query, op, criteria, doc, options, force, callback) {
- query.op = op;
- if (Query.canMerge(criteria)) {
- query.merge(criteria);
- }
- if (doc) {
- query._mergeUpdate(doc);
- }
- if (utils.isObject(options)) {
-
- query.setOptions(options);
- }
-
-
- if (!(force || callback)) {
- return query;
- }
- if (!query._update ||
- !query.options.overwrite && 0 === utils.keys(query._update).length) {
- callback && utils.soon(callback.bind(null, null, 0));
- return query;
- }
- options = query._optionsForExec();
- if (!callback) options.safe = false;
- criteria = query._conditions;
- doc = query._updateForExec();
- debug('update', query._collection.collectionName, criteria, doc, options);
- callback = query._wrapCallback(op, callback, {
- conditions: criteria,
- doc: doc,
- options: options
- });
- query._collection[op](criteria, doc, options, utils.tick(callback));
- return query;
- }
- Query.prototype.remove = function(criteria, callback) {
- this.op = 'remove';
- var force;
- if ('function' === typeof criteria) {
- callback = criteria;
- criteria = undefined;
- } else if (Query.canMerge(criteria)) {
- this.merge(criteria);
- } else if (true === criteria) {
- force = criteria;
- criteria = undefined;
- }
- if (!(force || callback))
- return this;
- var options = this._optionsForExec();
- if (!callback) options.safe = false;
- var conds = this._conditions;
- debug('remove', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('remove', callback, {
- conditions: conds,
- options: options
- });
- this._collection.remove(conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.deleteOne = function(criteria, callback) {
- this.op = 'deleteOne';
- var force;
- if ('function' === typeof criteria) {
- callback = criteria;
- criteria = undefined;
- } else if (Query.canMerge(criteria)) {
- this.merge(criteria);
- } else if (true === criteria) {
- force = criteria;
- criteria = undefined;
- }
- if (!(force || callback))
- return this;
- var options = this._optionsForExec();
- if (!callback) options.safe = false;
- delete options.justOne;
- var conds = this._conditions;
- debug('deleteOne', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('deleteOne', callback, {
- conditions: conds,
- options: options
- });
- this._collection.deleteOne(conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.deleteMany = function(criteria, callback) {
- this.op = 'deleteMany';
- var force;
- if ('function' === typeof criteria) {
- callback = criteria;
- criteria = undefined;
- } else if (Query.canMerge(criteria)) {
- this.merge(criteria);
- } else if (true === criteria) {
- force = criteria;
- criteria = undefined;
- }
- if (!(force || callback))
- return this;
- var options = this._optionsForExec();
- if (!callback) options.safe = false;
- delete options.justOne;
- var conds = this._conditions;
- debug('deleteOne', this._collection.collectionName, conds, options);
- callback = this._wrapCallback('deleteOne', callback, {
- conditions: conds,
- options: options
- });
- this._collection.deleteMany(conds, options, utils.tick(callback));
- return this;
- };
- Query.prototype.findOneAndUpdate = function(criteria, doc, options, callback) {
- this.op = 'findOneAndUpdate';
- this._validate();
- switch (arguments.length) {
- case 3:
- if ('function' == typeof options) {
- callback = options;
- options = {};
- }
- break;
- case 2:
- if ('function' == typeof doc) {
- callback = doc;
- doc = criteria;
- criteria = undefined;
- }
- options = undefined;
- break;
- case 1:
- if ('function' == typeof criteria) {
- callback = criteria;
- criteria = options = doc = undefined;
- } else {
- doc = criteria;
- criteria = options = undefined;
- }
- }
- if (Query.canMerge(criteria)) {
- this.merge(criteria);
- }
-
- if (doc) {
- this._mergeUpdate(doc);
- }
- options && this.setOptions(options);
- if (!callback) return this;
- return this._findAndModify('update', callback);
- };
- Query.prototype.findOneAndRemove = Query.prototype.findOneAndDelete = function(conditions, options, callback) {
- this.op = 'findOneAndRemove';
- this._validate();
- if ('function' == typeof options) {
- callback = options;
- options = undefined;
- } else if ('function' == typeof conditions) {
- callback = conditions;
- conditions = undefined;
- }
-
- if (Query.canMerge(conditions)) {
- this.merge(conditions);
- }
-
- options && this.setOptions(options);
- if (!callback) return this;
- return this._findAndModify('remove', callback);
- };
- Query.prototype._findAndModify = function(type, callback) {
- assert.equal('function', typeof callback);
- var options = this._optionsForExec();
- var fields;
- var doc;
- if ('remove' == type) {
- options.remove = true;
- } else {
- if (!('new' in options)) options.new = true;
- if (!('upsert' in options)) options.upsert = false;
- doc = this._updateForExec();
- if (!doc) {
- if (options.upsert) {
-
- doc = { $set: {} };
- } else {
- return this.findOne(callback);
- }
- }
- }
- fields = this._fieldsForExec();
- if (fields != null) {
- if (this.$useProjection) {
- options.projection = this._fieldsForExec();
- } else {
- options.fields = this._fieldsForExec();
- }
- }
- var conds = this._conditions;
- debug('findAndModify', this._collection.collectionName, conds, doc, options);
- callback = this._wrapCallback('findAndModify', callback, {
- conditions: conds,
- doc: doc,
- options: options
- });
- this._collection.findAndModify(conds, doc, options, utils.tick(callback));
- return this;
- };
- Query.prototype._wrapCallback = function(method, callback, queryInfo) {
- var traceFunction = this._traceFunction || Query.traceFunction;
- if (traceFunction) {
- queryInfo.collectionName = this._collection.collectionName;
- var traceCallback = traceFunction &&
- traceFunction.call(null, method, queryInfo, this);
- var startTime = new Date().getTime();
- return function wrapperCallback(err, result) {
- if (traceCallback) {
- var millis = new Date().getTime() - startTime;
- traceCallback.call(null, err, result, millis);
- }
- if (callback) {
- callback.apply(null, arguments);
- }
- };
- }
- return callback;
- };
- Query.prototype.setTraceFunction = function(traceFunction) {
- this._traceFunction = traceFunction;
- return this;
- };
- Query.prototype.exec = function exec(op, callback) {
- switch (typeof op) {
- case 'function':
- callback = op;
- op = null;
- break;
- case 'string':
- this.op = op;
- break;
- }
- assert.ok(this.op, 'Missing query type: (find, update, etc)');
- if ('update' == this.op || 'remove' == this.op) {
- callback || (callback = true);
- }
- var _this = this;
- if ('function' == typeof callback) {
- this[this.op](callback);
- } else {
- return new Query.Promise(function(success, error) {
- _this[_this.op](function(err, val) {
- if (err) error(err);
- else success(val);
- success = error = null;
- });
- });
- }
- };
- Query.prototype.thunk = function() {
- var _this = this;
- return function(cb) {
- _this.exec(cb);
- };
- };
- Query.prototype.then = function(resolve, reject) {
- var _this = this;
- var promise = new Query.Promise(function(success, error) {
- _this.exec(function(err, val) {
- if (err) error(err);
- else success(val);
- success = error = null;
- });
- });
- return promise.then(resolve, reject);
- };
- Query.prototype.stream = function(streamOptions) {
- if ('find' != this.op)
- throw new Error('stream() is only available for find');
- var conds = this._conditions;
- var options = this._optionsForExec();
- if (this.$useProjection) {
- options.projection = this._fieldsForExec();
- } else {
- options.fields = this._fieldsForExec();
- }
- debug('stream', this._collection.collectionName, conds, options, streamOptions);
- return this._collection.findStream(conds, options, streamOptions);
- };
- Query.prototype.selected = function selected() {
- return !!(this._fields && Object.keys(this._fields).length > 0);
- };
- Query.prototype.selectedInclusively = function selectedInclusively() {
- if (!this._fields) return false;
- var keys = Object.keys(this._fields);
- if (0 === keys.length) return false;
- for (var i = 0; i < keys.length; ++i) {
- var key = keys[i];
- if (0 === this._fields[key]) return false;
- if (this._fields[key] &&
- typeof this._fields[key] === 'object' &&
- this._fields[key].$meta) {
- return false;
- }
- }
- return true;
- };
- Query.prototype.selectedExclusively = function selectedExclusively() {
- if (!this._fields) return false;
- var keys = Object.keys(this._fields);
- if (0 === keys.length) return false;
- for (var i = 0; i < keys.length; ++i) {
- var key = keys[i];
- if (0 === this._fields[key]) return true;
- }
- return false;
- };
- Query.prototype._mergeUpdate = function(doc) {
- if (!this._update) this._update = {};
- if (doc instanceof Query) {
- if (doc._update) {
- utils.mergeClone(this._update, doc._update);
- }
- } else {
- utils.mergeClone(this._update, doc);
- }
- };
- Query.prototype._optionsForExec = function() {
- var options = utils.clone(this.options);
- return options;
- };
- Query.prototype._fieldsForExec = function() {
- return utils.clone(this._fields);
- };
- Query.prototype._updateForExec = function() {
- var update = utils.clone(this._update),
- ops = utils.keys(update),
- i = ops.length,
- ret = {};
- while (i--) {
- var op = ops[i];
- if (this.options.overwrite) {
- ret[op] = update[op];
- continue;
- }
- if ('$' !== op[0]) {
-
- if (!ret.$set) {
- if (update.$set) {
- ret.$set = update.$set;
- } else {
- ret.$set = {};
- }
- }
- ret.$set[op] = update[op];
- ops.splice(i, 1);
- if (!~ops.indexOf('$set')) ops.push('$set');
- } else if ('$set' === op) {
- if (!ret.$set) {
- ret[op] = update[op];
- }
- } else {
- ret[op] = update[op];
- }
- }
- this._compiledUpdate = ret;
- return ret;
- };
- Query.prototype._ensurePath = function(method) {
- if (!this._path) {
- var msg = method + '() must be used after where() '
- + 'when called with these arguments';
- throw new Error(msg);
- }
- };
- Query.permissions = require('./permissions');
- Query._isPermitted = function(a, b) {
- var denied = Query.permissions[b];
- if (!denied) return true;
- return true !== denied[a];
- };
- Query.prototype._validate = function(action) {
- var fail;
- var validator;
- if (undefined === action) {
- validator = Query.permissions[this.op];
- if ('function' != typeof validator) return true;
- fail = validator(this);
- } else if (!Query._isPermitted(action, this.op)) {
- fail = action;
- }
- if (fail) {
- throw new Error(fail + ' cannot be used with ' + this.op);
- }
- };
- Query.canMerge = function(conds) {
- return conds instanceof Query || utils.isObject(conds);
- };
- Query.setGlobalTraceFunction = function(traceFunction) {
- Query.traceFunction = traceFunction;
- };
- Query.utils = utils;
- Query.env = require('./env');
- Query.Collection = require('./collection');
- Query.BaseCollection = require('./collection/collection');
- Query.Promise = require('bluebird');
- module.exports = exports = Query;
|