You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
2.9 KiB
100 lines
2.9 KiB
"use strict"; |
|
|
|
Object.defineProperty(exports, "__esModule", { |
|
value: true |
|
}); |
|
exports.default = simplifyAccess; |
|
|
|
var _t = require("@babel/types"); |
|
|
|
const { |
|
LOGICAL_OPERATORS, |
|
assignmentExpression, |
|
binaryExpression, |
|
cloneNode, |
|
identifier, |
|
logicalExpression, |
|
numericLiteral, |
|
sequenceExpression, |
|
unaryExpression |
|
} = _t; |
|
|
|
function simplifyAccess(path, bindingNames, includeUpdateExpression = true) { |
|
path.traverse(simpleAssignmentVisitor, { |
|
scope: path.scope, |
|
bindingNames, |
|
seen: new WeakSet(), |
|
includeUpdateExpression |
|
}); |
|
} |
|
|
|
const simpleAssignmentVisitor = { |
|
UpdateExpression: { |
|
exit(path) { |
|
const { |
|
scope, |
|
bindingNames, |
|
includeUpdateExpression |
|
} = this; |
|
|
|
if (!includeUpdateExpression) { |
|
return; |
|
} |
|
|
|
const arg = path.get("argument"); |
|
if (!arg.isIdentifier()) return; |
|
const localName = arg.node.name; |
|
if (!bindingNames.has(localName)) return; |
|
|
|
if (scope.getBinding(localName) !== path.scope.getBinding(localName)) { |
|
return; |
|
} |
|
|
|
if (path.parentPath.isExpressionStatement() && !path.isCompletionRecord()) { |
|
const operator = path.node.operator == "++" ? "+=" : "-="; |
|
path.replaceWith(assignmentExpression(operator, arg.node, numericLiteral(1))); |
|
} else if (path.node.prefix) { |
|
path.replaceWith(assignmentExpression("=", identifier(localName), binaryExpression(path.node.operator[0], unaryExpression("+", arg.node), numericLiteral(1)))); |
|
} else { |
|
const old = path.scope.generateUidIdentifierBasedOnNode(arg.node, "old"); |
|
const varName = old.name; |
|
path.scope.push({ |
|
id: old |
|
}); |
|
const binary = binaryExpression(path.node.operator[0], identifier(varName), numericLiteral(1)); |
|
path.replaceWith(sequenceExpression([assignmentExpression("=", identifier(varName), unaryExpression("+", arg.node)), assignmentExpression("=", cloneNode(arg.node), binary), identifier(varName)])); |
|
} |
|
} |
|
|
|
}, |
|
AssignmentExpression: { |
|
exit(path) { |
|
const { |
|
scope, |
|
seen, |
|
bindingNames |
|
} = this; |
|
if (path.node.operator === "=") return; |
|
if (seen.has(path.node)) return; |
|
seen.add(path.node); |
|
const left = path.get("left"); |
|
if (!left.isIdentifier()) return; |
|
const localName = left.node.name; |
|
if (!bindingNames.has(localName)) return; |
|
|
|
if (scope.getBinding(localName) !== path.scope.getBinding(localName)) { |
|
return; |
|
} |
|
|
|
const operator = path.node.operator.slice(0, -1); |
|
|
|
if (LOGICAL_OPERATORS.includes(operator)) { |
|
path.replaceWith(logicalExpression(operator, path.node.left, assignmentExpression("=", cloneNode(path.node.left), path.node.right))); |
|
} else { |
|
path.node.right = binaryExpression(operator, cloneNode(path.node.left), path.node.right); |
|
path.node.operator = "="; |
|
} |
|
} |
|
|
|
} |
|
}; |