Skip to content

Commit

Permalink
add next option for babel-plugin-transform-react-jsx
Browse files Browse the repository at this point in the history
  • Loading branch information
lunaruan committed Feb 21, 2020
1 parent 1613418 commit ddc42e9
Show file tree
Hide file tree
Showing 170 changed files with 1,874 additions and 97 deletions.
4 changes: 3 additions & 1 deletion packages/babel-plugin-transform-react-jsx/package.json
Expand Up @@ -13,8 +13,10 @@
],
"dependencies": {
"@babel/helper-builder-react-jsx": "^7.8.3",
"@babel/helper-module-imports": "^7.8.3",
"@babel/helper-plugin-utils": "^7.8.3",
"@babel/plugin-syntax-jsx": "^7.8.3"
"@babel/plugin-syntax-jsx": "^7.8.3",
"esutils": "^2.0.0"
},
"peerDependencies": {
"@babel/core": "^7.0.0-0"
Expand Down
105 changes: 9 additions & 96 deletions packages/babel-plugin-transform-react-jsx/src/index.js
@@ -1,100 +1,13 @@
/* eslint-disable-next-line @babel/development/plugin-name */
import transformStable from "./transform-stable";
/* eslint-disable-next-line @babel/development/plugin-name */
import transformExperimental from "./transform-experimental";
import { declare } from "@babel/helper-plugin-utils";
import jsx from "@babel/plugin-syntax-jsx";
import helper from "@babel/helper-builder-react-jsx";
import { types as t } from "@babel/core";

export default declare((api, options) => {
api.assertVersion(7);

const THROW_IF_NAMESPACE =
options.throwIfNamespace === undefined ? true : !!options.throwIfNamespace;

const PRAGMA_DEFAULT = options.pragma || "React.createElement";
const PRAGMA_FRAG_DEFAULT = options.pragmaFrag || "React.Fragment";

const JSX_ANNOTATION_REGEX = /\*?\s*@jsx\s+([^\s]+)/;
const JSX_FRAG_ANNOTATION_REGEX = /\*?\s*@jsxFrag\s+([^\s]+)/;

// returns a closure that returns an identifier or memberExpression node
// based on the given id
const createIdentifierParser = (id: string) => () => {
return id
.split(".")
.map(name => t.identifier(name))
.reduce((object, property) => t.memberExpression(object, property));
};

const visitor = helper({
pre(state) {
const tagName = state.tagName;
const args = state.args;
if (t.react.isCompatTag(tagName)) {
args.push(t.stringLiteral(tagName));
} else {
args.push(state.tagExpr);
}
},

post(state, pass) {
state.callee = pass.get("jsxIdentifier")();
},

throwIfNamespace: THROW_IF_NAMESPACE,
});

visitor.Program = {
enter(path, state) {
const { file } = state;

let pragma = PRAGMA_DEFAULT;
let pragmaFrag = PRAGMA_FRAG_DEFAULT;
let pragmaSet = !!options.pragma;
let pragmaFragSet = !!options.pragmaFrag;

if (file.ast.comments) {
for (const comment of (file.ast.comments: Array<Object>)) {
const jsxMatches = JSX_ANNOTATION_REGEX.exec(comment.value);
if (jsxMatches) {
pragma = jsxMatches[1];
pragmaSet = true;
}
const jsxFragMatches = JSX_FRAG_ANNOTATION_REGEX.exec(comment.value);
if (jsxFragMatches) {
pragmaFrag = jsxFragMatches[1];
pragmaFragSet = true;
}
}
}

state.set("jsxIdentifier", createIdentifierParser(pragma));
state.set("jsxFragIdentifier", createIdentifierParser(pragmaFrag));
state.set("usedFragment", false);
state.set("pragmaSet", pragmaSet);
state.set("pragmaFragSet", pragmaFragSet);
},
exit(path, state) {
if (
state.get("pragmaSet") &&
state.get("usedFragment") &&
!state.get("pragmaFragSet")
) {
throw new Error(
"transform-react-jsx: pragma has been set but " +
"pragmaFrag has not been set",
);
}
},
};

visitor.JSXAttribute = function(path) {
if (t.isJSXElement(path.node.value)) {
path.node.value = t.jsxExpressionContainer(path.node.value);
}
};

return {
name: "transform-react-jsx",
inherits: jsx,
visitor,
};
if (options.version === "experimental") {
return transformExperimental(api);
} else {
return transformStable(api, options);
}
});

0 comments on commit ddc42e9

Please sign in to comment.