Migrate Babel from Flow to TypeScript (except Babel parser) (#11578)
Co-authored-by: Nicolò Ribaudo <nicolo.ribaudo@gmail.com>
This commit is contained in:
@@ -0,0 +1,96 @@
|
||||
/**
|
||||
* This adds {fileName, lineNumber, columnNumber} annotations to JSX tags.
|
||||
*
|
||||
* NOTE: lineNumber and columnNumber are both 1-based.
|
||||
*
|
||||
* == JSX Literals ==
|
||||
*
|
||||
* <sometag />
|
||||
*
|
||||
* becomes:
|
||||
*
|
||||
* var __jsxFileName = 'this/file.js';
|
||||
* <sometag __source={{fileName: __jsxFileName, lineNumber: 10, columnNumber: 1}}/>
|
||||
*/
|
||||
import { declare } from "@babel/helper-plugin-utils";
|
||||
import { types as t } from "@babel/core";
|
||||
|
||||
const TRACE_ID = "__source";
|
||||
const FILE_NAME_VAR = "_jsxFileName";
|
||||
|
||||
export default declare(api => {
|
||||
api.assertVersion(7);
|
||||
|
||||
function makeTrace(fileNameIdentifier, lineNumber, column0Based) {
|
||||
const fileLineLiteral =
|
||||
lineNumber != null ? t.numericLiteral(lineNumber) : t.nullLiteral();
|
||||
const fileColumnLiteral =
|
||||
column0Based != null
|
||||
? t.numericLiteral(column0Based + 1)
|
||||
: t.nullLiteral();
|
||||
const fileNameProperty = t.objectProperty(
|
||||
t.identifier("fileName"),
|
||||
fileNameIdentifier,
|
||||
);
|
||||
const lineNumberProperty = t.objectProperty(
|
||||
t.identifier("lineNumber"),
|
||||
fileLineLiteral,
|
||||
);
|
||||
const columnNumberProperty = t.objectProperty(
|
||||
t.identifier("columnNumber"),
|
||||
fileColumnLiteral,
|
||||
);
|
||||
return t.objectExpression([
|
||||
fileNameProperty,
|
||||
lineNumberProperty,
|
||||
columnNumberProperty,
|
||||
]);
|
||||
}
|
||||
|
||||
const visitor = {
|
||||
JSXOpeningElement(path, state) {
|
||||
const id = t.jsxIdentifier(TRACE_ID);
|
||||
const location = path.container.openingElement.loc;
|
||||
if (!location) {
|
||||
// the element was generated and doesn't have location information
|
||||
return;
|
||||
}
|
||||
|
||||
const attributes = path.container.openingElement.attributes;
|
||||
for (let i = 0; i < attributes.length; i++) {
|
||||
const name = attributes[i].name;
|
||||
if (name?.name === TRACE_ID) {
|
||||
// The __source attribute already exists
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if (!state.fileNameIdentifier) {
|
||||
const fileName = state.filename || "";
|
||||
|
||||
const fileNameIdentifier =
|
||||
path.scope.generateUidIdentifier(FILE_NAME_VAR);
|
||||
const scope = path.hub.getScope();
|
||||
if (scope) {
|
||||
scope.push({
|
||||
id: fileNameIdentifier,
|
||||
init: t.stringLiteral(fileName),
|
||||
});
|
||||
}
|
||||
state.fileNameIdentifier = fileNameIdentifier;
|
||||
}
|
||||
|
||||
const trace = makeTrace(
|
||||
t.cloneNode(state.fileNameIdentifier),
|
||||
location.start.line,
|
||||
location.start.column,
|
||||
);
|
||||
attributes.push(t.jsxAttribute(id, t.jsxExpressionContainer(trace)));
|
||||
},
|
||||
};
|
||||
|
||||
return {
|
||||
name: "transform-react-jsx-source",
|
||||
visitor,
|
||||
};
|
||||
});
|
||||
Reference in New Issue
Block a user