2016-03-16 20:06:09 -07:00

51 lines
1.3 KiB
JavaScript

/* eslint max-len: 0 */
export default function ({ types: t }) {
let JSX_ANNOTATION_REGEX = /\*?\s*@jsx\s+([^\s]+)/;
let visitor = require("babel-helper-builder-react-jsx")({
pre(state) {
let tagName = state.tagName;
let 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")();
}
});
visitor.Program = function (path, state) {
let { file } = state;
let id = state.opts.pragma || "React.createElement";
for (let comment of (file.ast.comments: Array<Object>)) {
let matches = JSX_ANNOTATION_REGEX.exec(comment.value);
if (matches) {
id = matches[1];
if (id === "React.DOM") {
throw file.buildCodeFrameError(comment, "The @jsx React.DOM pragma has been deprecated as of React 0.12");
} else {
break;
}
}
}
state.set(
"jsxIdentifier",
() => id.split(".").map((name) => t.identifier(name)).reduce(
(object, property) => t.memberExpression(object, property)
)
);
};
return {
inherits: require("babel-plugin-syntax-jsx"),
visitor
};
}