123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.hasOwnDecorators = hasOwnDecorators;
- exports.hasDecorators = hasDecorators;
- exports.buildDecoratedClass = buildDecoratedClass;
- var _core = require("@babel/core");
- var _helperReplaceSupers = require("@babel/helper-replace-supers");
- var _helperFunctionName = require("@babel/helper-function-name");
- function hasOwnDecorators(node) {
- return !!(node.decorators && node.decorators.length);
- }
- function hasDecorators(node) {
- return hasOwnDecorators(node) || node.body.body.some(hasOwnDecorators);
- }
- function prop(key, value) {
- if (!value) return null;
- return _core.types.objectProperty(_core.types.identifier(key), value);
- }
- function method(key, body) {
- return _core.types.objectMethod("method", _core.types.identifier(key), [], _core.types.blockStatement(body));
- }
- function takeDecorators(node) {
- let result;
- if (node.decorators && node.decorators.length > 0) {
- result = _core.types.arrayExpression(node.decorators.map(decorator => decorator.expression));
- }
- node.decorators = undefined;
- return result;
- }
- function getKey(node) {
- if (node.computed) {
- return node.key;
- } else if (_core.types.isIdentifier(node.key)) {
- return _core.types.stringLiteral(node.key.name);
- } else {
- return _core.types.stringLiteral(String(node.key.value));
- }
- }
- function extractElementDescriptor(classRef, superRef, path) {
- const {
- node,
- scope
- } = path;
- const isMethod = path.isClassMethod();
- if (path.isPrivate()) {
- throw path.buildCodeFrameError(`Private ${isMethod ? "methods" : "fields"} in decorated classes are not supported yet.`);
- }
- new _helperReplaceSupers.default({
- methodPath: path,
- objectRef: classRef,
- superRef,
- file: this,
- refToPreserve: classRef
- }).replace();
- const properties = [prop("kind", _core.types.stringLiteral(isMethod ? node.kind : "field")), prop("decorators", takeDecorators(node)), prop("static", node.static && _core.types.booleanLiteral(true)), prop("key", getKey(node))].filter(Boolean);
- if (isMethod) {
- const id = node.computed ? null : node.key;
- _core.types.toExpression(node);
- properties.push(prop("value", (0, _helperFunctionName.default)({
- node,
- id,
- scope
- }) || node));
- } else if (node.value) {
- properties.push(method("value", _core.template.statements.ast`return ${node.value}`));
- } else {
- properties.push(prop("value", scope.buildUndefinedNode()));
- }
- path.remove();
- return _core.types.objectExpression(properties);
- }
- function addDecorateHelper(file) {
- try {
- return file.addHelper("decorate");
- } catch (err) {
- if (err.code === "BABEL_HELPER_UNKNOWN") {
- err.message += "\n '@babel/plugin-transform-decorators' in non-legacy mode" + " requires '@babel/core' version ^7.0.2 and you appear to be using" + " an older version.";
- }
- throw err;
- }
- }
- function buildDecoratedClass(ref, path, elements, file) {
- const {
- node,
- scope
- } = path;
- const initializeId = scope.generateUidIdentifier("initialize");
- const isDeclaration = node.id && path.isDeclaration();
- const isStrict = path.isInStrictMode();
- const {
- superClass
- } = node;
- node.type = "ClassDeclaration";
- if (!node.id) node.id = _core.types.cloneNode(ref);
- let superId;
- if (superClass) {
- superId = scope.generateUidIdentifierBasedOnNode(node.superClass, "super");
- node.superClass = superId;
- }
- const classDecorators = takeDecorators(node);
- const definitions = _core.types.arrayExpression(elements.filter(element => !element.node.abstract).map(extractElementDescriptor.bind(file, node.id, superId)));
- const wrapperCall = _core.template.expression.ast`
- ${addDecorateHelper(file)}(
- ${classDecorators || _core.types.nullLiteral()},
- function (${initializeId}, ${superClass ? _core.types.cloneNode(superId) : null}) {
- ${node}
- return { F: ${_core.types.cloneNode(node.id)}, d: ${definitions} };
- },
- ${superClass}
- )
- `;
- if (!isStrict) {
- wrapperCall.arguments[1].body.directives.push(_core.types.directive(_core.types.directiveLiteral("use strict")));
- }
- let replacement = wrapperCall;
- let classPathDesc = "arguments.1.body.body.0";
- if (isDeclaration) {
- replacement = _core.template.statement.ast`let ${ref} = ${wrapperCall}`;
- classPathDesc = "declarations.0.init." + classPathDesc;
- }
- return {
- instanceNodes: [_core.template.statement.ast`${_core.types.cloneNode(initializeId)}(this)`],
- wrapClass(path) {
- path.replaceWith(replacement);
- return path.get(classPathDesc);
- }
- };
- }
|