123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.default = rewriteLiveReferences;
- var _core = require("@babel/core");
- function isInType(path) {
- do {
- switch (path.parent.type) {
- case "TSTypeAnnotation":
- case "TSTypeAliasDeclaration":
- case "TSTypeReference":
- case "TypeAnnotation":
- case "TypeAlias":
- return true;
- case "ExportSpecifier":
- return path.parentPath.parent.exportKind === "type";
- default:
- if (path.parentPath.isStatement() || path.parentPath.isExpression()) {
- return false;
- }
- }
- } while (path = path.parentPath);
- }
- function rewriteLiveReferences(programPath, metadata, wrapReference) {
- const imported = new Map();
- const exported = new Map();
- const requeueInParent = path => {
- programPath.requeue(path);
- };
- for (const [source, data] of metadata.source) {
- for (const [localName, importName] of data.imports) {
- imported.set(localName, [source, importName, null]);
- }
- for (const localName of data.importsNamespace) {
- imported.set(localName, [source, null, localName]);
- }
- }
- for (const [local, data] of metadata.local) {
- let exportMeta = exported.get(local);
- if (!exportMeta) {
- exportMeta = [];
- exported.set(local, exportMeta);
- }
- exportMeta.push(...data.names);
- }
- const rewriteBindingInitVisitorState = {
- metadata,
- requeueInParent,
- scope: programPath.scope,
- exported
- };
- programPath.traverse(rewriteBindingInitVisitor, rewriteBindingInitVisitorState);
- const rewriteReferencesVisitorState = {
- seen: new WeakSet(),
- metadata,
- requeueInParent,
- scope: programPath.scope,
- imported,
- exported,
- buildImportReference([source, importName, localName], identNode) {
- const meta = metadata.source.get(source);
- meta.referenced = true;
- if (localName) {
- if (meta.wrap) {
- var _wrapReference;
- identNode = (_wrapReference = wrapReference(identNode, meta.wrap)) != null ? _wrapReference : identNode;
- }
- return identNode;
- }
- let namespace = _core.types.identifier(meta.name);
- if (meta.wrap) {
- var _wrapReference2;
- namespace = (_wrapReference2 = wrapReference(namespace, meta.wrap)) != null ? _wrapReference2 : namespace;
- }
- if (importName === "default" && meta.interop === "node-default") {
- return namespace;
- }
- const computed = metadata.stringSpecifiers.has(importName);
- return _core.types.memberExpression(namespace, computed ? _core.types.stringLiteral(importName) : _core.types.identifier(importName), computed);
- }
- };
- programPath.traverse(rewriteReferencesVisitor, rewriteReferencesVisitorState);
- }
- const rewriteBindingInitVisitor = {
- Scope(path) {
- path.skip();
- },
- ClassDeclaration(path) {
- const {
- requeueInParent,
- exported,
- metadata
- } = this;
- const {
- id
- } = path.node;
- if (!id) throw new Error("Expected class to have a name");
- const localName = id.name;
- const exportNames = exported.get(localName) || [];
- if (exportNames.length > 0) {
- const statement = _core.types.expressionStatement(buildBindingExportAssignmentExpression(metadata, exportNames, _core.types.identifier(localName), path.scope));
- statement._blockHoist = path.node._blockHoist;
- requeueInParent(path.insertAfter(statement)[0]);
- }
- },
- VariableDeclaration(path) {
- const {
- requeueInParent,
- exported,
- metadata
- } = this;
- const isVar = path.node.kind === "var";
- for (const decl of path.get("declarations")) {
- const {
- id
- } = decl.node;
- let {
- init
- } = decl.node;
- if (_core.types.isIdentifier(id) && exported.has(id.name) && !_core.types.isArrowFunctionExpression(init) && (!_core.types.isFunctionExpression(init) || init.id) && (!_core.types.isClassExpression(init) || init.id)) {
- if (!init) {
- if (isVar) {
- continue;
- } else {
- init = path.scope.buildUndefinedNode();
- }
- }
- decl.node.init = buildBindingExportAssignmentExpression(metadata, exported.get(id.name), init, path.scope);
- requeueInParent(decl.get("init"));
- } else {
- for (const localName of Object.keys(decl.getOuterBindingIdentifiers())) {
- if (exported.has(localName)) {
- const statement = _core.types.expressionStatement(buildBindingExportAssignmentExpression(metadata, exported.get(localName), _core.types.identifier(localName), path.scope));
- statement._blockHoist = path.node._blockHoist;
- requeueInParent(path.insertAfter(statement)[0]);
- }
- }
- }
- }
- }
- };
- const buildBindingExportAssignmentExpression = (metadata, exportNames, localExpr, scope) => {
- const exportsObjectName = metadata.exportName;
- for (let currentScope = scope; currentScope != null; currentScope = currentScope.parent) {
- if (currentScope.hasOwnBinding(exportsObjectName)) {
- currentScope.rename(exportsObjectName);
- }
- }
- return (exportNames || []).reduce((expr, exportName) => {
- const {
- stringSpecifiers
- } = metadata;
- const computed = stringSpecifiers.has(exportName);
- return _core.types.assignmentExpression("=", _core.types.memberExpression(_core.types.identifier(exportsObjectName), computed ? _core.types.stringLiteral(exportName) : _core.types.identifier(exportName), computed), expr);
- }, localExpr);
- };
- const buildImportThrow = localName => {
- return _core.template.expression.ast`
- (function() {
- throw new Error('"' + '${localName}' + '" is read-only.');
- })()
- `;
- };
- const rewriteReferencesVisitor = {
- ReferencedIdentifier(path) {
- const {
- seen,
- buildImportReference,
- scope,
- imported,
- requeueInParent
- } = this;
- if (seen.has(path.node)) return;
- seen.add(path.node);
- const localName = path.node.name;
- const importData = imported.get(localName);
- if (importData) {
- if (isInType(path)) {
- throw path.buildCodeFrameError(`Cannot transform the imported binding "${localName}" since it's also used in a type annotation. ` + `Please strip type annotations using @babel/preset-typescript or @babel/preset-flow.`);
- }
- const localBinding = path.scope.getBinding(localName);
- const rootBinding = scope.getBinding(localName);
- if (rootBinding !== localBinding) return;
- const ref = buildImportReference(importData, path.node);
- ref.loc = path.node.loc;
- if ((path.parentPath.isCallExpression({
- callee: path.node
- }) || path.parentPath.isOptionalCallExpression({
- callee: path.node
- }) || path.parentPath.isTaggedTemplateExpression({
- tag: path.node
- })) && _core.types.isMemberExpression(ref)) {
- path.replaceWith(_core.types.sequenceExpression([_core.types.numericLiteral(0), ref]));
- } else if (path.isJSXIdentifier() && _core.types.isMemberExpression(ref)) {
- const {
- object,
- property
- } = ref;
- path.replaceWith(_core.types.jsxMemberExpression(_core.types.jsxIdentifier(object.name), _core.types.jsxIdentifier(property.name)));
- } else {
- path.replaceWith(ref);
- }
- requeueInParent(path);
- path.skip();
- }
- },
- UpdateExpression(path) {
- const {
- scope,
- seen,
- imported,
- exported,
- requeueInParent,
- buildImportReference
- } = this;
- if (seen.has(path.node)) return;
- seen.add(path.node);
- const arg = path.get("argument");
- if (arg.isMemberExpression()) return;
- const update = path.node;
- if (arg.isIdentifier()) {
- const localName = arg.node.name;
- if (scope.getBinding(localName) !== path.scope.getBinding(localName)) {
- return;
- }
- const exportedNames = exported.get(localName);
- const importData = imported.get(localName);
- if ((exportedNames == null ? void 0 : exportedNames.length) > 0 || importData) {
- if (importData) {
- path.replaceWith(_core.types.assignmentExpression(update.operator[0] + "=", buildImportReference(importData, arg.node), buildImportThrow(localName)));
- } else if (update.prefix) {
- path.replaceWith(buildBindingExportAssignmentExpression(this.metadata, exportedNames, _core.types.cloneNode(update), path.scope));
- } else {
- const ref = scope.generateDeclaredUidIdentifier(localName);
- path.replaceWith(_core.types.sequenceExpression([_core.types.assignmentExpression("=", _core.types.cloneNode(ref), _core.types.cloneNode(update)), buildBindingExportAssignmentExpression(this.metadata, exportedNames, _core.types.identifier(localName), path.scope), _core.types.cloneNode(ref)]));
- }
- }
- }
- requeueInParent(path);
- path.skip();
- },
- AssignmentExpression: {
- exit(path) {
- const {
- scope,
- seen,
- imported,
- exported,
- requeueInParent,
- buildImportReference
- } = this;
- if (seen.has(path.node)) return;
- seen.add(path.node);
- const left = path.get("left");
- if (left.isMemberExpression()) return;
- if (left.isIdentifier()) {
- const localName = left.node.name;
- if (scope.getBinding(localName) !== path.scope.getBinding(localName)) {
- return;
- }
- const exportedNames = exported.get(localName);
- const importData = imported.get(localName);
- if ((exportedNames == null ? void 0 : exportedNames.length) > 0 || importData) {
- const assignment = path.node;
- if (importData) {
- assignment.left = buildImportReference(importData, left.node);
- assignment.right = _core.types.sequenceExpression([assignment.right, buildImportThrow(localName)]);
- }
- const {
- operator
- } = assignment;
- let newExpr;
- if (operator === "=") {
- newExpr = assignment;
- } else if (operator === "&&=" || operator === "||=" || operator === "??=") {
- newExpr = _core.types.assignmentExpression("=", assignment.left, _core.types.logicalExpression(operator.slice(0, -1), _core.types.cloneNode(assignment.left), assignment.right));
- } else {
- newExpr = _core.types.assignmentExpression("=", assignment.left, _core.types.binaryExpression(operator.slice(0, -1), _core.types.cloneNode(assignment.left), assignment.right));
- }
- path.replaceWith(buildBindingExportAssignmentExpression(this.metadata, exportedNames, newExpr, path.scope));
- requeueInParent(path);
- path.skip();
- }
- } else {
- const ids = left.getOuterBindingIdentifiers();
- const programScopeIds = Object.keys(ids).filter(localName => scope.getBinding(localName) === path.scope.getBinding(localName));
- const id = programScopeIds.find(localName => imported.has(localName));
- if (id) {
- path.node.right = _core.types.sequenceExpression([path.node.right, buildImportThrow(id)]);
- }
- const items = [];
- programScopeIds.forEach(localName => {
- const exportedNames = exported.get(localName) || [];
- if (exportedNames.length > 0) {
- items.push(buildBindingExportAssignmentExpression(this.metadata, exportedNames, _core.types.identifier(localName), path.scope));
- }
- });
- if (items.length > 0) {
- let node = _core.types.sequenceExpression(items);
- if (path.parentPath.isExpressionStatement()) {
- node = _core.types.expressionStatement(node);
- node._blockHoist = path.parentPath.node._blockHoist;
- }
- const statement = path.insertAfter(node)[0];
- requeueInParent(statement);
- }
- }
- }
- },
- ForXStatement(path) {
- const {
- scope,
- node
- } = path;
- const {
- left
- } = node;
- const {
- exported,
- imported,
- scope: programScope
- } = this;
- if (!_core.types.isVariableDeclaration(left)) {
- let didTransformExport = false,
- importConstViolationName;
- const loopBodyScope = path.get("body").scope;
- for (const name of Object.keys(_core.types.getOuterBindingIdentifiers(left))) {
- if (programScope.getBinding(name) === scope.getBinding(name)) {
- if (exported.has(name)) {
- didTransformExport = true;
- if (loopBodyScope.hasOwnBinding(name)) {
- loopBodyScope.rename(name);
- }
- }
- if (imported.has(name) && !importConstViolationName) {
- importConstViolationName = name;
- }
- }
- }
- if (!didTransformExport && !importConstViolationName) {
- return;
- }
- path.ensureBlock();
- const bodyPath = path.get("body");
- const newLoopId = scope.generateUidIdentifierBasedOnNode(left);
- path.get("left").replaceWith(_core.types.variableDeclaration("let", [_core.types.variableDeclarator(_core.types.cloneNode(newLoopId))]));
- scope.registerDeclaration(path.get("left"));
- if (didTransformExport) {
- bodyPath.unshiftContainer("body", _core.types.expressionStatement(_core.types.assignmentExpression("=", left, newLoopId)));
- }
- if (importConstViolationName) {
- bodyPath.unshiftContainer("body", _core.types.expressionStatement(buildImportThrow(importConstViolationName)));
- }
- }
- }
- };
- //# sourceMappingURL=rewrite-live-references.js.map
|