@tanstack/router-plugin 1.167.1 → 1.167.3

package/dist/cjs/core/code-splitter/compilers.cjs

@@ -1,6 +1,7 @@
 const require_runtime = require("../../_virtual/_rolldown/runtime.cjs");
 const require_constants = require("../constants.cjs");
 const require_route_hmr_statement = require("../route-hmr-statement.cjs");
+const require_utils = require("../utils.cjs");
 const require_path_ids = require("./path-ids.cjs");
 const require_framework_options = require("./framework-options.cjs");
 let _tanstack_router_utils = require("@tanstack/router-utils");
@@ -166,10 +167,12 @@ function computeSharedBindings(opts) {
 	const splitGroupsPresent = /* @__PURE__ */ new Set();
 	let hasNonSplit = false;
 	for (const prop of routeOptions.properties) {
-		if (!_babel_types.isObjectProperty(prop) || !_babel_types.isIdentifier(prop.key)) continue;
-		if (prop.key.name === "codeSplitGroupings") continue;
+		if (!_babel_types.isObjectProperty(prop)) continue;
+		const key = require_utils.getObjectPropertyKeyName(prop);
+		if (!key) continue;
+		if (key === "codeSplitGroupings") continue;
 		if (_babel_types.isIdentifier(prop.value) && prop.value.name === "undefined") continue;
-		const groupIndex = findIndexForSplitNode(prop.key.name);
+		const groupIndex = findIndexForSplitNode(key);
 		if (groupIndex === -1) hasNonSplit = true;
 		else splitGroupsPresent.add(groupIndex);
 	}
@@ -181,8 +184,9 @@ function computeSharedBindings(opts) {
 	const fullDepGraph = buildDependencyGraph(declMap, allLocalBindings);
 	const refsByGroup = /* @__PURE__ */ new Map();
 	for (const prop of routeOptions.properties) {
-		if (!_babel_types.isObjectProperty(prop) || !_babel_types.isIdentifier(prop.key)) continue;
-		const key = prop.key.name;
+		if (!_babel_types.isObjectProperty(prop)) continue;
+		const key = require_utils.getObjectPropertyKeyName(prop);
+		if (!key) continue;
 		if (key === "codeSplitGroupings") continue;
 		const groupIndex = findIndexForSplitNode(key);
 		const directRefs = collectModuleLevelRefsFromNode(prop.value, localModuleLevelBindings);
@@ -362,6 +366,7 @@ function compileCodeSplitReferenceRoute(opts) {
 	const PACKAGE = frameworkOptions.package;
 	const LAZY_ROUTE_COMPONENT_IDENT = frameworkOptions.idents.lazyRouteComponent;
 	const LAZY_FN_IDENT = frameworkOptions.idents.lazyFn;
+	const stableRouteOptionKeys = [...new Set((opts.compilerPlugins ?? []).flatMap((plugin) => plugin.getStableRouteOptionKeys?.() ?? []))];
 	let createRouteFn;
 	let modified = false;
 	let hmrAdded = false;
@@ -385,20 +390,35 @@ function compileCodeSplitReferenceRoute(opts) {
 				const hasImportedOrDefinedIdentifier = (name) => {
 					return programPath.scope.hasBinding(name);
 				};
+				const addRouteHmr = (insertionPath, routeOptions) => {
+					if (!opts.addHmr || hmrAdded) return;
+					opts.compilerPlugins?.forEach((plugin) => {
+						if ((plugin.onAddHmr?.({
+							programPath,
+							callExpressionPath: path,
+							insertionPath,
+							routeOptions,
+							createRouteFn,
+							opts
+						}))?.modified) modified = true;
+					});
+					programPath.pushContainer("body", require_route_hmr_statement.createRouteHmrStatement(stableRouteOptionKeys));
+					modified = true;
+					hmrAdded = true;
+				};
 				if (_babel_types.isObjectExpression(routeOptions)) {
+					const insertionPath = path.getStatementParent() ?? path;
 					if (opts.deleteNodes && opts.deleteNodes.size > 0) routeOptions.properties = routeOptions.properties.filter((prop) => {
 						if (_babel_types.isObjectProperty(prop)) {
-							if (_babel_types.isIdentifier(prop.key)) {
-								if (opts.deleteNodes.has(prop.key.name)) {
-									modified = true;
-									return false;
-								}
+							const key = require_utils.getObjectPropertyKeyName(prop);
+							if (key && opts.deleteNodes.has(key)) {
+								modified = true;
+								return false;
 							}
 						}
 						return true;
 					});
 					if (!splittableCreateRouteFns.includes(createRouteFn)) {
-						const insertionPath = path.getStatementParent() ?? path;
 						opts.compilerPlugins?.forEach((plugin) => {
 							if ((plugin.onUnsplittableRoute?.({
 								programPath,
@@ -409,17 +429,13 @@ function compileCodeSplitReferenceRoute(opts) {
 								opts
 							}))?.modified) modified = true;
 						});
-						if (opts.addHmr && !hmrAdded) {
-							programPath.pushContainer("body", require_route_hmr_statement.routeHmrStatement);
-							modified = true;
-							hmrAdded = true;
-						}
+						addRouteHmr(insertionPath, routeOptions);
 						return programPath.stop();
 					}
 					routeOptions.properties.forEach((prop) => {
 						if (_babel_types.isObjectProperty(prop)) {
-							if (_babel_types.isIdentifier(prop.key)) {
-								const key = prop.key.name;
+							const key = require_utils.getObjectPropertyKeyName(prop);
+							if (key) {
 								const codeSplitGroupingByKey = findIndexForSplitNode(key);
 								if (codeSplitGroupingByKey === -1) return;
 								const codeSplitGroup = [...new Set(opts.codeSplitGroupings[codeSplitGroupingByKey])];
@@ -442,12 +458,26 @@ function compileCodeSplitReferenceRoute(opts) {
 									modified = true;
 									if (!hasImportedOrDefinedIdentifier(LAZY_ROUTE_COMPONENT_IDENT)) programPath.unshiftContainer("body", [_babel_template.statement(`import { ${LAZY_ROUTE_COMPONENT_IDENT} } from '${PACKAGE}'`)()]);
 									if (!hasImportedOrDefinedIdentifier(splitNodeMeta.localImporterIdent)) programPath.unshiftContainer("body", [_babel_template.statement(`const ${splitNodeMeta.localImporterIdent} = () => import('${splitUrl}')`)()]);
-									prop.value = _babel_template.expression(`${LAZY_ROUTE_COMPONENT_IDENT}(${splitNodeMeta.localImporterIdent}, '${splitNodeMeta.exporterIdent}')`)();
-									if (opts.addHmr && !hmrAdded) {
-										programPath.pushContainer("body", require_route_hmr_statement.routeHmrStatement);
+									const insertionPath = path.getStatementParent() ?? path;
+									let splitPropValue;
+									for (const plugin of opts.compilerPlugins ?? []) {
+										const pluginPropValue = plugin.onSplitRouteProperty?.({
+											programPath,
+											callExpressionPath: path,
+											insertionPath,
+											routeOptions,
+											prop,
+											splitNodeMeta,
+											lazyRouteComponentIdent: LAZY_ROUTE_COMPONENT_IDENT
+										});
+										if (!pluginPropValue) continue;
 										modified = true;
-										hmrAdded = true;
+										splitPropValue = pluginPropValue;
+										break;
 									}
+									if (splitPropValue) prop.value = splitPropValue;
+									else prop.value = _babel_template.expression(`${LAZY_ROUTE_COMPONENT_IDENT}(${splitNodeMeta.localImporterIdent}, '${splitNodeMeta.exporterIdent}')`)();
+									addRouteHmr(insertionPath, routeOptions);
 								} else {
 									const value = prop.value;
 									let shouldSplit = true;
@@ -469,6 +499,7 @@ function compileCodeSplitReferenceRoute(opts) {
 						}
 						programPath.scope.crawl();
 					});
+					addRouteHmr(insertionPath, routeOptions);
 				}
 			}
 			if (_babel_types.isCallExpression(path.parentPath.node)) babelHandleReference(resolveIdentifier(path, path.parentPath.node.arguments[0]));
@@ -544,7 +575,7 @@ function compileCodeSplitVirtualRoute(opts) {
 				if (_babel_types.isObjectExpression(options)) {
 					options.properties.forEach((prop) => {
 						if (_babel_types.isObjectProperty(prop)) KNOWN_SPLIT_ROUTE_IDENTS.forEach((splitType) => {
-							if (!_babel_types.isIdentifier(prop.key) || prop.key.name !== splitType) return;
+							if (require_utils.getObjectPropertyKeyName(prop) !== splitType) return;
 							const value = prop.value;
 							if (_babel_types.isIdentifier(value) && value.name === "undefined") return;
 							let isExported = false;
@@ -738,18 +769,16 @@ function detectCodeSplitGroupingsFromRoute(opts) {
 			function babelHandleSplittingGroups(routeOptions) {
 				if (_babel_types.isObjectExpression(routeOptions)) routeOptions.properties.forEach((prop) => {
 					if (_babel_types.isObjectProperty(prop)) {
-						if (_babel_types.isIdentifier(prop.key)) {
-							if (prop.key.name === "codeSplitGroupings") {
-								const value = prop.value;
-								if (_babel_types.isArrayExpression(value)) codeSplitGroupings = value.elements.map((group) => {
-									if (_babel_types.isArrayExpression(group)) return group.elements.map((node) => {
-										if (!_babel_types.isStringLiteral(node)) throw new Error("You must provide a string literal for the codeSplitGroupings");
-										return node.value;
-									});
-									throw new Error("You must provide arrays with codeSplitGroupings options.");
+						if (require_utils.getObjectPropertyKeyName(prop) === "codeSplitGroupings") {
+							const value = prop.value;
+							if (_babel_types.isArrayExpression(value)) codeSplitGroupings = value.elements.map((group) => {
+								if (_babel_types.isArrayExpression(group)) return group.elements.map((node) => {
+									if (!_babel_types.isStringLiteral(node)) throw new Error("You must provide a string literal for the codeSplitGroupings");
+									return node.value;
 								});
-								else throw new Error("You must provide an array of arrays for the codeSplitGroupings.");
-							}
+								throw new Error("You must provide arrays with codeSplitGroupings options.");
+							});
+							else throw new Error("You must provide an array of arrays for the codeSplitGroupings.");
 						}
 					}
 				});