UNPKG

3.4 kBJavaScriptView Raw
1"use strict";
2
3Object.defineProperty(exports, "__esModule", {
4 value: true
5});
6exports.default = void 0;
7
8var _helperPluginUtils = require("@babel/helper-plugin-utils");
9
10var _pluginSyntaxJsx = _interopRequireDefault(require("@babel/plugin-syntax-jsx"));
11
12var _helperBuilderReactJsx = _interopRequireDefault(require("@babel/helper-builder-react-jsx"));
13
14var _core = require("@babel/core");
15
16function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
17
18const DEFAULT = {
19 pragma: "React.createElement",
20 pragmaFrag: "React.Fragment"
21};
22
23var _default = (0, _helperPluginUtils.declare)((api, options) => {
24 const THROW_IF_NAMESPACE = options.throwIfNamespace === undefined ? true : !!options.throwIfNamespace;
25 const PRAGMA_DEFAULT = options.pragma || DEFAULT.pragma;
26 const PRAGMA_FRAG_DEFAULT = options.pragmaFrag || DEFAULT.pragmaFrag;
27 const PURE_ANNOTATION = options.pure;
28 const JSX_ANNOTATION_REGEX = /\*?\s*@jsx\s+([^\s]+)/;
29 const JSX_FRAG_ANNOTATION_REGEX = /\*?\s*@jsxFrag\s+([^\s]+)/;
30
31 const createIdentifierParser = id => () => {
32 return id.split(".").map(name => _core.types.identifier(name)).reduce((object, property) => _core.types.memberExpression(object, property));
33 };
34
35 const visitor = (0, _helperBuilderReactJsx.default)({
36 pre(state) {
37 const tagName = state.tagName;
38 const args = state.args;
39
40 if (_core.types.react.isCompatTag(tagName)) {
41 args.push(_core.types.stringLiteral(tagName));
42 } else {
43 args.push(state.tagExpr);
44 }
45 },
46
47 post(state, pass) {
48 state.callee = pass.get("jsxIdentifier")();
49 state.pure = PURE_ANNOTATION != null ? PURE_ANNOTATION : pass.get("pragma") === DEFAULT.pragma;
50 },
51
52 throwIfNamespace: THROW_IF_NAMESPACE
53 });
54 visitor.Program = {
55 enter(path, state) {
56 const {
57 file
58 } = state;
59 let pragma = PRAGMA_DEFAULT;
60 let pragmaFrag = PRAGMA_FRAG_DEFAULT;
61 let pragmaSet = !!options.pragma;
62 let pragmaFragSet = !!options.pragma;
63
64 if (file.ast.comments) {
65 for (const comment of file.ast.comments) {
66 const jsxMatches = JSX_ANNOTATION_REGEX.exec(comment.value);
67
68 if (jsxMatches) {
69 pragma = jsxMatches[1];
70 pragmaSet = true;
71 }
72
73 const jsxFragMatches = JSX_FRAG_ANNOTATION_REGEX.exec(comment.value);
74
75 if (jsxFragMatches) {
76 pragmaFrag = jsxFragMatches[1];
77 pragmaFragSet = true;
78 }
79 }
80 }
81
82 state.set("jsxIdentifier", createIdentifierParser(pragma));
83 state.set("jsxFragIdentifier", createIdentifierParser(pragmaFrag));
84 state.set("usedFragment", false);
85 state.set("pragma", pragma);
86 state.set("pragmaSet", pragmaSet);
87 state.set("pragmaFragSet", pragmaFragSet);
88 },
89
90 exit(path, state) {
91 if (state.get("pragmaSet") && state.get("usedFragment") && !state.get("pragmaFragSet")) {
92 throw new Error("transform-react-jsx: pragma has been set but " + "pragmaFrag has not been set");
93 }
94 }
95
96 };
97
98 visitor.JSXAttribute = function (path) {
99 if (_core.types.isJSXElement(path.node.value)) {
100 path.node.value = _core.types.jsxExpressionContainer(path.node.value);
101 }
102 };
103
104 return {
105 name: "transform-react-jsx",
106 inherits: _pluginSyntaxJsx.default,
107 visitor
108 };
109});
110
111exports.default = _default;
\No newline at end of file