forked from babel/babel
/
renamer.js
95 lines (76 loc) · 2.83 KB
/
renamer.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import Binding from "../binding";
import * as t from "babel-types";
const renameVisitor = {
ReferencedIdentifier({ node }, state) {
if (node.name === state.oldName) {
node.name = state.newName;
}
},
Scope(path, state) {
if (!path.scope.bindingIdentifierEquals(state.oldName, state.binding.identifier)) {
path.skip();
}
},
"AssignmentExpression|Declaration"(path, state) {
const ids = path.getOuterBindingIdentifiers();
for (const name in ids) {
if (name === state.oldName) ids[name].name = state.newName;
}
}
};
export default class Renamer {
constructor(binding: Binding, oldName: string, newName: string) {
this.newName = newName;
this.oldName = oldName;
this.binding = binding;
}
oldName: string;
newName: string;
binding: Binding;
maybeConvertFromExportDeclaration(parentDeclar) {
const exportDeclar = parentDeclar.parentPath.isExportDeclaration() && parentDeclar.parentPath;
if (!exportDeclar) return;
// build specifiers that point back to this export declaration
const isDefault = exportDeclar.isExportDefaultDeclaration();
if (isDefault && (parentDeclar.isFunctionDeclaration() ||
parentDeclar.isClassDeclaration()) && !parentDeclar.node.id) {
// Ensure that default class and function exports have a name so they have a identifier to
// reference from the export specifier list.
parentDeclar.node.id = parentDeclar.scope.generateUidIdentifier("default");
}
const bindingIdentifiers = parentDeclar.getOuterBindingIdentifiers();
const specifiers = [];
for (const name in bindingIdentifiers) {
const localName = name === this.oldName ? this.newName : name;
const exportedName = isDefault ? "default" : name;
specifiers.push(t.exportSpecifier(t.identifier(localName), t.identifier(exportedName)));
}
if (specifiers.length) {
const aliasDeclar = t.exportNamedDeclaration(null, specifiers);
// hoist to the top if it's a function
if (parentDeclar.isFunctionDeclaration()) {
aliasDeclar._blockHoist = 3;
}
exportDeclar.insertAfter(aliasDeclar);
exportDeclar.replaceWith(parentDeclar.node);
}
}
rename(block?) {
const { binding, oldName, newName } = this;
const { scope, path } = binding;
const parentDeclar = path.find((path) => path.isDeclaration() || path.isFunctionExpression());
if (parentDeclar) {
this.maybeConvertFromExportDeclaration(parentDeclar);
}
scope.traverse(block || scope.block, renameVisitor, this);
if (!block) {
scope.removeOwnBinding(oldName);
scope.bindings[newName] = binding;
this.binding.identifier.name = newName;
}
if (binding.type === "hoisted") {
// https://github.com/babel/babel/issues/2435
// todo: hoist and convert function to a let
}
}
}