@tanstack/router-plugin
Version:
Modern and scalable routing for React applications
33 lines (32 loc) • 1.4 kB
JavaScript
import { getUniqueProgramIdentifier } from "../../utils.js";
import * as t from "@babel/types";
//#region src/core/code-splitter/plugins/react-refresh-route-components.ts
var REACT_REFRESH_ROUTE_COMPONENT_IDENTS = new Set([
"component",
"pendingComponent",
"errorComponent",
"notFoundComponent"
]);
function createReactRefreshRouteComponentsPlugin() {
return {
name: "react-refresh-route-components",
onUnsplittableRoute(ctx) {
if (!ctx.opts.addHmr) return;
const hoistedDeclarations = [];
ctx.routeOptions.properties.forEach((prop) => {
if (!t.isObjectProperty(prop) || !t.isIdentifier(prop.key)) return;
if (!REACT_REFRESH_ROUTE_COMPONENT_IDENTS.has(prop.key.name)) return;
if (!t.isArrowFunctionExpression(prop.value) && !t.isFunctionExpression(prop.value)) return;
const hoistedIdentifier = getUniqueProgramIdentifier(ctx.programPath, `TSR${prop.key.name[0].toUpperCase()}${prop.key.name.slice(1)}`);
hoistedDeclarations.push(t.variableDeclaration("const", [t.variableDeclarator(hoistedIdentifier, t.cloneNode(prop.value, true))]));
prop.value = t.cloneNode(hoistedIdentifier);
});
if (hoistedDeclarations.length === 0) return;
ctx.insertionPath.insertBefore(hoistedDeclarations);
return { modified: true };
}
};
}
//#endregion
export { createReactRefreshRouteComponentsPlugin };
//# sourceMappingURL=react-refresh-route-components.js.map