command.js 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. const query_1 = require("./commands/query");
  4. const logic_1 = require("./commands/logic");
  5. const update_1 = require("./commands/update");
  6. const type_1 = require("./utils/type");
  7. const aggregate_1 = require("./aggregate");
  8. exports.Command = {
  9. eq(val) {
  10. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.EQ, [val]);
  11. },
  12. neq(val) {
  13. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.NEQ, [val]);
  14. },
  15. lt(val) {
  16. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.LT, [val]);
  17. },
  18. lte(val) {
  19. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.LTE, [val]);
  20. },
  21. gt(val) {
  22. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.GT, [val]);
  23. },
  24. gte(val) {
  25. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.GTE, [val]);
  26. },
  27. in(val) {
  28. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.IN, val);
  29. },
  30. nin(val) {
  31. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.NIN, val);
  32. },
  33. all(val) {
  34. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.ALL, val);
  35. },
  36. elemMatch(val) {
  37. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.ELEM_MATCH, [val]);
  38. },
  39. exists(val) {
  40. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.EXISTS, [val]);
  41. },
  42. size(val) {
  43. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.SIZE, [val]);
  44. },
  45. mod() {
  46. if (arguments.length == 1)
  47. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.MOD, [arguments[0]]);
  48. if (arguments.length == 2)
  49. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.MOD, [[arguments[0], arguments[1]]]);
  50. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.MOD, arguments);
  51. },
  52. geoNear(val) {
  53. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.GEO_NEAR, [val]);
  54. },
  55. geoWithin(val) {
  56. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.GEO_WITHIN, [val]);
  57. },
  58. geoIntersects(val) {
  59. return new query_1.QueryCommand(query_1.QUERY_COMMANDS_LITERAL.GEO_INTERSECTS, [val]);
  60. },
  61. and(...__expressions__) {
  62. const expressions = type_1.isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  63. return new logic_1.LogicCommand(logic_1.LOGIC_COMMANDS_LITERAL.AND, expressions);
  64. },
  65. nor(...__expressions__) {
  66. const expressions = type_1.isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  67. return new logic_1.LogicCommand(logic_1.LOGIC_COMMANDS_LITERAL.NOR, expressions);
  68. },
  69. or(...__expressions__) {
  70. const expressions = type_1.isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  71. return new logic_1.LogicCommand(logic_1.LOGIC_COMMANDS_LITERAL.OR, expressions);
  72. },
  73. not(...__expressions__) {
  74. const expressions = type_1.isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  75. return new logic_1.LogicCommand(logic_1.LOGIC_COMMANDS_LITERAL.NOT, expressions);
  76. },
  77. set(val) {
  78. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.SET, [val]);
  79. },
  80. remove() {
  81. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.REMOVE, []);
  82. },
  83. inc(val) {
  84. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.INC, [val]);
  85. },
  86. mul(val) {
  87. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.MUL, [val]);
  88. },
  89. push(...args) {
  90. let values;
  91. if (type_1.isObject(args[0]) && args[0].hasOwnProperty('each')) {
  92. values = {};
  93. const options = args[0];
  94. if (options.each !== undefined) {
  95. values['$each'] = options.each;
  96. }
  97. if (options.position !== undefined) {
  98. values['$position'] = options.position;
  99. }
  100. if (options.sort !== undefined) {
  101. values['$sort'] = options.sort;
  102. }
  103. if (options.slice !== undefined) {
  104. values['$slice'] = options.slice;
  105. }
  106. }
  107. else if (type_1.isArray(args[0])) {
  108. values = args[0];
  109. }
  110. else {
  111. values = Array.from(args);
  112. }
  113. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.PUSH, values);
  114. },
  115. pull(values) {
  116. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.PULL, values);
  117. },
  118. pullAll(values) {
  119. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.PULL_ALL, values);
  120. },
  121. pop() {
  122. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.POP, []);
  123. },
  124. shift() {
  125. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.SHIFT, []);
  126. },
  127. unshift(...__values__) {
  128. const values = type_1.isArray(arguments[0]) ? arguments[0] : Array.from(arguments);
  129. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.UNSHIFT, values);
  130. },
  131. addToSet(values) {
  132. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.ADD_TO_SET, values);
  133. },
  134. rename(values) {
  135. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.RENAME, [values]);
  136. },
  137. bit(values) {
  138. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.BIT, [values]);
  139. },
  140. max(values) {
  141. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.MAX, [values]);
  142. },
  143. min(values) {
  144. return new update_1.UpdateCommand(update_1.UPDATE_COMMANDS_LITERAL.MIN, [values]);
  145. },
  146. expr(values) {
  147. return {
  148. $expr: values
  149. };
  150. },
  151. jsonSchema(schema) {
  152. return {
  153. $jsonSchema: schema
  154. };
  155. },
  156. text(values) {
  157. if (type_1.isString(values)) {
  158. return {
  159. $search: values.search
  160. };
  161. }
  162. else {
  163. return {
  164. $search: values.search,
  165. $language: values.language,
  166. $caseSensitive: values.caseSensitive,
  167. $diacriticSensitive: values.diacriticSensitive
  168. };
  169. }
  170. },
  171. aggregate: {
  172. pipeline() {
  173. return new aggregate_1.default();
  174. },
  175. abs: param => new AggregationOperator('abs', param),
  176. add: param => new AggregationOperator('add', param),
  177. ceil: param => new AggregationOperator('ceil', param),
  178. divide: param => new AggregationOperator('divide', param),
  179. exp: param => new AggregationOperator('exp', param),
  180. floor: param => new AggregationOperator('floor', param),
  181. ln: param => new AggregationOperator('ln', param),
  182. log: param => new AggregationOperator('log', param),
  183. log10: param => new AggregationOperator('log10', param),
  184. mod: param => new AggregationOperator('mod', param),
  185. multiply: param => new AggregationOperator('multiply', param),
  186. pow: param => new AggregationOperator('pow', param),
  187. sqrt: param => new AggregationOperator('sqrt', param),
  188. subtract: param => new AggregationOperator('subtract', param),
  189. trunc: param => new AggregationOperator('trunc', param),
  190. arrayElemAt: param => new AggregationOperator('arrayElemAt', param),
  191. arrayToObject: param => new AggregationOperator('arrayToObject', param),
  192. concatArrays: param => new AggregationOperator('concatArrays', param),
  193. filter: param => new AggregationOperator('filter', param),
  194. in: param => new AggregationOperator('in', param),
  195. indexOfArray: param => new AggregationOperator('indexOfArray', param),
  196. isArray: param => new AggregationOperator('isArray', param),
  197. map: param => new AggregationOperator('map', param),
  198. range: param => new AggregationOperator('range', param),
  199. reduce: param => new AggregationOperator('reduce', param),
  200. reverseArray: param => new AggregationOperator('reverseArray', param),
  201. size: param => new AggregationOperator('size', param),
  202. slice: param => new AggregationOperator('slice', param),
  203. zip: param => new AggregationOperator('zip', param),
  204. and: param => new AggregationOperator('and', param),
  205. not: param => new AggregationOperator('not', param),
  206. or: param => new AggregationOperator('or', param),
  207. cmp: param => new AggregationOperator('cmp', param),
  208. eq: param => new AggregationOperator('eq', param),
  209. gt: param => new AggregationOperator('gt', param),
  210. gte: param => new AggregationOperator('gte', param),
  211. lt: param => new AggregationOperator('lt', param),
  212. lte: param => new AggregationOperator('lte', param),
  213. neq: param => new AggregationOperator('ne', param),
  214. cond: param => new AggregationOperator('cond', param),
  215. ifNull: param => new AggregationOperator('ifNull', param),
  216. switch: param => new AggregationOperator('switch', param),
  217. dateFromParts: param => new AggregationOperator('dateFromParts', param),
  218. dateFromString: param => new AggregationOperator('dateFromString', param),
  219. dayOfMonth: param => new AggregationOperator('dayOfMonth', param),
  220. dayOfWeek: param => new AggregationOperator('dayOfWeek', param),
  221. dayOfYear: param => new AggregationOperator('dayOfYear', param),
  222. isoDayOfWeek: param => new AggregationOperator('isoDayOfWeek', param),
  223. isoWeek: param => new AggregationOperator('isoWeek', param),
  224. isoWeekYear: param => new AggregationOperator('isoWeekYear', param),
  225. millisecond: param => new AggregationOperator('millisecond', param),
  226. minute: param => new AggregationOperator('minute', param),
  227. month: param => new AggregationOperator('month', param),
  228. second: param => new AggregationOperator('second', param),
  229. hour: param => new AggregationOperator('hour', param),
  230. week: param => new AggregationOperator('week', param),
  231. year: param => new AggregationOperator('year', param),
  232. literal: param => new AggregationOperator('literal', param),
  233. mergeObjects: param => new AggregationOperator('mergeObjects', param),
  234. objectToArray: param => new AggregationOperator('objectToArray', param),
  235. allElementsTrue: param => new AggregationOperator('allElementsTrue', param),
  236. anyElementTrue: param => new AggregationOperator('anyElementTrue', param),
  237. setDifference: param => new AggregationOperator('setDifference', param),
  238. setEquals: param => new AggregationOperator('setEquals', param),
  239. setIntersection: param => new AggregationOperator('setIntersection', param),
  240. setIsSubset: param => new AggregationOperator('setIsSubset', param),
  241. setUnion: param => new AggregationOperator('setUnion', param),
  242. concat: param => new AggregationOperator('concat', param),
  243. dateToString: param => new AggregationOperator('dateToString', param),
  244. indexOfBytes: param => new AggregationOperator('indexOfBytes', param),
  245. indexOfCP: param => new AggregationOperator('indexOfCP', param),
  246. split: param => new AggregationOperator('split', param),
  247. strLenBytes: param => new AggregationOperator('strLenBytes', param),
  248. strLenCP: param => new AggregationOperator('strLenCP', param),
  249. strcasecmp: param => new AggregationOperator('strcasecmp', param),
  250. substr: param => new AggregationOperator('substr', param),
  251. substrBytes: param => new AggregationOperator('substrBytes', param),
  252. substrCP: param => new AggregationOperator('substrCP', param),
  253. toLower: param => new AggregationOperator('toLower', param),
  254. toUpper: param => new AggregationOperator('toUpper', param),
  255. meta: param => new AggregationOperator('meta', param),
  256. addToSet: param => new AggregationOperator('addToSet', param),
  257. avg: param => new AggregationOperator('avg', param),
  258. first: param => new AggregationOperator('first', param),
  259. last: param => new AggregationOperator('last', param),
  260. max: param => new AggregationOperator('max', param),
  261. min: param => new AggregationOperator('min', param),
  262. push: param => new AggregationOperator('push', param),
  263. stdDevPop: param => new AggregationOperator('stdDevPop', param),
  264. stdDevSamp: param => new AggregationOperator('stdDevSamp', param),
  265. sum: param => new AggregationOperator('sum', param),
  266. let: param => new AggregationOperator('let', param)
  267. },
  268. project: {
  269. slice: param => new ProjectionOperator('slice', param),
  270. elemMatch: param => new ProjectionOperator('elemMatch', param)
  271. }
  272. };
  273. class AggregationOperator {
  274. constructor(name, param) {
  275. this['$' + name] = param;
  276. }
  277. }
  278. exports.AggregationOperator = AggregationOperator;
  279. class ProjectionOperator {
  280. constructor(name, param) {
  281. this['$' + name] = param;
  282. }
  283. }
  284. exports.ProjectionOperator = ProjectionOperator;
  285. exports.default = exports.Command;