1 | "use strict";
|
---|
2 |
|
---|
3 | Object.defineProperty(exports, "__esModule", {
|
---|
4 | value: true
|
---|
5 | });
|
---|
6 | exports.default = rewriteLiveReferences;
|
---|
7 |
|
---|
8 | var _assert = require("assert");
|
---|
9 |
|
---|
10 | var _t = require("@babel/types");
|
---|
11 |
|
---|
12 | var _template = require("@babel/template");
|
---|
13 |
|
---|
14 | var _helperSimpleAccess = require("@babel/helper-simple-access");
|
---|
15 |
|
---|
16 | const {
|
---|
17 | assignmentExpression,
|
---|
18 | callExpression,
|
---|
19 | cloneNode,
|
---|
20 | expressionStatement,
|
---|
21 | getOuterBindingIdentifiers,
|
---|
22 | identifier,
|
---|
23 | isMemberExpression,
|
---|
24 | isVariableDeclaration,
|
---|
25 | jsxIdentifier,
|
---|
26 | jsxMemberExpression,
|
---|
27 | memberExpression,
|
---|
28 | numericLiteral,
|
---|
29 | sequenceExpression,
|
---|
30 | stringLiteral,
|
---|
31 | variableDeclaration,
|
---|
32 | variableDeclarator
|
---|
33 | } = _t;
|
---|
34 |
|
---|
35 | function isInType(path) {
|
---|
36 | do {
|
---|
37 | switch (path.parent.type) {
|
---|
38 | case "TSTypeAnnotation":
|
---|
39 | case "TSTypeAliasDeclaration":
|
---|
40 | case "TSTypeReference":
|
---|
41 | case "TypeAnnotation":
|
---|
42 | case "TypeAlias":
|
---|
43 | return true;
|
---|
44 |
|
---|
45 | case "ExportSpecifier":
|
---|
46 | return path.parentPath.parent.exportKind === "type";
|
---|
47 |
|
---|
48 | default:
|
---|
49 | if (path.parentPath.isStatement() || path.parentPath.isExpression()) {
|
---|
50 | return false;
|
---|
51 | }
|
---|
52 |
|
---|
53 | }
|
---|
54 | } while (path = path.parentPath);
|
---|
55 | }
|
---|
56 |
|
---|
57 | function rewriteLiveReferences(programPath, metadata) {
|
---|
58 | const imported = new Map();
|
---|
59 | const exported = new Map();
|
---|
60 |
|
---|
61 | const requeueInParent = path => {
|
---|
62 | programPath.requeue(path);
|
---|
63 | };
|
---|
64 |
|
---|
65 | for (const [source, data] of metadata.source) {
|
---|
66 | for (const [localName, importName] of data.imports) {
|
---|
67 | imported.set(localName, [source, importName, null]);
|
---|
68 | }
|
---|
69 |
|
---|
70 | for (const localName of data.importsNamespace) {
|
---|
71 | imported.set(localName, [source, null, localName]);
|
---|
72 | }
|
---|
73 | }
|
---|
74 |
|
---|
75 | for (const [local, data] of metadata.local) {
|
---|
76 | let exportMeta = exported.get(local);
|
---|
77 |
|
---|
78 | if (!exportMeta) {
|
---|
79 | exportMeta = [];
|
---|
80 | exported.set(local, exportMeta);
|
---|
81 | }
|
---|
82 |
|
---|
83 | exportMeta.push(...data.names);
|
---|
84 | }
|
---|
85 |
|
---|
86 | const rewriteBindingInitVisitorState = {
|
---|
87 | metadata,
|
---|
88 | requeueInParent,
|
---|
89 | scope: programPath.scope,
|
---|
90 | exported
|
---|
91 | };
|
---|
92 | programPath.traverse(rewriteBindingInitVisitor, rewriteBindingInitVisitorState);
|
---|
93 | (0, _helperSimpleAccess.default)(programPath, new Set([...Array.from(imported.keys()), ...Array.from(exported.keys())]));
|
---|
94 | const rewriteReferencesVisitorState = {
|
---|
95 | seen: new WeakSet(),
|
---|
96 | metadata,
|
---|
97 | requeueInParent,
|
---|
98 | scope: programPath.scope,
|
---|
99 | imported,
|
---|
100 | exported,
|
---|
101 | buildImportReference: ([source, importName, localName], identNode) => {
|
---|
102 | const meta = metadata.source.get(source);
|
---|
103 |
|
---|
104 | if (localName) {
|
---|
105 | if (meta.lazy) identNode = callExpression(identNode, []);
|
---|
106 | return identNode;
|
---|
107 | }
|
---|
108 |
|
---|
109 | let namespace = identifier(meta.name);
|
---|
110 | if (meta.lazy) namespace = callExpression(namespace, []);
|
---|
111 |
|
---|
112 | if (importName === "default" && meta.interop === "node-default") {
|
---|
113 | return namespace;
|
---|
114 | }
|
---|
115 |
|
---|
116 | const computed = metadata.stringSpecifiers.has(importName);
|
---|
117 | return memberExpression(namespace, computed ? stringLiteral(importName) : identifier(importName), computed);
|
---|
118 | }
|
---|
119 | };
|
---|
120 | programPath.traverse(rewriteReferencesVisitor, rewriteReferencesVisitorState);
|
---|
121 | }
|
---|
122 |
|
---|
123 | const rewriteBindingInitVisitor = {
|
---|
124 | Scope(path) {
|
---|
125 | path.skip();
|
---|
126 | },
|
---|
127 |
|
---|
128 | ClassDeclaration(path) {
|
---|
129 | const {
|
---|
130 | requeueInParent,
|
---|
131 | exported,
|
---|
132 | metadata
|
---|
133 | } = this;
|
---|
134 | const {
|
---|
135 | id
|
---|
136 | } = path.node;
|
---|
137 | if (!id) throw new Error("Expected class to have a name");
|
---|
138 | const localName = id.name;
|
---|
139 | const exportNames = exported.get(localName) || [];
|
---|
140 |
|
---|
141 | if (exportNames.length > 0) {
|
---|
142 | const statement = expressionStatement(buildBindingExportAssignmentExpression(metadata, exportNames, identifier(localName)));
|
---|
143 | statement._blockHoist = path.node._blockHoist;
|
---|
144 | requeueInParent(path.insertAfter(statement)[0]);
|
---|
145 | }
|
---|
146 | },
|
---|
147 |
|
---|
148 | VariableDeclaration(path) {
|
---|
149 | const {
|
---|
150 | requeueInParent,
|
---|
151 | exported,
|
---|
152 | metadata
|
---|
153 | } = this;
|
---|
154 | Object.keys(path.getOuterBindingIdentifiers()).forEach(localName => {
|
---|
155 | const exportNames = exported.get(localName) || [];
|
---|
156 |
|
---|
157 | if (exportNames.length > 0) {
|
---|
158 | const statement = expressionStatement(buildBindingExportAssignmentExpression(metadata, exportNames, identifier(localName)));
|
---|
159 | statement._blockHoist = path.node._blockHoist;
|
---|
160 | requeueInParent(path.insertAfter(statement)[0]);
|
---|
161 | }
|
---|
162 | });
|
---|
163 | }
|
---|
164 |
|
---|
165 | };
|
---|
166 |
|
---|
167 | const buildBindingExportAssignmentExpression = (metadata, exportNames, localExpr) => {
|
---|
168 | return (exportNames || []).reduce((expr, exportName) => {
|
---|
169 | const {
|
---|
170 | stringSpecifiers
|
---|
171 | } = metadata;
|
---|
172 | const computed = stringSpecifiers.has(exportName);
|
---|
173 | return assignmentExpression("=", memberExpression(identifier(metadata.exportName), computed ? stringLiteral(exportName) : identifier(exportName), computed), expr);
|
---|
174 | }, localExpr);
|
---|
175 | };
|
---|
176 |
|
---|
177 | const buildImportThrow = localName => {
|
---|
178 | return _template.default.expression.ast`
|
---|
179 | (function() {
|
---|
180 | throw new Error('"' + '${localName}' + '" is read-only.');
|
---|
181 | })()
|
---|
182 | `;
|
---|
183 | };
|
---|
184 |
|
---|
185 | const rewriteReferencesVisitor = {
|
---|
186 | ReferencedIdentifier(path) {
|
---|
187 | const {
|
---|
188 | seen,
|
---|
189 | buildImportReference,
|
---|
190 | scope,
|
---|
191 | imported,
|
---|
192 | requeueInParent
|
---|
193 | } = this;
|
---|
194 | if (seen.has(path.node)) return;
|
---|
195 | seen.add(path.node);
|
---|
196 | const localName = path.node.name;
|
---|
197 | const importData = imported.get(localName);
|
---|
198 |
|
---|
199 | if (importData) {
|
---|
200 | if (isInType(path)) {
|
---|
201 | 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.`);
|
---|
202 | }
|
---|
203 |
|
---|
204 | const localBinding = path.scope.getBinding(localName);
|
---|
205 | const rootBinding = scope.getBinding(localName);
|
---|
206 | if (rootBinding !== localBinding) return;
|
---|
207 | const ref = buildImportReference(importData, path.node);
|
---|
208 | ref.loc = path.node.loc;
|
---|
209 |
|
---|
210 | if ((path.parentPath.isCallExpression({
|
---|
211 | callee: path.node
|
---|
212 | }) || path.parentPath.isOptionalCallExpression({
|
---|
213 | callee: path.node
|
---|
214 | }) || path.parentPath.isTaggedTemplateExpression({
|
---|
215 | tag: path.node
|
---|
216 | })) && isMemberExpression(ref)) {
|
---|
217 | path.replaceWith(sequenceExpression([numericLiteral(0), ref]));
|
---|
218 | } else if (path.isJSXIdentifier() && isMemberExpression(ref)) {
|
---|
219 | const {
|
---|
220 | object,
|
---|
221 | property
|
---|
222 | } = ref;
|
---|
223 | path.replaceWith(jsxMemberExpression(jsxIdentifier(object.name), jsxIdentifier(property.name)));
|
---|
224 | } else {
|
---|
225 | path.replaceWith(ref);
|
---|
226 | }
|
---|
227 |
|
---|
228 | requeueInParent(path);
|
---|
229 | path.skip();
|
---|
230 | }
|
---|
231 | },
|
---|
232 |
|
---|
233 | AssignmentExpression: {
|
---|
234 | exit(path) {
|
---|
235 | const {
|
---|
236 | scope,
|
---|
237 | seen,
|
---|
238 | imported,
|
---|
239 | exported,
|
---|
240 | requeueInParent,
|
---|
241 | buildImportReference
|
---|
242 | } = this;
|
---|
243 | if (seen.has(path.node)) return;
|
---|
244 | seen.add(path.node);
|
---|
245 | const left = path.get("left");
|
---|
246 | if (left.isMemberExpression()) return;
|
---|
247 |
|
---|
248 | if (left.isIdentifier()) {
|
---|
249 | const localName = left.node.name;
|
---|
250 |
|
---|
251 | if (scope.getBinding(localName) !== path.scope.getBinding(localName)) {
|
---|
252 | return;
|
---|
253 | }
|
---|
254 |
|
---|
255 | const exportedNames = exported.get(localName);
|
---|
256 | const importData = imported.get(localName);
|
---|
257 |
|
---|
258 | if ((exportedNames == null ? void 0 : exportedNames.length) > 0 || importData) {
|
---|
259 | _assert(path.node.operator === "=", "Path was not simplified");
|
---|
260 |
|
---|
261 | const assignment = path.node;
|
---|
262 |
|
---|
263 | if (importData) {
|
---|
264 | assignment.left = buildImportReference(importData, assignment.left);
|
---|
265 | assignment.right = sequenceExpression([assignment.right, buildImportThrow(localName)]);
|
---|
266 | }
|
---|
267 |
|
---|
268 | path.replaceWith(buildBindingExportAssignmentExpression(this.metadata, exportedNames, assignment));
|
---|
269 | requeueInParent(path);
|
---|
270 | }
|
---|
271 | } else {
|
---|
272 | const ids = left.getOuterBindingIdentifiers();
|
---|
273 | const programScopeIds = Object.keys(ids).filter(localName => scope.getBinding(localName) === path.scope.getBinding(localName));
|
---|
274 | const id = programScopeIds.find(localName => imported.has(localName));
|
---|
275 |
|
---|
276 | if (id) {
|
---|
277 | path.node.right = sequenceExpression([path.node.right, buildImportThrow(id)]);
|
---|
278 | }
|
---|
279 |
|
---|
280 | const items = [];
|
---|
281 | programScopeIds.forEach(localName => {
|
---|
282 | const exportedNames = exported.get(localName) || [];
|
---|
283 |
|
---|
284 | if (exportedNames.length > 0) {
|
---|
285 | items.push(buildBindingExportAssignmentExpression(this.metadata, exportedNames, identifier(localName)));
|
---|
286 | }
|
---|
287 | });
|
---|
288 |
|
---|
289 | if (items.length > 0) {
|
---|
290 | let node = sequenceExpression(items);
|
---|
291 |
|
---|
292 | if (path.parentPath.isExpressionStatement()) {
|
---|
293 | node = expressionStatement(node);
|
---|
294 | node._blockHoist = path.parentPath.node._blockHoist;
|
---|
295 | }
|
---|
296 |
|
---|
297 | const statement = path.insertAfter(node)[0];
|
---|
298 | requeueInParent(statement);
|
---|
299 | }
|
---|
300 | }
|
---|
301 | }
|
---|
302 |
|
---|
303 | },
|
---|
304 |
|
---|
305 | "ForOfStatement|ForInStatement"(path) {
|
---|
306 | const {
|
---|
307 | scope,
|
---|
308 | node
|
---|
309 | } = path;
|
---|
310 | const {
|
---|
311 | left
|
---|
312 | } = node;
|
---|
313 | const {
|
---|
314 | exported,
|
---|
315 | imported,
|
---|
316 | scope: programScope
|
---|
317 | } = this;
|
---|
318 |
|
---|
319 | if (!isVariableDeclaration(left)) {
|
---|
320 | let didTransformExport = false,
|
---|
321 | importConstViolationName;
|
---|
322 | const loopBodyScope = path.get("body").scope;
|
---|
323 |
|
---|
324 | for (const name of Object.keys(getOuterBindingIdentifiers(left))) {
|
---|
325 | if (programScope.getBinding(name) === scope.getBinding(name)) {
|
---|
326 | if (exported.has(name)) {
|
---|
327 | didTransformExport = true;
|
---|
328 |
|
---|
329 | if (loopBodyScope.hasOwnBinding(name)) {
|
---|
330 | loopBodyScope.rename(name);
|
---|
331 | }
|
---|
332 | }
|
---|
333 |
|
---|
334 | if (imported.has(name) && !importConstViolationName) {
|
---|
335 | importConstViolationName = name;
|
---|
336 | }
|
---|
337 | }
|
---|
338 | }
|
---|
339 |
|
---|
340 | if (!didTransformExport && !importConstViolationName) {
|
---|
341 | return;
|
---|
342 | }
|
---|
343 |
|
---|
344 | path.ensureBlock();
|
---|
345 | const bodyPath = path.get("body");
|
---|
346 | const newLoopId = scope.generateUidIdentifierBasedOnNode(left);
|
---|
347 | path.get("left").replaceWith(variableDeclaration("let", [variableDeclarator(cloneNode(newLoopId))]));
|
---|
348 | scope.registerDeclaration(path.get("left"));
|
---|
349 |
|
---|
350 | if (didTransformExport) {
|
---|
351 | bodyPath.unshiftContainer("body", expressionStatement(assignmentExpression("=", left, newLoopId)));
|
---|
352 | }
|
---|
353 |
|
---|
354 | if (importConstViolationName) {
|
---|
355 | bodyPath.unshiftContainer("body", expressionStatement(buildImportThrow(importConstViolationName)));
|
---|
356 | }
|
---|
357 | }
|
---|
358 | }
|
---|
359 |
|
---|
360 | }; |
---|