2023-12-18 13:12:25 +08:00
|
|
|
"use strict";
|
|
|
|
|
|
|
|
Object.defineProperty(exports, "__esModule", {
|
|
|
|
value: true
|
|
|
|
});
|
|
|
|
exports.evaluateTruthy = evaluateTruthy;
|
|
|
|
exports.evaluate = evaluate;
|
2024-01-16 21:26:16 +08:00
|
|
|
var VALID_CALLEES = ["String", "Number", "Math"];
|
|
|
|
var INVALID_METHODS = ["random"];
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
function evaluateTruthy() {
|
2024-01-16 21:26:16 +08:00
|
|
|
var res = this.evaluate();
|
2023-12-18 13:12:25 +08:00
|
|
|
if (res.confident) return !!res.value;
|
|
|
|
}
|
|
|
|
|
|
|
|
function deopt(path, state) {
|
|
|
|
if (!state.confident) return;
|
|
|
|
state.deoptPath = path;
|
|
|
|
state.confident = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
function evaluateCached(path, state) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var node = path.node;
|
|
|
|
var seen = state.seen;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (seen.has(node)) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var existing = seen.get(node);
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (existing.resolved) {
|
|
|
|
return existing.value;
|
|
|
|
} else {
|
|
|
|
deopt(path, state);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
} else {
|
2024-01-16 21:26:16 +08:00
|
|
|
var item = {
|
2023-12-18 13:12:25 +08:00
|
|
|
resolved: false
|
|
|
|
};
|
|
|
|
seen.set(node, item);
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var val = _evaluate(path, state);
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (state.confident) {
|
|
|
|
item.resolved = true;
|
|
|
|
item.value = val;
|
|
|
|
}
|
|
|
|
|
|
|
|
return val;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
function _evaluate(path, state) {
|
|
|
|
if (!state.confident) return;
|
2024-01-16 21:26:16 +08:00
|
|
|
var node = path.node;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (path.isSequenceExpression()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var exprs = path.get("expressions");
|
2023-12-18 13:12:25 +08:00
|
|
|
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()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var object = path.get("tag.object");
|
|
|
|
var name = object.node.name;
|
|
|
|
var property = path.get("tag.property");
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
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()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var testResult = evaluateCached(path.get("test"), state);
|
2023-12-18 13:12:25 +08:00
|
|
|
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
|
|
|
|
})) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var _property = path.get("property");
|
|
|
|
|
|
|
|
var _object = path.get("object");
|
2023-12-18 13:12:25 +08:00
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
if (_object.isLiteral() && _property.isIdentifier()) {
|
|
|
|
var value = _object.node.value;
|
|
|
|
var type = typeof value;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (type === "number" || type === "string") {
|
2024-01-16 21:26:16 +08:00
|
|
|
return value[_property.node.name];
|
2023-12-18 13:12:25 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (path.isReferencedIdentifier()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var binding = path.scope.getBinding(node.name);
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var resolved = path.resolve();
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (resolved === path) {
|
|
|
|
return deopt(path, state);
|
|
|
|
} else {
|
|
|
|
return evaluateCached(resolved, state);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (path.isUnaryExpression({
|
|
|
|
prefix: true
|
|
|
|
})) {
|
|
|
|
if (node.operator === "void") {
|
|
|
|
return undefined;
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var argument = path.get("argument");
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (node.operator === "typeof" && (argument.isFunction() || argument.isClass())) {
|
|
|
|
return "function";
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var arg = evaluateCached(argument, state);
|
2023-12-18 13:12:25 +08:00
|
|
|
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()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var arr = [];
|
|
|
|
var elems = path.get("elements");
|
2023-12-18 13:12:25 +08:00
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
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();
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (elemValue.confident) {
|
|
|
|
arr.push(elemValue.value);
|
|
|
|
} else {
|
|
|
|
return deopt(elem, state);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return arr;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (path.isObjectExpression()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
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;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
if (prop.isObjectMethod() || prop.isSpreadElement()) {
|
|
|
|
return deopt(prop, state);
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var keyPath = prop.get("key");
|
|
|
|
var key = keyPath;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var valuePath = prop.get("value");
|
|
|
|
|
|
|
|
var _value = valuePath.evaluate();
|
2023-12-18 13:12:25 +08:00
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
if (!_value.confident) {
|
2023-12-18 13:12:25 +08:00
|
|
|
return deopt(valuePath, state);
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
_value = _value.value;
|
|
|
|
obj[key] = _value;
|
2023-12-18 13:12:25 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
return obj;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (path.isLogicalExpression()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var wasConfident = state.confident;
|
|
|
|
var left = evaluateCached(path.get("left"), state);
|
|
|
|
var leftConfident = state.confident;
|
2023-12-18 13:12:25 +08:00
|
|
|
state.confident = wasConfident;
|
2024-01-16 21:26:16 +08:00
|
|
|
var right = evaluateCached(path.get("right"), state);
|
|
|
|
var rightConfident = state.confident;
|
|
|
|
state.confident = leftConfident && rightConfident;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
switch (node.operator) {
|
|
|
|
case "||":
|
2024-01-16 21:26:16 +08:00
|
|
|
if (left && leftConfident) {
|
|
|
|
state.confident = true;
|
|
|
|
return left;
|
|
|
|
}
|
|
|
|
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) return;
|
|
|
|
return left || right;
|
|
|
|
|
|
|
|
case "&&":
|
2024-01-16 21:26:16 +08:00
|
|
|
if (!left && leftConfident || !right && rightConfident) {
|
|
|
|
state.confident = true;
|
|
|
|
}
|
|
|
|
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) return;
|
|
|
|
return left && right;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (path.isBinaryExpression()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var _left = evaluateCached(path.get("left"), state);
|
|
|
|
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) return;
|
2024-01-16 21:26:16 +08:00
|
|
|
|
|
|
|
var _right = evaluateCached(path.get("right"), state);
|
|
|
|
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) return;
|
|
|
|
|
|
|
|
switch (node.operator) {
|
|
|
|
case "-":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left - _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "+":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left + _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "/":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left / _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "*":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left * _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "%":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left % _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "**":
|
2024-01-16 21:26:16 +08:00
|
|
|
return Math.pow(_left, _right);
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "<":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left < _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case ">":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left > _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "<=":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left <= _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case ">=":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left >= _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "==":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left == _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "!=":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left != _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "===":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left === _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "!==":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left !== _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "|":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left | _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "&":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left & _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "^":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left ^ _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case "<<":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left << _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case ">>":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left >> _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
case ">>>":
|
2024-01-16 21:26:16 +08:00
|
|
|
return _left >>> _right;
|
2023-12-18 13:12:25 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (path.isCallExpression()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var callee = path.get("callee");
|
|
|
|
var context;
|
|
|
|
var func;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
|
|
|
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()) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var _object2 = callee.get("object");
|
|
|
|
|
|
|
|
var _property2 = callee.get("property");
|
2023-12-18 13:12:25 +08:00
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
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];
|
2023-12-18 13:12:25 +08:00
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
if (_object2.isLiteral() && _property2.isIdentifier()) {
|
|
|
|
var _type = typeof _object2.node.value;
|
2023-12-18 13:12:25 +08:00
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
if (_type === "string" || _type === "number") {
|
|
|
|
context = _object2.node.value;
|
|
|
|
func = context[_property2.node.name];
|
2023-12-18 13:12:25 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (func) {
|
2024-01-16 21:26:16 +08:00
|
|
|
var args = path.get("arguments").map(function (arg) {
|
|
|
|
return evaluateCached(arg, state);
|
|
|
|
});
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) return;
|
|
|
|
return func.apply(context, args);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
deopt(path, state);
|
|
|
|
}
|
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
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;
|
|
|
|
}
|
2023-12-18 13:12:25 +08:00
|
|
|
|
2024-01-16 21:26:16 +08:00
|
|
|
var elem = _ref3;
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) break;
|
|
|
|
str += raw ? elem.value.raw : elem.value.cooked;
|
2024-01-16 21:26:16 +08:00
|
|
|
var expr = exprs[i++];
|
2023-12-18 13:12:25 +08:00
|
|
|
if (expr) str += String(evaluateCached(expr, state));
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!state.confident) return;
|
|
|
|
return str;
|
|
|
|
}
|
|
|
|
|
|
|
|
function evaluate() {
|
2024-01-16 21:26:16 +08:00
|
|
|
var state = {
|
2023-12-18 13:12:25 +08:00
|
|
|
confident: true,
|
|
|
|
deoptPath: null,
|
|
|
|
seen: new Map()
|
|
|
|
};
|
2024-01-16 21:26:16 +08:00
|
|
|
var value = evaluateCached(this, state);
|
2023-12-18 13:12:25 +08:00
|
|
|
if (!state.confident) value = undefined;
|
|
|
|
return {
|
|
|
|
confident: state.confident,
|
|
|
|
deopt: state.deoptPath,
|
|
|
|
value: value
|
|
|
|
};
|
|
|
|
}
|