123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.evaluateTruthy = evaluateTruthy;
- exports.evaluate = evaluate;
- var VALID_CALLEES = ["String", "Number", "Math"];
- var INVALID_METHODS = ["random"];
- function evaluateTruthy() {
- var res = this.evaluate();
- if (res.confident) return !!res.value;
- }
- function deopt(path, state) {
- if (!state.confident) return;
- state.deoptPath = path;
- state.confident = false;
- }
- function evaluateCached(path, state) {
- var node = path.node;
- var seen = state.seen;
- if (seen.has(node)) {
- var existing = seen.get(node);
- if (existing.resolved) {
- return existing.value;
- } else {
- deopt(path, state);
- return;
- }
- } else {
- var item = {
- resolved: false
- };
- seen.set(node, item);
- var val = _evaluate(path, state);
- if (state.confident) {
- item.resolved = true;
- item.value = val;
- }
- return val;
- }
- }
- function _evaluate(path, state) {
- if (!state.confident) return;
- var node = path.node;
- if (path.isSequenceExpression()) {
- var exprs = path.get("expressions");
- return evaluateCached(exprs[exprs.length - 1], state);
- }
- if (path.isStringLiteral() || path.isNumericLiteral() || path.isBooleanLiteral()) {
- return node.value;
- }
- if (path.isNullLiteral()) {
- return null;
- }
- if (path.isTemplateLiteral()) {
- return evaluateQuasis(path, node.quasis, state);
- }
- if (path.isTaggedTemplateExpression() && path.get("tag").isMemberExpression()) {
- var object = path.get("tag.object");
- var name = object.node.name;
- var property = path.get("tag.property");
- if (object.isIdentifier() && name === "String" && !path.scope.getBinding(name, true) && property.isIdentifier && property.node.name === "raw") {
- return evaluateQuasis(path, node.quasi.quasis, state, true);
- }
- }
- if (path.isConditionalExpression()) {
- var testResult = evaluateCached(path.get("test"), state);
- if (!state.confident) return;
- if (testResult) {
- return evaluateCached(path.get("consequent"), state);
- } else {
- return evaluateCached(path.get("alternate"), state);
- }
- }
- if (path.isExpressionWrapper()) {
- return evaluateCached(path.get("expression"), state);
- }
- if (path.isMemberExpression() && !path.parentPath.isCallExpression({
- callee: node
- })) {
- var _property = path.get("property");
- var _object = path.get("object");
- if (_object.isLiteral() && _property.isIdentifier()) {
- var value = _object.node.value;
- var type = typeof value;
- if (type === "number" || type === "string") {
- return value[_property.node.name];
- }
- }
- }
- if (path.isReferencedIdentifier()) {
- var binding = path.scope.getBinding(node.name);
- if (binding && binding.constantViolations.length > 0) {
- return deopt(binding.path, state);
- }
- if (binding && path.node.start < binding.path.node.end) {
- return deopt(binding.path, state);
- }
- if (binding && binding.hasValue) {
- return binding.value;
- } else {
- if (node.name === "undefined") {
- return binding ? deopt(binding.path, state) : undefined;
- } else if (node.name === "Infinity") {
- return binding ? deopt(binding.path, state) : Infinity;
- } else if (node.name === "NaN") {
- return binding ? deopt(binding.path, state) : NaN;
- }
- var resolved = path.resolve();
- if (resolved === path) {
- return deopt(path, state);
- } else {
- return evaluateCached(resolved, state);
- }
- }
- }
- if (path.isUnaryExpression({
- prefix: true
- })) {
- if (node.operator === "void") {
- return undefined;
- }
- var argument = path.get("argument");
- if (node.operator === "typeof" && (argument.isFunction() || argument.isClass())) {
- return "function";
- }
- var arg = evaluateCached(argument, state);
- if (!state.confident) return;
- switch (node.operator) {
- case "!":
- return !arg;
- case "+":
- return +arg;
- case "-":
- return -arg;
- case "~":
- return ~arg;
- case "typeof":
- return typeof arg;
- }
- }
- if (path.isArrayExpression()) {
- var arr = [];
- var elems = path.get("elements");
- for (var _iterator = elems, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : _iterator[Symbol.iterator]();;) {
- var _ref;
- if (_isArray) {
- if (_i >= _iterator.length) break;
- _ref = _iterator[_i++];
- } else {
- _i = _iterator.next();
- if (_i.done) break;
- _ref = _i.value;
- }
- var elem = _ref;
- var elemValue = elem.evaluate();
- if (elemValue.confident) {
- arr.push(elemValue.value);
- } else {
- return deopt(elem, state);
- }
- }
- return arr;
- }
- if (path.isObjectExpression()) {
- var obj = {};
- var props = path.get("properties");
- for (var _iterator2 = props, _isArray2 = Array.isArray(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : _iterator2[Symbol.iterator]();;) {
- var _ref2;
- if (_isArray2) {
- if (_i2 >= _iterator2.length) break;
- _ref2 = _iterator2[_i2++];
- } else {
- _i2 = _iterator2.next();
- if (_i2.done) break;
- _ref2 = _i2.value;
- }
- var prop = _ref2;
- if (prop.isObjectMethod() || prop.isSpreadElement()) {
- return deopt(prop, state);
- }
- var keyPath = prop.get("key");
- var key = keyPath;
- if (prop.node.computed) {
- key = key.evaluate();
- if (!key.confident) {
- return deopt(keyPath, state);
- }
- key = key.value;
- } else if (key.isIdentifier()) {
- key = key.node.name;
- } else {
- key = key.node.value;
- }
- var valuePath = prop.get("value");
- var _value = valuePath.evaluate();
- if (!_value.confident) {
- return deopt(valuePath, state);
- }
- _value = _value.value;
- obj[key] = _value;
- }
- return obj;
- }
- if (path.isLogicalExpression()) {
- var wasConfident = state.confident;
- var left = evaluateCached(path.get("left"), state);
- var leftConfident = state.confident;
- state.confident = wasConfident;
- var right = evaluateCached(path.get("right"), state);
- var rightConfident = state.confident;
- state.confident = leftConfident && rightConfident;
- switch (node.operator) {
- case "||":
- if (left && leftConfident) {
- state.confident = true;
- return left;
- }
- if (!state.confident) return;
- return left || right;
- case "&&":
- if (!left && leftConfident || !right && rightConfident) {
- state.confident = true;
- }
- if (!state.confident) return;
- return left && right;
- }
- }
- if (path.isBinaryExpression()) {
- var _left = evaluateCached(path.get("left"), state);
- if (!state.confident) return;
- var _right = evaluateCached(path.get("right"), state);
- if (!state.confident) return;
- switch (node.operator) {
- case "-":
- return _left - _right;
- case "+":
- return _left + _right;
- case "/":
- return _left / _right;
- case "*":
- return _left * _right;
- case "%":
- return _left % _right;
- case "**":
- return Math.pow(_left, _right);
- case "<":
- return _left < _right;
- case ">":
- return _left > _right;
- case "<=":
- return _left <= _right;
- case ">=":
- return _left >= _right;
- case "==":
- return _left == _right;
- case "!=":
- return _left != _right;
- case "===":
- return _left === _right;
- case "!==":
- return _left !== _right;
- case "|":
- return _left | _right;
- case "&":
- return _left & _right;
- case "^":
- return _left ^ _right;
- case "<<":
- return _left << _right;
- case ">>":
- return _left >> _right;
- case ">>>":
- return _left >>> _right;
- }
- }
- if (path.isCallExpression()) {
- var callee = path.get("callee");
- var context;
- var func;
- if (callee.isIdentifier() && !path.scope.getBinding(callee.node.name, true) && VALID_CALLEES.indexOf(callee.node.name) >= 0) {
- func = global[node.callee.name];
- }
- if (callee.isMemberExpression()) {
- var _object2 = callee.get("object");
- var _property2 = callee.get("property");
- if (_object2.isIdentifier() && _property2.isIdentifier() && VALID_CALLEES.indexOf(_object2.node.name) >= 0 && INVALID_METHODS.indexOf(_property2.node.name) < 0) {
- context = global[_object2.node.name];
- func = context[_property2.node.name];
- }
- if (_object2.isLiteral() && _property2.isIdentifier()) {
- var _type = typeof _object2.node.value;
- if (_type === "string" || _type === "number") {
- context = _object2.node.value;
- func = context[_property2.node.name];
- }
- }
- }
- if (func) {
- var args = path.get("arguments").map(function (arg) {
- return evaluateCached(arg, state);
- });
- if (!state.confident) return;
- return func.apply(context, args);
- }
- }
- deopt(path, state);
- }
- function evaluateQuasis(path, quasis, state, raw) {
- if (raw === void 0) {
- raw = false;
- }
- var str = "";
- var i = 0;
- var exprs = path.get("expressions");
- for (var _iterator3 = quasis, _isArray3 = Array.isArray(_iterator3), _i3 = 0, _iterator3 = _isArray3 ? _iterator3 : _iterator3[Symbol.iterator]();;) {
- var _ref3;
- if (_isArray3) {
- if (_i3 >= _iterator3.length) break;
- _ref3 = _iterator3[_i3++];
- } else {
- _i3 = _iterator3.next();
- if (_i3.done) break;
- _ref3 = _i3.value;
- }
- var elem = _ref3;
- if (!state.confident) break;
- str += raw ? elem.value.raw : elem.value.cooked;
- var expr = exprs[i++];
- if (expr) str += String(evaluateCached(expr, state));
- }
- if (!state.confident) return;
- return str;
- }
- function evaluate() {
- var state = {
- confident: true,
- deoptPath: null,
- seen: new Map()
- };
- var value = evaluateCached(this, state);
- if (!state.confident) value = undefined;
- return {
- confident: state.confident,
- deopt: state.deoptPath,
- value: value
- };
- }
|