/
index.ts
90 lines (80 loc) 路 2.49 KB
/
index.ts
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
import { declare } from "@babel/helper-plugin-utils";
import { types as t } from "@babel/core";
import type { NodePath } from "@babel/traverse";
export default declare(api => {
api.assertVersion(7);
return {
name: "transform-new-target",
visitor: {
MetaProperty(path) {
const meta = path.get("meta");
const property = path.get("property");
const { scope } = path;
if (
meta.isIdentifier({ name: "new" }) &&
property.isIdentifier({ name: "target" })
) {
const func = path.findParent(path => {
if (path.isClass()) return true;
if (path.isFunction() && !path.isArrowFunctionExpression()) {
if (path.isClassMethod({ kind: "constructor" })) {
return false;
}
return true;
}
return false;
}) as NodePath<
| t.FunctionDeclaration
| t.FunctionExpression
| t.Class
| t.ClassMethod
| t.ClassPrivateMethod
>;
if (!func) {
throw path.buildCodeFrameError(
"new.target must be under a (non-arrow) function or a class.",
);
}
const { node } = func;
if (t.isMethod(node)) {
path.replaceWith(scope.buildUndefinedNode());
return;
}
if (!node.id) {
node.id = scope.generateUidIdentifier("target");
} else {
// packages/babel-helper-create-class-features-plugin/src/fields.ts#L192 unshadow
let scope = path.scope;
while (
scope !== func.parentPath.scope &&
scope?.hasBinding(node.id.name) &&
!scope.bindingIdentifierEquals(node.id.name, node.id)
) {
scope.rename(node.id.name);
scope = scope.parent;
}
}
const constructor = t.memberExpression(
t.thisExpression(),
t.identifier("constructor"),
);
if (func.isClass()) {
path.replaceWith(constructor);
return;
}
path.replaceWith(
t.conditionalExpression(
t.binaryExpression(
"instanceof",
t.thisExpression(),
t.cloneNode(node.id),
),
constructor,
scope.buildUndefinedNode(),
),
);
}
},
},
};
});