index.js
1.48 KB
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
"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = void 0;
var _helperPluginUtils = require("@babel/helper-plugin-utils");
var _core = require("@babel/core");
var _default = (0, _helperPluginUtils.declare)(api => {
api.assertVersion(7);
return {
name: "transform-shorthand-properties",
visitor: {
ObjectMethod(path) {
const {
node
} = path;
if (node.kind === "method") {
const func = _core.types.functionExpression(null, node.params, node.body, node.generator, node.async);
func.returnType = node.returnType;
const computedKey = _core.types.toComputedKey(node);
if (_core.types.isStringLiteral(computedKey, {
value: "__proto__"
})) {
path.replaceWith(_core.types.objectProperty(computedKey, func, true));
} else {
path.replaceWith(_core.types.objectProperty(node.key, func, node.computed));
}
}
},
ObjectProperty(path) {
const {
node
} = path;
if (node.shorthand) {
const computedKey = _core.types.toComputedKey(node);
if (_core.types.isStringLiteral(computedKey, {
value: "__proto__"
})) {
path.replaceWith(_core.types.objectProperty(computedKey, node.value, true));
} else {
node.shorthand = false;
}
}
}
}
};
});
exports.default = _default;