65 lines
2.0 KiB
JavaScript
65 lines
2.0 KiB
JavaScript
|
"use strict";
|
||
|
|
|
||
|
|
exports.__esModule = true;
|
||
|
|
|
||
|
|
exports.default = function (opts) {
|
||
|
|
var visitor = {};
|
||
|
|
|
||
|
|
function isAssignment(node) {
|
||
|
|
return node && node.operator === opts.operator + "=";
|
||
|
|
}
|
||
|
|
|
||
|
|
function buildAssignment(left, right) {
|
||
|
|
return t.assignmentExpression("=", left, right);
|
||
|
|
}
|
||
|
|
|
||
|
|
visitor.ExpressionStatement = function (path, file) {
|
||
|
|
if (path.isCompletionRecord()) return;
|
||
|
|
|
||
|
|
var expr = path.node.expression;
|
||
|
|
if (!isAssignment(expr)) return;
|
||
|
|
|
||
|
|
var nodes = [];
|
||
|
|
var exploded = (0, _babelHelperExplodeAssignableExpression2.default)(expr.left, nodes, file, path.scope, true);
|
||
|
|
|
||
|
|
nodes.push(t.expressionStatement(buildAssignment(exploded.ref, opts.build(exploded.uid, expr.right))));
|
||
|
|
|
||
|
|
path.replaceWithMultiple(nodes);
|
||
|
|
};
|
||
|
|
|
||
|
|
visitor.AssignmentExpression = function (path, file) {
|
||
|
|
var node = path.node,
|
||
|
|
scope = path.scope;
|
||
|
|
|
||
|
|
if (!isAssignment(node)) return;
|
||
|
|
|
||
|
|
var nodes = [];
|
||
|
|
var exploded = (0, _babelHelperExplodeAssignableExpression2.default)(node.left, nodes, file, scope);
|
||
|
|
nodes.push(buildAssignment(exploded.ref, opts.build(exploded.uid, node.right)));
|
||
|
|
path.replaceWithMultiple(nodes);
|
||
|
|
};
|
||
|
|
|
||
|
|
visitor.BinaryExpression = function (path) {
|
||
|
|
var node = path.node;
|
||
|
|
|
||
|
|
if (node.operator === opts.operator) {
|
||
|
|
path.replaceWith(opts.build(node.left, node.right));
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
return visitor;
|
||
|
|
};
|
||
|
|
|
||
|
|
var _babelHelperExplodeAssignableExpression = require("babel-helper-explode-assignable-expression");
|
||
|
|
|
||
|
|
var _babelHelperExplodeAssignableExpression2 = _interopRequireDefault(_babelHelperExplodeAssignableExpression);
|
||
|
|
|
||
|
|
var _babelTypes = require("babel-types");
|
||
|
|
|
||
|
|
var t = _interopRequireWildcard(_babelTypes);
|
||
|
|
|
||
|
|
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
|
||
|
|
|
||
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
||
|
|
|
||
|
|
module.exports = exports["default"];
|