You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

252 lines
8.0 KiB
JavaScript

"use strict";
exports.__esModule = true;
var _getIterator2 = require("babel-runtime/core-js/get-iterator");
var _getIterator3 = _interopRequireDefault(_getIterator2);
exports.default = function (_ref) {
var t = _ref.types;
var findBareSupers = {
Super: function Super(path) {
if (path.parentPath.isCallExpression({ callee: path.node })) {
this.push(path.parentPath);
}
}
};
var referenceVisitor = {
ReferencedIdentifier: function ReferencedIdentifier(path) {
if (this.scope.hasOwnBinding(path.node.name)) {
this.collision = true;
path.skip();
}
}
};
var buildObjectDefineProperty = (0, _babelTemplate2.default)("\n Object.defineProperty(REF, KEY, {\n // configurable is false by default\n enumerable: true,\n writable: true,\n value: VALUE\n });\n ");
var buildClassPropertySpec = function buildClassPropertySpec(ref, _ref2) {
var key = _ref2.key,
value = _ref2.value,
computed = _ref2.computed;
return buildObjectDefineProperty({
REF: ref,
KEY: t.isIdentifier(key) && !computed ? t.stringLiteral(key.name) : key,
VALUE: value ? value : t.identifier("undefined")
});
};
var buildClassPropertyNonSpec = function buildClassPropertyNonSpec(ref, _ref3) {
var key = _ref3.key,
value = _ref3.value,
computed = _ref3.computed;
return t.expressionStatement(t.assignmentExpression("=", t.memberExpression(ref, key, computed || t.isLiteral(key)), value));
};
return {
inherits: require("babel-plugin-syntax-class-properties"),
visitor: {
Class: function Class(path, state) {
var buildClassProperty = state.opts.spec ? buildClassPropertySpec : buildClassPropertyNonSpec;
var isDerived = !!path.node.superClass;
var constructor = void 0;
var props = [];
var body = path.get("body");
for (var _iterator = body.get("body"), _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) {
var _ref4;
if (_isArray) {
if (_i >= _iterator.length) break;
_ref4 = _iterator[_i++];
} else {
_i = _iterator.next();
if (_i.done) break;
_ref4 = _i.value;
}
var _path = _ref4;
if (_path.isClassProperty()) {
props.push(_path);
} else if (_path.isClassMethod({ kind: "constructor" })) {
constructor = _path;
}
}
if (!props.length) return;
var nodes = [];
var ref = void 0;
if (path.isClassExpression() || !path.node.id) {
(0, _babelHelperFunctionName2.default)(path);
ref = path.scope.generateUidIdentifier("class");
} else {
ref = path.node.id;
}
var instanceBody = [];
for (var _iterator2 = props, _isArray2 = Array.isArray(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : (0, _getIterator3.default)(_iterator2);;) {
var _ref5;
if (_isArray2) {
if (_i2 >= _iterator2.length) break;
_ref5 = _iterator2[_i2++];
} else {
_i2 = _iterator2.next();
if (_i2.done) break;
_ref5 = _i2.value;
}
var _prop = _ref5;
var propNode = _prop.node;
if (propNode.decorators && propNode.decorators.length > 0) continue;
if (!state.opts.spec && !propNode.value) continue;
var isStatic = propNode.static;
if (isStatic) {
nodes.push(buildClassProperty(ref, propNode));
} else {
if (!propNode.value) continue;
instanceBody.push(buildClassProperty(t.thisExpression(), propNode));
}
}
if (instanceBody.length) {
if (!constructor) {
var newConstructor = t.classMethod("constructor", t.identifier("constructor"), [], t.blockStatement([]));
if (isDerived) {
newConstructor.params = [t.restElement(t.identifier("args"))];
newConstructor.body.body.push(t.returnStatement(t.callExpression(t.super(), [t.spreadElement(t.identifier("args"))])));
}
var _body$unshiftContaine = body.unshiftContainer("body", newConstructor);
constructor = _body$unshiftContaine[0];
}
var collisionState = {
collision: false,
scope: constructor.scope
};
for (var _iterator3 = props, _isArray3 = Array.isArray(_iterator3), _i3 = 0, _iterator3 = _isArray3 ? _iterator3 : (0, _getIterator3.default)(_iterator3);;) {
var _ref6;
if (_isArray3) {
if (_i3 >= _iterator3.length) break;
_ref6 = _iterator3[_i3++];
} else {
_i3 = _iterator3.next();
if (_i3.done) break;
_ref6 = _i3.value;
}
var prop = _ref6;
prop.traverse(referenceVisitor, collisionState);
if (collisionState.collision) break;
}
if (collisionState.collision) {
var initialisePropsRef = path.scope.generateUidIdentifier("initialiseProps");
nodes.push(t.variableDeclaration("var", [t.variableDeclarator(initialisePropsRef, t.functionExpression(null, [], t.blockStatement(instanceBody)))]));
instanceBody = [t.expressionStatement(t.callExpression(t.memberExpression(initialisePropsRef, t.identifier("call")), [t.thisExpression()]))];
}
if (isDerived) {
var bareSupers = [];
constructor.traverse(findBareSupers, bareSupers);
for (var _iterator4 = bareSupers, _isArray4 = Array.isArray(_iterator4), _i4 = 0, _iterator4 = _isArray4 ? _iterator4 : (0, _getIterator3.default)(_iterator4);;) {
var _ref7;
if (_isArray4) {
if (_i4 >= _iterator4.length) break;
_ref7 = _iterator4[_i4++];
} else {
_i4 = _iterator4.next();
if (_i4.done) break;
_ref7 = _i4.value;
}
var bareSuper = _ref7;
bareSuper.insertAfter(instanceBody);
}
} else {
constructor.get("body").unshiftContainer("body", instanceBody);
}
}
for (var _iterator5 = props, _isArray5 = Array.isArray(_iterator5), _i5 = 0, _iterator5 = _isArray5 ? _iterator5 : (0, _getIterator3.default)(_iterator5);;) {
var _ref8;
if (_isArray5) {
if (_i5 >= _iterator5.length) break;
_ref8 = _iterator5[_i5++];
} else {
_i5 = _iterator5.next();
if (_i5.done) break;
_ref8 = _i5.value;
}
var _prop2 = _ref8;
_prop2.remove();
}
if (!nodes.length) return;
if (path.isClassExpression()) {
path.scope.push({ id: ref });
path.replaceWith(t.assignmentExpression("=", ref, path.node));
} else {
if (!path.node.id) {
path.node.id = ref;
}
if (path.parentPath.isExportDeclaration()) {
path = path.parentPath;
}
}
path.insertAfter(nodes);
},
ArrowFunctionExpression: function ArrowFunctionExpression(path) {
var classExp = path.get("body");
if (!classExp.isClassExpression()) return;
var body = classExp.get("body");
var members = body.get("body");
if (members.some(function (member) {
return member.isClassProperty();
})) {
path.ensureBlock();
}
}
}
};
};
var _babelHelperFunctionName = require("babel-helper-function-name");
var _babelHelperFunctionName2 = _interopRequireDefault(_babelHelperFunctionName);
var _babelTemplate = require("babel-template");
var _babelTemplate2 = _interopRequireDefault(_babelTemplate);
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
module.exports = exports["default"];