arangodb/js/node/node_modules/aqb/index.js

/*jshint browserify: true */
/*globals console: false */
'use strict';
var AqlError = require('./errors').AqlError,
  assumptions = require('./assumptions'),
  types = require('./types'),
  toArray,
  warn;

toArray = Function.prototype.call.bind(Array.prototype.slice);

warn = (function () {
  if (typeof console !== 'undefined') {
    return function () {return console.warn.apply(console, arguments);};
  }
  try {
    var cons = require('console');
    return function () {return cons.warn.apply(cons, arguments);};
  } catch (err) {
    return function () {};
  }
}());

function QB(obj) {
  if (typeof obj === 'string') {
    return QB.str(obj);
  }
  if (obj === null || obj === undefined) {
    return new types.NullLiteral();
  }
  if (typeof obj === 'object') {
    if (obj instanceof Date) {
      return QB(JSON.stringify(obj));
    }
    if (obj instanceof Array) {
      return new types.ListLiteral(obj.map(QB));
    }
    var result = {};
    for (var key in obj) {
      if (obj.hasOwnProperty(key)) {
        result[key] = QB(obj[key]);
      }
    }
    return new types.ObjectLiteral(result);
  }
  return types.autoCastToken(obj);
}

Object.keys(types._PartialStatement.prototype).forEach(function (key) {
  if (key === 'constructor') return;
  QB[key] = types._PartialStatement.prototype[key].bind(null);
});

QB.bool = function (value) {return new types.BooleanLiteral(value);};
QB.num = function (value) {return new types.NumberLiteral(value);};
QB.int_ = QB.int = function (value) {return new types.IntegerLiteral(value);};
QB.str = function (value) {return new types.StringLiteral(value);};
QB.list = function (arr) {return new types.ListLiteral(arr);};
QB.obj = function (obj) {return new types.ObjectLiteral(obj);};
QB.range = function (start, end) {return new types.RangeExpression(start, end);};
QB.get = function (obj, key) {return new types.PropertyAccess(obj, key);};
QB.ref = function (value) {
  if (types.Identifier.re.exec(value)) {
    return new types.Identifier(value);
  }
  return new types.SimpleReference(value);
};
QB.expr = function (value) {return new types.RawExpression(value);};

QB.and = function (/* x, y... */) {return new types.NAryOperation('&&', toArray(arguments));};
QB.or = function (/* x, y... */) {return new types.NAryOperation('||', toArray(arguments));};
QB.plus = QB.add = function (/* x, y... */) {return new types.NAryOperation('+', toArray(arguments));};
QB.minus = QB.sub = function (/* x, y... */) {return new types.NAryOperation('-', toArray(arguments));};
QB.times = QB.mul = function (/* x, y... */) {return new types.NAryOperation('*', toArray(arguments));};
QB.div = function (/* x, y... */) {return new types.NAryOperation('/', toArray(arguments));};
QB.mod = function (/* x, y... */) {return new types.NAryOperation('%', toArray(arguments));};
QB.eq = function (x, y) {return new types.BinaryOperation('==', x, y);};
QB.gt = function (x, y) {return new types.BinaryOperation('>', x, y);};
QB.gte = function (x, y) {return new types.BinaryOperation('>=', x, y);};
QB.lt = function (x, y) {return new types.BinaryOperation('<', x, y);};
QB.lte = function (x, y) {return new types.BinaryOperation('<=', x, y);};
QB.neq = function (x, y) {return new types.BinaryOperation('!=', x, y);};
QB.not = function (x) {return new types.UnaryOperation('!', x);};
QB.neg = function (x) {return new types.UnaryOperation('-', x);};
QB.in_ = QB.in = function (x, y) {return new types.BinaryOperation('in', x, y);};
QB.if_ = QB.if = function (x, y, z) {return new types.TernaryOperation('?', ':', x, y, z);};

QB.fn = function (functionName, arity) {
  if (typeof arity === 'number') {
    arity = [arity];
  }
  return function () {
    var args = Array.prototype.slice.call(arguments), valid, i;
    if (arity) {
      valid = false;
      for (i = 0; !valid && i < arity.length; i++) {
        if (typeof arity[i] === 'number') {
          if (args.length === arity[i]) {
            valid = true;
          }
        } else if (
          Object.prototype.toString.call(arity[i]) === '[object Array]' &&
            args.length >= arity[i][0] && args.length <= arity[i][1]
        ) {
          valid = true;
        }
      }
      if (!valid) {
        throw new AqlError(
          'Invalid number of arguments for function ' +
            functionName + ': ' + args.length
        );
      }
    }
    return new types.FunctionCall(functionName, args);
  };
};

function deprecateAqlFunction(fn, functionName) {
  return function () {
    warn('The AQL function ' + functionName + ' is deprecated!');
    return fn.apply(this, arguments);
  };
}

Object.keys(assumptions.builtins).forEach(function (key) {
  QB[key] = QB.fn(key, assumptions.builtins[key]);
  if (assumptions.deprecatedBuiltins.indexOf(key) !== -1) {
    QB[key] = deprecateAqlFunction(QB[key], key);
  }
});

module.exports = QB;