index.js
2.77 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
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.default = void 0;
var _helperPluginUtils = require("@babel/helper-plugin-utils");
var _path = require("path");
var _core = require("@babel/core");
var _default = (0, _helperPluginUtils.declare)(api => {
api.assertVersion(7);
function addDisplayName(id, call) {
const props = call.arguments[0].properties;
let safe = true;
for (let i = 0; i < props.length; i++) {
const prop = props[i];
if (_core.types.isSpreadElement(prop)) {
continue;
}
const key = _core.types.toComputedKey(prop);
if (_core.types.isStringLiteral(key, {
value: "displayName"
})) {
safe = false;
break;
}
}
if (safe) {
props.unshift(_core.types.objectProperty(_core.types.identifier("displayName"), _core.types.stringLiteral(id)));
}
}
const isCreateClassCallExpression = _core.types.buildMatchMemberExpression("React.createClass");
const isCreateClassAddon = callee => _core.types.isIdentifier(callee, {
name: "createReactClass"
});
function isCreateClass(node) {
if (!node || !_core.types.isCallExpression(node)) return false;
if (!isCreateClassCallExpression(node.callee) && !isCreateClassAddon(node.callee)) {
return false;
}
const args = node.arguments;
if (args.length !== 1) return false;
const first = args[0];
if (!_core.types.isObjectExpression(first)) return false;
return true;
}
return {
name: "transform-react-display-name",
visitor: {
ExportDefaultDeclaration({
node
}, state) {
if (isCreateClass(node.declaration)) {
const filename = state.filename || "unknown";
let displayName = _path.basename(filename, _path.extname(filename));
if (displayName === "index") {
displayName = _path.basename(_path.dirname(filename));
}
addDisplayName(displayName, node.declaration);
}
},
CallExpression(path) {
const {
node
} = path;
if (!isCreateClass(node)) return;
let id;
path.find(function (path) {
if (path.isAssignmentExpression()) {
id = path.node.left;
} else if (path.isObjectProperty()) {
id = path.node.key;
} else if (path.isVariableDeclarator()) {
id = path.node.id;
} else if (path.isStatement()) {
return true;
}
if (id) return true;
});
if (!id) return;
if (_core.types.isMemberExpression(id)) {
id = id.property;
}
if (_core.types.isIdentifier(id)) {
addDisplayName(id.name, node);
}
}
}
};
});
exports.default = _default;