1 | 'use strict';
|
2 |
|
3 | Object.defineProperty(exports, '__esModule', { value: true });
|
4 |
|
5 | var path = require('path');
|
6 | var _templateBuilder = require('@babel/template');
|
7 |
|
8 | function _interopDefaultLegacy (e) { return e && typeof e === 'object' && 'default' in e ? e : { 'default': e }; }
|
9 |
|
10 | var path__default = _interopDefaultLegacy(path);
|
11 | var _templateBuilder__default = _interopDefaultLegacy(_templateBuilder);
|
12 |
|
13 | function isAtom(t, callee) {
|
14 | if (t.isIdentifier(callee) && atomFunctionNames.includes(callee.name)) {
|
15 | return true;
|
16 | }
|
17 |
|
18 | if (t.isMemberExpression(callee)) {
|
19 | var property = callee.property;
|
20 |
|
21 | if (t.isIdentifier(property) && atomFunctionNames.includes(property.name)) {
|
22 | return true;
|
23 | }
|
24 | }
|
25 |
|
26 | return false;
|
27 | }
|
28 | var atomFunctionNames = ['atom', 'atomFamily', 'atomWithDefault', 'atomWithHash', 'atomWithImmer', 'atomWithInfiniteQuery', 'atomWithMachine', 'atomWithMutation', 'atomWithObservable', 'atomWithProxy', 'atomWithQuery', 'atomWithReducer', 'atomWithReset', 'atomWithSubscription', 'atomWithStorage', 'atomWithStore', 'freezeAtom', 'loadable', 'selectAtom', 'splitAtom'];
|
29 |
|
30 | var templateBuilder = _templateBuilder__default["default"].default || _templateBuilder__default["default"];
|
31 | function debugLabelPlugin(_ref) {
|
32 | var t = _ref.types;
|
33 | return {
|
34 | visitor: {
|
35 | ExportDefaultDeclaration: function ExportDefaultDeclaration(nodePath, state) {
|
36 | var node = nodePath.node;
|
37 |
|
38 | if (t.isCallExpression(node.declaration) && isAtom(t, node.declaration.callee)) {
|
39 | var filename = state.filename || 'unknown';
|
40 | var displayName = path__default["default"].basename(filename, path__default["default"].extname(filename));
|
41 |
|
42 | if (displayName === 'index') {
|
43 | displayName = path__default["default"].basename(path__default["default"].dirname(filename));
|
44 | }
|
45 |
|
46 | var buildExport = templateBuilder("\n const %%atomIdentifier%% = %%atom%%;\n export default %%atomIdentifier%%\n ");
|
47 | var ast = buildExport({
|
48 | atomIdentifier: t.identifier(displayName),
|
49 | atom: node.declaration
|
50 | });
|
51 | nodePath.replaceWithMultiple(ast);
|
52 | }
|
53 | },
|
54 | VariableDeclarator: function VariableDeclarator(path) {
|
55 | if (t.isIdentifier(path.node.id) && t.isCallExpression(path.node.init) && isAtom(t, path.node.init.callee)) {
|
56 | path.parentPath.insertAfter(t.expressionStatement(t.assignmentExpression('=', t.memberExpression(t.identifier(path.node.id.name), t.identifier('debugLabel')), t.stringLiteral(path.node.id.name))));
|
57 | }
|
58 | }
|
59 | }
|
60 | };
|
61 | }
|
62 |
|
63 | exports["default"] = debugLabelPlugin;
|